package testftp;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Path;
import java.nio.file.Paths;
import org.apache.commons.net.ftp.FTP;
import org.apache.commons.net.ftp.FTPClient;
import org.apache.commons.net.ftp.FTPFile;
import static java.lang.System.out;
import static java.nio.file.Files.copy;
import static java.nio.file.Files.createDirectory;
import static java.nio.file.Files.exists;
import static java.nio.file.StandardCopyOption.REPLACE_EXISTING;

public class TestFtp {

    private static final String //
            SERVER = "127.0.0.1",
            USER = "Andrei",
            PASS = "qw",
            FTP_FOLDER = "/public/rank2/";
//            SERVER = "127.0.0.1",
//            USER = "anonymous",
//            PASS = "",
//            FTP_FOLDER = "/pub/Windows/internet/";
    //
    private static final int //
            PORT = 2121;
//            PORT = 21;

    public static void main(String[] args) throws IOException {
        FTPClient ftpClient = new FTPClient();
        ftpClient.connect(SERVER, PORT);
        try {
            ftpClient.login(USER, PASS);
            try {
//                ftpClient.enterLocalPassiveMode();
                out.println("Connected");
                TestFtp testFtp = new TestFtp(ftpClient);

                testFtp.downloadDirectory(FTP_FOLDER, Paths.get("D:\\test"));
            } finally {
                ftpClient.logout();
            }
        } finally {
            ftpClient.disconnect();
        }

    }

    private final FTPClient ftpClient;
    public TestFtp(FTPClient ftpClient) {
        this.ftpClient = ftpClient;
    }

    public void downloadDirectory(String dirFtp, Path dirLocal) throws IOException {
        for (FTPFile ftpFile : ftpClient.listFiles(dirFtp)) {
            String fileName = ftpFile.getName();
            if (equalsWithAny(fileName, ".", "..")) {
                continue;
            }

            String subPathFtp = dirFtp + "/" + fileName;
            Path subPathLocal = dirLocal.resolve(fileName);

            if (ftpFile.isDirectory()) {
                if (!exists(subPathLocal)) {
                    createDirectory(subPathLocal);
                }
                downloadDirectory(subPathFtp, subPathLocal);
            } else {
                try (BufferedOutputStream os = new BufferedOutputStream(new FileOutputStream(subPathLocal.toFile()))) {
                    ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
                    ftpClient.retrieveFile(dirFtp, os);
                }
                //
                //
//                try (InputStream is = ftpClient.retrieveFileStream(subPathFtp)) {
//                    if (is != null) {
//                        ftpClient.setFileType(FTP.BINARY_FILE_TYPE);
//                        copy(is, subPathLocal, REPLACE_EXISTING);
//                    }
//                }
            }
        }
    }

    public static boolean equalsWithAny(String s, String... someStrings) {
        for (String aString : someStrings) {
            if (s.equals(aString)) {
                return true;
            }
        }
        return false;
    }
}