package org.apache.maven.scm.provider.cvslib.cvsjava.util;

 /*

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

  */

 import ch.ethz.ssh2.Connection;

 import ch.ethz.ssh2.Session;

 import ch.ethz.ssh2.StreamGobbler;

 import org.netbeans.lib.cvsclient.CVSRoot;

 import org.netbeans.lib.cvsclient.command.CommandAbortedException;

 import org.netbeans.lib.cvsclient.connection.AbstractConnection;

 import org.netbeans.lib.cvsclient.connection.AuthenticationException;

 import org.netbeans.lib.cvsclient.connection.ConnectionModifier;

 import org.netbeans.lib.cvsclient.util.LoggedDataInputStream;

 import org.netbeans.lib.cvsclient.util.LoggedDataOutputStream;

 import java.io.BufferedReader;

 import java.io.File;

 import java.io.IOException;

 import java.io.InputStream;

 import java.io.InputStreamReader;

 /**

  * @author <a href="mailto:evenisse@apache.org">Emmanuel Venisse</a>

  * @version $Id: ExtConnection.java 685548 2008-08-13 13:35:49Z vsiveton $

  */

 public class ExtConnection

     extends AbstractConnection

 {

     private String host;

     private int port;

     private String userName;

     private String password;

     private Connection connection;

     private Session session;

     private BufferedReader stderrReader;

     public ExtConnection( CVSRoot cvsRoot )

     {

         this( cvsRoot.getHostName(), cvsRoot.getPort(), cvsRoot.getUserName(), cvsRoot.getPassword(),

               cvsRoot.getRepository() );

     }

     public ExtConnection( String host, int port, String username, String password, String repository )

     {

         this.userName = username;

         if ( this.userName == null )

         {

             this.userName = System.getProperty( "user.name" );

         }

         this.password = password;

         this.host = host;

         setRepository( repository );

         this.port = port;

         if ( this.port == 0 )

         {

             this.port = 22;

         }

     }

     /** {@inheritDoc} */

     public void open()

         throws AuthenticationException, CommandAbortedException

     {

         connection = new Connection( host, port );

         /* TODO: add proxy support

         ProxyData proxy = new HTTPProxyData( proxyHost, proxyPort, proxyUserName, proxyPassword );

         connection.setProxyData( proxy );

         */

         try

         {

             // TODO: connection timeout?

             connection.connect();

         }

         catch ( IOException e )

         {

             String message = "Cannot connect. Reason: " + e.getMessage();

             throw new AuthenticationException( message, e, message );

         }

         File privateKey = getPrivateKey();

         try

         {

             boolean authenticated;

             if ( privateKey != null && privateKey.exists() )

             {

                 authenticated = connection.authenticateWithPublicKey( userName, privateKey, getPassphrase() );

             }

             else

             {

                 authenticated = connection.authenticateWithPassword( userName, password );

             }

             if ( !authenticated )

             {

                 String message = "Authentication failed.";

                 throw new AuthenticationException( message, message );

             }

         }

         catch ( IOException e )

         {

             closeConnection();

             String message = "Cannot authenticate. Reason: " + e.getMessage();

             throw new AuthenticationException( message, e, message );

         }

         try

         {

             session = connection.openSession();

         }

         catch ( IOException e )

         {

             String message = "Cannot open session. Reason: " + e.getMessage();

             throw new CommandAbortedException( message, message );

         }

         String command = "cvs server";

         try

         {

             session.execCommand( command );

         }

         catch ( IOException e )

         {

             String message = "Cannot execute remote command: " + command;

             throw new CommandAbortedException( message, message );

         }

         InputStream stdout = new StreamGobbler( session.getStdout() );

         InputStream stderr = new StreamGobbler( session.getStderr() );

         stderrReader = new BufferedReader( new InputStreamReader( stderr ) );

         setInputStream( new LoggedDataInputStream( stdout ) );

         setOutputStream( new LoggedDataOutputStream( session.getStdin() ) );

     }

     /** {@inheritDoc} */

     public void verify()

         throws AuthenticationException

     {

         try

         {

             open();

             verifyProtocol();

             close();

         }

         catch ( Exception e )

         {

             String message = "Failed to verify the connection: " + e.getMessage();

             throw new AuthenticationException( message, e, message );

         }

     }

     private void closeConnection()

     {

         try

         {

             if ( stderrReader != null )

             {

                 while ( true )

                 {

                     String line = stderrReader.readLine();

                     if ( line == null )

                     {

                         break;

                     }

                     System.err.println( line );

                 }

             }

         }

         catch ( IOException e )

         {

             //nothing to do

         }

         if ( session != null )

         {

             System.out.println( "Exit code:" + session.getExitStatus().intValue() );

             session.close();

         }

         if ( connection != null )

         {

             connection.close();

         }

         reset();

     }

     private void reset()

     {

         connection = null;

         session = null;

         stderrReader = null;

         setInputStream( null );

         setOutputStream( null );

     }

     /** {@inheritDoc} */

     public void close()

         throws IOException

     {

         closeConnection();

     }

     /** {@inheritDoc} */

     public boolean isOpen()

     {

         return connection != null;

     }

     /** {@inheritDoc} */

     public int getPort()

     {

         return port;

     }

     /** {@inheritDoc} */

     public void modifyInputStream( ConnectionModifier modifier )

         throws IOException

     {

         modifier.modifyInputStream( getInputStream() );

     }

     /** {@inheritDoc} */

     public void modifyOutputStream( ConnectionModifier modifier )

         throws IOException

     {

         modifier.modifyOutputStream( getOutputStream() );

     }

     private File getPrivateKey()

     {

         // If user don't define a password, he want to use a private key

         File privateKey = null;

         if ( password == null )

         {

             String pk = System.getProperty( "maven.scm.cvs.java.ssh.privateKey" );

             if ( pk != null )

             {

                 privateKey = new File( pk );

             }

             else

             {

                 privateKey = findPrivateKey();

             }

         }

         return privateKey;

     }

     private String getPassphrase()

     {

         String passphrase = System.getProperty( "maven.scm.cvs.java.ssh.passphrase" );

         if ( passphrase == null )

         {

             passphrase = "";

         }

         return passphrase;

     }

     private File findPrivateKey()

     {

         String privateKeyDirectory = System.getProperty( "maven.scm.ssh.privateKeyDirectory" );

         if ( privateKeyDirectory == null )

         {

             privateKeyDirectory = System.getProperty( "user.home" );

         }

         File privateKey = new File( privateKeyDirectory, ".ssh/id_dsa" );

         if ( !privateKey.exists() )

         {

             privateKey = new File( privateKeyDirectory, ".ssh/id_rsa" );

         }

         return privateKey;

     }

 }