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

Go to download
/*
 * 
 */

package com.automationrockstars.gir.cli;

import com.google.common.base.Throwables;
import net.schmizz.sshj.common.IOUtils;
import net.schmizz.sshj.common.StreamCopier;
import net.schmizz.sshj.connection.channel.direct.Session;
import net.schmizz.sshj.connection.channel.direct.Session.Command;
import net.schmizz.sshj.connection.channel.direct.Session.Shell;

import java.io.ByteArrayOutputStream;
import java.io.Closeable;
import java.io.IOException;
import java.util.concurrent.TimeUnit;

public class Terminal implements Closeable {

    final ByteArrayOutputStream output = new ByteArrayOutputStream();
    final ByteArrayOutputStream error = new ByteArrayOutputStream();
    private final Session session;
    Shell shell;
    Terminal(Session session) {
        this.session = session;
        try {
            session.allocateDefaultPTY();
            shell = session.startShell();
            new StreamCopier(shell.getInputStream(), output, net.schmizz.sshj.common.LoggerFactory.DEFAULT)
                    .bufSize(shell.getLocalMaxPacketSize())
                    .spawn("stdout");
            new StreamCopier(shell.getErrorStream(), error, net.schmizz.sshj.common.LoggerFactory.DEFAULT)
                    .bufSize(shell.getLocalMaxPacketSize())
                    .spawn("stderr");

        } catch (IOException e) {
            Throwables.propagate(e);
        }

    }

    void write(String text) {
        try {
            shell.getOutputStream().write(text.getBytes());
        } catch (IOException e) {
            Throwables.propagate(e);
        }
    }

    ExecResult exec(String command, long timeout) throws IOException {
        final Command cmd = session.exec(command);
        String result = IOUtils.readFully(cmd.getInputStream()).toString();
        String error = IOUtils.readFully(cmd.getErrorStream()).toString();
        cmd.join(timeout, TimeUnit.MILLISECONDS);
        int exit = cmd.getExitStatus();
        return new ExecResult(result, error, exit);
    }

    String getStdOut() {
        return output.toString();
    }

    String getStdErr() {
        return error.toString();
    }

    @Override
    public void close() throws IOException {
        session.close();
    }


}