/*******************************************************************************
 * 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.mapping;

import com.google.gson.Gson;
import com.google.gson.GsonBuilder;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import cuchaz.enigma.json.*;

public class MappingsWriter {


    public void write(File file, Mappings mappings) throws IOException {
        if (!file.isDirectory()) {
            //TODO Error
        }

        Gson gson = new GsonBuilder().setPrettyPrinting().create();
        for (ClassMapping classMapping : sorted(mappings.classes())) {
            JsonClass jsonClass = new JsonClass(classMapping.getObfSimpleName(), classMapping.getDeobfName());
            write(jsonClass, classMapping);

            File f = new File(file, jsonClass.getName() + ".json");
            f.getParentFile().mkdirs();
            f.createNewFile();
            FileWriter writer = new FileWriter(f);
            writer.write(gson.toJson(jsonClass));
            writer.close();
        }
    }

    private void write(JsonClass jsonClass, ClassMapping classMapping) throws IOException {
        if (classMapping.getDeobfName() != null && !classMapping.getDeobfName().equalsIgnoreCase("") && !classMapping.getDeobfName().equalsIgnoreCase("null")) {

            for (ClassMapping innerClassMapping : sorted(classMapping.innerClasses())) {
                JsonClass innerClass = new JsonClass(classMapping.getObfSimpleName() + "$" + innerClassMapping.getObfSimpleName().replace("nome/", ""), innerClassMapping.getDeobfName());
                write(innerClass, innerClassMapping);
                jsonClass.addInnerClass(innerClass);
            }

            for (FieldMapping fieldMapping : sorted(classMapping.fields())) {
                jsonClass.addField(new JsonField(fieldMapping.getObfName(), fieldMapping.getDeobfName(), fieldMapping.getObfType().toString()));
            }

            for (MethodMapping methodMapping : sorted(classMapping.methods())) {
                List<JsonArgument> args = new ArrayList<>();
                for (ArgumentMapping argumentMapping : sorted(methodMapping.arguments())) {
                    args.add(new JsonArgument(argumentMapping.getIndex(), argumentMapping.getName()));
                }
                if (methodMapping.getObfName().contains("<init>") || methodMapping.getObfName().contains("<clinit>")) {
                    jsonClass.addConstructor(new JsonConstructor(methodMapping.getObfSignature().toString(), args, methodMapping.getObfName().contains("<clinit>")));
                } else {
                    jsonClass.addMethod(new JsonMethod(methodMapping.getObfName(), methodMapping.getDeobfName(), methodMapping.getObfSignature().toString(), args));
                }
            }
        }
    }

    private <T extends Comparable<T>> List<T> sorted(Iterable<T> classes) {
        List<T> out = new ArrayList<T>();
        for (T t : classes) {
            out.add(t);
        }
        Collections.sort(out);
        return out;
    }
}