package org.argeo.ssh;
   
   import java.io.IOException;
   import java.net.URI;
   import java.nio.file.FileSystem;
   import java.nio.file.Path;
   
   import org.apache.sshd.client.subsystem.sftp.fs.SftpFileSystem;
   
  /** Create an SFTP {@link FileSystem}. */
  public class Sftp extends AbstractSsh {
  	private URI uri;
  
  	private SftpFileSystem fileSystem;
  
  	public Sftp(String username, String host, int port) {
  		this(AbstractSsh.toUri(username, host, port));
  	}
  
  	public Sftp(URI uri) {
  		this.uri = uri;
  		openSession(uri);
  	}
  
  	public FileSystem getFileSystem() {
  		if (fileSystem == null) {
  			try {
  				authenticate();
  				fileSystem = getSftpFileSystemProvider().newFileSystem(getSession());
  			} catch (IOException e) {
  				throw new IllegalStateException(e);
  			}
  		}
  		return fileSystem;
  	}
  
  	public Path getBasePath() {
  		String p = uri.getPath() != null ? uri.getPath() : "/";
  		return getFileSystem().getPath(p);
  	}
  
  }