Open Source Projects Europe

package net.timbusproject.extractors.modules.tavernaextractor.utils;

import com.jcraft.jsch.*;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.lang.Exception;import java.lang.String;import java.nio.file.Path;
import java.nio.file.Paths;

/**
 * Created by marco unterberger on 28.08.2014.
 *
 * Copy input files from a remote location to use them for the extraction.
 *
 *  * munterberger@sba-research.org
 */
public class SSHManager {

    private final String username;
    private final String password;
    private final String fqdn;
    private final int port;

    public SSHManager(String fqdn, int port, String username, String password){
        this.username = username;
        this.password = password;
        this.fqdn = fqdn;
        this.port = port;
    }

    public Session createSession(int timeout) throws SSHManagerException{

        JSch jsch = new JSch();

        Session session = null;
        try {
            session = jsch.getSession(username, fqdn, port);
            session.setPassword(password);
            session.setConfig("StrictHostKeyChecking", "no");
            session.connect(timeout);

        } catch (JSchException e) {
            e.printStackTrace();
        }
        return session;
    }

    public File readFile(Session session, String path) throws SSHManagerException{

        if(!session.isConnected()){
            try{
                session.connect();
            }
            catch (JSchException e){
                e.printStackTrace();
                throw new SSHManagerException(e.getLocalizedMessage());
            }
        }

        Channel channel = null;
        try {
            channel = session.openChannel("sftp");
            channel.connect();
        } catch (JSchException e) {
            e.printStackTrace();
            throw new SSHManagerException(e.getLocalizedMessage());
        }

        ChannelSftp sftpChannel = (ChannelSftp) channel;

        File file = null;
        try {
            file = File.createTempFile("tavernaExtractor-", ".input");
        } catch (IOException e) {
            e.printStackTrace();
            throw new SSHManagerException(e.getLocalizedMessage());
        }

        try {
            sftpChannel.get(path, file.getAbsolutePath());
        } catch (SftpException e) {
            e.printStackTrace();
            throw new SSHManagerException(e.getLocalizedMessage());
        }

       sftpChannel.disconnect();
       session.disconnect();
       return file;
    }

    public void sendFile(Session session, String from, String to) throws SSHManagerException{

        if(!session.isConnected()){
            try{
                session.connect();
            }
            catch (JSchException e){
                e.printStackTrace();
                throw new SSHManagerException(e.getLocalizedMessage());
            }
        }

        Channel channel = null;
        try {
            channel = session.openChannel("sftp");
            channel.connect();
        } catch (JSchException e) {
            e.printStackTrace();
            throw new SSHManagerException(e.getLocalizedMessage());
        }

        ChannelSftp sftpChannel = (ChannelSftp) channel;

        try {
            sftpChannel.put(from, to);
        } catch (SftpException e) {
            e.printStackTrace();
            throw new SSHManagerException(e.getLocalizedMessage());
        }

        session.disconnect();
    }
  }