Command line args processing

Former-commit-id: 5ddfc091

plugins/org.jkiss.dbeaver.core.application/META-INF/MANIFEST.MF

@@ -24,4 +24,5 @@ Require-Bundle: org.jkiss.dbeaver.core,
  org.eclipse.e4.ui.css.core,
  org.eclipse.e4.ui.css.swt,
  org.eclipse.e4.ui.workbench.renderers.swt,
- org.w3c.css.sac
+ org.w3c.css.sac,
+ org.apache.commons.cli

plugins/org.jkiss.dbeaver.core.application/src/org/jkiss/dbeaver/core/application/DBeaverApplication.java

@@ -17,6 +17,7 @@
  */
 package org.jkiss.dbeaver.core.application;
 
+import org.apache.commons.cli.*;
 import org.eclipse.core.runtime.Platform;
 import org.eclipse.equinox.app.IApplication;
 import org.eclipse.equinox.app.IApplicationContext;
@@ -28,11 +29,17 @@ import org.eclipse.swt.widgets.Shell;
 import org.eclipse.ui.IWorkbench;
 import org.eclipse.ui.IWorkbenchPreferenceConstants;
 import org.eclipse.ui.PlatformUI;
+import org.jkiss.code.Nullable;
 import org.jkiss.dbeaver.Log;
 import org.jkiss.dbeaver.core.DBeaverCore;
+import org.jkiss.utils.ArrayUtils;
 
 import java.io.File;
 import java.net.URL;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.Comparator;
+import java.util.List;
 
 /**
  * This class controls all aspects of the application's execution
@@ -42,9 +49,6 @@ public class DBeaverApplication implements IApplication
     private static final Log log = Log.getLog(DBeaverApplication.class);
     public static final String DBEAVER_DEFAULT_DIR = ".dbeaver"; //$NON-NLS-1$
 
-    /* (non-Javadoc)
-    * @see org.eclipse.equinox.app.IApplication#start(org.eclipse.equinox.app.IApplicationContext)
-    */
     @Override
     public Object start(IApplicationContext context)
     {
@@ -53,7 +57,6 @@ public class DBeaverApplication implements IApplication
         Location instanceLoc = Platform.getInstanceLocation();
         String defaultHomePath = getDefaultWorkspaceLocation().getAbsolutePath();
         try {
-
             URL defaultHomeURL = new URL(
                 "file",  //$NON-NLS-1$
                 null,
@@ -62,6 +65,9 @@ public class DBeaverApplication implements IApplication
             Shell shell = null;
             while (keepTrying) {
                 if (!instanceLoc.set(defaultHomeURL, true)) {
+                    if (handleCommandLine(instanceLoc)) {
+                        return IApplication.EXIT_OK;
+                    }
                     // Can't lock specified path
                     if (shell == null) {
                         shell = new Shell(display, SWT.ON_TOP);
@@ -125,6 +131,44 @@ public class DBeaverApplication implements IApplication
         }
     }
 
+    private boolean handleCommandLine(Location instanceLoc) {
+        CommandLine commandLine = getCommandLine();
+        if (commandLine == null) {
+            return false;
+        }
+        String[] files = commandLine.getOptionValues(DBeaverCommandLine.PARAM_FILE);
+        String[] fileArgs = commandLine.getArgs();
+        if (!ArrayUtils.isEmpty(files) || !ArrayUtils.isEmpty(fileArgs)) {
+            List<String> fileNames = new ArrayList<>();
+            if (!ArrayUtils.isEmpty(files)) {
+                Collections.addAll(fileNames, files);
+            }
+            if (!ArrayUtils.isEmpty(fileArgs)) {
+                Collections.addAll(fileNames, fileArgs);
+            }
+            return true;
+        }
+        if (commandLine.hasOption(DBeaverCommandLine.PARAM_STOP)) {
+            return true;
+        }
+        if (commandLine.hasOption(DBeaverCommandLine.PARAM_THREAD_DUMP)) {
+            return true;
+        }
+        if (commandLine.hasOption(DBeaverCommandLine.PARAM_HELP)) {
+            HelpFormatter helpFormatter = new HelpFormatter();
+            helpFormatter.setWidth(120);
+            helpFormatter.setOptionComparator(new Comparator<Option>() {
+                @Override
+                public int compare(Option o1, Option o2) {
+                    return 0;
+                }
+            });
+            helpFormatter.printHelp("dbeaver", "Universal Database Manager", DBeaverCommandLine.ALL_OPTIONS, "(C) 2016 Serge Rider", true);
+            return true;
+        }
+        return false;
+    }
+
     private static File getDefaultWorkspaceLocation() {
         return new File(
             System.getProperty("user.home"),
@@ -153,5 +197,14 @@ public class DBeaverApplication implements IApplication
         });
     }
 
+    @Nullable
+    public static CommandLine getCommandLine() {
+        try {
+            return new DefaultParser().parse(DBeaverCommandLine.ALL_OPTIONS, Platform.getApplicationArgs(), false);
+        } catch (Exception e) {
+            log.error("Error parsing command line", e);
+            return null;
+        }
+    }
 
 }

plugins/org.jkiss.dbeaver.core.application/src/org/jkiss/dbeaver/core/application/DBeaverCommandLine.java

+/*
+ * DBeaver - Universal Database Manager
+ * Copyright (C) 2010-2016 Serge Rieder (serge@jkiss.org)
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License (version 2)
+ * as published by the Free Software Foundation.
+ *
+ * This program 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 for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+package org.jkiss.dbeaver.core.application;
+
+import org.apache.commons.cli.Options;
+
+/**
+ * command line options
+ */
+public class DBeaverCommandLine
+{
+    public static final String PARAM_HELP = "help";
+    public static final String PARAM_FILE = "f";
+    public static final String PARAM_STOP = "stop";
+    public static final String PARAM_THREAD_DUMP = "dump";
+
+    public final static Options ALL_OPTIONS = new Options()
+        .addOption(PARAM_HELP, false, "Help")
+
+        .addOption(PARAM_FILE, "file", true, "File top open")
+        .addOption(PARAM_STOP, "quit", true, "Stop DBeaver running instance")
+        .addOption(PARAM_THREAD_DUMP, "thread-dump", true, "Print instance thread dump")
+        ;
+}