package ftbsc.lll.mixin;

import ftbsc.lll.IInjector;
import ftbsc.lll.exceptions.InjectionException;
import org.objectweb.asm.tree.ClassNode;
import org.spongepowered.asm.mixin.extensibility.IMixinConfigPlugin;
import org.spongepowered.asm.mixin.extensibility.IMixinInfo;

import java.util.*;

public class LilleroMixin implements IMixinConfigPlugin {

	private final Map<String, List<IInjector>> injectorMap = new HashMap<>();

	/**
	 * Called after the plugin is instantiated, do any setup here.
	 * @param mixinPackage The mixin root package from the config
	 */
	@Override
	public void onLoad(String mixinPackage) {
		for (IInjector inj : ServiceLoader.load(IInjector.class, this.getClass().getClassLoader())) {
			//LOGGER.info(RESOURCE, "Registering injector {}", inj.name());
			List<IInjector> injectors = this.injectorMap.get(inj.targetClass());
			if(injectors == null) {
				injectors = new ArrayList<>();
				injectorMap.put(inj.targetClass(), injectors);
			}
			injectors.add(inj);
		}
	}

	/**
	 * Called only if the "referenceMap" key in the config is <b>not</b> set.
	 * This allows the refmap file name to be supplied by the plugin
	 * programatically if desired. Returning <code>null</code> will revert to
	 * the default behaviour of using the default refmap json file.
	 *
	 * @return Path to the refmap resource or null to revert to the default
	 */
	@Override //TODO ?
	public String getRefMapperConfig() {
		return null;
	}

	/**
	 * Called during mixin intialisation, allows this plugin to control whether
	 * a specific will be applied to the specified target. Returning false will
	 * remove the target from the mixin's target set, and if all targets are
	 * removed then the mixin will not be applied at all.
	 *
	 * @param targetClassName Fully qualified class name of the target class
	 * @param mixinClassName  Fully qualified class name of the mixin
	 * @return True to allow the mixin to be applied, or false to remove it from
	 * target's mixin set
	 */
	@Override
	public boolean shouldApplyMixin(String targetClassName, String mixinClassName) {
		return true;
	}

	/**
	 * Called after all configurations are initialised, this allows this plugin
	 * to observe classes targetted by other mixin configs and optionally remove
	 * targets from its own set. The set myTargets is a direct view of the
	 * targets collection in this companion config and keys may be removed from
	 * this set to suppress mixins in this config which target the specified
	 * class. Adding keys to the set will have no effect.
	 *
	 * @param myTargets    Target class set from the companion config
	 * @param otherTargets Target class set incorporating targets from all other
	 *                     configs, read-only
	 */
	@Override
	public void acceptTargets(Set<String> myTargets, Set<String> otherTargets) {

	}

	/**
	 * After mixins specified in the configuration have been processed, this
	 * method is called to allow the plugin to add any additional mixins to
	 * load. It should return a list of mixin class names or return null if the
	 * plugin does not wish to append any mixins of its own.
	 *
	 * @return additional mixins to apply
	 */
	@Override
	public List<String> getMixins() {
		return null;
	}

	/**
	 * Called immediately <b>before</b> a mixin is applied to a target class,
	 * allows any pre-application transformations to be applied.
	 *
	 * @param targetClassName Transformed name of the target class
	 * @param targetClass     Target class tree
	 * @param mixinClassName  Name of the mixin class
	 * @param mixinInfo       Information about this mixin
	 */
	@Override
	public void preApply(String targetClassName, ClassNode targetClass, String mixinClassName, IMixinInfo mixinInfo) {
		injectorMap.get(targetClassName).forEach((inj) -> targetClass.methods.stream()
			.filter(m -> m.name.equals(inj.methodName()) && m.desc.equals(inj.methodDesc()))
			.forEach(m -> {
				try {
					inj.inject(targetClass, m);
				} catch (InjectionException ignored) {} //TODO log
			}));
	}

	/**
	 * Called immediately <b>after</b> a mixin is applied to a target class,
	 * allows any post-application transformations to be applied.
	 *
	 * @param targetClassName Transformed name of the target class
	 * @param targetClass     Target class tree
	 * @param mixinClassName  Name of the mixin class
	 * @param mixinInfo       Information about this mixin
	 */
	@Override
	public void postApply(String targetClassName, ClassNode targetClass, String mixinClassName, IMixinInfo mixinInfo) {

	}
}