package org.codehaus.mojo.appassembler.daemon.daemontools;
   
   /*
    * The MIT License
    *
    * Copyright (c) 2006-2012, The Codehaus
    *
    * Permission is hereby granted, free of charge, to any person obtaining a copy of
    * this software and associated documentation files (the "Software"), to deal in
   * the Software without restriction, including without limitation the rights to
   * use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies
   * of the Software, and to permit persons to whom the Software is furnished to do
   * so, subject to the following conditions:
   *
   * The above copyright notice and this permission notice shall be included in all
   * copies or substantial portions of the Software.
   *
   * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
   * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
   * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
   * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
   * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
   * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
   * SOFTWARE.
   */
  
  import java.io.File;
  import java.io.FileWriter;
  import java.io.IOException;
  import java.io.InputStream;
  import java.io.InputStreamReader;
  import java.io.Reader;
  import java.io.Writer;
  import java.util.HashMap;
  import java.util.Map;
  
  import org.codehaus.mojo.appassembler.daemon.DaemonGenerationRequest;
  import org.codehaus.mojo.appassembler.daemon.DaemonGenerator;
  import org.codehaus.mojo.appassembler.daemon.DaemonGeneratorException;
  import org.codehaus.mojo.appassembler.model.Daemon;
  import org.codehaus.plexus.logging.AbstractLogEnabled;
  import org.codehaus.plexus.util.FileUtils;
  import org.codehaus.plexus.util.IOUtil;
  import org.codehaus.plexus.util.InterpolationFilterReader;
  
  /**
   * @author Andrew Williams
   * @version $Id$
   * @plexus.component role-hint="daemontools"
   */
  public class DaemonToolsDaemonGenerator
      extends AbstractLogEnabled
      implements DaemonGenerator
  {
      // -----------------------------------------------------------------------
      // DaemonGenerator Implementation
      // -----------------------------------------------------------------------
  
      /*
       * (non-Javadoc)
       * @see org.codehaus.mojo.appassembler.daemon.DaemonGenerator#generate(org.codehaus.mojo.appassembler.daemon.
       * DaemonGenerationRequest)
       */
      public void generate( DaemonGenerationRequest request )
          throws DaemonGeneratorException
      {
          Daemon daemon = request.getDaemon();
  
          try
          {
              FileUtils.forceMkdir( request.getOutputDirectory() );
          }
          catch ( IOException e )
          {
              throw new DaemonGeneratorException( "Error creating output directory: " + request.getOutputDirectory(), e );
          }
  
          File envDir = new File( request.getOutputDirectory(), "env" );
          envDir.mkdir();
  
          copyEnvFile( "JAVA_HOME", envDir );
          copyEnvFile( "USER", envDir );
  
          File logDir = new File( request.getOutputDirectory(), "logs" );
          logDir.mkdir();
  
          File serviceDir = new File( request.getOutputDirectory(), "service" );
          serviceDir.mkdir();
  
          // -----------------------------------------------------------------------
          //
          // -----------------------------------------------------------------------
  
          InputStream in = this.getClass().getResourceAsStream( "run.sh.template" );
  
          if ( in == null )
          {
              throw new DaemonGeneratorException( "Could not load template." );
          }
 
         InputStreamReader reader = new InputStreamReader( in );
 
         Map<Object, Object> context = new HashMap<Object, Object>();
         context.put( "MAINCLASS", daemon.getMainClass() );
         context.put( "NAME", daemon.getId() );
 
         InterpolationFilterReader interpolationFilterReader = new InterpolationFilterReader( reader, context, "@", "@" );
 
         File runFile = new File( request.getOutputDirectory(), "run" );
         FileWriter out = null;
 
         try
         {
             // -----------------------------------------------------------------------
             // Write the file
             // -----------------------------------------------------------------------
 
             out = new FileWriter( runFile );
 
             IOUtil.copy( interpolationFilterReader, out );
         }
         catch ( IOException e )
         {
             throw new DaemonGeneratorException( "Error writing output file: " + runFile.getAbsolutePath(), e );
         }
         finally
         {
             IOUtil.close( interpolationFilterReader );
             IOUtil.close( out );
         }
 
     }
 
     private void copyEnvFile( String envName, File envDir )
         throws DaemonGeneratorException
     {
         Writer out = null;
         Reader envReader = null;
 
         File envFile = new File( envDir, envName );
 
         try
         {
             envReader = new InputStreamReader( this.getClass().getResourceAsStream( "env/" + envName ) );
 
             // -----------------------------------------------------------------------
             // Write the file
             // -----------------------------------------------------------------------
 
             out = new FileWriter( envFile );
 
             IOUtil.copy( envReader, out );
         }
         catch ( IOException e )
         {
             throw new DaemonGeneratorException( "Error writing environment file: " + envFile, e );
         }
         finally
         {
             IOUtil.close( envReader );
             IOUtil.close( out );
         }
     }
 }