/**
    *
    * Copyright 2004 The Apache Software Foundation
    *
    *  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 org.apache.geronimo.system.main;
  
  import java.io.IOException;
  import java.net.JarURLConnection;
  import java.net.URI;
  import java.net.URISyntaxException;
  import java.net.URL;
  import java.util.ArrayList;
  import java.util.Collections;
  import java.util.List;
  import java.util.StringTokenizer;
  import java.util.jar.Attributes;
  import java.util.jar.Manifest;
  
  import org.apache.geronimo.gbean.AbstractNameQuery;
  import org.apache.geronimo.kernel.repository.Artifact;
  
  /**
   * @version $Rev: 406440 $ $Date: 2006-05-14 14:45:03 -0700 (Sun, 14 May 2006) $
   */
  public class CommandLineManifest {
      public static final Attributes.Name MAIN_GBEAN = new Attributes.Name("Main-GBean");
      public static final Attributes.Name MAIN_METHOD = new Attributes.Name("Main-Method");
      public static final Attributes.Name CONFIGURATIONS = new Attributes.Name("Configurations");
      public static final Attributes.Name ENDORSED_DIRS = new Attributes.Name("Endorsed-Dirs");
      public static final Attributes.Name EXTENSION_DIRS = new Attributes.Name("Extension-Dirs");
  
      public static CommandLineManifest getManifestEntries() {
          // find the startup jar
          ClassLoader classLoader = CommandLine.class.getClassLoader();
          URL url = classLoader.getResource("META-INF/startup-jar");
          if (url == null) {
              throw new IllegalArgumentException("Unable to determine location of startup jar");
          }
  
          // extract the manifest
          Manifest manifest;
          try {
              JarURLConnection jarConnection = (JarURLConnection) url.openConnection();
              manifest = jarConnection.getManifest();
          } catch (IOException e) {
              System.err.println("Startup jar does not contain a manifest: " + url);
              System.exit(1);
              throw new AssertionError();
          }
          Attributes mainAttributes = manifest.getMainAttributes();
  
          // get the main gbean class
          String mainGBeanString = mainAttributes.getValue(MAIN_GBEAN);
  
          AbstractNameQuery mainGBean = null;
          if (mainGBeanString != null) {
              try {
                  mainGBean = new AbstractNameQuery(new URI(mainGBeanString));
              } catch (URISyntaxException e) {
                  System.err.println("Invalid Main-GBean name: " + mainGBeanString);
                  System.exit(1);
                  throw new AssertionError();
              }
          }
  
          // get the main method
          String mainMethod = mainAttributes.getValue(MAIN_METHOD);
  
          // get the list of extra configurations to load
          List configurations = new ArrayList();
          String configurationsString = mainAttributes.getValue(CONFIGURATIONS);
          if (configurationsString != null) {
              for (StringTokenizer tokenizer = new StringTokenizer(configurationsString, " "); tokenizer.hasMoreTokens();)
              {
                  String configuration = tokenizer.nextToken();
                  configurations.add(Artifact.create(configuration));
              }
          }
  
          // get the list of endorsed directories
          List endorsedDirs = new ArrayList();
          String endorsedDirsString = mainAttributes.getValue(ENDORSED_DIRS);
          if (endorsedDirsString != null) {
              for (StringTokenizer tokenizer = new StringTokenizer(endorsedDirsString, " "); tokenizer.hasMoreTokens();) {
                  String directory = tokenizer.nextToken();
                  endorsedDirs.add(directory);
              }
         }
 
         // get the list of extension directories
         List extensionDirs = new ArrayList();
         String extensionDirsString = mainAttributes.getValue(EXTENSION_DIRS);
         if (extensionDirsString != null) {
             for (StringTokenizer tokenizer = new StringTokenizer(extensionDirsString, " "); tokenizer.hasMoreTokens();)
             {
                 String directory = tokenizer.nextToken();
                 extensionDirs.add(directory);
             }
         }
 
         CommandLineManifest commandLineManifest = new CommandLineManifest(mainGBean, mainMethod, configurations, endorsedDirs, extensionDirs);
         return commandLineManifest;
     }
 
     private final AbstractNameQuery mainGBean;
     private final String mainMethod;
     private final List configurations;
     private final List endorsedDirs;
     private final List extensionDirs;
 
     public CommandLineManifest(AbstractNameQuery mainGBean, String mainMethod, List configurations, List endorsedDirs, List extensionDirs) {
         this.mainGBean = mainGBean;
         this.mainMethod = mainMethod;
         this.configurations = Collections.unmodifiableList(configurations);
         this.endorsedDirs = endorsedDirs;
         this.extensionDirs = extensionDirs;
     }
 
     public AbstractNameQuery getMainGBeanQuery() {
         return mainGBean;
     }
 
     public String getMainMethod() {
         return mainMethod;
     }
 
     public List getConfigurations() {
         return configurations;
     }
 
     public List getEndorsedDirs() {
         return endorsedDirs;
     }
 
     public List getExtensionDirs() {
         return extensionDirs;
     }
 }