it.uniroma2.art.semanticturkey.extension.impl.deployer.sftp.SFTPDeployer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of st-sftp-deployer Show documentation
The newest version!
package it.uniroma2.art.semanticturkey.extension.impl.deployer.sftp;

import java.io.BufferedOutputStream;
import java.io.IOException;
import java.net.SocketAddress;
import java.security.PublicKey;
import java.util.AbstractMap.SimpleImmutableEntry;

import org.apache.sshd.client.SshClient;
import org.apache.sshd.client.session.ClientSession;
import org.apache.sshd.common.config.keys.KeyUtils;
import org.apache.sshd.common.digest.Digest;

import it.uniroma2.art.semanticturkey.extension.extpts.deployer.Deployer;
import it.uniroma2.art.semanticturkey.extension.extpts.deployer.FormattedResourceSource;
import it.uniroma2.art.semanticturkey.extension.extpts.deployer.StreamSourcedDeployer;
import org.apache.sshd.sftp.client.SftpClient;
import org.apache.sshd.sftp.client.SftpClientFactory;

/**
 * Implementation of the {@link Deployer} extension point that uses the SFTP protocol. This implementation can deploy data
 * provided by a {@link FormattedResourceSource}.
 * 
 * Manuel Fiorelli
 */
public class SFTPDeployer implements StreamSourcedDeployer {

	private SFTPDeployerConfiguration conf;

	public SFTPDeployer(SFTPDeployerConfiguration conf) {
		this.conf = conf;
	}

	@Override
	public void deploy(FormattedResourceSource source) throws IOException {
		try (SshClient client = SshClient.setUpDefaultClient()) {
			client.start();

			try (ClientSession session = client.connect(conf.username, conf.host, conf.port)
					.verify(conf.timeout).getSession()) {
				session.setServerKeyVerifier(
						(ClientSession clientSession, SocketAddress remoteAddress, PublicKey serverKey) -> {
							try {
								SimpleImmutableEntry rv = KeyUtils.checkFingerPrint(
										conf.serverKeyFingerprint, (Digest) null, serverKey);
								return rv.getKey();
							} catch (Exception e) {
								return false;
							}
						});
				if (conf.password != null) {
					session.addPasswordIdentity(conf.password);
				}
				session.auth().verify(conf.timeout);

				SftpClient sftpClient = SftpClientFactory.instance().createSftpClient(session);

				try (BufferedOutputStream os = new BufferedOutputStream(
						sftpClient.write(conf.destinationPath, SftpClient.OpenMode.Write, SftpClient.OpenMode.Create))) {
					source.getSourceFormattedResource().writeTo(os);
				}
			}
		}
	}
}