com.llamalad7.mixinextras.utils.MixinAPInternals Maven / Gradle / Ivy

Go to download
package com.llamalad7.mixinextras.utils;

import org.spongepowered.tools.obfuscation.mirror.AnnotationHandle;
import org.spongepowered.tools.obfuscation.struct.InjectorRemap;

import javax.annotation.processing.ProcessingEnvironment;
import javax.lang.model.element.ExecutableElement;
import javax.lang.model.element.TypeElement;

public class MixinAPInternals {
    private static final String ANNOTATED_MIXINS = "org.spongepowered.tools.obfuscation.AnnotatedMixins";
    private static final String ANNOTATED_MIXIN = "org.spongepowered.tools.obfuscation.AnnotatedMixin";
    private static final Class ANNOTATED_MIXIN_CLASS;

    static {
        try {
            ANNOTATED_MIXIN_CLASS = Class.forName(ANNOTATED_MIXIN);
        } catch (ClassNotFoundException e) {
            throw new RuntimeException("Could not find class! Please report to LlamaLad7:", e);
        }
    }

    private static final InternalMethod ANNOTATED_MIXINS_GET_FOR_ENV =
            InternalMethod.of(ANNOTATED_MIXINS, "getMixinsForEnvironment", ProcessingEnvironment.class);
    private static final InternalMethod ANNOTATED_MIXINS_GET_MIXIN =
            InternalMethod.of(ANNOTATED_MIXINS, "getMixin", TypeElement.class);
    private static final InternalMethod ANNOTATED_MIXINS_SHOULD_REMAP =
            InternalMethod.of(ANNOTATED_MIXINS, "shouldRemap", ANNOTATED_MIXIN_CLASS, AnnotationHandle.class);
    private static final InternalMethod ANNOTATED_MIXIN_REGISTER_INJECTION_POINT =
            InternalMethod.of(ANNOTATED_MIXIN, "registerInjectionPoint",
                    ExecutableElement.class, AnnotationHandle.class, String.class,
                    AnnotationHandle.class, InjectorRemap.class, String.class
            );
    private static final InternalMethod ANNOTATED_MIXINS_WRITE_REFERENCES =
            InternalMethod.of(ANNOTATED_MIXINS, "writeReferences");

    public static void registerInjectionPoint(
            ProcessingEnvironment env,
            TypeElement mixin,
            ExecutableElement handler,
            AnnotationHandle injector,
            AnnotationHandle at
    ) {
        Object annotatedMixin = ANNOTATED_MIXINS_GET_MIXIN.call(ANNOTATED_MIXINS_GET_FOR_ENV.call(null, env), mixin);
        InjectorRemap remap = new InjectorRemap(ANNOTATED_MIXINS_SHOULD_REMAP.call(null, annotatedMixin, injector));
        ANNOTATED_MIXIN_REGISTER_INJECTION_POINT.call(annotatedMixin, handler, injector, "at", at, remap, "@At(%s)");
    }

    public static void writeReferences(ProcessingEnvironment env) {
        ANNOTATED_MIXINS_WRITE_REFERENCES.call(ANNOTATED_MIXINS_GET_FOR_ENV.call(null, env));
    }
}