/**
    * The MIT License
    * 
    * Copyright 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.
   */
  package org.codehaus.mojo.appassembler.daemon.daemontools;
  
  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;
  
  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;
  
  /**
   * @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 context = new HashMap ( );
         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 );
         }
     }
 }