package org.apache.maven.plugin.assembly.utils;

 /*

  * The MIT License

  *

  * Copyright (c) 2004, 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.BufferedReader;

 import java.io.IOException;

 import java.io.InputStreamReader;

 import java.util.Properties;

 /**

  * @author <a href="mailto:trygvis@inamo.no">Trygve Laugst&oslash;l </a>

  * @version $Id: CommandLineUtils.java 728546 2008-12-21 22:56:51Z bentmann $

  */

 public abstract class CommandLineUtils

 {

     /**

      * Return the shell environment variables. If <code>caseSensitive == true</code>, then envar

      * keys will all be upper-case.

      *

      * @param caseSensitive Whether environment variable keys should be treated case-sensitively.

      * @return Properties object of (possibly modified) envar keys mapped to their values.

      * @throws IOException

      */

     public static Properties getSystemEnvVars( boolean caseSensitive )

         throws IOException

     {

         Process p = null;

         Properties envVars = new Properties();

         Runtime r = Runtime.getRuntime();

         String os = System.getProperty( "os.name" ).toLowerCase();

         //If this is windows set the shell to command.com or cmd.exe with correct arguments.

         if ( os.indexOf( "windows" ) != -1 )

         {

             if ( os.indexOf( "95" ) != -1 || os.indexOf( "98" ) != -1 || os.indexOf( "Me" ) != -1 )

             {

                 p = r.exec( "command.com /c set" );

             }

             else

             {

                 p = r.exec( "cmd.exe /c set" );

             }

         }

         else

         {

             p = r.exec( "env" );

         }

         BufferedReader br = new BufferedReader( new InputStreamReader( p.getInputStream() ) );

         String line;

         String lastKey = null;

         String lastVal = null;

         while ( ( line = br.readLine() ) != null )

         {

             int idx = line.indexOf( '=' );

             if ( idx > 1 )

             {

                 lastKey = line.substring( 0, idx );

                 if ( !caseSensitive )

                 {

                     lastKey = lastKey.toUpperCase();

                 }

                 lastVal = line.substring( idx + 1 );

                 envVars.setProperty( lastKey, lastVal );

             }

             else if ( lastKey != null )

             {

                 lastVal += "\n" + line;

                 envVars.setProperty( lastKey, lastVal );

             }

         }

         return envVars;

     }

 }