1 files changed, 346 insertions, 0 deletions
diff --git a/sca-java-2.x/tags/2.0.1-RC1/modules/launcher/src/main/java/org/apache/tuscany/sca/launcher/LauncherMain.java b/sca-java-2.x/tags/2.0.1-RC1/modules/launcher/src/main/java/org/apache/tuscany/sca/launcher/LauncherMain.java
new file mode 100644
index 0000000000..275e0a75c7
--- /dev/null
+++ b/sca-java-2.x/tags/2.0.1-RC1/modules/launcher/src/main/java/org/apache/tuscany/sca/launcher/LauncherMain.java
@@ -0,0 +1,346 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you 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 org.apache.tuscany.sca.launcher;
+
+import java.io.File;
+import java.io.FileFilter;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.FilenameFilter;
+import java.io.IOException;
+import java.lang.reflect.InvocationTargetException;
+import java.lang.reflect.Method;
+import java.net.MalformedURLException;
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.net.URL;
+import java.net.URLClassLoader;
+import java.util.Enumeration;
+import java.util.HashSet;
+import java.util.Properties;
+import java.util.Set;
+import java.util.jar.JarFile;
+import java.util.jar.Manifest;
+
+public class LauncherMain {
+    
+    private static final String DEFAULT_PROPERTY_FILENAME = "default.config";
+    private static final String CONFIG_CLASSPATH = "classpath";
+    private static final String CONFIG_MAIN_CLASS = "mainClass";
+    private static final String CONFIG_ARG_JAR_MAIN = "[firstArgJarManifestMainClass]";
+    private static final String LAUNCHER_ARGS = "launcherArgs";
+
+    public static void main(String[] args) throws SecurityException, IllegalArgumentException, ClassNotFoundException, NoSuchMethodException, IllegalAccessException, InvocationTargetException, URISyntaxException, IOException {
+        Properties launcherProperties = getLauncherProperties(args);
+        setSystemProperties(launcherProperties);
+        ClassLoader classLoader = getClassLoader(launcherProperties);
+        String mainClassName = getMainClass(launcherProperties, classLoader);
+        String[] mainArgs = getMainArgs(launcherProperties);
+        invokeMainMethod(mainClassName, classLoader, mainArgs);
+    }
+
+    private static void setSystemProperties(Properties launcherProperties) throws URISyntaxException {
+        for (Enumeration<?> e = launcherProperties.propertyNames(); e.hasMoreElements();) {
+            String pn = (String) e.nextElement();
+            if (pn.startsWith("-D")) {
+                System.setProperty(pn.substring(2), keywordExpand(launcherProperties.getProperty(pn)));
+            }
+        }
+    }
+
+    private static String keywordExpand(String property) throws URISyntaxException {
+        if (property.contains("{TUSCANY_HOME}")) {
+            property = property.replace("{TUSCANY_HOME}", getLauncherFolder().getParentFile().getAbsolutePath());
+        }
+        return property;
+    }
+
+    private static String[] getMainArgs(Properties launcherProperties) {
+        String[] mainArgs = (String[])launcherProperties.get(LAUNCHER_ARGS);
+        if (mainArgs == null) {
+            mainArgs = new String[0];
+        }
+        return mainArgs;
+    }
+
+    private static String getMainClass(Properties launcherProperties, ClassLoader classLoader) {
+        String mainClassName;
+        String[] args = getMainArgs(launcherProperties);
+        if (args.length > 0) {
+            try {
+                Class.forName(args[0], true, classLoader);
+                mainClassName = args[0];
+                String[] args2 = new String[args.length-1];
+                System.arraycopy(args, 1, args2, 0, args.length-1);
+                launcherProperties.put(LAUNCHER_ARGS, args2);
+            } catch (ClassNotFoundException e) {
+                mainClassName = launcherProperties.getProperty(CONFIG_MAIN_CLASS);
+            }
+        } else {
+            mainClassName = launcherProperties.getProperty(CONFIG_MAIN_CLASS);
+        }
+
+        if (mainClassName.startsWith(CONFIG_ARG_JAR_MAIN + "|")) {
+            mainClassName = mainClassName.substring(CONFIG_ARG_JAR_MAIN.length()+1);
+        }
+        
+        return mainClassName;
+    }
+
+    private static void invokeMainMethod(String className, ClassLoader classLoader, String[] args) throws ClassNotFoundException, SecurityException, NoSuchMethodException, IllegalArgumentException, IllegalAccessException, InvocationTargetException {
+        ClassLoader tccl = Thread.currentThread().getContextClassLoader();
+        try {
+
+            Thread.currentThread().setContextClassLoader(classLoader);
+            Class mainClass = Class.forName(className, true, classLoader);
+            Method m = mainClass.getMethod("main", new Class[]{ args.getClass() });
+
+            m.invoke(null, new Object[]{args});
+            
+        } finally {
+            Thread.currentThread().setContextClassLoader(tccl);
+        }
+    }
+
+    private static ClassLoader getClassLoader(Properties launcherProperties) throws URISyntaxException, IOException {
+        Set<URL> jarURLs = new HashSet<URL>(); 
+        for (Enumeration<?> e = launcherProperties.propertyNames(); e.hasMoreElements();) {
+            String pn = (String) e.nextElement();
+            if (pn.startsWith(CONFIG_CLASSPATH)) {
+                jarURLs.addAll(getJARs(launcherProperties.getProperty(pn), launcherProperties));
+            } else if (pn.equals(CONFIG_MAIN_CLASS) && launcherProperties.getProperty(pn).startsWith(CONFIG_ARG_JAR_MAIN)) {
+                if (firstArgJarHasManifestMainClass(launcherProperties)) {
+                    jarURLs.add(firstArgJarManifestMainClass(launcherProperties));
+                }
+            }
+        }
+        ClassLoader parentCL = Thread.currentThread().getContextClassLoader();
+        if (parentCL == null) {
+            parentCL = LauncherMain.class.getClassLoader();
+        }
+        return new URLClassLoader(jarURLs.toArray(new URL[]{}), parentCL);
+    }
+
+    private static void argJars(Set<URL> jarURLs, Properties launcherProperties) throws IOException {
+        String[] args = (String[])launcherProperties.get(LAUNCHER_ARGS);
+        if (args != null && args.length > 0) {
+            for (String a : args) {
+                File f = new File(a);
+                if (!f.exists()) {
+                    throw new FileNotFoundException(args[0]);
+                }
+                jarURLs.add(f.toURI().toURL());
+            }
+        }
+        launcherProperties.put(LAUNCHER_ARGS, new String[]{});
+    }
+
+    private static URL firstArgJarManifestMainClass(Properties launcherProperties) throws IOException {
+        String[] args = (String[])launcherProperties.get(LAUNCHER_ARGS);
+        if (args.length < 1) {
+            throw new IllegalArgumentException("must specifiy a jar file");
+        }
+        File f = new File(args[0]);
+        if (!f.exists()) {
+            throw new FileNotFoundException(args[0]);
+        }
+        JarFile jar = new JarFile(f);
+        String mfc = jar.getManifest().getMainAttributes().getValue("Main-Class");
+        if (mfc == null || mfc.length() < 1) {
+            throw new IllegalArgumentException("first jar file missing manifest Main-Class attribute");
+        }
+        launcherProperties.setProperty(CONFIG_MAIN_CLASS, mfc);
+        
+        return f.toURL();
+    }
+
+    private static boolean firstArgJarHasManifestMainClass(Properties launcherProperties) throws IOException {
+        String[] args = (String[])launcherProperties.get(LAUNCHER_ARGS);
+        if (args.length < 1) {
+            return false;
+        }
+        File f = new File(args[0]);
+        if (!f.exists() || f.isDirectory()) {
+            return false;
+        }
+        JarFile jar = new JarFile(f);
+        Manifest mf = jar.getManifest();
+        if (mf != null) {
+            String mfc = jar.getManifest().getMainAttributes().getValue("Main-Class");
+            return mfc != null && mfc.length() > 0;
+        } else {
+            return false;
+        }
+    }
+    /**
+     * Gets the jars matching a config classpath property
+     * property values may be an explicit jar name or use an asterix wildcard for
+     * all jars in a folder, or a double asterix '**' for all jars in a folder and its subfolders
+     * @throws URISyntaxException 
+     */
+    private static Set<URL> getJARs(String classpathValue, Properties launcherProperties) throws URISyntaxException {
+        Set<URL> jarURLs = new HashSet<URL>();
+        
+        if (classpathValue.endsWith("**")) {
+            File folder = new File(classpathValue.substring(0, classpathValue.length()-2));
+            if (!folder.isAbsolute()) {
+                folder = new File(getLauncherFolder().getParent(), folder.getName());
+            }
+            jarURLs.addAll(getFolderJars(folder));
+            jarURLs.addAll(getSubFolderJars(folder));
+        } else if (classpathValue.endsWith("*")) {
+            File folder = new File(classpathValue.substring(0, classpathValue.length()-1));
+            if (!folder.isAbsolute()) {
+                folder = new File(getLauncherFolder(), folder.getName());
+            }
+            jarURLs.addAll(getFolderJars(folder));
+        } else {
+            File f = new File(classpathValue);
+            if (!f.isAbsolute()) {
+                f = new File(getLauncherFolder(), classpathValue);
+            }
+            try {
+                jarURLs.add(f.toURI().toURL());
+            } catch (MalformedURLException e) {
+                throw new RuntimeException("Exception getting JAR URL", e);
+            }
+        }
+        return jarURLs;
+    }
+
+    /**
+     * Gets all the jars in a folder
+     */
+    private static Set<URL> getFolderJars(File folder) {
+        Set<URL> jarURLs = new HashSet<URL>(); 
+        if (folder.exists() && folder.isDirectory()) {
+            File[] files = folder.listFiles(new FilenameFilter(){
+                public boolean accept(File dir, String name) {
+                    return name.endsWith(".jar");
+                }});
+            for (File f : files) {
+                try {
+                    jarURLs.add(f.toURI().toURL());
+                } catch (MalformedURLException e) {
+                    throw new RuntimeException("Exception getting JAR URL", e);
+                }
+            }
+        }
+        return jarURLs;
+    }
+
+    /**
+     * Recursively gets all the jars in a folder and its subfolders
+     */
+    private static Set<URL> getSubFolderJars(File folder) {
+        Set<URL> jarURLs = new HashSet<URL>(); 
+        if (folder.exists() && folder.isDirectory()) {
+            File[] files = folder.listFiles(new FileFilter(){
+                public boolean accept(File pathname) {
+                    return pathname.isDirectory();
+                }});
+            for (File f : files) {
+                jarURLs.addAll(getFolderJars(f));
+                jarURLs.addAll(getSubFolderJars(f));
+            }
+        }
+        return jarURLs;
+    }
+
+    /**
+     * Read the config properties for this launcher invocation
+     * (Either default.config or the 1st cmd line argument suffixed with ".config" if that file exists 
+     */
+    private static Properties getLauncherProperties(String[] args) throws URISyntaxException, FileNotFoundException {
+
+        Properties properties = new Properties();
+        
+        File f;
+        if (args.length > 0) {
+            f = new File(getLauncherFolder(), args[0] + ".config");
+            if (f.exists()) {
+                String[] args2 = new String[args.length-1];
+                System.arraycopy(args, 1, args2, 0, args.length-1);
+                args = args2;
+            } else {
+                f = new File(getLauncherFolder(), DEFAULT_PROPERTY_FILENAME);
+            }
+        } else {
+            f = new File(getLauncherFolder(), DEFAULT_PROPERTY_FILENAME);
+        }
+        
+        if (!f.exists()) {
+//            throw new FileNotFoundException(f.getName());
+            if (args.length > 0 && "client".equals(args[0])) {
+                properties.put("mainClass", "org.apache.tuscany.sca.client.javascript.TuscanyShell");
+                String[] args2 = new String[args.length-1];
+                System.arraycopy(args, 1, args2, 0, args.length-1);
+                args = args2;
+            } else {
+                properties.put("mainClass", "[firstArgJarManifestMainClass]|org.apache.tuscany.sca.domain.node.DomainNodeMain");
+            }
+        } else {
+            try {
+                FileInputStream fis = new FileInputStream(f);
+                properties.load(fis);
+                fis.close();
+            } catch (IOException e) {
+                throw new RuntimeException(e);
+            }
+        }
+
+        properties.put(LAUNCHER_ARGS, args);
+        
+        return properties;
+    }
+
+    /**
+     * Find the folder that contains the launcher jar
+     */
+    private static File getLauncherFolder() throws URISyntaxException {
+
+        File folder = null;
+
+        String resource = LauncherMain.class.getName().replace('.', '/') + ".class"; 
+        URL url = LauncherMain.class.getClassLoader().getResource(resource);
+        if (url != null) {
+            URI uri = url.toURI();
+            String scheme = uri.getScheme();
+            if (uri.getScheme().equals("jar")) {
+                String path = uri.toString().substring(4);
+                int i = path.indexOf("!/");
+                if (i != -1) {
+                    path = path.substring(0, i);
+                    uri = URI.create(path);
+                }
+        
+                File file = new File(uri);
+                if (file.exists()) {
+                    File jarDirectory = file.getParentFile();
+                    if (jarDirectory != null && jarDirectory.exists()) {
+                        folder = file;
+                    }
+                }
+            }
+        }
+        folder = folder.getParentFile();
+        return folder;
+    }
+}