/*
 * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.yangtools.yang2sources.plugin.it;

import static org.hamcrest.CoreMatchers.containsString;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.junit.Assert.fail;

import com.google.common.base.Joiner;
import com.google.common.io.Resources;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.net.URL;
import java.nio.file.Files;
import java.nio.file.Path;
import java.util.Arrays;
import java.util.Optional;
import java.util.Properties;
import org.apache.maven.it.VerificationException;
import org.apache.maven.it.Verifier;
import org.junit.BeforeClass;
import org.junit.Test;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class YangToSourcesPluginTestIT {
    private static final Logger LOG = LoggerFactory.getLogger(YangToSourcesPluginTestIT.class);
    private static final String MAVEN_OPTS = "MAVEN_OPTS";
    private static final String DESTFILE = "destfile=";

    private static String ARGLINE_PREFIX;
    private static String ARGLINE_SUFFIX;

    @BeforeClass
    public static void beforeClass() {
        final String argLine = System.getProperty("itArgPath");
        if (argLine == null || argLine.isBlank()) {
            return;
        }

        final int destFileIndex = argLine.indexOf(DESTFILE);
        if (destFileIndex == -1) {
            LOG.warn("Cannot find \"{} in \"{}\", disabling integration", DESTFILE, argLine);
            return;
        }

        ARGLINE_PREFIX = argLine.substring(0, destFileIndex + DESTFILE.length());
        final int commaIndex = argLine.indexOf(',', ARGLINE_PREFIX.length());
        ARGLINE_SUFFIX = commaIndex == -1 ? "" : argLine.substring(commaIndex);
    }

    // TODO Test yang files in transitive dependencies

    @Test
    public void testYangRootNotExist() throws Exception {
        setUp("test-parent/YangRootNotExist/", false)
            .verifyTextInLog("[WARNING] yang-to-sources: YANG source directory");
    }

    @Test
    public void testCorrect() throws VerificationException, URISyntaxException, IOException {
        verifyCorrectLog(setUp("test-parent/Correct/", false));
    }

    @Test
    public void testAdditionalConfiguration() throws Exception {
        Verifier vrf = setUp("test-parent/AdditionalConfig/", false);
        vrf.verifyTextInLog("[DEBUG] yang-to-sources: Additional configuration picked up for : "
                + "org.opendaylight.yangtools.yang2sources.spi.CodeGeneratorTestImpl: "
                + "{nm1=abcd=a.b.c.d, nm2=abcd2=a.b.c.d.2}");
        vrf.verifyTextInLog("[DEBUG] yang-to-sources: Additional configuration picked up for : "
                + "org.opendaylight.yangtools.yang2sources.spi.CodeGeneratorTestImpl: {c1=config}");
        vrf.verifyTextInLog("[DEBUG] yang-to-sources: YANG files marked as resources:");
        vrf.verifyTextInLog(Joiner.on(File.separator).join(Arrays.asList("target", "generated-sources", "spi"))
                + " marked as resources for generator: org.opendaylight.yangtools.yang2sources.spi."
                + "CodeGeneratorTestImpl");
    }

    @Test
    public void testMissingYangInDep() throws Exception {
        try {
            setUp("test-parent/MissingYangInDep/", false);
            fail("Verification exception should have been thrown");
        } catch (VerificationException e) {
            assertVerificationException(e, "Imported module [unknownDep] was not found.");
        }
    }

    static void verifyCorrectLog(final Verifier vrf) throws VerificationException {
        vrf.verifyErrorFreeLog();
        vrf.verifyTextInLog("[INFO] yang-to-sources: Project model files found: ");
        vrf.verifyTextInLog("[INFO] yang-to-sources: Code generator instantiated from "
                + "org.opendaylight.yangtools.yang2sources.spi.CodeGeneratorTestImpl");
        vrf.verifyTextInLog("[INFO] yang-to-sources: Sources generated by "
                + "org.opendaylight.yangtools.yang2sources.spi.CodeGeneratorTestImpl: 0");
    }

    @Test
    public void testNoGenerators() throws Exception {
        Verifier vrf = setUp("test-parent/NoGenerators/", false);
        vrf.verifyErrorFreeLog();
        vrf.verifyTextInLog("[WARNING] yang-to-sources: No code generators provided");
    }

    @Test
    public void testInvalidVersion() throws Exception {
        Verifier vrf = setUp("test-parent/InvalidVersion/", false);
        vrf.verifyErrorFreeLog();
        vrf.verifyTextInLog("[WARNING] yang-to-sources: Dependency resolution conflict:");
    }

    @Test
    public void testUnknownGenerator() throws Exception {
        Verifier vrf = setUp("test-parent/UnknownGenerator/", true);
        vrf.verifyTextInLog("[INFO] yang-to-sources: Code generator instantiated from "
                + "org.opendaylight.yangtools.yang2sources.spi.CodeGeneratorTestImpl");
        vrf.verifyTextInLog("on project unknown-generator: Failed to find code generator class unknown");
        vrf.verifyTextInLog("MojoFailureException: Failed to find code generator class unknown");
        vrf.verifyTextInLog("java.lang.ClassNotFoundException: unknown");
    }

    @Test
    public void testNoYangFiles() throws Exception {
        setUp("test-parent/NoYangFiles/", false).verifyTextInLog("[INFO] yang-to-sources: No input files found");
    }

    static void assertVerificationException(final VerificationException ex, final String string) {
        assertThat(ex.getMessage(), containsString(string));
    }

    static Verifier setUp(final String project, final boolean ignoreF)
            throws VerificationException, URISyntaxException, IOException {
        final URL path = YangToSourcesPluginTestIT.class.getResource("/" + project + "pom.xml");
        final File parent = new File(path.toURI()).getParentFile();
        final Verifier verifier = new Verifier(parent.toString());
        if (ignoreF) {
            verifier.addCliOption("-fn");
        }

        getEffectiveSettingsXML().ifPresent(settings -> {
            verifier.addCliOption("-gs");
            verifier.addCliOption(settings);
        });

        verifier.setForkJvm(true);
        verifier.setMavenDebug(true);

        if (ARGLINE_PREFIX != null) {
            final String argLine =
                ARGLINE_PREFIX + parent.getParent() + "/" + parent.getName() + ".exec" + ARGLINE_SUFFIX;
            final String mavenOpts = System.getenv(MAVEN_OPTS);
            final String newMavenOpts = mavenOpts == null ? argLine : mavenOpts + " " + argLine;
            LOG.debug("Adjusted {} to \"{}\"", MAVEN_OPTS, newMavenOpts);
            verifier.setEnvironmentVariable(MAVEN_OPTS, newMavenOpts);
        }

        verifier.executeGoal("generate-sources");
        return verifier;
    }

    @Test
    public void testNoOutputDir() throws Exception {
        verifyCorrectLog(YangToSourcesPluginTestIT.setUp("test-parent/NoOutputDir/", false));
    }

    @Test
    public void testFindResourceOnCp() throws Exception {
        Verifier v1 = setUp("test-parent/GenerateTest1/", false);
        v1.executeGoal("clean");
        v1.executeGoal("package");

        String buildDir = getMavenBuildDirectory(v1);
        v1.assertFilePresent(buildDir + "/classes/META-INF/yang/types1@2013-02-27.yang");
        v1.assertFilePresent(buildDir + "/classes/META-INF/yang/types2@2013-02-27.yang");
        v1.assertFilePresent(buildDir + "/classes/META-INF/yang/types3@2013-02-27.yang");

        Verifier v2 = setUp("test-parent/GenerateTest2/", false);
        v2.executeGoal("clean");
        v2.executeGoal("package");

        buildDir = getMavenBuildDirectory(v2);
        v2.assertFilePresent(buildDir + "/classes/META-INF/yang/private@2013-02-27.yang");
        v2.assertFileNotPresent(buildDir + "/classes/META-INF/yang/types1@2013-02-27.yang");
        v2.assertFileNotPresent(buildDir + "/classes/META-INF/yang/types2@2013-02-27.yang");
        v2.assertFileNotPresent(buildDir + "/classes/META-INF/yang/types3@2013-02-27.yang");
    }

    @Test
    public void testFileGenerator() throws Exception {
        Verifier v1 = setUp("test-parent/FileGenerator/", false);
        v1.executeGoal("clean");
        v1.executeGoal("package");

        String buildDir = getMavenBuildDirectory(v1);

        v1.assertFilePresent(buildDir + "/generated-sources/"
            + "org.opendaylight.yangtools.yang2sources.spi.TestFileGenerator/fooGenSource.test");
        v1.assertFilePresent(buildDir + "/generated-resources/"
            + "org.opendaylight.yangtools.yang2sources.spi.TestFileGenerator/foo-gen-resource");
        v1.assertFilePresent(buildDir + "/../src/main/java/fooSource.test");
        v1.assertFilePresent(buildDir + "/../src/main/resources/foo-resource");
    }

    private static String getMavenBuildDirectory(final Verifier verifier) throws IOException {
        final Properties sp = new Properties();
        final Path path = new File(verifier.getBasedir() + "/it-project.properties").toPath();
        try (InputStream is = Files.newInputStream(path)) {
            sp.load(is);
        }
        return sp.getProperty("target.dir");
    }

    private static Optional<String> getEffectiveSettingsXML() throws URISyntaxException, VerificationException,
            IOException {
        final URL path = Resources.getResource(YangToSourcesPluginTestIT.class, "/test-parent/pom.xml");
        final File buildDir = new File(path.toURI()).getParentFile().getParentFile().getParentFile();
        final File effectiveSettingsXML = new File(buildDir, "effective-settings.xml");
        if (effectiveSettingsXML.exists()) {
            return Optional.of(effectiveSettingsXML.getAbsolutePath());
        }

        fail(effectiveSettingsXML.getAbsolutePath());
        return Optional.empty();
    }
}