package org.apache.maven.wagon.providers.ssh;

 /*

  * 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 org.apache.mina.core.session.IoSession;

 import org.apache.sshd.SshServer;

 import org.apache.sshd.common.Session;

 import org.apache.sshd.common.session.AbstractSession;

 import org.apache.sshd.server.Command;

 import org.apache.sshd.server.CommandFactory;

 import org.apache.sshd.server.FileSystemFactory;

 import org.apache.sshd.server.FileSystemView;

 import org.apache.sshd.server.SshFile;

 import org.apache.sshd.server.auth.UserAuthPassword;

 import org.apache.sshd.server.auth.UserAuthPublicKey;

 import org.apache.sshd.server.filesystem.NativeSshFile;

 import org.apache.sshd.server.keyprovider.PEMGeneratorHostKeyProvider;

 import org.apache.sshd.server.session.SessionFactory;

 import org.apache.sshd.server.shell.ProcessShellFactory;

 import org.codehaus.plexus.util.FileUtils;

 import java.io.File;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.Arrays;

 import java.util.List;

 /**

  * @author Olivier Lamy

  */

 public class SshServerEmbedded

 {

     private String wagonProtocol;

     private int port;

     private SshServer sshd;

     private List<String> sshKeysResources = new ArrayList<String>();

     public TestPublickeyAuthenticator publickeyAuthenticator;

     public TestPasswordAuthenticator passwordAuthenticator = new TestPasswordAuthenticator();

     private boolean keyAuthz;

     /**

      * @param wagonProtocol    scp scpexe

      * @param sshKeysResources paths in the classlaoder with ssh keys

      */

     public SshServerEmbedded( String wagonProtocol, List<String> sshKeysResources, boolean keyAuthz )

     {

         this.wagonProtocol = wagonProtocol;

         this.sshKeysResources = sshKeysResources;

         this.sshd = SshServer.setUpDefaultServer();

         //this.sshd.setKeyExchangeFactories(  );

         this.keyAuthz = keyAuthz;

         publickeyAuthenticator = new TestPublickeyAuthenticator( this.keyAuthz );

     }

     /**

      * @return random port used

      */

     public int start()

         throws IOException

     {

         sshd.setPort( 0 );

         sshd.setUserAuthFactories( Arrays.asList( new UserAuthPublicKey.Factory(), new UserAuthPassword.Factory() ) );

         sshd.setPublickeyAuthenticator( this.publickeyAuthenticator );

         sshd.setPasswordAuthenticator( this.passwordAuthenticator );

         sshd.setUserAuthFactories( Arrays.asList( new UserAuthPublicKey.Factory(), new UserAuthPassword.Factory() ) );

         //ResourceKeyPairProvider resourceKeyPairProvider =

         //    new ResourceKeyPairProvider( sshKeysResources.toArray( new String[sshKeysResources.size()] ) );

         File path = new File( "target/keys" );

         path.mkdirs();

         path = new File( path, "simple.key" );

         path.delete();

         PEMGeneratorHostKeyProvider provider = new PEMGeneratorHostKeyProvider();

         provider.setAlgorithm( "RSA" );

         provider.setKeySize( 1024 );

         provider.setPath( path.getPath() );

         sshd.setKeyPairProvider( provider );

         SessionFactory sessionFactory = new SessionFactory()

         {

             @Override

             protected AbstractSession doCreateSession( IoSession ioSession )

                 throws Exception

             {

                 return super.doCreateSession( ioSession );

             }

         };

         sshd.setSessionFactory( sessionFactory );

         //sshd.setFileSystemFactory(  );

         final ProcessShellFactory processShellFactory =

             new ProcessShellFactory( new String[]{ "/bin/sh", "-i", "-l" } );

         sshd.setShellFactory( processShellFactory );

         CommandFactory delegateCommandFactory = new CommandFactory()

         {

             public Command createCommand( String command )

             {

                 return new ShellCommand( command );

             }

         };

         ScpCommandFactory commandFactory = new ScpCommandFactory( delegateCommandFactory );

         sshd.setCommandFactory( commandFactory );

         FileSystemFactory fileSystemFactory = new FileSystemFactory()

         {

             public FileSystemView createFileSystemView( Session session )

                 throws IOException

             {

                 return new FileSystemView()

                 {

                     public SshFile getFile( String file )

                     {

                         file = file.replace( "\\", "" );

                         file = file.replace( "\"", "" );

                         File f = new File( FileUtils.normalize( file ) );

                         return new SshServerEmbedded.TestSshFile( f.getAbsolutePath(), f,

                                                                   System.getProperty( "user.name" ) );

                     }

                     public SshFile getFile( SshFile baseDir, String file )

                     {

                         file = file.replace( "\\", "" );

                         file = file.replace( "\"", "" );

                         File f = new File( FileUtils.normalize( file ) );

                         return new SshServerEmbedded.TestSshFile( f.getAbsolutePath(), f,

                                                                   System.getProperty( "user.name" ) );

                     }

                 };

             }

         };

         sshd.setNioWorkers( 0 );

         //sshd.setScheduledExecutorService(  );

         sshd.setFileSystemFactory( fileSystemFactory );

         sshd.start();

         this.port = sshd.getPort();

         return this.port;

     }

     public void stop()

         throws InterruptedException

     {

         sshd.stop( Boolean.getBoolean( "sshd.stopImmediatly" ) );

     }

     public int getPort()

     {

         return port;

     }

     public static class TestSshFile

         extends NativeSshFile

     {

         public TestSshFile( String fileName, File file, String userName )

         {

             super( FileUtils.normalize( fileName ), file, userName );

         }

     }

 }