Server.openBrowser no longer writes to System.out directly if opening the URL failed.

h2/src/main/org/h2/server/web/WebServer.java

@@ -253,7 +253,7 @@ public class WebServer implements Service {
     public void init(String... args) {
         // set the serverPropertiesDir, because it's used in loadProperties()
         for (int i = 0; args != null && i < args.length; i++) {
-            if (args[i].equals("-properties")) {
+            if ("-properties".equals(args[i])) {
                 serverPropertiesDir = args[++i];
             }
         }
@@ -276,7 +276,7 @@ public class WebServer implements Service {
                 SysProperties.setBaseDir(baseDir);
             } else if ("-ifExists".equals(a)) {
                 ifExists = true;
-            } else if (args[i].equals("-properties")) {
+            } else if ("-properties".equals(args[i])) {
                 // already set
                 i++;
             } else if ("-trace".equals(a)) {
@@ -519,6 +519,9 @@ public class WebServer implements Service {
 
     private Properties loadProperties() {
         try {
+            if ("null".equals(serverPropertiesDir)) {
+                return new Properties();
+            }
             return SortedProperties.loadProperties(serverPropertiesDir + "/" + Constants.SERVER_PROPERTIES_NAME);
         } catch (Exception e) {
             TraceSystem.traceThrowable(e);
@@ -595,9 +598,11 @@ public class WebServer implements Service {
                     prop.setProperty(String.valueOf(len - i - 1), info.getString());
                 }
             }
+            if (!"null".equals(serverPropertiesDir)) {
                 OutputStream out = IOUtils.openFileOutputStream(serverPropertiesDir + "/" + Constants.SERVER_PROPERTIES_NAME, false);
                 prop.store(out, "H2 Server Properties");
                 out.close();
+            }
         } catch (Exception e) {
             TraceSystem.traceThrowable(e);
         }

h2/src/main/org/h2/tools/Console.java

@@ -170,7 +170,7 @@ ShutdownHandler {
         // because some people don't look at the output,
         // but are wondering why nothing happens
         if (browserStart) {
-            Server.openBrowser(web.getURL());
+            openBrowser(web.getURL());
         }
 
         if (tcpStart) {
@@ -421,12 +421,20 @@ ShutdownHandler {
             long now = System.currentTimeMillis();
             if (lastOpen == 0 || lastOpen + 100 < now) {
                 lastOpen = now;
-                Server.openBrowser(url);
+                openBrowser(url);
             }
         }
     }
 //## AWT end ##
 
+    private void openBrowser(String url) {
+        try {
+            Server.openBrowser(url);
+        } catch (Exception e) {
+            out.println(e.getMessage());
+        }
+    }
+
     /**
      * INTERNAL
      */

h2/src/main/org/h2/tools/Server.java

@@ -6,7 +6,6 @@
  */
 package org.h2.tools;
 
-import java.io.IOException;
 import java.net.URI;
 import java.sql.Connection;
 import java.sql.SQLException;
@@ -21,6 +20,7 @@ import org.h2.server.pg.PgServer;
 import org.h2.server.web.WebServer;
 import org.h2.util.StringUtils;
 import org.h2.util.Tool;
+import org.h2.util.Utils;
 
 /**
  * Starts the H2 Console (web-) server, TCP, and PG server.
@@ -31,6 +31,7 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
     private Service service;
     private Server web, tcp, pg;
     private ShutdownHandler shutdownHandler;
+    private boolean started;
 
     public Server() {
         // nothing to do
@@ -95,7 +96,7 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
      * <tr><td>[-pgPort &lt;port&gt;]</td>
      * <td>The port (default: 5435)</td></tr>
      * <tr><td>[-properties "&lt;dir&gt;"]</td>
-     * <td>The server properties directory (default: ~)</td></tr>
+     * <td>Server properties (default: ~, disable: null)</td></tr>
      * <tr><td>[-baseDir &lt;dir&gt;]</td>
      * <td>The base directory for H2 databases (all servers)</td></tr>
      * <tr><td>[-ifExists]</td>
@@ -140,8 +141,6 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
                     // no parameters
                 } else if ("-webPort".equals(arg)) {
                     i++;
-                } else if ("-webScript".equals(arg)) {
-                    i++;
                 } else {
                     throwUnsupportedOption(arg);
                 }
@@ -221,7 +220,11 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
             // because some people don't look at the output,
             // but are wondering why nothing happens
             if (browserStart) {
-                Server.openBrowser(web.getURL());
+                try {
+                    openBrowser(web.getURL());
+                } catch (Exception e) {
+                    out.println(e.getMessage());
+                }
             }
             if (result != null) {
                 throw result;
@@ -269,7 +272,9 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
      */
     public String getStatus() {
         StringBuilder buff = new StringBuilder();
-        if (isRunning(false)) {
+        if (!started) {
+            buff.append("Not started");
+        } else if (isRunning(false)) {
             buff.append(service.getType()).
                 append(" server running on ").
                 append(service.getURL()).
@@ -344,6 +349,7 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
      */
     public Server start() throws SQLException {
         try {
+            started = true;
             service.start();
             Thread t = new Thread(this);
             t.setDaemon(service.isDaemon());
@@ -404,6 +410,7 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
      * Stops the server.
      */
     public void stop() {
+        started = false;
         service.stop();
     }
 
@@ -468,13 +475,16 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
      *
      * @param url the URL to open
      */
-    public static void openBrowser(String url) {
+    public static void openBrowser(String url) throws Exception {
+        try {
             String osName = SysProperties.getStringSetting("os.name", "linux").toLowerCase();
             Runtime rt = Runtime.getRuntime();
-        try {
-            String browser = SysProperties.BROWSER;
+            String browser = System.getProperty(SysProperties.H2_BROWSER);
             if (browser != null) {
-                if (browser.indexOf("%url") >= 0) {
+                if (browser.startsWith("call:")) {
+                    browser = browser.substring("call:".length());
+                    Utils.callStaticMethod(browser, url);
+                } else if (browser.indexOf("%url") >= 0) {
                     String[] args = StringUtils.arraySplit(browser, ',', false);
                     for (int i = 0; i < args.length; i++) {
                         args[i] = StringUtils.replaceAll(args[i], "%url", url);
@@ -525,12 +535,11 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
                 }
                 if (!ok) {
                     // No success in detection.
-                    System.out.println("Please open a browser and go to " + url);
+                    throw new Exception("Browser detection failed and system property " + SysProperties.H2_BROWSER + " not set");
                 }
             }
-        } catch (IOException e) {
-            System.out.println("Failed to start a browser to open the URL " + url);
-            e.printStackTrace();
+        } catch (Exception e) {
+            throw new Exception("Failed to start a browser to open the URL " + url + ": " + e.getMessage());
         }
     }
 
@@ -550,13 +559,13 @@ public class Server extends Tool implements Runnable, ShutdownHandler {
         server.web = web;
         webServer.setShutdownHandler(server);
         String url = webServer.addSession(conn);
+        try {
             Server.openBrowser(url);
             while (!webServer.isStopped()) {
-            try {
                 Thread.sleep(1000);
-            } catch (InterruptedException e) {
-                // ignore
             }
+        } catch (Exception e) {
+            // ignore
         }
     }
 

h2/src/test/org/h2/test/server/TestWeb.java

@@ -6,14 +6,22 @@
  */
 package org.h2.test.server;
 
+import java.io.ByteArrayOutputStream;
+import java.io.IOException;
+import java.io.PrintStream;
+import java.sql.Connection;
+import org.h2.constant.SysProperties;
 import org.h2.test.TestBase;
 import org.h2.tools.Server;
+import org.h2.util.Task;
 
 /**
  * Tests the H2 Console application.
  */
 public class TestWeb extends TestBase {
 
+    private static volatile String lastUrl;
+
     /**
      * Run just this test.
      *
@@ -24,10 +32,32 @@ public class TestWeb extends TestBase {
     }
 
     public void test() throws Exception {
-        Server server = Server.createWebServer("-webPort", "8182", "-properties", getBaseDir());
+        testAlreadyRunning();
+        testStartWebServerWithConnection();
+        testAutoComplete();
+    }
+
+    private void testAlreadyRunning() throws Exception {
+        Server server = Server.createWebServer("-webPort", "8182", "-properties", "null");
         server.start();
+        assertTrue(server.getStatus().indexOf("server running") >= 0);
+        Server server2 = Server.createWebServer("-webPort", "8182", "-properties", "null");
+        assertEquals("Not started", server2.getStatus());
         try {
-            String url = server.getURL();
+            server2.start();
+            fail();
+        } catch (Exception e) {
+            assertTrue(e.toString().indexOf("port may be in use") >= 0);
+            assertTrue(server2.getStatus().indexOf("could not be started") >= 0);
+        }
+        server.stop();
+    }
+
+    private void testAutoComplete() throws Exception {
+        Server server = new Server();
+        server.setOut(new PrintStream(new ByteArrayOutputStream()));
+        server.runTool("-web", "-webPort", "8182", "-properties", "null", "-tcp", "-tcpPort", "9001");
+        String url = "http://localhost:8182";
         WebClient client = new WebClient();
         String result = client.get(url);
         client.readSessionId(result);
@@ -69,12 +99,12 @@ public class TestWeb extends TestBase {
 
         result = client.get(url, "autoCompleteList.do?query=se");
 
-    //        long time = System.currentTimeMillis();
-    //        for (int i=0; i<1000; i++) {
-    //            if(System.currentTimeMillis()-time > 15000) {
-    //                break;
-    //            }
-    //        result = client.get(url, "autoCompleteList.do?query=select * from ");
+//        long time = System.currentTimeMillis();
+//        for (int i=0; i<1000; i++) {
+//            if(System.currentTimeMillis()-time > 15000) {
+//                break;
+//            }
+//        result = client.get(url, "autoCompleteList.do?query=select * from ");
 
         assertContains(result, "select");
         assertContains(result, "set");
@@ -121,11 +151,65 @@ public class TestWeb extends TestBase {
         result = client.get(url, "settingRemove.do?name=_test_");
 
         client.get(url, "admin.do");
-            // this would also stop the server
-            // client.get(url, "adminShutdown.do");
-        } finally {
+        try {
+            client.get(url, "adminShutdown.do");
+        } catch (IOException e) {
+            // expected
+            Thread.sleep(100);
+        }
+        server.shutdown();
+        // it should be stopped now
+        server = Server.createTcpServer("-tcpPort", "9001");
+        server.start();
         server.stop();
     }
+
+    private void testStartWebServerWithConnection() throws Exception {
+        String old = System.getProperty(SysProperties.H2_BROWSER);
+        try {
+            System.setProperty(SysProperties.H2_BROWSER, "call:" + TestWeb.class.getName() + ".openBrowser");
+            Server.openBrowser("testUrl");
+            assertEquals("testUrl", lastUrl);
+            String oldUrl = lastUrl;
+            final Connection conn = getConnection("testWeb");
+            Task t = new Task() {
+                public void call() throws Exception {
+                    Server.startWebServer(conn);
+                }
+            };
+            t.execute();
+            for (int i = 0; lastUrl == oldUrl; i++) {
+                if (i > 100) {
+                    throw new Exception("Browser not started");
+                }
+                Thread.sleep(100);
+            }
+            String url = lastUrl;
+            WebClient client = new WebClient();
+            client.readSessionId(url);
+            url = client.getBaseUrl(url);
+            try {
+                client.get(url, "logout.do");
+            } catch (Exception e) {
+                // the server stopps on logout
+            }
+            t.get();
+        } finally {
+            if (old != null) {
+                System.setProperty(SysProperties.H2_BROWSER, old);
+            } else {
+                System.clearProperty(SysProperties.H2_BROWSER);
+            }
+        }
+    }
+
+    /**
+     * This method is called via reflection.
+     *
+     * @param url the browser url
+     */
+    public static void openBrowser(String url) {
+        lastUrl = url;
     }
 
 }

h2/src/test/org/h2/test/server/WebClient.java

@@ -46,11 +46,12 @@ public class WebClient {
      * Read the session ID from a URL.
      *
      * @param url the URL
+     * @return the session id
      */
-    void readSessionId(String url) {
+    String readSessionId(String url) {
         int idx = url.indexOf("jsessionid=");
         String id = url.substring(idx + "jsessionid=".length());
-        for (int i = 0; i < url.length(); i++) {
+        for (int i = 0; i < id.length(); i++) {
             char ch = id.charAt(i);
             if (!Character.isLetterOrDigit(ch)) {
                 id = id.substring(0, i);
@@ -58,6 +59,7 @@ public class WebClient {
             }
         }
         this.sessionId = id;
+        return id;
     }
 
     /**
@@ -83,4 +85,13 @@ public class WebClient {
         return get(url);
     }
 
+    String getBaseUrl(String url) {
+        int idx = url.indexOf("//");
+        idx = url.indexOf("/", idx + 2);
+        if (idx >= 0) {
+            return url.substring(0, idx);
+        }
+        return url;
+    }
+
 }

h2/src/test/org/h2/test/unit/TestTools.java

@@ -40,9 +40,9 @@ import org.h2.tools.RunScript;
 import org.h2.tools.Script;
 import org.h2.tools.Server;
 import org.h2.tools.SimpleResultSet;
-import org.h2.util.Task;
 import org.h2.util.IOUtils;
 import org.h2.util.JdbcUtils;
+import org.h2.util.Task;
 
 /**
  * Tests the database tools.