FTPClient.h

#pragma once
 
#include "Arduino.h"
#include "Client.h"
#include "FTPBasicAPI.h"
#include "FTPFile.h"
#include "FTPFileIterator.h"
#include "FTPSessionMgr.h"
#include "IPAddress.h"
#include "Stream.h"
 
namespace ftp_client {
 
template <class ClientType>
class FTPClient {
 public:
  FTPClient(int port = FTP_COMMAND_PORT, bool useType = false) {
    FTPLogger::writeLog(LOG_DEBUG, "FTPClient");
    setUseTypeCommand(useType);
    setPort(port);
  }
 
  void setUseTypeCommand(bool useType) {
    use_type_command = useType;
  }
 
  bool begin(IPAddress remote_addr, const char *user = "anonymous",
             const char *password = nullptr) {
    FTPLogger::writeLog(LOG_INFO, "FTPClient", "begin");
    this->userid = user;
    this->password = password;
    this->remote_addr = remote_addr;
    return mgr.begin(remote_addr, this->port, user, password);
  }
 
  void end() {
    FTPLogger::writeLog(LOG_INFO, "FTPClient", "end");
    mgr.end();
  }
 
  FTPFile open(const char *filename, FileMode mode = READ_MODE,
               bool autoClose = false) {
    char msg[200];
    snprintf(msg, sizeof(msg), "open: %s", filename);
    FTPLogger::writeLog(LOG_INFO, "FTPClient", msg);
 
    FTPBasicAPI &api = mgr.session().api();
 
    // Open new data connection
    api.passv();
 
    return FTPFile(&api, filename, mode, autoClose);
  }
 
  bool mkdir(const char *filepath) {
    FTPLogger::writeLog(LOG_INFO, "FTPClient", "mkdir");
    FTPBasicAPI &api = mgr.session().api();
    if (!api) return false;
    return api.mkdir(filepath);
  }
 
  bool remove(const char *filepath) {
    FTPLogger::writeLog(LOG_INFO, "FTPClient", "remove");
    FTPBasicAPI &api = mgr.session().api();
    if (!api) return false;
    return api.del(filepath);
  }
 
  bool rmdir(const char *filepath) {
    FTPLogger::writeLog(LOG_INFO, "FTPClient", "rmdir");
    FTPBasicAPI &api = mgr.session().api();
    if (!api) return false;
    return api.rmd(filepath);
  }
 
  FTPFileIterator ls(const char *path, FileMode mode = WRITE_MODE) {
    FTPLogger::writeLog(LOG_INFO, "FTPClient", "ls");
    FTPBasicAPI &api = mgr.session().api();
 
    // Open new data connection
    api.passv();
 
    FTPFileIterator it(&api, path, mode);
    return it;
  }
 
  bool binary() {
    FTPBasicAPI &api = mgr.session().api();
    if (!api) return false;
    api.setUseTypeCommand(use_type_command);
    return api.binary();
  }
 
  bool ascii() {
    FTPBasicAPI &api = mgr.session().api();
    if (!api) return false;
    api.setUseTypeCommand(use_type_command);
    return api.ascii();
  }
 
  bool type(const char *str) {
    FTPBasicAPI &api = mgr.session().api();
    if (!api) return false;
    return api.type(str);
  }
 
  void setPort(int port) {
    this->port = port;
  }
 
  bool abort(CurrentOperation op) {
    return mgr.abort(op);
  } 
 
  FTPSessionMgr<ClientType> &sessionMgr() {
    return mgr;
  }
 protected:
  FTPSessionMgr<ClientType> mgr;
  IPAddress remote_addr;
  const char *userid = nullptr;
  const char *password = nullptr;
  int port;
  bool cleanup_clients;
  bool auto_close = true;
  bool use_type_command = false;
 
};
 
}  // namespace ftp_client