X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?p=it%2Fotf.git;a=blobdiff_plain;f=otf-camunda%2Fsrc%2Fmain%2Fjava%2Forg%2Foran%2Fotf%2Fcommon%2Futility%2Fsftp%2FSftpUtility.java;fp=otf-camunda%2Fsrc%2Fmain%2Fjava%2Forg%2Foran%2Fotf%2Fcommon%2Futility%2Fsftp%2FSftpUtility.java;h=cdaf8fd090b5c6dedde79f4d115425b2673cb794;hp=0000000000000000000000000000000000000000;hb=14f6f95c84a4a1fa8774190db4a03fd0214ec55f;hpb=f49bd1efeaaddd4891c1f329b18d8cfb28b3e75b

diff --git a/otf-camunda/src/main/java/org/oran/otf/common/utility/sftp/SftpUtility.java b/otf-camunda/src/main/java/org/oran/otf/common/utility/sftp/SftpUtility.java
new file mode 100644
index 0000000..cdaf8fd
--- /dev/null
+++ b/otf-camunda/src/main/java/org/oran/otf/common/utility/sftp/SftpUtility.java
@@ -0,0 +1,94 @@
+/*  Copyright (c) 2019 AT&T Intellectual Property.                             #
+#                                                                              #
+#   Licensed under the Apache License, Version 2.0 (the "License");            #
+#   you may not use this file except in compliance with the License.           #
+#   You may obtain a copy of the License at                                    #
+#                                                                              #
+#       http://www.apache.org/licenses/LICENSE-2.0                             #
+#                                                                              #
+#   Unless required by applicable law or agreed to in writing, software        #
+#   distributed under the License is distributed on an "AS IS" BASIS,          #
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.   #
+#   See the License for the specific language governing permissions and        #
+#   limitations under the License.                                             #
+##############################################################################*/
+
+
+package org.oran.otf.common.utility.sftp;
+
+import org.apache.commons.io.IOUtils;
+import org.apache.commons.vfs2.*;
+import org.apache.commons.vfs2.provider.sftp.IdentityInfo;
+import org.apache.commons.vfs2.provider.sftp.SftpFileSystemConfigBuilder;
+
+import java.io.File;
+import java.io.InputStream;
+
+
+public class SftpUtility {
+
+    public static byte[] getFile(String host, String artifactPath, String privateKeyPath, String privateKeyUsername, String privateKeyPasspharase) throws Exception {
+        String remoteURI = "sftp://" + privateKeyUsername + "@" + host + "/" + artifactPath;
+
+        FileSystemOptions fsOptions = new FileSystemOptions();
+        FileSystemManager fsManager = null;
+        byte[] bytes = null;
+        SftpFileSystemConfigBuilder builder = SftpFileSystemConfigBuilder.getInstance();
+        builder.setUserDirIsRoot(fsOptions, false);
+        builder.setStrictHostKeyChecking(fsOptions, "no");
+        IdentityInfo identityInfo = new IdentityInfo(new File(privateKeyPath), privateKeyPasspharase.getBytes());
+        builder.setIdentityInfo(fsOptions, identityInfo);
+        fsManager = VFS.getManager();
+        FileObject remoteFileObject = fsManager.resolveFile(remoteURI, fsOptions);
+        if(!remoteFileObject.isFile()) {
+            remoteFileObject.close();
+            throw new Exception("Expected a file, but supplied filePath was not a file.");
+        }
+        InputStream is = remoteFileObject.getContent().getInputStream();
+        bytes = IOUtils.toByteArray(is);
+        remoteFileObject.close();
+        return bytes;
+
+    }
+
+    public static FileObject getDirectory(String host, String artifactPath, String privateKeyPath, String privateKeyUsername, String privateKeyPasspharase) throws Exception {
+        String remoteURI = "sftp://" + privateKeyUsername + "@" + host + "/" + artifactPath;
+
+        FileSystemOptions fsOptions = new FileSystemOptions();
+        FileSystemManager fsManager = null;
+        SftpFileSystemConfigBuilder builder = SftpFileSystemConfigBuilder.getInstance();
+        builder.setUserDirIsRoot(fsOptions, false);
+        builder.setStrictHostKeyChecking(fsOptions, "no");
+        IdentityInfo identityInfo = new IdentityInfo(new File(privateKeyPath), privateKeyPasspharase.getBytes());
+        builder.setIdentityInfo(fsOptions, identityInfo);
+        fsManager = VFS.getManager();
+        FileObject fileObject = fsManager.resolveFile(remoteURI, fsOptions);
+        if(fileObject.isFolder()) {
+            return fileObject;
+        }
+        fileObject.close();
+        throw new Exception("Expected a folder, but supplied filePath was not a folder.");
+    }
+
+    public static void uploadFile(String host, String artifactPath, String privateKeyPath, String privateKeyUsername, String privateKeyPasspharase, File tempFile) throws Exception {
+        String remoteURI = "sftp://" + privateKeyUsername + "@" + host + "/" + artifactPath;
+
+        FileSystemOptions fsOptions = new FileSystemOptions();
+        FileSystemManager fsManager = null;
+        SftpFileSystemConfigBuilder builder = SftpFileSystemConfigBuilder.getInstance();
+        builder.setUserDirIsRoot(fsOptions, false);
+        builder.setStrictHostKeyChecking(fsOptions, "no");
+        IdentityInfo identityInfo = new IdentityInfo(new File(privateKeyPath), privateKeyPasspharase.getBytes());
+        builder.setIdentityInfo(fsOptions, identityInfo);
+        fsManager = VFS.getManager();
+        //resolve file location
+        FileObject remoteFileObject = fsManager.resolveFile(remoteURI, fsOptions);
+        FileObject sourceFile = fsManager.resolveFile(tempFile.getAbsolutePath());
+        //if file exists then override, else create file
+        remoteFileObject.copyFrom(sourceFile, Selectors.SELECT_SELF);
+        remoteFileObject.close();
+        sourceFile.close();
+    }
+
+
+}