com.llamalad7.mixinextras.injector.LateInjectionApplicatorExtension Maven / Gradle / Ivy

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

import org.objectweb.asm.tree.ClassNode;
import org.spongepowered.asm.mixin.MixinEnvironment;
import org.spongepowered.asm.mixin.transformer.ext.IExtension;
import org.spongepowered.asm.mixin.transformer.ext.ITargetClassContext;

import java.util.*;

/**
 * This extension is responsible for actually injecting all late-applying injectors which were queued up during the
 * normal injection phase. Applying them here means we are guaranteed to run after every other injector, which is
 * crucial.
 */
public class LateInjectionApplicatorExtension implements IExtension {
    private static final Map>> QUEUED_INJECTIONS = Collections.synchronizedMap(new HashMap<>());

    static void offerInjection(ITargetClassContext targetClassContext, LateApplyingInjectorInfo injectorInfo) {
        Map> map = QUEUED_INJECTIONS.computeIfAbsent(targetClassContext, k -> initializeMap());
        map.get(injectorInfo.getLateInjectionType()).add(new Runnable[]{injectorInfo::lateInject, injectorInfo::latePostInject});
    }

    @Override
    public boolean checkActive(MixinEnvironment environment) {
        return true;
    }

    @Override
    public void preApply(ITargetClassContext context) {
    }

    @Override
    public void postApply(ITargetClassContext context) {
        Map> relevant = QUEUED_INJECTIONS.get(context);
        if (relevant == null) {
            return;
        }
        for (List queuedInjections : relevant.values()) {
            for (Runnable[] injection : queuedInjections) {
                injection[0].run();
            }
            for (Runnable[] injection : queuedInjections) {
                injection[1].run();
            }
        }
        QUEUED_INJECTIONS.remove(context);
    }

    @Override
    public void export(MixinEnvironment env, String name, boolean force, ClassNode classNode) {
    }

    private static Map> initializeMap() {
        Map> result = new LinkedHashMap<>();
        result.put("ModifyExpressionValue", new ArrayList<>());
        result.put("WrapWithCondition", new ArrayList<>());
        result.put("WrapOperation", new ArrayList<>());
        return result;
    }
}