guru.nidi.ftpsync.fs.FtpFileSystem.java Source code

Introduction

Here is the source code for guru.nidi.ftpsync.fs.FtpFileSystem.java

Source

/*
 * Copyright (C) 2014 Stefan Niederhauser (nidin@gmx.ch)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package guru.nidi.ftpsync.fs;

import guru.nidi.ftpsync.Config;
import guru.nidi.ftpsync.FtpException;
import org.apache.commons.net.ftp.*;

import java.io.*;
import java.util.ArrayList;
import java.util.List;

/**
 *
 */
public class FtpFileSystem extends FileSystemBase {
    private final FTPClient client;

    public FtpFileSystem(String basedir, Config config) throws IOException {
        super(basedir);
        client = new FTPClient();
        client.connect(config.getHost());
        int reply = client.getReplyCode();

        if (!FTPReply.isPositiveCompletion(reply)) {
            client.disconnect();
            throw new IOException("FTP server refused connection.");
        }
        final boolean login = client.login(config.getUsername(), config.getPassword());
        if (!login) {
            throw new FtpException("Could not login", client.getReplyStrings());
        }
        client.setFileType(FTP.BINARY_FILE_TYPE);
    }

    public void deleteFile(String name) throws IOException {
        name = expand(name);
        if (!client.deleteFile(name)) {
            throw new FtpException("Could not delete file " + name, client.getReplyStrings());
        }
    }

    public void deleteDirectory(String name) throws IOException {
        name = expand(name);
        if (!client.removeDirectory(name)) {
            throw new FtpException("Could not delete directory " + name, client.getReplyStrings());
        }
    }

    public void putFile(File local, String dest) throws IOException {
        dest = expand(dest);
        try (InputStream in = new FileInputStream(local)) {
            if (!client.storeFile(dest, in)) {
                throw new FtpException("Could put copy file " + local + " to " + dest, client.getReplyStrings());
            }
        }
    }

    public void getFile(File local, String dest) throws IOException {
        dest = expand(dest);
        try (OutputStream out = new FileOutputStream(local)) {
            if (!client.retrieveFile(dest, out)) {
                throw new FtpException("Could get copy file " + local + " from " + dest, client.getReplyStrings());
            }
        }
    }

    public void createDirectory(String name) throws IOException {
        name = expand(name);
        if (!client.makeDirectory(name)) {
            throw new FtpException("Could not create directory " + name, client.getReplyStrings());
        }
    }

    @Override
    public List<AbstractFile> listFiles(String dir, AbstractFileFilter filter) throws IOException {
        List<AbstractFile> res = new ArrayList<>();
        for (FTPFile file : client.listFiles(expand(dir), new FTPFileFilterImpl(filter))) {
            res.add(new AbstractFileImpl(file));
        }
        return res;
    }

    @Override
    public void close() {
    }

    private static class FTPFileFilterImpl implements FTPFileFilter {
        private final AbstractFileFilter filter;

        private FTPFileFilterImpl(AbstractFileFilter filter) {
            this.filter = filter;
        }

        @Override
        public boolean accept(FTPFile ftpFile) {
            return filter.accept(new AbstractFileImpl(ftpFile));
        }
    }

    private static class AbstractFileImpl implements AbstractFile {
        private final FTPFile file;

        private AbstractFileImpl(FTPFile file) {
            this.file = file;
        }

        @Override
        public boolean isFile() {
            return file.isFile();
        }

        @Override
        public boolean isDirectory() {
            return file.isDirectory();
        }

        @Override
        public String getName() {
            return file.getName();
        }

        @Override
        public File asFile() {
            throw new UnsupportedOperationException();
        }
    }
}