/*

 * Copyright (c) 2002, 2017, 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.BufferedWriter;

import java.io.ByteArrayOutputStream;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileWriter;

import java.io.FilenameFilter;

import java.io.IOException;

import java.io.PrintStream;

import java.io.PrintWriter;

import java.io.StringWriter;

import java.lang.annotation.Annotation;

import java.lang.annotation.Retention;

import java.lang.annotation.RetentionPolicy;

import java.lang.ref.SoftReference;

import java.lang.reflect.InvocationTargetException;

import java.lang.reflect.Method;

import java.nio.charset.Charset;

import java.nio.charset.UnsupportedCharsetException;

import java.nio.file.Files;

import java.util.Arrays;

import java.util.ArrayList;

import java.util.Collection;

import java.util.Collections;

import java.util.EnumMap;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import java.util.Objects;

import java.util.function.Function;

/**

 * Test framework for running javadoc and performing tests on the resulting output.

 *

 * <p>

 * Tests are typically written as subtypes of JavadocTester, with a main

 * method that creates an instance of the test class and calls the runTests()

 * method. The runTests() methods calls all the test methods declared in the class,

 * and then calls a method to print a summary, and throw an exception if

 * any of the test methods reported a failure.

 *

 * <p>

 * Test methods are identified with a @Test annotation. They have no parameters.

 * The name of the method is not important, but if you have more than one, it is

 * recommended that the names be meaningful and suggestive of the test case

 * contained therein.

 *

 * <p>

 * Typically, a test method will invoke javadoc, and then perform various

 * checks on the results. The standard checks are:

 *

 * <dl>

 * <dt>checkExitCode

 * <dd>Check the exit code returned from javadoc.

 * <dt>checkOutput

 * <dd>Perform a series of checks on the contents on a file or output stream

 *     generated by javadoc.

 *     The checks can be either that a series of strings are found or are not found.

 * <dt>checkFiles

 * <dd>Perform a series of checks on the files generated by javadoc.

 *     The checks can be that a series of files are found or are not found.

 * </dl>

 *

 * <pre><code>

 *  public class MyTester extends JavadocTester {

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

 *          MyTester tester = new MyTester();

 *          tester.runTests();

 *      }

 *

 *      // test methods...

 *      @Test

 *      void test() {

 *          javadoc(<i>args</i>);

 *          checkExit(Exit.OK);

 *          checkOutput(<i>file</i>, true,

 *              <i>strings-to-find</i>);

 *          checkOutput(<i>file</i>, false,

 *              <i>strings-to-not-find</i>);

 *      }

 *  }

 * </code></pre>

 *

 * <p>

 * If javadoc is run more than once in a test method, you can compare the

 * results that are generated with the diff method. Since files written by

 * javadoc typically contain a timestamp, you may want to use the -notimestamp

 * option if you are going to compare the results from two runs of javadoc.

 *

 * <p>

 * If you have many calls of checkOutput that are very similar, you can write

 * your own check... method to reduce the amount of duplication. For example,

 * if you want to check that many files contain the same string, you could

 * write a method that takes a varargs list of files and calls checkOutput

 * on each file in turn with the string to be checked.

 *

 * <p>

 * You can also write you own custom check methods, which can use

 * readFile to get the contents of a file generated by javadoc,

 * and then use pass(...) or fail(...) to report whether the check

 * succeeded or not.

 *

 * <p>

 * You can have many separate test methods, each identified with a @Test

 * annotation. However, you should <b>not</b> assume they will be called

 * in the order declared in your source file.  If the order of a series

 * of javadoc invocations is important, do that within a single method.

 * If the invocations are independent, for better clarity, use separate

 * test methods, each with their own set of checks on the results.

 *

 * @author Doug Kramer

 * @author Jamie Ho

 * @author Jonathan Gibbons (rewrite)

 */

public abstract class JavadocTester {

    public static final String FS = System.getProperty("file.separator");

    public static final String PS = System.getProperty("path.separator");

    public static final String NL = System.getProperty("line.separator");

    public enum Output {

        /** The name of the output stream from javadoc. */

        OUT,

        /** The name for any output written to System.out. */

        STDOUT,

        /** The name for any output written to System.err. */

        STDERR

    }

    /** The output directory used in the most recent call of javadoc. */

    protected File outputDir;

    /** The output charset used in the most recent call of javadoc. */

    protected Charset charset = Charset.defaultCharset();

    /** The exit code of the most recent call of javadoc. */

    private int exitCode;

    /** The output generated by javadoc to the various writers and streams. */

    private final Map<Output, String> outputMap = new EnumMap<>(Output.class);

    /** A cache of file content, to avoid reading files unnecessarily. */

    private final Map<File,SoftReference<String>> fileContentCache = new HashMap<>();

    /** The charset used for files in the fileContentCache. */

    private Charset fileContentCacheCharset = null;

    /** Stream used for logging messages. */

    protected final PrintStream out = System.out;

    /** The directory containing the source code for the test. */

    public static final String testSrc = System.getProperty("test.src");

    /**

     * Get the path for a source file in the test source directory.

     * @param path the path of a file or directory in the source directory

     * @return the full path of the specified file

     */

    public static String testSrc(String path) {

        return new File(testSrc, path).getPath();

    }

    /**

     * Alternatives for checking the contents of a directory.

     */

    public enum DirectoryCheck {

        /**

         * Check that the directory is empty.

         */

        EMPTY((file, name) -> true),

        /**

         * Check that the directory does not contain any HTML files,

         * such as may have been generated by a prior run of javadoc

         * using this directory.

         * For now, the check is only performed on the top level directory.

         */

        NO_HTML_FILES((file, name) -> name.endsWith(".html")),

        /**

         * No check is performed on the directory contents.

         */

        NONE(null) { @Override void check(File dir) { } };

        /** The filter used to detect that files should <i>not</i> be present. */

        FilenameFilter filter;

        DirectoryCheck(FilenameFilter f) {

            filter = f;

        }

        void check(File dir) {

            if (dir.isDirectory()) {

                String[] contents = dir.list(filter);

                if (contents == null)

                    throw new Error("cannot list directory: " + dir);

                if (contents.length > 0) {

                    System.err.println("Found extraneous files in dir:" + dir.getAbsolutePath());

                    for (String x : contents) {

                        System.err.println(x);

                    }

                    throw new Error("directory has unexpected content: " + dir);

                }

            }

        }

    }

    private DirectoryCheck outputDirectoryCheck = DirectoryCheck.EMPTY;

    /** The current subtest number. Incremented when checking(...) is called. */

    private int numTestsRun = 0;

    /** The number of subtests passed. Incremented when passed(...) is called. */

    private int numTestsPassed = 0;

    /** The current run of javadoc. Incremented when javadoc is called. */

    private int javadocRunNum = 0;

    /** Marker annotation for test methods to be invoked by runTests. */

    @Retention(RetentionPolicy.RUNTIME)

    @interface Test { }

    /**

     * Run all methods annotated with @Test, followed by printSummary.

     * Typically called on a tester object in main()

     * @throws Exception if any errors occurred

     */

    public void runTests() throws Exception {

        runTests(m -> new Object[0]);

    }

    /**

     * Run all methods annotated with @Test, followed by printSummary.

     * Typically called on a tester object in main()

     * @param f a function which will be used to provide arguments to each

     *          invoked method

     * @throws Exception if any errors occurred

     */

    public void runTests(Function<Method, Object[]> f) throws Exception {

        for (Method m: getClass().getDeclaredMethods()) {

            Annotation a = m.getAnnotation(Test.class);

            if (a != null) {

                try {

                    out.println("Running test " + m.getName());

                    m.invoke(this, f.apply(m));

                } catch (InvocationTargetException e) {

                    Throwable cause = e.getCause();

                    throw (cause instanceof Exception) ? ((Exception) cause) : e;

                }

                out.println();

            }

        }

        printSummary();

    }

    /**

     * Run javadoc.

     * The output directory used by this call and the final exit code

     * will be saved for later use.

     * To aid the reader, it is recommended that calls to this method

     * put each option and the arguments it takes on a separate line.

     *

     * Example:

     * <pre><code>

     *  javadoc("-d", "out",

     *          "-sourcepath", testSrc,

     *          "-notimestamp",

     *          "pkg1", "pkg2", "pkg3/C.java");

     * </code></pre>

     *

     * @param args the arguments to pass to javadoc

     */

    public void javadoc(String... args) {

        outputMap.clear();

        fileContentCache.clear();

        javadocRunNum++;

        if (javadocRunNum == 1) {

            out.println("Running javadoc...");

        } else {

        }

        outputDir = new File(".");

        String charsetArg = null;

        String docencodingArg = null;

        String encodingArg = null;

        for (int i = 0; i < args.length - 2; i++) {

            switch (args[i]) {

                case "-d":

                    outputDir = new File(args[++i]);

                    break;

                case "-charset":

                    charsetArg = args[++i];

                    break;

                case "-docencoding":

                    docencodingArg = args[++i];

                    break;

                case "-encoding":

                    encodingArg = args[++i];

                    break;

            }

        }

        // The following replicates HtmlConfiguration.finishOptionSettings0

        // and sets up the charset used to read files.

        String cs;

        if (docencodingArg == null) {

            if (charsetArg == null) {

                cs = (encodingArg == null) ? "UTF-8" : encodingArg;

            } else {

                cs = charsetArg;

            }

        } else {

           cs = docencodingArg;

        }

        try {

            charset = Charset.forName(cs);

        } catch (UnsupportedCharsetException e) {

            charset = Charset.defaultCharset();

        }

        out.println("args: " + Arrays.toString(args));

//        log.setOutDir(outputDir);

        outputDirectoryCheck.check(outputDir);

        // This is the sole stream used by javadoc

        WriterOutput outOut = new WriterOutput();

        // These are to catch output to System.out and System.err,

        // in case these are used instead of the primary streams

        StreamOutput sysOut = new StreamOutput(System.out, System::setOut);

        StreamOutput sysErr = new StreamOutput(System.err, System::setErr);

        try {

            exitCode = jdk.javadoc.internal.tool.Main.execute(args, outOut.pw);

        } finally {

            outputMap.put(Output.STDOUT, sysOut.close());

            outputMap.put(Output.STDERR, sysErr.close());

            outputMap.put(Output.OUT, outOut.close());

        }

        outputMap.forEach((name, text) -> {

            if (!text.isEmpty()) {

                out.println("javadoc " + name + ":");

                out.println(text);

            }

        });

    }

    /**

     * Set the kind of check for the initial contents of the output directory

     * before javadoc is run.

     * The filter should return true for files that should <b>not</b> appear.

     * @param c the kind of check to perform

     */

    public void setOutputDirectoryCheck(DirectoryCheck c) {

        outputDirectoryCheck = c;

    }

    /**

     * The exit codes returned by the javadoc tool.

     * @see jdk.javadoc.internal.tool.Main.Result

     */

    public enum Exit {

        OK(0),        // Javadoc completed with no errors.

        ERROR(1),     // Completed but reported errors.

        CMDERR(2),    // Bad command-line arguments

        SYSERR(3),    // System error or resource exhaustion.

        ABNORMAL(4);  // Javadoc terminated abnormally

        Exit(int code) {

            this.code = code;

        }

        final int code;

        @Override

        public String toString() {

            return name() + '(' + code + ')';

        }

    }

    /**

     * Check the exit code of the most recent call of javadoc.

     *

     * @param expected the exit code that is required for the test

     * to pass.

     */

    public void checkExit(Exit expected) {

        checking("check exit code");

        if (exitCode == expected.code) {

            passed("return code " + exitCode);

        } else {

            failed("return code " + exitCode +"; expected " + expected);

        }

    }

    /**

     * Check for content in (or not in) the generated output.

     * Within the search strings, the newline character \n

     * will be translated to the platform newline character sequence.

     * @param path a path within the most recent output directory

     *  or the name of one of the output buffers, identifying

     *  where to look for the search strings.

     * @param expectedFound true if all of the search strings are expected

     *  to be found, or false if the file is not expected to be found

     * @param strings the strings to be searched for

     */

    public void checkFileAndOutput(String path, boolean expectedFound, String... strings) {

        if (expectedFound) {

            checkOutput(path, true, strings);

        } else {

            checkFiles(false, path);

        }

    }

    /**

     * Check for content in (or not in) the generated output.

     * Within the search strings, the newline character \n

     * will be translated to the platform newline character sequence.

     *  where to look for the search strings.

     * @param expectedFound true if all of the search strings are expected

     *  to be found, or false if all of the strings are expected to be

     *  not found

     * @param strings the strings to be searched for

     */

    public void checkOutput(String path, boolean expectedFound, String... strings) {

        // Read contents of file

        try {

        } catch (Error e) {

            checking("Read file");

            failed("Error reading file: " + e);

        }

    }

    /**

     * Check for content in (or not in) the one of the output streams written by

     * javadoc. Within the search strings, the newline character \n

     * will be translated to the platform newline character sequence.

     * @param output the output stream to check

     * @param expectedFound true if all of the search strings are expected

     *  to be found, or false if all of the strings are expected to be

     *  not found

     * @param strings the strings to be searched for

     */

    public void checkOutput(Output output, boolean expectedFound, String... strings) {

        checkOutput(output.toString(), outputMap.get(output), expectedFound, strings);

    }

    private void checkOutput(String path, String fileString, boolean expectedFound, String... strings) {

        for (String stringToFind : strings) {

//            log.logCheckOutput(path, expectedFound, stringToFind);

            checking("checkOutput");

            // Find string in file's contents

            boolean isFound = findString(fileString, stringToFind);

            if (isFound == expectedFound) {

                passed(path + ": following text " + (isFound ? "found:" : "not found:") + "\n"

            } else {

                failed(path + ": following text " + (isFound ? "found:" : "not found:") + "\n"