package insertGenerics;

import de.dhbwstuttgart.bytecode.constraint.TPHConstraint.Relation;
import de.dhbwstuttgart.bytecode.genericsGeneratorTypes.GenericGenratorResultForSourceFile;
import de.dhbwstuttgart.bytecode.insertGenerics.ClassConstraint;
import de.dhbwstuttgart.bytecode.insertGenerics.FamilyOfGeneratedGenerics;
import de.dhbwstuttgart.bytecode.insertGenerics.MethodConstraint;
import de.dhbwstuttgart.core.JavaTXCompiler;
import de.dhbwstuttgart.syntaxtree.SourceFile;
import de.dhbwstuttgart.syntaxtree.visual.ASTPrinter;
import de.dhbwstuttgart.syntaxtree.visual.ASTTypePrinter;
import de.dhbwstuttgart.typedeployment.TypeInsert;
import de.dhbwstuttgart.typedeployment.TypeInsertFactory;
import de.dhbwstuttgart.typeinference.result.ResultSet;

import org.junit.BeforeClass;
import org.junit.Test;

import static org.junit.Assert.assertEquals;

import java.io.File;
import java.io.IOException;
import java.lang.reflect.InvocationTargetException;
import java.net.URL;
import java.net.URLClassLoader;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Set;

public class TestGGFinder {

    private static final String rootDirectory = System.getProperty("user.dir")+"/resources/insertGenericsJav/";
    private String pathToClassFile = System.getProperty("user.dir")+"/resources/testBytecode/generatedBC/";
    private static ClassLoader loader;
    private static Class<?> classToTest;
	private static Object instanceOfClass;
	private static String className = "TestGGFinder";

    @BeforeClass
    public static void resetNamesOfTypePlaceholder() {
    	de.dhbwstuttgart.syntaxtree.factory.NameGenerator.reset();
    }
    
    @Test
    public void ggFinder() throws IOException, ClassNotFoundException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, NoSuchMethodException, SecurityException {
        execute(new File(rootDirectory+className+".jav"));
    }

    private static class TestResultSet{

    }

    public TestResultSet execute(File fileToTest) throws IOException, ClassNotFoundException, InstantiationException, IllegalAccessException, IllegalArgumentException, InvocationTargetException, NoSuchMethodException, SecurityException {
        JavaTXCompiler compiler = new JavaTXCompiler(fileToTest);
        List<ResultSet> results = compiler.typeInference();
        List<GenericGenratorResultForSourceFile> simplifyResultsForAllSourceFiles = compiler.getGeneratedGenericResultsForAllSourceFiles(results);
        
        List<ClassConstraint> classConstraintsTest = new ArrayList<>();
        classConstraintsTest.add(new ClassConstraint("S", "java/lang/Object", Relation.EXTENDS));
        HashMap<String, Set<MethodConstraint>> methodConstraintsWithPositionTest = new HashMap<>();
        Set<MethodConstraint> lmc;
        lmc = new HashSet<>();
        lmc.add(new MethodConstraint("P", "AC", Relation.EXTENDS));
        lmc.add(new MethodConstraint("AC", "java/lang/Object", Relation.EXTENDS));
        methodConstraintsWithPositionTest.put("TPH ACid(TPH P)",  lmc);
        lmc = new HashSet<>();
        lmc.add(new MethodConstraint("T", "S", Relation.EXTENDS));
        methodConstraintsWithPositionTest.put("TPH SsetA(TPH T)",  lmc);
        lmc = new HashSet<>();
        lmc.add(new MethodConstraint("AC", "Y", Relation.EXTENDS));
        lmc.add(new MethodConstraint("Z", "P", Relation.EXTENDS));
        lmc.add(new MethodConstraint("P", "AC", Relation.EXTENDS));
        lmc.add(new MethodConstraint("Y", "java/lang/Object", Relation.EXTENDS));
        methodConstraintsWithPositionTest.put("voidm(TPH YTPH Z)",  lmc);
        
        FamilyOfGeneratedGenerics fogg = compiler.fogg;
        Set<ClassConstraint> computedClassCons = new HashSet<>(fogg.classConstraints);
        Set<ClassConstraint> expectedClassCons = new HashSet<>(classConstraintsTest);
        
        assertEquals(expectedClassCons, computedClassCons);
        
        HashMap<String, Set<MethodConstraint>> methodConstraintsWithPositionComputed = new HashMap<>();
        fogg.methodConstraintsWithPosition.forEach((s, l) -> methodConstraintsWithPositionComputed.put(s,new HashSet<>(l)));
        
        assertEquals(methodConstraintsWithPositionTest, methodConstraintsWithPositionComputed);
        
        compiler.generateBytecode(new File(pathToClassFile), results, simplifyResultsForAllSourceFiles);
        loader = new URLClassLoader(new URL[] {new URL("file://"+pathToClassFile)});
        classToTest = loader.loadClass(className);
		instanceOfClass = classToTest.getDeclaredConstructor().newInstance();
        return new TestResultSet();
    }
}