package com.sunsheen.blockchain.admin.utils;

import java.io.BufferedReader;

import java.io.File;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import java.io.OutputStream;

import java.io.PrintWriter;

import java.util.Properties;

import org.apache.http.util.Asserts;

import ch.ethz.ssh2.ChannelCondition;

import ch.ethz.ssh2.Connection;

import ch.ethz.ssh2.StreamGobbler;

import com.jcraft.jsch.Channel;

import com.jcraft.jsch.ChannelSftp;

import com.jcraft.jsch.ChannelShell;

import com.jcraft.jsch.JSch;

import com.jcraft.jsch.JSchException;

import com.jcraft.jsch.Session;

import com.sunsheen.blockchain.admin.common.ServersConstant;

public class SSHUtil {

    /**

     * 连接服务器

     * @param host

     * @param user

     * @param pwd

     * @return

     * @throws JSchException

     */

    public static synchronized Session connect(String host,String user,String pwd) throws JSchException {

        JSch jsch = new JSch();

        Session session = jsch.getSession(user, host, 22);

        session.setConfig("StrictHostKeyChecking", "no");

        // java.util.Properties config = new java.util.Properties();

        // config.put("StrictHostKeyChecking", "no");

        session.setPassword(pwd);

        session.connect();

        return session;

    }

    /**

     * 执行命令集

     * @param session

     * @param cmds

     * @return

     * @throws IOException

     * @throws JSchException

     */

    public static String execCommandByShell(Session session,String[] cmds)

            throws IOException, JSchException {

        String result = "";

        // 2.尝试解决 远程ssh只能执行一句命令的情况

        ChannelShell channelShell = (ChannelShell) session.openChannel("shell");

        InputStream inputStream = channelShell.getInputStream();// 从远端到达的数据都能从这个流读取到

        channelShell.setPty(true);

        channelShell.connect();

        OutputStream outputStream = channelShell.getOutputStream();// 写入该流的数据

                                                                    // 都将发送到远程端

        // 使用PrintWriter 就是为了使用println 这个方法

        // 好处就是不需要每次手动给字符加\n

        PrintWriter printWriter = new PrintWriter(outputStream);

        for (String cmd:cmds) {

            printWriter.println(cmd);

        }

        printWriter.println("exit");// 为了结束本次交互

        printWriter.flush();// 把缓冲区的数据强行输出

        return result;

    }

    /**

     * 单个文件上传

     * @param file        上传的文件

     * @param remoteFolder    服务器上存放当前文件的文件夹

     * @param uploadFileName    上传文件的名字

     */

    public static void postFile(InputStream fileStream,String remoteFolder,String uploadFileName) throws Exception{

        //上传文件的个数应该跟对应文件夹相同

        if((null==fileStream || null==remoteFolder))

            return;

        String username = ServersConstant.USERNAME;

        String password = ServersConstant.PASSWORD;

        String address = ServersConstant.ADDRES;

        int port = ServersConstant.PORT;

        ChannelSftp sftp = null;

        Channel channel = null;

        Session sshSession = null;

        try {

            //创建连接

            JSch jsch = new JSch();

            sshSession = jsch.getSession(username, address, port);

            sshSession.setPassword(password);

            //获取session

            Properties sshConfig = new Properties();

            sshConfig.put("StrictHostKeyChecking", "no");

            sshSession.setConfig(sshConfig);

            sshSession.connect();

            //得到sftp

            channel = sshSession.openChannel("sftp");

            channel.connect();

            sftp = (ChannelSftp) channel;

            sftp.cd(remoteFolder);//进入对应存放日志文件的目录

            sftp.put(fileStream, uploadFileName);//写入文件

        } finally {

            //关闭sftp信道

            if (sftp != null) {

                if (sftp.isConnected()) {

                    sftp.disconnect();

                }

            }

            //关闭channel管道

            if (channel != null) {

                if (channel.isConnected()) {

                    channel.disconnect();

                }

            }

            //关闭session

            if (sshSession != null) {

                if (sshSession.isConnected()) {

                    sshSession.disconnect();

                }

            }

        }

    }

    /**

     * 转换指令到服务器执行

     * @param command    要执行的指令

     */

    public static void transferCommand(String... commands){

        String romoteAddr = ServersConstant.ADDRES;

        String username = ServersConstant.USERNAME;

        String password = ServersConstant.PASSWORD;

        try {

              Connection connection = new Connection(romoteAddr);// 创建一个连接实例

              connection.connect();// Now connect

              boolean isAuthenticated = connection.authenticateWithPassword(username, password);//認證

              Asserts.check(isAuthenticated, "用戶名或密碼錯誤！");

              ch.ethz.ssh2.Session sess = connection.openSession();// 創建一個會話

              sess.requestPTY("bash");

              sess.startShell();

              InputStream stdout = new StreamGobbler(sess.getStdout());

              InputStream stderr = new StreamGobbler(sess.getStderr());

              BufferedReader stdoutReader = new BufferedReader(new InputStreamReader(stdout));

              BufferedReader stderrReader = new BufferedReader(new InputStreamReader(stderr));

              //向服务器上输入命令

              PrintWriter out = new PrintWriter(sess.getStdin());

              for(String command : commands){

                  out.println(command);

              }

              out.close();

              sess.waitForCondition(ChannelCondition.CLOSED|ChannelCondition.EOF | ChannelCondition.EXIT_STATUS,100);

              //关闭连接

              sess.close();

              connection.close();

              stderrReader.close();

              stdoutReader.close();

          } catch (IOException e) {

              e.printStackTrace();

          }

    }

    /**

     * 下载指定文件到本地指定文件夹oo

     * @param serversFolder    服务器文件所在目录  /opt/log

     * @param fileName            需要下载的文件名        monitor.log

     * @param localFolder        本地存放文件的文件夹 d:\\logs

     */

    public static void download(String serversFolder,String fileName,String localFolder)  {

        String username = ServersConstant.USERNAME;

        String password = ServersConstant.PASSWORD;

        String address = ServersConstant.ADDRES;

        int port = ServersConstant.PORT;

        ChannelSftp sftp = null;

        Channel channel = null;

        Session sshSession = null;

        try {

            // 创建连接

            JSch jsch = new JSch();

            sshSession = jsch.getSession(username, address, port);

            sshSession.setPassword(password);

            // 获取session

            Properties sshConfig = new Properties();

            sshConfig.put("StrictHostKeyChecking", "no");

            sshSession.setConfig(sshConfig);

            sshSession.connect();

            // 得到sftp

            channel = sshSession.openChannel("sftp");

            channel.connect();

            sftp = (ChannelSftp) channel;

            // 进入服务器文件夹

            sftp.cd(serversFolder);

            //创建本地文件夹

            File downloadFile = new File(localFolder);

            if(!downloadFile.exists())

                downloadFile.mkdirs();

            // 下载

            String serversFile = serversFolder +"/"+ fileName;

            sftp.get(serversFile,localFolder);

            System.out.println(fileName+"已下载到:"+localFolder);

        } catch (Exception e) {

            e.printStackTrace();

        } finally {

            // 关闭sftp信道

            if (sftp != null) {

                if (sftp.isConnected()) {

                    sftp.disconnect();

                }

            }

            // 关闭channel管道

            if (channel != null) {

                if (channel.isConnected()) {

                    channel.disconnect();

                }

            }

            // 关闭session

            if (sshSession != null) {

                if (sshSession.isConnected()) {

                    sshSession.disconnect();

                }

            }

        }

    }

}