/*

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

 *

 */

package sun.jvm.hotspot;

import java.util.ArrayList;

import java.util.Arrays;

import sun.jvm.hotspot.tools.JStack;

import sun.jvm.hotspot.tools.JMap;

import sun.jvm.hotspot.tools.JInfo;

import sun.jvm.hotspot.tools.JSnap;

public class SALauncher {

    private static boolean launcherHelp() {

        System.out.println("    clhsdb       \tcommand line debugger");

        System.out.println("    hsdb         \tui debugger");

        System.out.println("    debugd --help\tto get more information");

        System.out.println("    jstack --help\tto get more information");

        System.out.println("    jmap   --help\tto get more information");

        System.out.println("    jinfo  --help\tto get more information");

        System.out.println("    jsnap  --help\tto get more information");

        return false;

    }

        // --pid <pid>

        // --exe <exe>

        // --core <core>

        return false;

    }

    private static boolean debugdHelp() {

        // [options] <pid> [server-id]

        // [options] <executable> <core> [server-id]

    }

    private static boolean jinfoHelp() {

        // --flags -> -flags

        // --sysprops -> -sysprops

    }

    private static boolean jmapHelp() {

        // --heap -> -heap

        // --binaryheap -> -heap:format=b

        // --histo -> -histo

        // --clstats -> -clstats

        // --finalizerinfo -> -finalizerinfo

    }

    private static boolean jstackHelp() {

        // --locks -> -l

        // --mixed -> -m

    }

    private static boolean jsnapHelp() {

    }

    private static boolean toolHelp(String toolName) {

        if (toolName.equals("jstack")) {

            return jstackHelp();

        }

        if (toolName.equals("jinfo")) {

            return jinfoHelp();

        }

        if (toolName.equals("jmap")) {

            return jmapHelp();

        }

        if (toolName.equals("jsnap")) {

            return jsnapHelp();

        }

        if (toolName.equals("debugd")) {

            return debugdHelp();

        }

        }

        return launcherHelp();

    }

    private static void buildAttachArgs(ArrayList<String> newArgs, String pid,

                                  String exe, String core, boolean allowEmpty) {

        if (!allowEmpty && (pid == null) && (exe == null)) {

            throw new SAGetoptException("You have to set --pid or --exe.");

        }

        if (pid != null) { // Attach to live process

            if (exe != null) {

                throw new SAGetoptException("Unnecessary argument: --exe");

            } else if (core != null) {

                throw new SAGetoptException("Unnecessary argument: --core");

            } else if (!pid.matches("^\\d+$")) {

                throw new SAGetoptException("Invalid pid: " + pid);

            }

            newArgs.add(pid);

        } else if (exe != null) {

            if (exe.length() == 0) {

                throw new SAGetoptException("You have to set --exe.");

            }

            newArgs.add(exe);

            if ((core == null) || (core.length() == 0)) {

                throw new SAGetoptException("You have to set --core.");

            }

            newArgs.add(core);

        }

    }

    private static void runCLHSDB(String[] oldArgs) {

        SAGetopt sg = new SAGetopt(oldArgs);

        String[] longOpts = {"exe=", "core=", "pid="};

        ArrayList<String> newArgs = new ArrayList();

        String pid = null;

        String exe = null;

        String core = null;

        String s = null;

        while((s = sg.next(null, longOpts)) != null) {

            if (s.equals("exe")) {

                exe = sg.getOptarg();

                continue;

            }

            if (s.equals("core")) {

                core = sg.getOptarg();

                continue;

            }

            if (s.equals("pid")) {

                pid = sg.getOptarg();

                continue;

            }

        }

        buildAttachArgs(newArgs, pid, exe, core, true);

        CLHSDB.main(newArgs.toArray(new String[newArgs.size()]));

    }

    private static void runHSDB(String[] oldArgs) {

        SAGetopt sg = new SAGetopt(oldArgs);

        String[] longOpts = {"exe=", "core=", "pid="};

        ArrayList<String> newArgs = new ArrayList();

        String pid = null;

        String exe = null;

        String core = null;

        String s = null;

        while((s = sg.next(null, longOpts)) != null) {

            if (s.equals("exe")) {

                exe = sg.getOptarg();

                continue;

            }

            if (s.equals("core")) {

                core = sg.getOptarg();

                continue;

            }

            if (s.equals("pid")) {

                pid = sg.getOptarg();

                continue;

            }

        }

        buildAttachArgs(newArgs, pid, exe, core, true);

        HSDB.main(newArgs.toArray(new String[newArgs.size()]));

    }

    private static void runJSTACK(String[] oldArgs) {

        SAGetopt sg = new SAGetopt(oldArgs);

        String[] longOpts = {"exe=", "core=", "pid=",

                                 "mixed", "locks"};

        ArrayList<String> newArgs = new ArrayList();

        String pid = null;

        String exe = null;

        String core = null;

        String s = null;

        while((s = sg.next(null, longOpts)) != null) {

            if (s.equals("exe")) {

                exe = sg.getOptarg();

                continue;

            }

            if (s.equals("core")) {

                core = sg.getOptarg();

                continue;

            }

            if (s.equals("pid")) {

                pid = sg.getOptarg();

                continue;

            }

            if (s.equals("mixed")) {

                newArgs.add("-m");

                continue;

            }

            if (s.equals("locks")) {

                newArgs.add("-l");

                continue;

            }

        }

        buildAttachArgs(newArgs, pid, exe, core, false);

        JStack jstack = new JStack(false, false);

        jstack.runWithArgs(newArgs.toArray(new String[newArgs.size()]));

    }

    private static void runJMAP(String[] oldArgs) {

        SAGetopt sg = new SAGetopt(oldArgs);

        String[] longOpts = {"exe=", "core=", "pid=",

              "heap", "binaryheap", "dumpfile=", "histo", "clstats", "finalizerinfo"};

        ArrayList<String> newArgs = new ArrayList();

        String pid = null;

        String exe = null;

        String core = null;

        String s = null;

        String dumpfile = null;

        boolean requestHeapdump = false;

        while((s = sg.next(null, longOpts)) != null) {

            if (s.equals("exe")) {

                exe = sg.getOptarg();

                continue;

            }

            if (s.equals("core")) {

                core = sg.getOptarg();

                continue;

            }

            if (s.equals("pid")) {

                pid = sg.getOptarg();

                continue;

            }

            if (s.equals("heap")) {

                newArgs.add("-heap");

                continue;

            }

            if (s.equals("binaryheap")) {

                requestHeapdump = true;

                continue;

            }

            if (s.equals("dumpfile")) {

                dumpfile = sg.getOptarg();

                continue;

            }

            if (s.equals("histo")) {

                newArgs.add("-histo");

                continue;

            }

            if (s.equals("clstats")) {

                newArgs.add("-clstats");

                continue;

            }

            if (s.equals("finalizerinfo")) {

                newArgs.add("-finalizerinfo");

                continue;

            }

        }

        if (!requestHeapdump && (dumpfile != null)) {

            throw new IllegalArgumentException("Unexpected argument dumpfile");

        }

        if (requestHeapdump) {

            if (dumpfile == null) {

                newArgs.add("-heap:format=b");

            } else {

                newArgs.add("-heap:format=b,file=" + dumpfile);

            }

        }

        buildAttachArgs(newArgs, pid, exe, core, false);

        JMap.main(newArgs.toArray(new String[newArgs.size()]));

    }

    private static void runJINFO(String[] oldArgs) {

        SAGetopt sg = new SAGetopt(oldArgs);

        String[] longOpts = {"exe=", "core=", "pid=",

                                     "flags", "sysprops"};

        ArrayList<String> newArgs = new ArrayList();

        String exe = null;

        String pid = null;

        String core = null;

        String s = null;

        while((s = sg.next(null, longOpts)) != null) {

            if (s.equals("exe")) {

                exe = sg.getOptarg();

                continue;

            }

            if (s.equals("core")) {

                core = sg.getOptarg();

                continue;

            }

            if (s.equals("pid")) {

                pid = sg.getOptarg();

                continue;

            }

            if (s.equals("flags")) {

                newArgs.add("-flags");

                continue;

            }

            if (s.equals("sysprops")) {

                newArgs.add("-sysprops");

                continue;

            }

        }

        buildAttachArgs(newArgs, pid, exe, core, false);

        JInfo.main(newArgs.toArray(new String[newArgs.size()]));

    }

    private static void runJSNAP(String[] oldArgs) {

        SAGetopt sg = new SAGetopt(oldArgs);

        String[] longOpts = {"exe=", "core=", "pid=", "all"};

        ArrayList<String> newArgs = new ArrayList();

        String exe = null;

        String pid = null;

        String core = null;

        String s = null;

        while((s = sg.next(null, longOpts)) != null) {

            if (s.equals("exe")) {

                exe = sg.getOptarg();

                continue;

            }

            if (s.equals("core")) {

                core = sg.getOptarg();

                continue;

            }

            if (s.equals("pid")) {

                pid = sg.getOptarg();

                continue;

            }

            if (s.equals("all")) {

                newArgs.add("-a");

                continue;

            }

        }

        buildAttachArgs(newArgs, pid, exe, core, false);

        JSnap.main(newArgs.toArray(new String[newArgs.size()]));

    }

    private static void runDEBUGD(String[] oldArgs) {

        // By default SA agent classes prefer Windows process debugger

        // to windbg debugger. SA expects special properties to be set

        // to choose other debuggers. We will set those here before

        // attaching to SA agent.

        System.setProperty("sun.jvm.hotspot.debugger.useWindbgDebugger", "true");

        SAGetopt sg = new SAGetopt(oldArgs);

        String[] longOpts = {"exe=", "core=", "pid=", "serverid="};

        ArrayList<String> newArgs = new ArrayList<>();

        String exe = null;

        String pid = null;

        String core = null;

        String s = null;

        String serverid = null;

        while((s = sg.next(null, longOpts)) != null) {

          if (s.equals("exe")) {

              exe = sg.getOptarg();

              continue;

          }

          if (s.equals("core")) {

              core = sg.getOptarg();

              continue;

          }

          if (s.equals("pid")) {

              pid = sg.getOptarg();

              continue;

          }

          if (s.equals("serverid")) {

              serverid = sg.getOptarg();

              continue;

          }

        }

        buildAttachArgs(newArgs, pid, exe, core, false);

        if (serverid != null) {

            newArgs.add(serverid);

        }

        // delegate to the actual SA debug server.

        sun.jvm.hotspot.DebugServer.main(newArgs.toArray(new String[newArgs.size()]));

    }

    public static void main(String[] args) {

        // Provide a help

        if (args.length == 0) {

            launcherHelp();

            return;

        }

        // No arguments imply help for debugd, jstack, jmap, jinfo but launch clhsdb and hsdb

        if (args.length == 1 && !args[0].equals("clhsdb") && !args[0].equals("hsdb")) {

            toolHelp(args[0]);

            return;

        }

        for (String arg : args) {

            if (arg.equals("-h") || arg.equals("-help") || arg.equals("--help")) {

                toolHelp(args[0]);

                return;

            }

        }

        String[] oldArgs = Arrays.copyOfRange(args, 1, args.length);

        try {

            // Run SA interactive mode

            if (args[0].equals("clhsdb")) {

                runCLHSDB(oldArgs);

                return;

            }

            if (args[0].equals("hsdb")) {

                runHSDB(oldArgs);

                return;

            }

            // Run SA tmtools mode

            if (args[0].equals("jstack")) {

                runJSTACK(oldArgs);

                return;

            }

            if (args[0].equals("jmap")) {

                runJMAP(oldArgs);

                return;

            }

            if (args[0].equals("jinfo")) {

                runJINFO(oldArgs);

                return;

            }

            if (args[0].equals("jsnap")) {

                runJSNAP(oldArgs);

                return;

            }

            if (args[0].equals("debugd")) {

                runDEBUGD(oldArgs);

                return;