package com.example.codegen.codeGenerator;

import org.eclipse.emf.codegen.ecore.generator.Generator;
import org.eclipse.emf.codegen.ecore.genmodel.GenModel;
import org.eclipse.emf.codegen.ecore.genmodel.GenModelFactory;
import org.eclipse.emf.codegen.ecore.genmodel.GenModelPackage;
import org.eclipse.emf.codegen.ecore.genmodel.generator.GenBaseGeneratorAdapter;
import org.eclipse.emf.codegen.ecore.genmodel.generator.GenModelGeneratorAdapterFactory;
import org.eclipse.emf.common.util.BasicMonitor;
import org.eclipse.emf.common.util.URI;
import org.eclipse.emf.ecore.EPackage;
import org.eclipse.emf.ecore.resource.Resource;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.ecore.resource.impl.ResourceSetImpl;
import org.eclipse.emf.ecore.xmi.impl.EcoreResourceFactoryImpl;
import org.eclipse.emf.ecore.xmi.impl.XMLResourceFactoryImpl;

import java.io.File;
import java.io.IOException;
import java.util.Collections;


public class CodeGenerator {

    public static void main(String[] args) {

        String ecorePath = "/Users/adielt./Documents/demo/src/main/resources/model/Family.ecore";
        String genModelPath = "/Users/adielt./Documents/demo/src/main/resources/codegen/Family.genmodel";
        String outputDirectory = "/Users/adielt./Documents/demo/src/main/java";
        String modelName = "Family";

        // Verify and load the .ecore file
        ResourceSet resourceSet = createResourceSet();
        Resource ecoreResource = loadEcoreModel(resourceSet, ecorePath);
        if (ecoreResource == null) return;

        // Create and configure the GenModel
        GenModel genModel = createGenModel(ecoreResource, outputDirectory, modelName);
        saveGenModel(resourceSet, genModel, genModelPath);

        // Load the saved GenModel
        GenModel loadedGenModel = loadGenModel(resourceSet, genModelPath);
        if (loadedGenModel == null) return;

        // Generate the code
        generateCode(loadedGenModel, outputDirectory);
    }

    private static ResourceSet createResourceSet() {
        ResourceSet resourceSet = new ResourceSetImpl();
        resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put("ecore", new EcoreResourceFactoryImpl());
        resourceSet.getResourceFactoryRegistry().getExtensionToFactoryMap().put("genmodel", new XMLResourceFactoryImpl());
        return resourceSet;
    }

    private static Resource loadEcoreModel(ResourceSet resourceSet, String ecorePath) {
        System.out.println("Loading the Ecore file...");
        URI ecoreURI = URI.createFileURI(ecorePath);
        Resource ecoreResource = resourceSet.getResource(ecoreURI, true);
        try {
            ecoreResource.load(null);
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }

        if (ecoreResource.getContents().isEmpty()) {
            System.err.println("Error: The Ecore file does not contain any packages.");
            return null;
        }

        System.out.println("Ecore file loaded successfully.");
        return ecoreResource;
    }

    private static GenModel createGenModel(Resource ecoreResource, String outputDirectory, String modelName) {
        EPackage ePackage = (EPackage) ecoreResource.getContents().get(0);
        GenModel genModel = GenModelFactory.eINSTANCE.createGenModel();
        genModel.initialize(Collections.singleton(ePackage));
        genModel.setModelName(modelName);
        genModel.setModelDirectory(outputDirectory.replace("\\", "/"));
        genModel.setCanGenerate(true);
        genModel.setValidateModel(true);

        System.out.println("GenModel configuration completed.");
        return genModel;
    }

    private static void saveGenModel(ResourceSet resourceSet, GenModel genModel, String genModelPath) {
        URI genModelURI = URI.createFileURI(genModelPath);
        Resource genModelResource = resourceSet.createResource(genModelURI);
        genModelResource.getContents().add(genModel);
        try {
            genModelResource.save(null);
        } catch (IOException e) {
            e.printStackTrace();
        }
        System.out.println("GenModel saved successfully.");
    }

    private static GenModel loadGenModel(ResourceSet resourceSet, String genModelPath) {
        System.out.println("Loading the saved GenModel...");
        URI genModelURI = URI.createFileURI(genModelPath);
        Resource genModelResource = resourceSet.getResource(genModelURI, true);
        try {
            genModelResource.load(null);
        } catch (IOException e) {
            e.printStackTrace();
            return null;
        }

        if (genModelResource.getContents().isEmpty()) {
            System.err.println("Error: The saved GenModel does not contain any data.");
            return null;
        }

        System.out.println("Saved GenModel loaded successfully.");
        return (GenModel) genModelResource.getContents().get(0);
    }

    private static void generateCode(GenModel genModel, String outputDirectory) {
        genModel.reconcile();
        genModel.setCanGenerate(true);
        genModel.setValidateModel(true);
        genModel.setForceOverwrite(true);

        Generator generator = new Generator();
        generator.setInput(genModel);
        generator.getAdapterFactoryDescriptorRegistry().addDescriptor(GenModelPackage.eNS_URI, GenModelGeneratorAdapterFactory.DESCRIPTOR);

        // Generate the MODEL_PROJECT_TYPE code
        System.out.println("Starting MODEL_PROJECT_TYPE generation...");
        generator.generate(genModel, GenBaseGeneratorAdapter.MODEL_PROJECT_TYPE, new BasicMonitor.Printing(System.out));
        System.out.println("MODEL_PROJECT_TYPE generation completed.");

        // Verify if the files have been generated
        File generatedDir = new File(outputDirectory);
        File[] generatedFiles = generatedDir.listFiles();
        if (generatedFiles != null && generatedFiles.length > 0) {
            System.out.println("Java code generated successfully in the directory: " + outputDirectory);
            for (File file : generatedFiles) {
                System.out.println("Generated: " + file.getAbsolutePath());
            }
        } else {
            System.err.println("Error: No files generated in the output directory.");
        }
    }

    public static String getFileExtension(String fullName) {
        String fileName = new File(fullName).getName();
        int dotIndex = fileName.lastIndexOf('.');
        return (dotIndex == -1) ? "" : fileName.substring(dotIndex + 1);
    }
}

private static void setGenPackageBasePackage(GenModel genModel, String basePackage) {
    for (GenPackage genPackage : genModel.getGenPackages()) {
        genPackage.setBasePackage(basePackage);
    }
}