package ru.gravit.launcher;

import java.io.IOException;
import java.lang.instrument.Instrumentation;
import java.lang.instrument.UnmodifiableClassException;
import java.util.ArrayList;
import java.util.jar.JarFile;

import javassist.bytecode.ClassFile;
import ru.gravit.utils.helper.LogHelper;
import ru.gravit.launcher.transformers.SystemClassLoaderTransformer;

@LauncherAPI
public class LauncherAgent {
	private static final boolean enabled = false;
	private static boolean isAgentStarted=false;
    public static Instrumentation inst;
    
    public static void addJVMClassPath(String path) throws IOException {
        LogHelper.debug("Launcher Agent addJVMClassPath");
        inst.appendToSystemClassLoaderSearch(new JarFile(path));
    }
	public boolean isAgentStarted()
	{
		return isAgentStarted;
	}
    public static long getObjSize(Object obj) {
    	return inst.getObjectSize(obj);
    }
    
    public static void premain(String agentArgument, Instrumentation instrumentation) {
        System.out.println("Launcher Agent");
        inst = instrumentation;
        isAgentStarted = true;

        if(ClassFile.MAJOR_VERSION > ClassFile.JAVA_8 || enabled) {
	        	inst.addTransformer(new SystemClassLoaderTransformer());
	        Class<?>[] classes = inst.getAllLoadedClasses(); // Получаем список уже загруженных классов, которые могут быть изменены. Классы, которые ещё не загружены, будут изменены при загрузке
	        ArrayList<Class<?>> classList = new ArrayList<>();
	        for (Class<?> classe : classes)
				if (inst.isModifiableClass(classe))
					classList.add(classe);
	        // Reload classes, if possible.
	        Class<?>[] workaround = new Class[classList.size()];
	        try {
	        	inst.retransformClasses(classList.toArray(workaround)); // Запускаем процесс трансформации
	        } catch (UnmodifiableClassException e) {
	        	System.err.println("MainClass was unable to retransform early loaded classes: " + e);
	        }
        }
    }
}