package org.argeo.ssh;
   
   import java.net.URI;
   import java.util.ArrayList;
   import java.util.List;
   
   import org.apache.commons.cli.CommandLine;
   import org.apache.commons.cli.CommandLineParser;
   import org.apache.commons.cli.DefaultParser;
  import org.apache.commons.cli.HelpFormatter;
  import org.apache.commons.cli.Option;
  import org.apache.commons.cli.Options;
  
  /** Create an SSH shell. */
  public class Ssh extends AbstractSsh {
  	private final URI uri;
  
  	public Ssh(String username, String host, int port) {
  		this(AbstractSsh.toUri(username, host, port));
  	}
  
  	public Ssh(URI uri) {
  		this.uri = uri;
  		openSession(uri);
  	}
  
  	public static void main(String[] args) {
  		Options options = getOptions();
  		CommandLineParser parser = new DefaultParser();
  		try {
  			CommandLine line = parser.parse(options, args);
  			List<String> remaining = line.getArgList();
  			if (remaining.size() == 0) {
  				System.err.println("There must be at least one argument");
  				printHelp(options);
  				System.exit(1);
  			}
  			URI uri = new URI("ssh://" + remaining.get(0));
  			List<String> command = new ArrayList<>();
  			if (remaining.size() > 1) {
  				for (int i = 1; i < remaining.size(); i++) {
  					command.add(remaining.get(i));
  				}
  			}
  
  			// auth
  			Ssh ssh = new Ssh(uri);
  			ssh.authenticate();
  
  			if (command.size() == 0) {// shell
  				AbstractSsh.openShell(ssh.getSession());
  			} else {// execute command
  
  			}
  			ssh.closeSession();
  		} catch (Exception exp) {
  			exp.printStackTrace();
  			printHelp(options);
  			System.exit(1);
  		} finally {
  
  		}
  	}
  
  	public URI getUri() {
  		return uri;
  	}
  
  	public static Options getOptions() {
  		Options options = new Options();
  //		options.addOption("p", true, "port");
  		options.addOption(Option.builder("p").hasArg().argName("port").desc("port of the SSH server").build());
  
  		return options;
  	}
  
  	public static void printHelp(Options options) {
  		HelpFormatter formatter = new HelpFormatter();
  		formatter.printHelp("ssh [username@]hostname", options, true);
  	}
  }