package com.test.utils;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.FileFilter;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import com.jcraft.jsch.ChannelSftp;
import com.jcraft.jsch.JSch;
import com.jcraft.jsch.Session;
public class Ftp {
  //打印log日志
  private static final Log logger = LogFactory.getLog(Ftp.class);
  private static Date last_push_date = null;
  private Session sshSession;
  private ChannelSftp channel;
  private static ThreadLocal<Ftp> sftpLocal = new ThreadLocal<Ftp>();
  private Ftp(String host, int port, String username, String password) throws Exception {
    JSch jsch = new JSch();
    jsch.getSession(username, host, port);
    //根據(jù)用戶名，密碼，端口號獲取session
    sshSession = jsch.getSession(username, host, port);
    sshSession.setPassword(password);
    //修改服務(wù)器/etc/ssh/sshd_config 中 GSSAPIAuthentication的值yes為no，解決用戶不能遠程登錄
    sshSession.setConfig("userauth.gssapi-with-mic", "no");
    //為session對象設(shè)置properties,第一次訪問服務(wù)器時不用輸入yes
    sshSession.setConfig("StrictHostKeyChecking", "no");
    sshSession.connect();
    //獲取sftp通道
    channel = (ChannelSftp)sshSession.openChannel("sftp");
    channel.connect();
    logger.info("連接ftp成功!" + sshSession);
  }
  /**
   * 是否已連接
   *
   * @return
   */
  private boolean isConnected() {
    return null != channel && channel.isConnected();
  }
  /**
   * 獲取本地線程存儲的sftp客戶端
   *
   * @return
   * @throws Exception
   */
  public static Ftp getSftpUtil(String host, int port, String username, String password) throws Exception {
    //獲取本地線程
    Ftp sftpUtil = sftpLocal.get();
    if (null == sftpUtil || !sftpUtil.isConnected()) {
      //將新連接防止本地線程，實現(xiàn)并發(fā)處理
      sftpLocal.set(new Ftp(host, port, username, password));
    }
    return sftpLocal.get();
  }
  /**
   * 釋放本地線程存儲的sftp客戶端
   */
  public static void release() {
    if (null != sftpLocal.get()) {
      sftpLocal.get().closeChannel();
      logger.info("關(guān)閉連接" + sftpLocal.get().sshSession);
      sftpLocal.set(null);
    }
  }
  /**
   * 關(guān)閉通道
   *
   * @throws Exception
   */
  public void closeChannel() {
    if (null != channel) {
      try {
        channel.disconnect();
      } catch (Exception e) {
        logger.error("關(guān)閉SFTP通道發(fā)生異常:", e);
      }
    }
    if (null != sshSession) {
      try {
        sshSession.disconnect();
      } catch (Exception e) {
        logger.error("SFTP關(guān)閉 session異常:", e);
      }
    }
  }
  /**
   * @param directory 上傳ftp的目錄
   * @param uploadFile 本地文件目錄
   *
   */
  public void upload(String directory, String uploadFile) throws Exception {
    try {<br>    //執(zhí)行列表展示ls 命令
    channel.ls(directory);<br>    //執(zhí)行盤符切換cd 命令
    channel.cd(directory);
    List<File> files = getFiles(uploadFile, new ArrayList<File>());
    for (int i = 0; i < files.size(); i++) {
      File file = files.get(i);
      InputStream input = new BufferedInputStream(new FileInputStream(file));
      channel.put(input, file.getName());
      try {
        if (input != null) input.close();
      } catch (Exception e) {
        e.printStackTrace();
        logger.error(file.getName() + "關(guān)閉文件時.....異常!" + e.getMessage());
      }
      if (file.exists()) {
        boolean b = file.delete();
        logger.info(file.getName() + "文件上傳完畢!刪除標識:" + b);
      }
    }
    }catch (Exception e) {
      logger.error("【子目錄創(chuàng)建中】：",e);
            //創(chuàng)建子目錄
      channel.mkdir(directory);
    }
  }
  //獲取文件
  public List<File> getFiles(String realpath, List<File> files) {
    File realFile = new File(realpath);
    if (realFile.isDirectory()) {
      File[] subfiles = realFile.listFiles(new FileFilter() {
        @Override
        public boolean accept(File file) {
          if (null == last_push_date ) {
            return true;
          } else {
            long modifyDate = file.lastModified();
            return modifyDate > last_push_date.getTime();
          }
        }
      });
      for (File file : subfiles) {
        if (file.isDirectory()) {
          getFiles(file.getAbsolutePath(), files);
        } else {
          files.add(file);
        }
        if (null == last_push_date) {
          last_push_date = new Date(file.lastModified());
        } else {
          long modifyDate = file.lastModified();
          if (modifyDate > last_push_date.getTime()) {
            last_push_date = new Date(modifyDate);
          }
        }
      }
    }
    return files;
  }
}