/*

 * Copyright 2006-2008 Sun Microsystems, Inc.  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.  Sun designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Sun in the LICENSE file that accompanied this code.

 *

 * 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 Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,

 * CA 95054 USA or visit www.sun.com if you need additional information or

 * have any questions.

 */

package com.sun.tools.javac.main;

import com.sun.tools.javac.code.Lint;

import com.sun.tools.javac.code.Source;

import com.sun.tools.javac.code.Type;

import com.sun.tools.javac.jvm.Target;

import com.sun.tools.javac.main.JavacOption.HiddenOption;

import com.sun.tools.javac.main.JavacOption.Option;

import com.sun.tools.javac.main.JavacOption.XOption;

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

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

import com.sun.tools.javac.processing.JavacProcessingEnvironment;

import java.io.File;

import java.io.FileWriter;

import java.io.PrintWriter;

import java.util.Collection;

import java.util.EnumSet;

import java.util.LinkedHashSet;

import java.util.Set;

import javax.lang.model.SourceVersion;

import static com.sun.tools.javac.main.OptionName.*;

/**

 * TODO: describe com.sun.tools.javac.main.RecognizedOptions

 *

 * <p><b>This is NOT part of any API supported by Sun Microsystems.

 * If you write code that depends on this, you do so at your own

 * risk.  This code and its internal interfaces are subject to change

 * or deletion without notice.</b></p>

 */

public class RecognizedOptions {

    private RecognizedOptions() {}

    public interface OptionHelper {

        void setOut(PrintWriter out);

        void error(String key, Object... args);

        void printVersion();

        void printFullVersion();

        void printHelp();

        void printXhelp();

        void addFile(File f);

        void addClassName(String s);

    }

    public static class GrumpyHelper implements OptionHelper {

        public void setOut(PrintWriter out) {

            throw new IllegalArgumentException();

        }

        public void error(String key, Object... args) {

            throw new IllegalArgumentException(Main.getLocalizedString(key, args));

        }

        public void printVersion() {

            throw new IllegalArgumentException();

        }

        public void printFullVersion() {

            throw new IllegalArgumentException();

        }

        public void printHelp() {

            throw new IllegalArgumentException();

        }

        public void printXhelp() {

            throw new IllegalArgumentException();

        }

        public void addFile(File f) {

            throw new IllegalArgumentException(f.getPath());

        }

        public void addClassName(String s) {

            throw new IllegalArgumentException(s);

        }

    }

    static Set<OptionName> javacOptions = EnumSet.of(

        G,

        G_NONE,

        G_CUSTOM,

        XLINT,

        XLINT_CUSTOM,

        NOWARN,

        VERBOSE,

        DEPRECATION,

        CLASSPATH,

        CP,

        SOURCEPATH,

        BOOTCLASSPATH,

        XBOOTCLASSPATH_PREPEND,

        XBOOTCLASSPATH_APPEND,

        XBOOTCLASSPATH,

        EXTDIRS,

        DJAVA_EXT_DIRS,

        ENDORSEDDIRS,

        DJAVA_ENDORSED_DIRS,

        PROC,

        PROCESSOR,

        PROCESSORPATH,

        D,

        S,

        IMPLICIT,

        ENCODING,

        SOURCE,

        TARGET,

        VERSION,

        FULLVERSION,

        HELP,

        A,

        X,

        J,

        MOREINFO,

        WERROR,

        // COMPLEXINFERENCE,

        PROMPT,

        DOE,

        PRINTSOURCE,

        WARNUNCHECKED,

        XMAXERRS,

        XMAXWARNS,

        XSTDOUT,

        XPRINT,

        XPRINTROUNDS,

        XPRINTPROCESSORINFO,

        XPREFER,

        O,

        XJCOV,

        XD,

        SOURCEFILE);

    static Set<OptionName> javacFileManagerOptions = EnumSet.of(

        CLASSPATH,

        CP,

        SOURCEPATH,

        BOOTCLASSPATH,

        XBOOTCLASSPATH_PREPEND,

        XBOOTCLASSPATH_APPEND,

        XBOOTCLASSPATH,

        EXTDIRS,

        DJAVA_EXT_DIRS,

        ENDORSEDDIRS,

        DJAVA_ENDORSED_DIRS,

        PROCESSORPATH,

        D,

        S,

        ENCODING,

        SOURCE);

    static Set<OptionName> javacToolOptions = EnumSet.of(

        G,

        G_NONE,

        G_CUSTOM,

        XLINT,

        XLINT_CUSTOM,

        NOWARN,

        VERBOSE,

        DEPRECATION,

        PROC,

        PROCESSOR,

        IMPLICIT,

        SOURCE,

        TARGET,

        // VERSION,

        // FULLVERSION,

        // HELP,

        A,

        // X,

        // J,

        MOREINFO,

        WERROR,

        // COMPLEXINFERENCE,

        PROMPT,

        DOE,

        PRINTSOURCE,

        WARNUNCHECKED,

        XMAXERRS,

        XMAXWARNS,

        // XSTDOUT,

        XPRINT,

        XPRINTROUNDS,

        XPRINTPROCESSORINFO,

        XPREFER,

        O,

        XJCOV,

        XD);

    static Option[] getJavaCompilerOptions(OptionHelper helper) {

        return getOptions(helper, javacOptions);

    }

    public static Option[] getJavacFileManagerOptions(OptionHelper helper) {

        return getOptions(helper, javacFileManagerOptions);

    }

    public static Option[] getJavacToolOptions(OptionHelper helper) {

        return getOptions(helper, javacToolOptions);

    }

    static Option[] getOptions(OptionHelper helper, Set<OptionName> desired) {

        ListBuffer<Option> options = new ListBuffer<Option>();

        for (Option option : getAll(helper))

            if (desired.contains(option.getName()))

                options.append(option);

        return options.toArray(new Option[options.length()]);

    }

    /**

     * Get all the recognized options.

     * @param helper an {@code OptionHelper} to help when processing options

     * @return an array of options

     */

    public static Option[] getAll(final OptionHelper helper) {

        return new Option[] {

        new Option(G,                                           "opt.g"),

        new Option(G_NONE,                                      "opt.g.none") {

            @Override

            public boolean process(Options options, String option) {

                options.put("-g:", "none");

                return false;

            }

        },

        new Option(G_CUSTOM,                                    "opt.g.lines.vars.source",

                Option.ChoiceKind.ANYOF, "lines", "vars", "source"),

        new XOption(XLINT,                                      "opt.Xlint"),

        new XOption(XLINT_CUSTOM,                               "opt.Xlint.suboptlist",

                Option.ChoiceKind.ANYOF, getXLintChoices()),

        // -nowarn is retained for command-line backward compatibility

        new Option(NOWARN,                                      "opt.nowarn") {

            @Override

            public boolean process(Options options, String option) {

                options.put("-Xlint:none", option);

                return false;

            }

        },

        new Option(VERBOSE,                                     "opt.verbose"),

        // -deprecation is retained for command-line backward compatibility

        new Option(DEPRECATION,                                 "opt.deprecation") {

            @Override

            public boolean process(Options options, String option) {

                options.put("-Xlint:deprecation", option);

                return false;

            }

        },

        new Option(CLASSPATH,              "opt.arg.path",      "opt.classpath"),

        new Option(CP,                     "opt.arg.path",      "opt.classpath") {

            @Override

            public boolean process(Options options, String option, String arg) {

                return super.process(options, "-classpath", arg);

            }

        },

        new Option(SOURCEPATH,             "opt.arg.path",      "opt.sourcepath"),

        new Option(BOOTCLASSPATH,          "opt.arg.path",      "opt.bootclasspath") {

            @Override

            public boolean process(Options options, String option, String arg) {

                options.remove("-Xbootclasspath/p:");

                options.remove("-Xbootclasspath/a:");

                return super.process(options, option, arg);

            }

        },

        new XOption(XBOOTCLASSPATH_PREPEND,"opt.arg.path", "opt.Xbootclasspath.p"),

        new XOption(XBOOTCLASSPATH_APPEND, "opt.arg.path", "opt.Xbootclasspath.a"),

        new XOption(XBOOTCLASSPATH,        "opt.arg.path", "opt.bootclasspath") {

            @Override

            public boolean process(Options options, String option, String arg) {

                options.remove("-Xbootclasspath/p:");

                options.remove("-Xbootclasspath/a:");

                return super.process(options, "-bootclasspath", arg);

            }

        },

        new Option(EXTDIRS,                "opt.arg.dirs",      "opt.extdirs"),

        new XOption(DJAVA_EXT_DIRS,        "opt.arg.dirs",      "opt.extdirs") {

            @Override

            public boolean process(Options options, String option, String arg) {

                return super.process(options, "-extdirs", arg);

            }

        },

        new Option(ENDORSEDDIRS,            "opt.arg.dirs",     "opt.endorseddirs"),

        new XOption(DJAVA_ENDORSED_DIRS,    "opt.arg.dirs",     "opt.endorseddirs") {

            @Override

            public boolean process(Options options, String option, String arg) {

                return super.process(options, "-endorseddirs", arg);

            }

        },

        new Option(PROC,                                 "opt.proc.none.only",

                Option.ChoiceKind.ONEOF, "none", "only"),

        new Option(PROCESSOR,           "opt.arg.class.list",   "opt.processor"),

        new Option(PROCESSORPATH,       "opt.arg.path",         "opt.processorpath"),

        new Option(D,                   "opt.arg.directory",    "opt.d"),

        new Option(S,                   "opt.arg.directory",    "opt.sourceDest"),

        new Option(IMPLICIT,                                    "opt.implicit",

                Option.ChoiceKind.ONEOF, "none", "class"),

        new Option(ENCODING,            "opt.arg.encoding",     "opt.encoding"),

        new Option(SOURCE,              "opt.arg.release",      "opt.source") {

            @Override

            public boolean process(Options options, String option, String operand) {

                Source source = Source.lookup(operand);

                if (source == null) {

                    helper.error("err.invalid.source", operand);

                    return true;

                }

                return super.process(options, option, operand);

            }

        },

        new Option(TARGET,              "opt.arg.release",      "opt.target") {

            @Override

            public boolean process(Options options, String option, String operand) {

                Target target = Target.lookup(operand);

                if (target == null) {

                    helper.error("err.invalid.target", operand);

                    return true;

                }

                return super.process(options, option, operand);

            }

        },

        new Option(VERSION,                                     "opt.version") {

            @Override

            public boolean process(Options options, String option) {

                helper.printVersion();

                return super.process(options, option);

            }

        },

        new HiddenOption(FULLVERSION) {

            @Override

            public boolean process(Options options, String option) {

                helper.printFullVersion();

                return super.process(options, option);

            }

        },

        new Option(HELP,                                        "opt.help") {

            @Override

            public boolean process(Options options, String option) {

                helper.printHelp();

                return super.process(options, option);

            }

        },

        new Option(A,                "opt.arg.key.equals.value","opt.A") {

            @Override

            String helpSynopsis() {

                hasSuffix = true;

                return super.helpSynopsis();

            }

            @Override

            public boolean matches(String arg) {

                return arg.startsWith("-A");

            }

            @Override

            public boolean hasArg() {

                return false;

            }

            // Mapping for processor options created in

            // JavacProcessingEnvironment

            @Override

            public boolean process(Options options, String option) {

                int argLength = option.length();

                if (argLength == 2) {

                    helper.error("err.empty.A.argument");

                    return true;

                }

                int sepIndex = option.indexOf('=');

                String key = option.substring(2, (sepIndex != -1 ? sepIndex : argLength) );

                if (!JavacProcessingEnvironment.isValidOptionName(key)) {

                    helper.error("err.invalid.A.key", option);

                    return true;

                }

                return process(options, option, option);

            }

        },

        new Option(X,                                           "opt.X") {

            @Override

            public boolean process(Options options, String option) {

                helper.printXhelp();

                return super.process(options, option);

            }

        },

        // This option exists only for the purpose of documenting itself.

        // It's actually implemented by the launcher.

        new Option(J,                   "opt.arg.flag",         "opt.J") {

            @Override

            String helpSynopsis() {

                hasSuffix = true;

                return super.helpSynopsis();

            }

            @Override

            public boolean process(Options options, String option) {

                throw new AssertionError

                    ("the -J flag should be caught by the launcher.");

            }

        },

        // stop after parsing and attributing.

        // new HiddenOption("-attrparseonly"),

        // new Option("-moreinfo",                                      "opt.moreinfo") {

        new HiddenOption(MOREINFO) {

            @Override

            public boolean process(Options options, String option) {

                Type.moreInfo = true;

                return super.process(options, option);

            }

        },

        // treat warnings as errors

        new HiddenOption(WERROR),

        // use complex inference from context in the position of a method call argument

        new HiddenOption(COMPLEXINFERENCE),

        // generare source stubs

        // new HiddenOption("-stubs"),

        // relax some constraints to allow compiling from stubs

        // new HiddenOption("-relax"),

        // output source after translating away inner classes

        // new Option("-printflat",                             "opt.printflat"),

        // new HiddenOption("-printflat"),

        // display scope search details

        // new Option("-printsearch",                           "opt.printsearch"),

        // new HiddenOption("-printsearch"),

        // prompt after each error

        // new Option("-prompt",                                        "opt.prompt"),

        new HiddenOption(PROMPT),

        // dump stack on error

        new HiddenOption(DOE),

        // output source after type erasure

        // new Option("-s",                                     "opt.s"),

        new HiddenOption(PRINTSOURCE),

        // output shrouded class files

        // new Option("-scramble",                              "opt.scramble"),

        // new Option("-scrambleall",                           "opt.scrambleall"),

        // display warnings for generic unchecked operations

        new HiddenOption(WARNUNCHECKED) {

            @Override

            public boolean process(Options options, String option) {

                options.put("-Xlint:unchecked", option);

                return false;

            }

        },

        new XOption(XMAXERRS,           "opt.arg.number",       "opt.maxerrs"),

        new XOption(XMAXWARNS,          "opt.arg.number",       "opt.maxwarns"),

        new XOption(XSTDOUT,            "opt.arg.file",         "opt.Xstdout") {

            @Override

            public boolean process(Options options, String option, String arg) {

                try {

                    helper.setOut(new PrintWriter(new FileWriter(arg), true));

                } catch (java.io.IOException e) {

                    helper.error("err.error.writing.file", arg, e);

                    return true;

                }

                return super.process(options, option, arg);

            }

        },

        new XOption(XPRINT,                                     "opt.print"),

        new XOption(XPRINTROUNDS,                               "opt.printRounds"),

        new XOption(XPRINTPROCESSORINFO,                        "opt.printProcessorInfo"),

        new XOption(XPREFER,                                    "opt.prefer",

                Option.ChoiceKind.ONEOF, "source", "newer"),

        /* -O is a no-op, accepted for backward compatibility. */