/*

 * Copyright (c) 2015, 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.

 */

import java.io.IOException;

import java.io.OutputStream;

import java.io.PrintWriter;

import java.nio.file.Paths;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.HashMap;

import java.util.HashSet;

import java.util.LinkedHashSet;

import java.util.List;

import java.util.Locale;

import java.util.Map;

import java.util.Set;

import java.util.function.Supplier;

import java.util.stream.Collectors;

import java.util.stream.Stream;

/**

 * @test

 * @bug 8080608

 * @summary Test that jdeps verbose output has a summary line when dependencies

 *          are found within the same archive. For each testcase, compare the

 *          result obtained from jdeps with the expected result.

 * @build use.indirect.DontUseUnsafe2

 * @build use.indirect.UseUnsafeIndirectly

 * @build use.indirect2.DontUseUnsafe3

 * @build use.indirect2.UseUnsafeIndirectly2

 * @build use.unsafe.DontUseUnsafe

 * @build use.unsafe.UseClassWithUnsafe

 * @build use.unsafe.UseUnsafeClass

 * @build use.unsafe.UseUnsafeClass2

 * @run main JdepsDependencyClosure --test:0

 * @run main JdepsDependencyClosure --test:1

 * @run main JdepsDependencyClosure --test:2

 * @run main JdepsDependencyClosure --test:3

 */

public class JdepsDependencyClosure {

    static boolean VERBOSE = false;

    static boolean COMPARE_TEXT = true;

    static final String JDEPS_SUMMARY_TEXT_FORMAT = "%s -> %s%n";

    static final String JDEPS_VERBOSE_TEXT_FORMAT = "   %-50s -> %-50s %s%n";

    /**

     * Helper class used to store arguments to pass to

     * {@code JdepsDependencyClosure.test} as well as expected

     * results.

     */

    static class TestCaseData {

        final Map<String, Set<String>> expectedDependencies;

        final String expectedText;

        final String[] args;

        final boolean closure;

        TestCaseData(Map<String, Set<String>> expectedDependencies,

                        String expectedText,

                        boolean closure,

                        String[] args) {

            this.expectedDependencies = expectedDependencies;

            this.expectedText = expectedText;

            this.closure = closure;

            this.args = args;

        }

        public void test() {

            if (expectedDependencies != null) {

                String format = closure

                        ? "Running (closure): jdeps %s %s %s %s"

                        : "Running: jdeps %s %s %s %s";

                System.out.println(String.format(format, (Object[])args));

            }

            JdepsDependencyClosure.test(args, expectedDependencies, expectedText, closure);

        }

        /**

         * Make a new test case data to invoke jdeps and test its output.

         * @param pattern The pattern that will passed through to jdeps -e

         *                This is expected to match only one class.

         * @param arcPath The archive to analyze. A jar or a class directory.

         * @param classes For each reported archive dependency couple, the

         *                expected list of classes in the source that will

         *                be reported as having a dependency on the class

         *                in the target that matches the given pattern.

         * @param dependencies For each archive dependency couple, a singleton list

         *                containing the name of the class in the target that

         *                matches the pattern. It is expected that the pattern

         *                will match only one class in the target.

         *                If the pattern matches several classes the

         *                expected text may no longer match the jdeps output.

         * @param archives A list of archive dependency couple in the form

         *               {{sourceName1, sourcePath1, targetDescription1, targetPath1}

         *                {sourceName2, sourcePath2, targetDescription2, targetPath2}

         *                ... }

         *               For a JDK module - e.g. java.base, the targetDescription

         *               is usually something like "JDK internal API (java.base)"

         *               and the targetPath is usually the module name "java.base".

         * @param closure Whether jdeps should be recursively invoked to build

         *                the closure.

         * @return An instance of TestCaseData containing all the information

         *         needed to perform the jdeps invokation and test its output.

         */

        public static TestCaseData make(String pattern, String arcPath, String[][] classes,

                String[][] dependencies, String[][] archives, boolean closure) {

            final String[] args = new String[] {

                "-e", pattern, "-v", arcPath

            };

            Map<String, Set<String>> expected = new HashMap<>();

            String expectedText = "";

            for (int i=0; i<classes.length; i++) {

                final int index = i;

                expectedText += Stream.of(classes[i])

                    .map((cn) -> String.format(JDEPS_VERBOSE_TEXT_FORMAT, cn,

                            dependencies[index][0], archives[index][2]))

                    .reduce(String.format(JDEPS_SUMMARY_TEXT_FORMAT, archives[i][0],

                            archives[index][3]), (s1,s2) -> s1.concat(s2));

                for (String cn : classes[index]) {

                    expected.putIfAbsent(cn, new HashSet<>());

                    expected.get(cn).add(dependencies[index][0]);

                }

            }

            return new TestCaseData(expected, expectedText, closure, args);

        }

        public static TestCaseData valueOf(String[] args) {

            if (args.length == 1 && args[0].startsWith("--test:")) {

                // invoked from jtreg. build test case data for selected test.

                int index = Integer.parseInt(args[0].substring("--test:".length()));

                if (index >= dataSuppliers.size()) {

                    throw new RuntimeException("No such test case: " + index

                            + " - available testcases are [0.."

                            + (dataSuppliers.size()-1) + "]");

                }

                return dataSuppliers.get(index).get();

            } else {

                // invoked in standalone. just take the given argument

                // and perform no validation on the output (except that it

                // must start with a summary line)

                return new TestCaseData(null, null, true, args);

            }

        }

    }

    static TestCaseData makeTestCaseOne() {

        final String arcPath = System.getProperty("test.classes", "build/classes");

        final String arcName = Paths.get(arcPath).getFileName().toString();

        final String[][] classes = new String[][] {

            {"use.indirect2.UseUnsafeIndirectly2", "use.unsafe.UseClassWithUnsafe"},

        };

        final String[][] dependencies = new String[][] {

            {"use.unsafe.UseUnsafeClass"},

        };

        final String[][] archives = new String[][] {

            {arcName, arcPath, arcName, arcPath},

        };

        return TestCaseData.make("use.unsafe.UseUnsafeClass", arcPath, classes,

                dependencies, archives, false);

    }

    static TestCaseData makeTestCaseTwo() {

        String arcPath = System.getProperty("test.classes", "build/classes");

        String arcName = Paths.get(arcPath).getFileName().toString();

        String[][] classes = new String[][] {

            {"use.unsafe.UseUnsafeClass", "use.unsafe.UseUnsafeClass2"}

        };

        String[][] dependencies = new String[][] {

            {"sun.misc.Unsafe"}

        };

        String[][] archive = new String[][] {

            {arcName, arcPath, "JDK internal API (java.base)", "java.base"},

        };

        return TestCaseData.make("sun.misc.Unsafe", arcPath, classes,

                dependencies, archive, false);

    }

    static TestCaseData makeTestCaseThree() {

        final String arcPath = System.getProperty("test.classes", "build/classes");

        final String arcName = Paths.get(arcPath).getFileName().toString();

        final String[][] classes = new String[][] {

            {"use.indirect2.UseUnsafeIndirectly2", "use.unsafe.UseClassWithUnsafe"},

            {"use.indirect.UseUnsafeIndirectly"}

        };

        final String[][] dependencies = new String[][] {

            {"use.unsafe.UseUnsafeClass"},

            {"use.unsafe.UseClassWithUnsafe"}

        };

        final String[][] archives = new String[][] {

            {arcName, arcPath, arcName, arcPath},

            {arcName, arcPath, arcName, arcPath}

        };

        return TestCaseData.make("use.unsafe.UseUnsafeClass", arcPath, classes,

                dependencies, archives, true);

    }

    static TestCaseData makeTestCaseFour() {

        final String arcPath = System.getProperty("test.classes", "build/classes");

        final String arcName = Paths.get(arcPath).getFileName().toString();

        final String[][] classes = new String[][] {

            {"use.unsafe.UseUnsafeClass", "use.unsafe.UseUnsafeClass2"},

            {"use.indirect2.UseUnsafeIndirectly2", "use.unsafe.UseClassWithUnsafe"},

            {"use.indirect.UseUnsafeIndirectly"}

        };

        final String[][] dependencies = new String[][] {

            {"sun.misc.Unsafe"},

            {"use.unsafe.UseUnsafeClass"},

            {"use.unsafe.UseClassWithUnsafe"}

        };

        final String[][] archives = new String[][] {

            {arcName, arcPath, "JDK internal API (java.base)", "java.base"},

            {arcName, arcPath, arcName, arcPath},

            {arcName, arcPath, arcName, arcPath}

        };

        return TestCaseData.make("sun.misc.Unsafe", arcPath, classes, dependencies,

                archives, true);

    }

    static final List<Supplier<TestCaseData>> dataSuppliers = Arrays.asList(

        JdepsDependencyClosure::makeTestCaseOne,

        JdepsDependencyClosure::makeTestCaseTwo,

        JdepsDependencyClosure::makeTestCaseThree,

        JdepsDependencyClosure::makeTestCaseFour

    );

    /**

     * The OutputStreamParser is used to parse the format of jdeps.

     * It is thus dependent on that format.

     */

    static class OutputStreamParser extends OutputStream {

        // OutputStreamParser will populate this map:

        //

        // For each archive, a list of class in where dependencies where

        //     found...

        final Map<String, Set<String>> deps;

        final StringBuilder text = new StringBuilder();

        StringBuilder[] lines = { new StringBuilder(), new StringBuilder() };

        int line = 0;

        int sepi = 0;

        char[] sep;

        public OutputStreamParser(Map<String, Set<String>> deps) {

            this.deps = deps;

            this.sep = System.getProperty("line.separator").toCharArray();

        }

        @Override

        public void write(int b) throws IOException {

            lines[line].append((char)b);

            if (b == sep[sepi]) {

                if (++sepi == sep.length) {

                    text.append(lines[line]);

                    if (lines[0].toString().startsWith("  ")) {

                        throw new RuntimeException("Bad formatting: "

                                + "summary line missing for\n"+lines[0]);

                    }

                    // Usually the output looks like that:

                    // <archive-1> -> java.base

                    //   <class-1>      -> <dependency> <dependency description>

                    //   <class-2>      -> <dependency> <dependency description>

                    //   ...

                    // <archive-2> -> java.base

                    //   <class-3>      -> <dependency> <dependency description>

                    //   <class-4>      -> <dependency> <dependency description>

                    //   ...

                    //

                    // We want to keep the <archive> line in lines[0]

                    // and have the ith <class-i> line in lines[1]

                    if (line == 1) {

                        // we have either a <class> line or an <archive> line.

                        String line1 = lines[0].toString();

                        String line2 = lines[1].toString();

                        if (line2.startsWith("  ")) {

                            // we have a class line, record it.

                            parse(line1, line2);

                            // prepare for next <class> line.

                            lines[1] = new StringBuilder();

                        } else {

                            // We have an archive line: We are switching to the next archive.

                            // put the new <archive> line in lines[0], and prepare

                            // for reading the next <class> line

                            lines[0] = lines[1];

                            lines[1] = new StringBuilder();

                         }

                    } else {

                        // we just read the first <archive> line.

                        // prepare to read <class> lines.

                        line = 1;

                    }

                    sepi = 0;

                }

            } else {

                sepi = 0;

            }

        }

        // Takes a couple of lines, where line1 is an <archive> line and

        // line 2 is a <class> line. Parses the line to extract the archive

        // name and dependent class name, and record them in the map...

        void parse(String line1, String line2) {

            String archive = line1.substring(0, line1.indexOf(" -> "));

            int l2ArrowIndex = line2.indexOf(" -> ");

            String className = line2.substring(2, l2ArrowIndex).replace(" ", "");

            String depdescr = line2.substring(l2ArrowIndex + 4);

            String depclass = depdescr.substring(0, depdescr.indexOf(" "));

            deps.computeIfAbsent(archive, (k) -> new HashSet<>());

            deps.get(archive).add(className);

            if (VERBOSE) {

                System.out.println(archive+": "+className+" depends on "+depclass);

            }

        }

    }

    /**

     * The main method.

     *

     * Can be run in two modes:

     * <ul>

     * <li>From jtreg: expects 1 argument in the form {@code --test:<test-nb>}</li>

     * <li>From command line: expected syntax is {@code -e <pattern> -v jar [jars..]}</li>

     * </ul>

     * <p>When called from the command line this method will call jdeps recursively

     * to build a closure of the dependencies on {@code <pattern>} and print a summary.

     * <p>When called from jtreg - it will call jdeps either once only or

     * recursively depending on the pattern.

     * @param args either {@code --test:<test-nb>} or {@code -e <pattern> -v jar [jars..]}.

     */

    public static void main(String[] args) {

        runWithLocale(Locale.ENGLISH, TestCaseData.valueOf(args)::test);

    }

    private static void runWithLocale(Locale loc, Runnable run) {

        final Locale defaultLocale = Locale.getDefault();

        Locale.setDefault(loc);

        try {

            run.run();

        } finally {

            Locale.setDefault(defaultLocale);

        }

    }

    public static void test(String[] args, Map<String, Set<String>> expected,

            String expectedText, boolean closure) {

        try {

            doTest(args, expected, expectedText, closure);

        } catch (Throwable t) {

            try {

                printDiagnostic(args, expectedText, t, closure);

            } catch(Throwable tt) {

                throw t;

            }

            throw t;

        }

    }

    static class TextFormatException extends RuntimeException {

        final String expected;

        final String actual;

        TextFormatException(String message, String expected, String actual) {

            super(message);

            this.expected = expected;

            this.actual = actual;

        }

    }

    public static void printDiagnostic(String[] args, String expectedText,

            Throwable t, boolean closure) {

        if (expectedText != null || t instanceof TextFormatException) {

            System.err.println("=====   TEST FAILED   =======");

            System.err.println("command: " + Stream.of(args)

                    .reduce("jdeps", (s1,s2) -> s1.concat(" ").concat(s2)));

            System.err.println("===== Expected Output =======");

            System.err.append(expectedText);

            System.err.println("===== Command  Output =======");

            if (t instanceof TextFormatException) {

                System.err.print(((TextFormatException)t).actual);

            } else {

                com.sun.tools.jdeps.Main.run(args, new PrintWriter(System.err));

                if (closure) System.err.println("... (closure not available) ...");

            }

            System.err.println("=============================");

        }

    }

    public static void doTest(String[] args, Map<String, Set<String>> expected,

            String expectedText, boolean closure) {

        if (args.length < 3 || !"-e".equals(args[0]) || !"-v".equals(args[2])) {

            System.err.println("Syntax: -e <classname> -v [list of jars or directories]");

            return;

        }

        Map<String, Map<String, Set<String>>> alldeps = new HashMap<>();

        String depName = args[1];

        List<String> search = new ArrayList<>();

        search.add(depName);

        Set<String> searched = new LinkedHashSet<>();

        StringBuilder text = new StringBuilder();

        while(!search.isEmpty()) {

            args[1] = search.remove(0);

            if (VERBOSE) {

                System.out.println("Looking for " + args[1]);

            }

            searched.add(args[1]);

            Map<String, Set<String>> deps =

                    alldeps.computeIfAbsent(args[1], (k) -> new HashMap<>());

            OutputStreamParser parser = new OutputStreamParser(deps);

            PrintWriter writer = new PrintWriter(parser);

            com.sun.tools.jdeps.Main.run(args, writer);

            if (VERBOSE) {

                System.out.println("Found: " + deps.values().stream()

                        .flatMap(s -> s.stream()).collect(Collectors.toSet()));

            }

            if (expectedText != null) {

                text.append(parser.text.toString());

            }

            search.addAll(deps.values().stream()

                    .flatMap(s -> s.stream())

                    .filter(k -> !searched.contains(k))

                    .collect(Collectors.toSet()));

            if (!closure) break;

        }

        // Print summary...

        final Set<String> classes = alldeps.values().stream()

                .flatMap((m) -> m.values().stream())

                .flatMap(s -> s.stream()).collect(Collectors.toSet());

        Map<String, Set<String>> result = new HashMap<>();

        for (String c : classes) {

            Set<String> archives = new HashSet<>();

            Set<String> dependencies = new HashSet<>();

            for (String d : alldeps.keySet()) {

                Map<String, Set<String>> m = alldeps.get(d);

                for (String a : m.keySet()) {

                    Set<String> s = m.get(a);

                    if (s.contains(c)) {

                        archives.add(a);

                        dependencies.add(d);

                    }

                }

            }

            result.put(c, dependencies);

            System.out.println(c + " " + archives + " depends on " + dependencies);

        }

        // If we're in jtreg, then check result (expectedText != null)

        if (expectedText != null && COMPARE_TEXT) {

            //text.append(String.format("%n"));

            if (text.toString().equals(expectedText)) {

                System.out.println("SUCCESS - got expected text");

            } else {

                throw new TextFormatException("jdeps output is not as expected",

                                expectedText, text.toString());

            }

        }

        if (expected != null) {

            if (expected.equals(result)) {

                System.out.println("SUCCESS - found expected dependencies");

            } else if (expectedText == null) {

                throw new RuntimeException("Bad dependencies: Expected " + expected

                        + " but found " + result);

            } else {

                throw new TextFormatException("Bad dependencies: Expected "

                        + expected

                        + " but found " + result,

                        expectedText, text.toString());

            }

        }

    }

}