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

/*
 * @test
 * @bug 6958836 8002168
 * @summary javadoc should support -Xmaxerrs and -Xmaxwarns
 * @modules jdk.javadoc
 */

import java.io.*;
import java.util.*;

import com.sun.javadoc.DocErrorReporter;
import com.sun.javadoc.RootDoc;



public class Test {
    private static final String ERROR_MARKER = "Error-count";
    private static final String WARNING_MARKER = "Warning-count";

    public static void main(String... args) throws Exception {
        new Test().run();
    }

    void run() throws Exception {
        javadoc("Errors",  list(),                   10,  0);
        javadoc("Errors",  list("-Xmaxerrs",   "0"), 10,  0);
        javadoc("Errors",  list("-Xmaxerrs",   "2"),  2,  0);
        javadoc("Errors",  list("-Xmaxerrs",   "4"),  4,  0);
        javadoc("Errors",  list("-Xmaxerrs",  "20"), 10,  0);

        javadoc("Warnings", list(),                    0, 10);
        javadoc("Warnings", list("-Xmaxwarns",  "0"),  0, 10);
        javadoc("Warnings", list("-Xmaxwarns",  "2"),  0,  2);
        javadoc("Warnings", list("-Xmaxwarns",  "4"),  0,  4);
        javadoc("Warnings", list("-Xmaxwarns", "20"),  0, 10);

        if (errors > 0)
            throw new Exception(errors + " errors occurred.");
    }

    void javadoc(String selector, List<String> testOpts,
                int expectErrs, int expectWarns) {
        System.err.println("Test " + (++count) + ": " + selector + " " + testOpts);
        File testOutDir = new File("test" + count);

        List<String> opts = new ArrayList<String>();
        // Force en_US locale in lieu of something like -XDrawDiagnostics.
        // For some reason, this must be the first option when used.
        opts.addAll(list("-locale", "en_US"));
        opts.add(new File(System.getProperty("test.src"),
                Test.class.getSimpleName() + ".java").getPath());
        opts.addAll(testOpts);
        opts.add("-gen" + selector);

        StringWriter errSW = new StringWriter();
        PrintWriter errPW = new PrintWriter(errSW);
        StringWriter warnSW = new StringWriter();
        PrintWriter warnPW = new PrintWriter(warnSW);
        StringWriter noteSW = new StringWriter();
        PrintWriter notePW = new PrintWriter(noteSW);

        int rc = com.sun.tools.javadoc.Main.execute("javadoc",
                              errPW, warnPW, notePW,
                              "Test$TestDoclet",
                              getClass().getClassLoader(),
                              opts.toArray(new String[opts.size()]));
        System.err.println("rc: " + rc);

        errPW.close();
        String errOut = errSW.toString();
        System.err.println("Errors:\n" + errOut);
        warnPW.close();
        String warnOut = warnSW.toString();
        System.err.println("Warnings:\n" + warnOut);
        notePW.close();
        String noteOut = noteSW.toString();
        System.err.println("Notes:\n" + noteOut);

        check(errOut, ERROR_MARKER, expectErrs);
        check(warnOut, WARNING_MARKER, expectWarns); // requires -locale en_US
    }

    void check(String text, String expectText, int expectCount) {
        int foundCount = 0;
        for (String line: text.split("[\r\n]+")) {
            if (line.contains(expectText))
                foundCount++;
        }
        if (foundCount != expectCount) {
            error("incorrect number of matches found: " + foundCount
                  + ", expected: " + expectCount);
        }
    }

    private List<String> list(String... args) {
        return Arrays.asList(args);
    }

    void error(String msg) {
        System.err.println(msg);
        errors++;
    }

    int count;
    int errors;

    public static class TestDoclet {
        static boolean genErrors = false;
        static boolean genWarnings = false;

        public static boolean start(RootDoc root) {
            // generate 10 errors or warnings
            for (int i = 1 ; i <= 10 ; i++) {
                if (genErrors)
                    root.printError(ERROR_MARKER + " " + i);
                if (genWarnings)
                    root.printWarning(WARNING_MARKER + " " + i);
            }
            return true;
        }

        public static int optionLength(String option) {
            if (option == null) {
                throw new Error("invalid usage: ");
            }
            System.out.println("option: " + option);
            switch (option.trim()) {
                case "-genErrors":
                    return 1;
                case "-genWarnings":
                    return 1;
                default:
                    return 0;
            }
        }

        public static boolean validOptions(String[][] options, DocErrorReporter reporter) {
            for (int i = 0 ; i < options.length; i++) {
               String opt = options[i][0].trim();
               switch (opt) {
                   case "-genErrors":
                       genErrors = true;
                       genWarnings = false;
                       break;
                   case "-genWarnings":
                       genErrors = false;
                       genWarnings = true;
                       break;
               }
            }
            return true;
        }
    }
}