org.hpccsystems.ws.client.utils.Sftp Maven / Gradle / Ivy

Go to download
package org.hpccsystems.ws.client.utils;

import java.io.File;
import java.io.FileInputStream;
import java.util.Properties;
import java.util.Vector;

import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;

import com.jcraft.jsch.Channel;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.Session;

public class Sftp
{
    private static final Logger log = LogManager.getLogger(Sftp.class);

    static public void lzPut(String localFileName, String hostname, String landingZonePath, String targetFileName, String machineLoginUsername, String password, boolean isLZLinux, Properties connconfig) throws Exception
    {
        char lzSep;
        if (isLZLinux)
            lzSep = Utils.LINUX_SEP;
        else
            lzSep = Utils.WIN_SEP;

        if (hostname == null || hostname.length() == 0)
            throw new Exception("Target HPCC landingzone hostname is required");
        if (landingZonePath == null || landingZonePath.length()==0)
            throw new Exception("Target HPCC landing zone path is required");
        if (localFileName == null || localFileName.length()==0)
            throw new Exception ("Fully qualified local file path is required");

        File fileToUpload = new File(localFileName);
        if (!fileToUpload.isFile())
            throw new Exception ("Target file name does not appear to be a file: " + localFileName);

        if (targetFileName == null || targetFileName.length()==0)
            targetFileName = fileToUpload.getName();

        if (connconfig == null)
            connconfig = new java.util.Properties();

        if (!connconfig.containsKey("StrictHostKeyChecking"))
            connconfig.put("StrictHostKeyChecking", "no");

        Session session = null;
        Channel channel = null;
        try
        {
            JSch ssh = new JSch();
            session = ssh.getSession(machineLoginUsername, hostname, 22);
            session.setConfig(connconfig);
            session.setPassword(password);
            session.connect();

            channel = session.openChannel("sftp");
            channel.connect();

            ChannelSftp sftp = (ChannelSftp) channel;
            Vector files;
            try
            {
                files = (Vector)sftp.ls(landingZonePath);
                if (files == null || files.size() == 0)
                    throw new Exception("Could not excute LS command on target landingzone path: " + landingZonePath);
                else
                    log.debug("Found landing zone " + landingZonePath + " with " + files.size() + " files\n");
            }
            catch (Exception e)
            {
                throw new Exception("Could not contact Target landingzone\n" + e.getLocalizedMessage());
            }

            try
            {
                log.debug("Attempting to sftp " + localFileName + " to " + hostname + ":" + landingZonePath + lzSep +  targetFileName);
                sftp.put(new FileInputStream(fileToUpload), landingZonePath + lzSep + targetFileName );
                log.debug("Finished Attempt to sftp " + localFileName + " to " + hostname + ":" + landingZonePath + lzSep +  targetFileName + " please verify");
            }
            catch (Exception e)
            {
                throw new Exception ("Could not sftp " + localFileName + " to " + hostname + ":" + landingZonePath + lzSep +  targetFileName + "\n" + e.getLocalizedMessage());
            }
        }
        catch (Exception e)
        {
            throw new Exception ("Unexpected problem while attempting to perform LandingZone file transfer\n" + e.getLocalizedMessage());
        }
        finally
        {
            if (channel != null)
                channel.disconnect();
            if (session != null)
                session.disconnect();
        }
    }
}