package com.tianyang.task.utils;

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

import ch.ethz.ssh2.Connection;

import com.jcraft.jsch.*;
import com.tianyang.biz.constants.TYEnumConstants;
import com.tianyang.task.constants.TYTaskConstants;
 
import org.apache.log4j.Logger;

public class SftpClientUtil {
    protected static Logger logger = Logger.getLogger(SftpClientUtil.class);

    /** 主機 */
    private final static String host = TYTaskConstants.tyTaskConfig.getACQUISITION_SYSTEM_SFTP();
    /** 埠 */
    private final static int port = Integer.valueOf(TYTaskConstants.tyTaskConfig.getACQUISITION_SYSTEM_SFTP_PORT());
 
    /** 使用者名稱 */
    private final static String username =  TYTaskConstants.tyTaskConfig.getACQUISITION_SYSTEM_USERNAME();
    /** 密碼 */
    private final static String password = TYTaskConstants.tyTaskConfig.getACQUISITION_SYSTEM_PASSWORD();
    /** 目錄 */
    private final static String directory = TYTaskConstants.tyTaskConfig.getACQUISITION_SYSTEM_DIRECTORY();
 
    /** url地址 */
    private final static String baseUrl = TYTaskConstants.tyTaskConfig.getACQUISITION_SYSTEM_BASEURL();

    /**
     * 上傳單個檔案
     * @param remoteFolder
     *            上傳到SFTP伺服器的路徑
     * @param sourceFile
     *            上傳的檔案地址
     * @param remoteFileName
     *            上傳到SFTP伺服器後的檔名
     *
     * @throws Exception
     */
    public static String uploadFile(String remoteFolder,String sourceFile, String remoteFileName) throws Exception {
        long start = System.currentTimeMillis();
        remoteFolder = directory + remoteFolder;

        File file = new File(sourceFile);
        InputStream fileStream = new FileInputStream(file); // 提升作用域
        ChannelSftp sftp = connect();
        try{
            //如果資料夾不存在，則建立資料夾
            if(sftp.ls(remoteFolder) == null){
                sftp.mkdir(remoteFolder);
            }
            //切換到指定資料夾
            sftp.cd(remoteFolder);
        }catch (SftpException e){
            //建立不存在的資料夾，並切換到資料夾
            sftp.mkdir(remoteFolder);
            sftp.cd(remoteFolder);
        }
        sftp.put(fileStream, remoteFileName);
        disconnect(sftp);
        logger.info("檔案上傳成功！！ 耗時：{"+(System.currentTimeMillis() - start)+"}ms");
        return baseUrl + remoteFileName;
    }

    /**
     * 刪除檔案
     * @param deleteFile
     *            要刪除的檔案
     *
     * @throws Exception
     */
    public void delete(String deleteFile) throws Exception {
        ChannelSftp sftp = connect();
        sftp.cd(directory);
        sftp.rm(deleteFile);
        disconnect(sftp);
    }

    /**
     * 連線sftp伺服器
     *
     * @throws Exception
     */
    public static ChannelSftp connect() throws Exception {
        JSch jsch = new JSch();
        try{
            //採用指定的埠連線伺服器
            Session session = jsch.getSession(username,host,port);
            if(password != null){
                //設定登陸主機的密碼
                session.setPassword(password);
            }

            Properties sshConfig = new Properties();
            sshConfig.put("StrictHostKeyChecking", "no");
            session.setConfig(sshConfig);
            session.connect();

            //建立sftp通訊通道
            Channel channel = session.openChannel("sftp");
            channel.connect();
            logger.info("sftp server connect success !!");
            return (ChannelSftp) channel;
        }catch (JSchException e){
            logger.error("SFTP伺服器連線異常！！", e);
            throw new Exception("SFTP伺服器連線異常！！",e);
        }
    }

    /**
     * Disconnect with server
     *
     * @throws Exception
     */
    public static void disconnect(ChannelSftp sftp) throws Exception {
        if (sftp != null) {
            if (sftp.isConnected()) {
                sftp.disconnect();
            } else if (sftp.isClosed()) {
            }
        }
    }

    public static void main(String[] args){
        //資料檔名
        String dataFileName = "008_ZQQK_U_20200707_I.txt";
        String dyw_key1 = "/Users/lpl/Documents/myself/tempfile/008_ZQQK_U_20181108_I.txt";
        String resultDto1 = null;
        try {

            resultDto1 = new SftpClientUtil().uploadFile("20181108", dyw_key1, dataFileName);
        } catch (Exception e) {
            e.printStackTrace();
        }
//        if(resultDto1 != null && resultDto1.getCode().equals("0000")) {
            System.out.println("--上送ZQQK檔案-----成功， 檔案key = " + resultDto1);
//        }else{
//            System.out.println("--上送ZQQK檔案-----失敗， 檔案key = " + resultDto1.getMessage());
//        }
    }

}