com.automationrockstars.gir.cli.SshConnector Maven / Gradle / Ivy

Go to download
package com.automationrockstars.gir.cli;

import java.io.IOException;
import java.nio.file.Path;
import java.util.List;

import com.google.common.collect.Lists;

import net.schmizz.sshj.SSHClient;
import net.schmizz.sshj.transport.verification.HostKeyVerifier;
import java.security.PublicKey;
public class SshConnector {


	private static List clients = Lists.newCopyOnWriteArrayList();

	public static void close(){
		for (SSHClient client : clients){
			try {
				client.close();
			} catch (IOException e) {

			}
		}
	}

	private static SSHClient openConnection(String host) throws IOException{
		final SSHClient ssh = new SSHClient();
        ssh.addHostKeyVerifier(
                new HostKeyVerifier() {
                    @Override
                    public boolean verify(String s, int i, PublicKey publicKey) {
                        return true;
                    }
                });
		clients.add(ssh);
		ssh.connect(host);
		return ssh;
	}

	static SSHClient simpleConnect(String host, String username, String password) throws IOException {
		SSHClient ssh = openConnection(host);
		ssh.authPassword(username, password);
		return ssh;
	}

	static SSHClient simpleConnect(String host, String username, Path authFile) throws IOException{
		SSHClient ssh = openConnection(host);
		ssh.authPublickey(username, new String[]{authFile.toFile().getAbsolutePath()});
		return ssh;
	}

	public static Terminal connect(String host, String username, String password) throws IOException{
		return new Terminal(simpleConnect(host, username, password).startSession());
	}

	public static Terminal connect(String host, String username, Path authFile) throws IOException{
		return new Terminal(simpleConnect(host, username, authFile).startSession());
	}

}