package org.apache.maven.plugin.gpg;

import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;

import org.apache.commons.lang.SystemUtils;
import org.apache.maven.plugin.MojoExecutionException;
import org.apache.maven.project.MavenProject;
import org.codehaus.plexus.util.cli.CommandLineException;
import org.codehaus.plexus.util.cli.CommandLineUtils;
import org.codehaus.plexus.util.cli.Commandline;
import org.codehaus.plexus.util.cli.DefaultConsumer;

/*
 * 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.
 */

public class GpgSigner 
{
    public static final String SIGNATURE_EXTENSION = ".asc";
    
    
    private boolean useAgent;
    private boolean isInteractive = true;
    private String keyname;
    private File outputDir;
    private File buildDir;
    private File baseDir;
    
    public GpgSigner()
    {
    }
    
    
    public void setInteractive(boolean b)
    {
        isInteractive = b;
    }
    
    public void setUseAgent(boolean b)
    {
        useAgent = b;
    }
    
    public void setKeyName(String s) 
    {
        keyname = s;
    }
    
    public void setOutputDirectory(File out) 
    {
        outputDir = out;
    }
    public void setBuildDirectory(File out) 
    {
        buildDir = out;
    }
    public void setBaseDirectory(File out) 
    {
        baseDir = out;
    }
    
    
    public File generateSignatureForArtifact( File file , String pass)
        throws MojoExecutionException
    {
        File signature = new File( file + SIGNATURE_EXTENSION );
    
        boolean isInBuildDir = false;
        if ( buildDir != null )
        {
            File parent = signature.getParentFile();
            if ( buildDir.equals(parent) ) 
            {
                isInBuildDir = true;
            }
        }
        if ( !isInBuildDir
            && outputDir != null ) 
        {
            String fileDirectory = "";
            File signatureDirectory = signature;
            
            while( ( signatureDirectory = signatureDirectory.getParentFile() ) != null )
            {
                if( !signatureDirectory.equals( baseDir ) )
                {
                    fileDirectory = signatureDirectory.getName() + File.separatorChar + fileDirectory;
                }
                else
                {
                    break;
                }
            }
            signatureDirectory = new File( outputDir, fileDirectory );
            if( !signatureDirectory.exists() )
            {
                signatureDirectory.mkdirs();
            }
            signature = new File( signatureDirectory, file.getName() + SIGNATURE_EXTENSION );
        }
        
        if ( signature.exists() )
        {
            signature.delete();
        }
        
        Commandline cmd = new Commandline();
    
        cmd.setExecutable( "gpg" + ( SystemUtils.IS_OS_WINDOWS ? ".exe" : "" ) );
    
        if ( useAgent )
        {
            cmd.createArgument().setValue( "--use-agent" );
        }
        else
        {
            cmd.createArgument().setValue( "--no-use-agent" );
        }
    
        InputStream in = null;
        if ( null != pass) 
        {
            cmd.createArgument().setValue( "--passphrase-fd" );
    
            cmd.createArgument().setValue( "0" );
    
            // Prepare the input stream which will be used to pass the passphrase to the executable
            in = new ByteArrayInputStream( pass.getBytes() );
        }
    
        if ( null != keyname)
        {
            cmd.createArgument().setValue( "--local-user" );
    
            cmd.createArgument().setValue( keyname );
        }
    
        cmd.createArgument().setValue( "--armor" );
    
        cmd.createArgument().setValue( "--detach-sign" );
        
        if ( !isInteractive )
        {
            cmd.createArgument().setValue( "--no-tty" );
        }
        
        cmd.createArgument().setValue( "--output" );
        cmd.createArgument().setFile( signature );
        
        cmd.createArgument().setFile( file );
    
    
        try
        {
            int exitCode = CommandLineUtils.executeCommandLine( cmd, in, new DefaultConsumer(), new DefaultConsumer() );
    
            if ( exitCode != 0 )
            {
                throw new MojoExecutionException( "Exit code: " + exitCode );
            }
        }
        catch ( CommandLineException e )
        {
            throw new MojoExecutionException( "Unable to execute gpg command", e );
        }
    
        return signature;
    }
    
    
    private MavenProject findReactorProject(MavenProject prj) {
        if ( prj.getParent() != null )
        {
            if ( prj.getParent().getBasedir() != null && prj.getParent().getBasedir().exists() )
            {
                return findReactorProject( prj.getParent() );
            }
        }
        return prj;
    }
    
    
    public String getPassphrase(MavenProject project) throws IOException
    {
        String pass = null;
        
        if (project != null) {
            pass = project.getProperties().getProperty("gpg.passphrase");
            if (pass == null) 
            {
                MavenProject prj2 = findReactorProject(project);
                pass = prj2.getProperties().getProperty("gpg.passphrase");
            }
        }
        if (pass == null) 
        {
            //TODO: with JDK 1.6, we could call System.console().readPassword("GPG Passphrase: ", null);
            
            BufferedReader in = new BufferedReader(new InputStreamReader(System.in));
            while (System.in.available() != 0)
            {
                //there's some junk already on the input stream, consume it
                //so we can get the real passphrase
                System.in.read();
            }
            
            System.out.print("GPG Passphrase:  ");
            MaskingThread thread = new MaskingThread();
            thread.start();
    
    
            pass = in.readLine();
    
            // stop masking
            thread.stopMasking();
        }
        if (project != null) {
            findReactorProject(project).getProperties().setProperty("gpg.passphrase", pass);
        }
        return pass;
    }
    
    
    // based on ideas from http://java.sun.com/developer/technicalArticles/Security/pwordmask/
    class MaskingThread extends Thread
    {
        private volatile boolean stop;

       /**
        * Begin masking until asked to stop.
        */
        public void run()
        {
            //this needs to be high priority to make sure the characters don't
            //really get to the screen.
            
            int priority = Thread.currentThread().getPriority();
            Thread.currentThread().setPriority(Thread.MAX_PRIORITY);

            try {
                stop = false;
                while(!stop)
                { 
                    //print a backspace + * to overwrite anything they type
                    System.out.print("\010*");
                    try
                    {
                        //attempt masking at this rate
                        Thread.sleep(1);
                    }
                    catch (InterruptedException iex)
                    {
                        Thread.currentThread().interrupt();
                        return;
                    }
                }
            }
            finally
            {
                // restore the original priority
                Thread.currentThread().setPriority(priority);
            }
        }

        /**
         * Instruct the thread to stop masking.
         */
        public void stopMasking() {
            this.stop = true;
        }
    }    
    

}