package com.app.base.service.internal;

import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.io.PrintStream;
import org.apache.commons.lang3.StringUtils;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPReply;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.core.env.Environment;
import org.springframework.stereotype.Service;

import com.app.base.data.ApiResponseResult;
import com.app.base.service.FtpClientService;

@Service
public class FtpClientImpl implements FtpClientService {
    private static final Logger logger = LoggerFactory.getLogger(FtpClientImpl.class);

    @Autowired
    private Environment env;

    private FTPClient login() {
        FTPClient ftp = new FTPClient();
        try {
            FtpConfig config = getConfig();
            if (config == null) {
                return null;
            }

            ftp.connect(config.url, config.port);

            ftp.login(config.username, config.password);
            ftp.enterLocalPassiveMode();
            ftp.setFileType(2);
            ftp.setControlEncoding("gbk");
            int reply = ftp.getReplyCode();

            if (!FTPReply.isPositiveCompletion(reply)) {
                logger.info("FTP连接失败");
                ftp.disconnect();
                return null;
            }
            logger.info("FTP连接成功");
            return ftp;
        } catch (IOException e) {
            logger.debug("连接FTP 服务器失败.", e);
            closeCon(ftp);
        }
        return null;
    }

    private void closeCon(FTPClient ftp) {
        if ((ftp != null) &&
                (ftp.isConnected())) {
            try {
                ftp.logout();
                ftp.disconnect();
            } catch (IOException e) {
                logger.error("关闭FTP 链接失败", e);
            }
        }
    }

    private FtpConfig getConfig() {
        try {
            FtpConfig ftpConfig = new FtpConfig();
            ftpConfig.url = this.env.getProperty("fs.ftp.url");
            ftpConfig.port = Integer.parseInt(this.env.getProperty("fs.ftp.port"));
            ftpConfig.username = this.env.getProperty("fs.ftp.username");
            ftpConfig.password = this.env.getProperty("fs.ftp.password");
            return ftpConfig;
        } catch (Exception e) {
            logger.error("获取FTP 配置文件失败", e);
        }
        return null;
    }

    public ApiResponseResult uploadFile(String path, String filename, InputStream input) {
        String url = this.env.getProperty("fs.ftp.url");
        int port = Integer.parseInt(this.env.getProperty("fs.ftp.port"));
        String username = this.env.getProperty("fs.ftp.username");
        String password = this.env.getProperty("fs.ftp.password");
        if (logger.isDebugEnabled()) {
            logger.debug("FTP 上传 url:" + url + " user name:" + username + " password:" + password);
        }
        return uploadFile(url, port, username, password, path, filename, input);
    }

    public ApiResponseResult uploadFile(String url, int port, String username, String password, String path, String filename, InputStream input) {
        String rootPath = this.env.getProperty("fs.ftp.rootPath", "");
        if (!StringUtils.startsWith(path, "/")) {
            path = File.separator + path;
        }

        String savePath = rootPath + path;
        FTPClient ftp = new FTPClient();
        try {
            ftp.connect(url, port);
            ApiResponseResult localApiResponseResult1;
            if (!ftp.login(username, password)) {
                return ApiResponseResult.failure("ftp登录认证失败");
            }

            int reply = ftp.getReplyCode();
            if (!FTPReply.isPositiveCompletion(reply)) {
                ftp.disconnect();
                return ApiResponseResult.failure("上传失败");
            }
            ftp.enterLocalPassiveMode();
            ftp.setFileType(2);
            ftp.setControlEncoding("gbk");
            if (!changeWorkingDirectory(ftp, new String(savePath.getBytes("gbk"), "iso-8859-1"))) {
                return ApiResponseResult.failure("上传目录不存在或创建目录失败");
            }
            boolean success = ftp.storeFile(new String(filename.getBytes("gbk"), "iso-8859-1"), input);

            input.close();
            ftp.logout();
            if (!success) {
                return ApiResponseResult.failure("上传失败");
            }
            return ApiResponseResult.success("上传成功").data(path);
        } catch (IOException e) {
            boolean success;
            logger.error("ftp上传文件异常:" + savePath, e);
            return ApiResponseResult.failure("上传异常");
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException localIOException7) {
                }
            }
        }
    }

    public ApiResponseResult download(String path, String fileName) {
        String url = this.env.getProperty("fs.ftp.url");
        int port = Integer.parseInt(this.env.getProperty("fs.ftp.port"));
        String username = this.env.getProperty("fs.ftp.username");
        String password = this.env.getProperty("fs.ftp.password");
        if (logger.isInfoEnabled()) {
            logger.info("FTP 下载 url:" + url + " user name:" + username + " password:" + password);
        }
        return download(url, port, username, password, path, fileName);
    }

    public ApiResponseResult download(String url, int port, String username, String password, String path, String fileName) {
        logger.info("下载地址: " + url + " port: " + port + " username: " + username + " password: " + password);
        String rootPath = this.env.getProperty("fs.ftp.rootPath", "");
        if (!StringUtils.startsWith(path, "/")) {
            path = File.separator + path;
        }
        path = rootPath + path;
        FTPClient ftp = new FTPClient();
        try {
            ftp.connect(url, port);
            ApiResponseResult localApiResponseResult1;
            if (!ftp.login(username, password)) {
                return ApiResponseResult.failure("ftp登录认证失败");
            }
            logger.info("ftp 登录成功");

            ftp.setFileType(2);
            ftp.setControlEncoding("gbk");
            int reply = ftp.getReplyCode();
            logger.info("ftp.getReplyCode() success");
            if (!FTPReply.isPositiveCompletion(reply)) {
                System.out.println("!FTPReply.isPositiveCompletion(reply)");
                ftp.disconnect();
                return ApiResponseResult.failure("下载失败");
            }
            ftp.enterLocalPassiveMode();

            ByteArrayOutputStream bos = new ByteArrayOutputStream();
            boolean flag = ftp.retrieveFile(new String((path + "/" + fileName).getBytes("gbk"), "iso-8859-1"), (OutputStream) bos);

            logger.info("ftp.retrieveFile success");
            ApiResponseResult localApiResponseResult2;
            if (!flag) {
                return ApiResponseResult.failure("文件不存在或已被删除");
            }
            ftp.logout();
            return ApiResponseResult.success("下载成功").data(((ByteArrayOutputStream) bos).toByteArray());
        } catch (IOException e) {
            Object bos;
            logger.error("ftp下载文件异常", e);
            return ApiResponseResult.failure("下载失败");
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException localIOException6) {
                }
            }
        }
    }

    protected boolean changeWorkingDirectory(FTPClient ftp, String path) throws IOException {
        boolean changed = ftp.changeWorkingDirectory(path);
        if (changed) {
            return true;
        }
        String[] dirs = StringUtils.split(path, "/");
        if (null == dirs) {
            return true;
        }
        String pathname = "";
        for (String dir : dirs) {
            pathname = pathname + File.separator + dir;
            if (ftp.makeDirectory(pathname)) {
                logger.debug("create ftp directory with {}", pathname);
            }
        }

        return ftp.changeWorkingDirectory(path);
    }

    public ApiResponseResult downloadFile(String path, String fileName) {
        return download(path, fileName);
    }

    public ApiResponseResult downloadFile(String url, int port, String username, String password, String path, String fileName) {
        return download(url, port, username, password, path, fileName);
    }

    public ApiResponseResult remove(String path, String fileName) {
        String ftpPath = getPath(path);
        String pathName = ftpPath + File.separator + fileName;
        logger.debug("删除FTP 文件:" + pathName);
        FTPClient ftp = login();
        if (ftp == null)
            return ApiResponseResult.failure().message("链接FTP服务器失败");
        boolean flag = false;
        if (ftp != null) {
            try {
                if (!changeWorkingDirectory(ftp, new String(ftpPath.getBytes("gbk"), "iso-8859-1"))) {
                    logger.error("切换到目录" + ftpPath + "失败,请确认目录是否存在.");
                    return ApiResponseResult.failure().message("操作失败,请确认目录是否存在.");
                }

                flag = ftp.deleteFile(fileName);
            } catch (IOException e) {
                logger.error("文件删除出现异常", e);
            } finally {
                closeCon(ftp);
            }
        }
        return flag ? ApiResponseResult.success() : ApiResponseResult.failure().message("文件删除失败,请检查FTP是否连接正常.");
    }

    public ApiResponseResult rename(String path, String fileName, String newFileName) {
        String ftpPath = getPath(path);
        logger.debug("rename ftp 文件:" + ftpPath + " fileName:" + fileName + " new fileName:" + newFileName);
        FTPClient ftpClient = login();
        boolean flag = false;
        if (ftpClient != null) {
            try {
                if (!changeWorkingDirectory(ftpClient, new String(ftpPath
                        .getBytes("gbk"), "iso-8859-1"))) {
                    logger.error("切换到目录" + ftpPath + "失败,请确认目录是否存在.");
                    return ApiResponseResult.failure().message("操作失败,请确认目录是否存在.");
                }

                flag = ftpClient.rename(fileName, newFileName);
            } catch (IOException e) {
                logger.error("文件重命名出现异常", e);
                closeCon(ftpClient);
            }
        }
        return flag ? ApiResponseResult.success() : ApiResponseResult.failure().message("文件重命名失败,请检查FTP是否连接正常.");
    }

    @Override
    public ApiResponseResult download(String url, int port, String username, String password, String path, String fileName,String root) {
        String pwd = password.replace(" ", "+");

        logger.info("下载地址: " + url + " port: " + port + " username: " + username + " password: " + pwd);
//        String rootPath = this.env.getProperty("fs.ftp.rootPath", "");
//        String rootPath = "/task";
        String rootPath = root;

        if (!StringUtils.startsWith(path, "/")) {
            path = File.separator + path;
        }
        path = rootPath + path;
        FTPClient ftp = new FTPClient();
        try {
            ftp.connect(url, port);
            ApiResponseResult localApiResponseResult1;
            if (!ftp.login(username, pwd)) {
                return ApiResponseResult.failure("ftp登录认证失败");
            }
            logger.info("ftp 登录成功");

            ftp.setFileType(2);
            ftp.setControlEncoding("gbk");
            int reply = ftp.getReplyCode();
            logger.info("ftp.getReplyCode() success");
            if (!FTPReply.isPositiveCompletion(reply)) {
                System.out.println("!FTPReply.isPositiveCompletion(reply)");
                ftp.disconnect();
                return ApiResponseResult.failure("下载失败");
            }
            ftp.enterLocalPassiveMode();

            ByteArrayOutputStream bos = new ByteArrayOutputStream();
            boolean flag = ftp.retrieveFile(new String((path + "/" + fileName).getBytes("gbk"), "iso-8859-1"), (OutputStream) bos);

            logger.info("ftp.retrieveFile success");
            ApiResponseResult localApiResponseResult2;
            if (!flag) {
                logger.info("文件不存在或已被删除");
                return ApiResponseResult.failure("文件不存在或已被删除");
            }
            ftp.logout();
            return ApiResponseResult.success("下载成功").data(((ByteArrayOutputStream) bos).toByteArray());
        } catch (IOException e) {
            Object bos;
            logger.error("ftp下载文件异常", e);
            return ApiResponseResult.failure("下载失败");
        } finally {
            if (ftp.isConnected()) {
                try {
                    ftp.disconnect();
                } catch (IOException localIOException6) {
                }
            }
        }
    }

    private String getPath(String path) {
        String rootPath = this.env.getProperty("fs.ftp.rootPath", "");
        if (!StringUtils.startsWith(path, "/")) {
            path = File.separator + path;
        }
        path = rootPath + path;
        return path;
    }

    private class FtpConfig {
        public int port;
        public String url;
        public String username;
        public String password;

        private FtpConfig() {
        }
    }
}