/*******************************************************************************
 * Copyright (c) 2015 Jeff Martin.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public
 * License v3.0 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl.html
 * <p>
 * Contributors:
 * Jeff Martin - initial API and implementation
 ******************************************************************************/

package cuchaz.enigma;

import cuchaz.enigma.convert.*;
import cuchaz.enigma.gui.ClassMatchingGui;
import cuchaz.enigma.gui.MemberMatchingGui;
import cuchaz.enigma.mapping.*;
import cuchaz.enigma.throwables.MappingConflict;
import cuchaz.enigma.throwables.MappingParseException;

import java.io.File;
import java.io.IOException;
import java.util.jar.JarFile;

public class ConvertMain {

	public static void main(String[] args)
		throws IOException, MappingParseException {
		try {
			//Get all are args
			String JarOld = getArg(args, 1, "Path to Old Jar", true);
			String JarNew = getArg(args, 2, "Path to New Jar", true);
			String OldMappings = getArg(args, 3, "Path to old .mappings file", true);
			String NewMappings = getArg(args, 4, "Path to new .mappings file", true);
			String ClassMatches = getArg(args, 5, "Path to Class .matches file", true);
			String FieldMatches = getArg(args, 6, "Path to Field .matches file", true);
			String MethodMatches = getArg(args, 7, "Path to Method .matches file", true);
			//OldJar
			JarFile sourceJar = new JarFile(new File(JarOld));
			//NewJar
			JarFile destJar = new JarFile(new File(JarNew));
			//Get the mapping files
			File inMappingsFile = new File(OldMappings);
			File outMappingsFile = new File(NewMappings);
			Mappings mappings = new MappingsEnigmaReader().read(inMappingsFile);
			//Make the Match Files..
			File classMatchesFile = new File(ClassMatches);
			File fieldMatchesFile = new File(FieldMatches);
			File methodMatchesFile = new File(MethodMatches);

			String command = getArg(args, 0, "command", true);

			if (command.equalsIgnoreCase("computeClassMatches")) {
				computeClassMatches(classMatchesFile, sourceJar, destJar, mappings);
				convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile);
			} else if (command.equalsIgnoreCase("editClassMatches")) {
				editClasssMatches(classMatchesFile, sourceJar, destJar, mappings);
				convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile);
			} else if (command.equalsIgnoreCase("computeFieldMatches")) {
				computeFieldMatches(fieldMatchesFile, destJar, outMappingsFile, classMatchesFile);
				convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile);
			} else if (command.equalsIgnoreCase("editFieldMatches")) {
				editFieldMatches(sourceJar, destJar, outMappingsFile, mappings, classMatchesFile, fieldMatchesFile);
				convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile);
			} else if (command.equalsIgnoreCase("computeMethodMatches")) {
				computeMethodMatches(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile, methodMatchesFile);
				convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile, methodMatchesFile);
			} else if (command.equalsIgnoreCase("editMethodMatches")) {
				editMethodMatches(sourceJar, destJar, outMappingsFile, mappings, classMatchesFile, methodMatchesFile);
				convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile, methodMatchesFile);
			} else if (command.equalsIgnoreCase("convertMappings")) {
				convertMappings(outMappingsFile, sourceJar, destJar, mappings, classMatchesFile, fieldMatchesFile, methodMatchesFile);
			}
		} catch (MappingConflict ex) {
			System.out.println(ex.getMessage());
			ex.printStackTrace();
		} catch (IllegalArgumentException ex) {
			System.out.println(ex.getMessage());
			printHelp();
		}
	}

	private static void printHelp() {
		System.out.println(String.format("%s - %s", Constants.NAME, Constants.VERSION));
		System.out.println("Usage:");
		System.out.println("\tjava -cp enigma.jar cuchaz.enigma.ConvertMain <command> <old-jar> <new-jar> <old-mappings> <new-mappings> <class-matches> <field-matches> <method-matches>");
		System.out.println("\tWhere <command> is one of:");
		System.out.println("\t\tcomputeClassMatches");
		System.out.println("\t\teditClassMatches");
		System.out.println("\t\tcomputeFieldMatches");
		System.out.println("\t\teditFieldMatches");
		System.out.println("\t\teditMethodMatches");
		System.out.println("\t\tconvertMappings");
		System.out.println("\tWhere <old-jar> is the already mapped jar.");
		System.out.println("\tWhere <new-jar> is the unmapped jar.");
		System.out.println("\tWhere <old-mappings> is the path to the mappings for the old jar.");
		System.out.println("\tWhere <new-mappings> is the new mappings. (Where you want to save them and there name)");
		System.out.println("\tWhere <class-matches> is the class matches file.");
		System.out.println("\tWhere <field-matches> is the field matches file.");
		System.out.println("\tWhere <method-matches> is the method matches file.");
	}

	//Copy of getArg from CommandMain.... Should make a utils class.
	private static String getArg(String[] args, int i, String name, boolean required) {
		if (i >= args.length) {
			if (required) {
				throw new IllegalArgumentException(name + " is required");
			} else {
				return null;
			}
		}
		return args[i];
	}

	private static void computeClassMatches(File classMatchesFile, JarFile sourceJar, JarFile destJar, Mappings mappings)
		throws IOException {
		ClassMatches classMatches = MappingsConverter.computeClassMatches(sourceJar, destJar, mappings);
		MatchesWriter.writeClasses(classMatches, classMatchesFile);
		System.out.println("Wrote:\n\t" + classMatchesFile.getAbsolutePath());
	}

	private static void editClasssMatches(final File classMatchesFile, JarFile sourceJar, JarFile destJar, Mappings mappings)
		throws IOException {
		System.out.println("Reading class matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
		deobfuscators.source.setMappings(mappings);
		System.out.println("Starting GUI...");
		new ClassMatchingGui(classMatches, deobfuscators.source, deobfuscators.dest).setSaveListener(matches ->
		{
			try {
				MatchesWriter.writeClasses(matches, classMatchesFile);
			} catch (IOException ex) {
				throw new Error(ex);
			}
		});
	}

	@SuppressWarnings("unused")
	private static void convertMappings(File outMappingsFile, JarFile sourceJar, JarFile destJar, Mappings mappings, File classMatchesFile)
		throws IOException, MappingConflict {
		System.out.println("Reading class matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
		deobfuscators.source.setMappings(mappings);

		Mappings newMappings = MappingsConverter.newMappings(classMatches, mappings, deobfuscators.source, deobfuscators.dest);
		new MappingsEnigmaWriter().write(outMappingsFile, newMappings, true);
		System.out.println("Write converted mappings to: " + outMappingsFile.getAbsolutePath());
	}

	private static void computeFieldMatches(File memberMatchesFile, JarFile destJar, File destMappingsFile, File classMatchesFile)
		throws IOException, MappingParseException {

		System.out.println("Reading class matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		System.out.println("Reading mappings...");
		Mappings destMappings = new MappingsEnigmaReader().read(destMappingsFile);
		System.out.println("Indexing dest jar...");
		Deobfuscator destDeobfuscator = new Deobfuscator(destJar);

		System.out.println("Writing matches...");

		// get the matched and unmatched mappings
		MemberMatches<FieldEntry> fieldMatches = MappingsConverter.computeMemberMatches(
			destDeobfuscator,
			destMappings,
			classMatches,
			MappingsConverter.getFieldDoer()
		);

		MatchesWriter.writeMembers(fieldMatches, memberMatchesFile);
		System.out.println("Wrote:\n\t" + memberMatchesFile.getAbsolutePath());
	}

	private static void editFieldMatches(JarFile sourceJar, JarFile destJar, File destMappingsFile, Mappings sourceMappings, File classMatchesFile, final File fieldMatchesFile)
		throws IOException, MappingParseException {

		System.out.println("Reading matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		MemberMatches<FieldEntry> fieldMatches = MatchesReader.readMembers(fieldMatchesFile);

		// prep deobfuscators
		Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
		deobfuscators.source.setMappings(sourceMappings);
		Mappings destMappings = new MappingsEnigmaReader().read(destMappingsFile);
		MappingsChecker checker = new MappingsChecker(deobfuscators.dest.getJarIndex());
		checker.dropBrokenMappings(destMappings);
		deobfuscators.dest.setMappings(destMappings);

		new MemberMatchingGui<>(classMatches, fieldMatches, deobfuscators.source, deobfuscators.dest).setSaveListener(
			matches ->
			{
				try {
					MatchesWriter.writeMembers(matches, fieldMatchesFile);
				} catch (IOException ex) {
					throw new Error(ex);
				}
			});
	}

	@SuppressWarnings("unused")
	private static void convertMappings(File outMappingsFile, JarFile sourceJar, JarFile destJar, Mappings mappings, File classMatchesFile, File fieldMatchesFile)
		throws IOException, MappingConflict {

		System.out.println("Reading matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		MemberMatches<FieldEntry> fieldMatches = MatchesReader.readMembers(fieldMatchesFile);

		Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
		deobfuscators.source.setMappings(mappings);

		// apply matches
		Mappings newMappings = MappingsConverter.newMappings(classMatches, mappings, deobfuscators.source, deobfuscators.dest);
		MappingsConverter.applyMemberMatches(newMappings, classMatches, fieldMatches, MappingsConverter.getFieldDoer());

		// write out the converted mappings

		new MappingsEnigmaWriter().write(outMappingsFile, newMappings, true);
		System.out.println("Wrote converted mappings to:\n\t" + outMappingsFile.getAbsolutePath());
	}

	private static void computeMethodMatches(File outMappingsFile, JarFile sourceJar, JarFile destJar, Mappings sourceMappings, File classMatchesFile, File fieldMatchesFile, File methodMatchesFile)
		throws IOException, MappingParseException {

		System.out.println("Reading class matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		System.out.println("Reading dest mappings...");
		Mappings destMappings = new MappingsEnigmaReader().read(outMappingsFile);
		System.out.println("Indexing dest jar...");
		Deobfuscator destDeobfuscator = new Deobfuscator(destJar);
		System.out.println("Indexing source jar...");
		Deobfuscator sourceDeobfuscator = new Deobfuscator(sourceJar);

		System.out.println("Writing method matches...");

		// get the matched and unmatched mappings
		MemberMatches<BehaviorEntry> methodMatches = MappingsConverter.computeMethodsMatches(
			destDeobfuscator,
			destMappings,
			sourceDeobfuscator,
			sourceMappings,
			classMatches,
			MappingsConverter.getMethodDoer()
		);

		MatchesWriter.writeMembers(methodMatches, methodMatchesFile);
		System.out.println("Wrote:\n\t" + methodMatchesFile.getAbsolutePath());
	}

	private static void editMethodMatches(JarFile sourceJar, JarFile destJar, File destMappingsFile, Mappings sourceMappings, File classMatchesFile, final File methodMatchesFile)
		throws IOException, MappingParseException {

		System.out.println("Reading matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		MemberMatches<BehaviorEntry> methodMatches = MatchesReader.readMembers(methodMatchesFile);

		// prep deobfuscators
		Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
		deobfuscators.source.setMappings(sourceMappings);
		Mappings destMappings = new MappingsEnigmaReader().read(destMappingsFile);
		MappingsChecker checker = new MappingsChecker(deobfuscators.dest.getJarIndex());
		checker.dropBrokenMappings(destMappings);
		deobfuscators.dest.setMappings(destMappings);

		new MemberMatchingGui<>(classMatches, methodMatches, deobfuscators.source, deobfuscators.dest).setSaveListener(
			matches ->
			{
				try {
					MatchesWriter.writeMembers(matches, methodMatchesFile);
				} catch (IOException ex) {
					throw new Error(ex);
				}
			});
	}

	private static void convertMappings(File outMappingsFile, JarFile sourceJar, JarFile destJar, Mappings mappings, File classMatchesFile, File fieldMatchesFile, File methodMatchesFile)
		throws IOException, MappingConflict {

		System.out.println("Reading matches...");
		ClassMatches classMatches = MatchesReader.readClasses(classMatchesFile);
		MemberMatches<FieldEntry> fieldMatches = MatchesReader.readMembers(fieldMatchesFile);
		MemberMatches<BehaviorEntry> methodMatches = MatchesReader.readMembers(methodMatchesFile);

		Deobfuscators deobfuscators = new Deobfuscators(sourceJar, destJar);
		deobfuscators.source.setMappings(mappings);

		// apply matches
		Mappings newMappings = MappingsConverter.newMappings(classMatches, mappings, deobfuscators.source, deobfuscators.dest);
		MappingsConverter.applyMemberMatches(newMappings, classMatches, fieldMatches, MappingsConverter.getFieldDoer());
		MappingsConverter.applyMemberMatches(newMappings, classMatches, methodMatches, MappingsConverter.getMethodDoer());

		// check the final mappings
		MappingsChecker checker = new MappingsChecker(deobfuscators.dest.getJarIndex());
		checker.dropBrokenMappings(newMappings);

		for (java.util.Map.Entry<ClassEntry, ClassMapping> mapping : checker.getDroppedClassMappings().entrySet()) {
			System.out.println("WARNING: Broken class entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
		}
		for (java.util.Map.Entry<ClassEntry, ClassMapping> mapping : checker.getDroppedInnerClassMappings().entrySet()) {
			System.out.println("WARNING: Broken inner class entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
		}
		for (java.util.Map.Entry<FieldEntry, FieldMapping> mapping : checker.getDroppedFieldMappings().entrySet()) {
			System.out.println("WARNING: Broken field entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
		}
		for (java.util.Map.Entry<BehaviorEntry, MethodMapping> mapping : checker.getDroppedMethodMappings().entrySet()) {
			System.out.println("WARNING: Broken behavior entry " + mapping.getKey() + " (" + mapping.getValue().getDeobfName() + ")");
		}

		// write out the converted mappings
		new MappingsEnigmaWriter().write(outMappingsFile, newMappings, true);
		System.out.println("Wrote converted mappings to:\n\t" + outMappingsFile.getAbsolutePath());
	}

	private static class Deobfuscators {

		public Deobfuscator source;
		public Deobfuscator dest;

		public Deobfuscators(JarFile sourceJar, JarFile destJar) {
			System.out.println("Indexing source jar...");
			IndexerThread sourceIndexer = new IndexerThread(sourceJar);
			sourceIndexer.start();
			System.out.println("Indexing dest jar...");
			IndexerThread destIndexer = new IndexerThread(destJar);
			destIndexer.start();
			sourceIndexer.joinOrBail();
			destIndexer.joinOrBail();
			source = sourceIndexer.deobfuscator;
			dest = destIndexer.deobfuscator;
		}
	}

	private static class IndexerThread extends Thread {

		public Deobfuscator deobfuscator;
		private JarFile jarFile;

		public IndexerThread(JarFile jarFile) {
			this.jarFile = jarFile;
			deobfuscator = null;
		}

		public void joinOrBail() {
			try {
				join();
			} catch (InterruptedException ex) {
				throw new Error(ex);
			}
		}

		@Override
		public void run() {
			deobfuscator = new Deobfuscator(jarFile);
		}
	}
}