Class ChasmProcessor

chasm/src/main/java/org/quiltmc/chasm/api/ChasmProcessor.java:34–165

Transforms the added classes using the added Transformers.

 * Transforms the added classes using the added {@link Transformer}s.
 */
public class ChasmProcessor {
    private static final Logger LOGGER = LoggerFactory.getLogger(ChasmProcessor.class);

    private final Context context;

    private final List<Transformer> transformers = new ArrayList<>();

    private final List<ClassData> classes = new ArrayList<>();

    /**
     * Creates a new {@link ChasmProcessor} that uses the given {@link Context}.
     *
     * @param context A {@code Context} to supply parents of classes that are not being
     *            transformed.
     */
    public ChasmProcessor(Context context) {
        this.context = context;
    }

    /**
     * Adds the passed {@link Transformer} to this {@link ChasmProcessor}'s
     * list of {@code Transformer}s.
     *
     * @param transformer A {@code Transformer} to add to this {@code ChasmProcessor}'s
     *            list of {@code Transformer}s to transform classes with.
     */
    public void addTransformer(Transformer transformer) {
        transformers.add(transformer);
    }

    /**
     * Adds the passed class data to this {@link ChasmProcessor}'s
     * list of classes to transform.
     *
     * @param classBytes The bytes of the class.
     * @param metadata The metadata associated with the class.
     */
    public void addClass(byte @NotNull [] classBytes, @NotNull Metadata metadata) {
        this.classes.add(new ClassData(classBytes, metadata));
    }

    /**
     * Transforms the {@link ClassData} passed via {@link #addClass} using
     * the {@link Transformer Transformers} passed via {@link #addTransformer}.
     *
     * @return The transformed {@link ClassData}, wrapped in {@link ClassResult}.
     */
    public List<ClassResult> process() {
        LOGGER.info("Processing {} classes...", classes.size());

        ListNode classes = Ast.emptyList();
        Context context = new ChasmContext(this.context, classes);
        Map<String, ClassData> nameToData = new HashMap<>();
        for (ClassData classData : this.classes) {
            ClassReader classReader = new ClassReader(classData.getClassBytes());
            ClassNode classNode = new ClassNode(classReader, context, classes.size());
            classNode.getMetadata().putAll(classData.getMetadata());
            classes.add(classNode);