/*

 * Copyright (c) 2014, 2016, Oracle and/or its affiliates. All rights reserved.

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

/*

 * @test

 * @bug 8043643

 * @summary Run the langtools coding rules over the langtools source code.

 * @modules jdk.compiler/com.sun.tools.javac.util

 */

import java.io.*;

import java.lang.reflect.Method;

import java.net.URL;

import java.net.URLClassLoader;

import java.nio.file.Files;

import java.nio.file.Path;

import java.nio.file.Paths;

import java.nio.file.StandardOpenOption;

import java.util.*;

import java.util.stream.Collectors;

import java.util.stream.Stream;

import javax.tools.Diagnostic;

import javax.tools.DiagnosticListener;

import javax.tools.JavaCompiler;

import javax.tools.JavaFileObject;

import javax.tools.StandardJavaFileManager;

import javax.tools.ToolProvider;

import com.sun.tools.javac.util.Assert;

/**

 * This is a test to verify specific coding standards for source code in the langtools repository.

 *

 * As such, it is not a standard unit, regression or functional test, and will

 * automatically skip if the langtools source code is not available.

 *

 * If the source code is available, it will find and compile the coding

 * style analyzers found in langtools/make/tools/crules/*.java, and run the resulting

 * code on all source files under langtools/src/share/classes. Any coding style

 * violations will cause the test to fail.

 */

public class RunCodingRules {

    public static void main(String... args) throws Exception {

        new RunCodingRules().run();

    }

    public void run() throws Exception {

        Path testSrc = Paths.get(System.getProperty("test.src", "."));

        Path targetDir = Paths.get(".");

        List<Path> sourceDirs = null;

        Path crulesDir = null;

        Path mainSrcDir = null;

        for (Path d = testSrc; d != null; d = d.getParent()) {

            if (Files.exists(d.resolve("TEST.ROOT"))) {

                d = d.getParent();

                Path toolsPath = d.resolve("make/tools");

                if (Files.exists(toolsPath)) {

                    mainSrcDir = d.resolve("src");

                    crulesDir = toolsPath;

                    sourceDirs = Files.walk(mainSrcDir, 1)

                                      .map(p -> p.resolve("share/classes"))

                                      .filter(p -> Files.isDirectory(p))

                                      .collect(Collectors.toList());

                    break;

                }

            }

        }

        if (sourceDirs == null || crulesDir == null) {

            System.err.println("Warning: sources not found, test skipped.");

            return ;

        }

        JavaCompiler javaCompiler = ToolProvider.getSystemJavaCompiler();

        try (StandardJavaFileManager fm = javaCompiler.getStandardFileManager(null, null, null)) {

            DiagnosticListener<JavaFileObject> noErrors = diagnostic -> {

                Assert.check(diagnostic.getKind() != Diagnostic.Kind.ERROR, diagnostic.toString());

            };

            String FS = File.separator;

            String PS = File.pathSeparator;

            //compile crules:

            List<File> crulesFiles = Files.walk(crulesDir)

                                          .filter(entry -> entry.getFileName().toString().endsWith(".java"))

                                          .filter(entry -> entry.getParent().endsWith("crules"))

                                          .map(entry -> entry.toFile())

                                          .collect(Collectors.toList());

            Path crulesTarget = targetDir.resolve("crules");

            Files.createDirectories(crulesTarget);

            List<String> crulesOptions = Arrays.asList(

                    "-d", crulesTarget.toString());

            javaCompiler.getTask(null, fm, noErrors, crulesOptions, null,

                    fm.getJavaFileObjectsFromFiles(crulesFiles)).call();

            Path registration = crulesTarget.resolve("META-INF/services/com.sun.source.util.Plugin");

            Files.createDirectories(registration.getParent());

            try (Writer metaInfServices = Files.newBufferedWriter(registration, StandardOpenOption.CREATE, StandardOpenOption.TRUNCATE_EXISTING)) {

                metaInfServices.write("crules.CodingRulesAnalyzerPlugin\n");

            }

            //generate CompilerProperties.java:

            List<File> propertiesParserFiles =

                    Files.walk(crulesDir.resolve("propertiesparser"))

                         .filter(entry -> entry.getFileName().toString().endsWith(".java"))

                         .map(entry -> entry.toFile())

                         .collect(Collectors.toList());

            Path propertiesParserTarget = targetDir.resolve("propertiesParser");

            Files.createDirectories(propertiesParserTarget);

            List<String> propertiesParserOptions = Arrays.asList(

                    "-d", propertiesParserTarget.toString());

            javaCompiler.getTask(null, fm, noErrors, propertiesParserOptions, null,

                    fm.getJavaFileObjectsFromFiles(propertiesParserFiles)).call();

            Path genSrcTarget = targetDir.resolve("gensrc");

            ClassLoader propertiesParserLoader = new URLClassLoader(new URL[] {

                propertiesParserTarget.toUri().toURL(),

                crulesDir.toUri().toURL()

            });

            Class propertiesParserClass =

                    Class.forName("propertiesparser.PropertiesParser", false, propertiesParserLoader);

            Method propertiesParserRun =

                    propertiesParserClass.getDeclaredMethod("run", String[].class, PrintStream.class);

            String compilerProperties =

                    "jdk.compiler/share/classes/com/sun/tools/javac/resources/compiler.properties";

            Path propertiesPath = mainSrcDir.resolve(compilerProperties.replace("/", FS));

            Path genSrcTargetDir = genSrcTarget.resolve(mainSrcDir.relativize(propertiesPath.getParent()));

            Files.createDirectories(genSrcTargetDir);

            String[] propertiesParserRunOptions = new String[] {

                "-compile", propertiesPath.toString(), genSrcTargetDir.toString()

            };

            Object result = propertiesParserRun.invoke(null, propertiesParserRunOptions, System.err);

            if (!(result instanceof Boolean) || !(Boolean) result) {

                throw new AssertionError("Cannot parse properties: " + result);

            }

            //compile langtools sources with crules enabled:

            List<File> sources = sourceDirs.stream()

                                           .flatMap(dir -> silentFilesWalk(dir))

                                           .filter(entry -> entry.getFileName().toString().endsWith(".java"))

                                           .map(p -> p.toFile())

                                           .collect(Collectors.toList());

            Path sourceTarget = targetDir.resolve("classes");

            Files.createDirectories(sourceTarget);

            String processorPath = crulesTarget + PS + crulesDir;

            List<String> options = Arrays.asList(

                    "-d", sourceTarget.toString(),

                                       + genSrcTarget + FS + "*" + FS + "share" + FS + "classes",

                    "-XDaccessInternalAPI",

                    "-processorpath", processorPath,

                    "-Xplugin:coding_rules");

            javaCompiler.getTask(null, fm, noErrors, options, null,

                    fm.getJavaFileObjectsFromFiles(sources)).call();

        }

    }

    Stream<Path> silentFilesWalk(Path dir) throws IllegalStateException {

        try {

            return Files.walk(dir);

        } catch (IOException ex) {

            throw new IllegalStateException(ex);

        }

    }

}