1 files changed, 125 insertions, 0 deletions
diff --git a/trunk/infrastructure/net.appjet.oui/monitoring.scala b/trunk/infrastructure/net.appjet.oui/monitoring.scala
new file mode 100644
index 0000000..e380b84
--- /dev/null
+++ b/trunk/infrastructure/net.appjet.oui/monitoring.scala
@@ -0,0 +1,125 @@
+/**
+ * Copyright 2009 Google Inc.
+ * 
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS-IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package net.appjet.oui;
+
+object monitoring {
+
+  def startMonitoringServer() {
+    // remote JMX monitoring
+    // see: http://java.sun.com/javase/6/docs/technotes/guides/management/agent.html
+
+    import java.rmi.registry.LocateRegistry;
+    import java.lang.management.ManagementFactory;
+    import javax.management.ObjectName;
+    import javax.management.remote.{JMXServiceURL, JMXConnectorServerFactory};
+    
+    def REGISTRY_PORT = config.listenMonitoringPrimaryPort;
+    def SECONDARY_PORT = config.listenMonitoringSecondaryPort;
+    System.setProperty("java.rmi.server.randomIDs", "true");
+    
+    // we must set 'java.rmi.server.hostname' to the host that the client machine
+    // should connect to; in production, it will be the dashboard host, but the property
+    // can also be specified on the command-line, in which case it takes precedence.
+    var listenHost = config.listenMonitoringHost
+    if (listenHost == null || listenHost.length == 0) listenHost = "localhost";
+    if (System.getProperty("java.rmi.server.hostname") == null) {
+      System.setProperty("java.rmi.server.hostname", listenHost);
+    }
+    else {
+      listenHost = System.getProperty("java.rmi.server.hostname");
+    }
+    
+    LocateRegistry.createRegistry(REGISTRY_PORT);
+    val mbs = ManagementFactory.getPlatformMBeanServer();
+
+    mbs.createMBean(classOf[JSExecutor].getName, new ObjectName("net.appjet:type=JSExecutor"));
+    
+    val env = new java.util.HashMap[String,Object]();
+    //val csf = new javax.rmi.ssl.SslRMIClientSocketFactory();
+    //val ssf = new javax.rmi.ssl.SslRMIServerSocketFactory();
+    //env.put(javax.management.remote.rmi.RMIConnectorServer.RMI_CLIENT_SOCKET_FACTORY_ATTRIBUTE, csf);
+    //env.put(javax.management.remote.rmi.RMIConnectorServer.RMI_SERVER_SOCKET_FACTORY_ATTRIBUTE, ssf);
+    val PASSWORD_FILE_PATH = "data/jconsole-password.properties";
+    val ACCESS_FILE_PATH = "data/jconsole-access.properties";
+    def writeStringToFile(str: String, path: String) {
+      val out = new java.io.PrintStream(new java.io.BufferedOutputStream(
+	new java.io.FileOutputStream(path)));
+      out.println(str);
+      out.close;
+    }
+    if (! new java.io.File(PASSWORD_FILE_PATH).exists) {
+      System.err.println("Creating "+PASSWORD_FILE_PATH+"...");
+      writeStringToFile("appjet foo", PASSWORD_FILE_PATH);
+    }
+    if (! new java.io.File(ACCESS_FILE_PATH).exists) {
+      System.err.println("Creating "+ACCESS_FILE_PATH+"...");
+      writeStringToFile("appjet readwrite", ACCESS_FILE_PATH);
+    }
+    env.put("jmx.remote.x.password.file", PASSWORD_FILE_PATH);
+    env.put("jmx.remote.x.access.file", ACCESS_FILE_PATH);
+    val url = new JMXServiceURL(
+      "service:jmx:rmi://localhost:"+SECONDARY_PORT+"/jndi/rmi://localhost:"+REGISTRY_PORT+"/server");
+    try {
+      val cs = JMXConnectorServerFactory.newJMXConnectorServer(url, env, mbs);
+      cs.start();
+      System.err.println("Monitor server listening on "+listenHost+":{"+REGISTRY_PORT+
+	","+SECONDARY_PORT+"}");
+    }
+    catch {
+      case e => {
+	System.err.println("!!Could not start monitor server on "+listenHost+":{"+REGISTRY_PORT+
+	  ","+SECONDARY_PORT+"} due to:");
+	e.printStackTrace(System.err);
+      }
+    }
+  }
+  
+}
+
+trait JSExecutorMXBean {
+  def executeJS(code: String): String;
+}
+
+class JSExecutor extends JSExecutorMXBean {
+  import org.mozilla.javascript.{Context,ContextFactory,ContextAction};
+  import org.mozilla.javascript.tools.ToolErrorReporter;
+  import org.mozilla.javascript.tools.shell.{Global, ShellContextFactory};
+  
+  def executeJS(code: String): String = {
+    val outStream = new java.io.ByteArrayOutputStream;
+    val out = new java.io.PrintStream(outStream, true, "UTF-8");
+
+    val contextFactory = new ShellContextFactory;
+    try {
+      contextFactory.call(new ContextAction { def run(cx: Context): Object = {
+	val global = new Global(cx);
+	global.setOut(out);
+	global.setErr(out);
+	val errorReporter = new ToolErrorReporter(false, global.getErr);
+	val result = cx.evaluateString(global, code, "<script>", 1, null);
+	out.println(Context.toString(result));
+	null;
+      } });
+    }
+    catch {
+      case e => {
+	e.printStackTrace(out);
+      }
+    }
+    return new String(outStream.toByteArray, "UTF-8");
+  }
+}