FTPSessionMgr.h

#pragma once
#include "FTPSession.h"
#include "IPAddress.h"
 
namespace ftp_client {
template <class ClientType>
class FTPSessionMgr {
 public:
  FTPSessionMgr() { FTPLogger::writeLog(LOG_DEBUG, "FTPSessionMgr"); }
 
  ~FTPSessionMgr() {
    FTPLogger::writeLog(LOG_DEBUG, "~FTPSessionMgr");
    end();
  }
 
  bool begin(IPAddress &address, int port, const char *username,
             const char *password) {
    FTPLogger::writeLog(LOG_DEBUG, "FTPSessionMgr", "createSession");
    this->address = address;
    this->port = port;
    this->username = username;
    this->password = password;
    return true;
  }
 
  void end() {
    FTPLogger::writeLog(LOG_DEBUG, "FTPSessionMgr", "end");
    for (int i = 0; i < FTP_MAX_SESSIONS; i++) {
      if (sessions[i] != nullptr) {
        FTPSession<ClientType> &session = *sessions[i];
        session.api().quit();  // Send QUIT command to the server
        session.end();
        delete sessions[i];
        sessions[i] = nullptr;
      }
    }
  }
 
  FTPSession<ClientType> &session() {
    for (int i = 0; i < FTP_MAX_SESSIONS; i++) {
      if (sessions[i] == nullptr) {
        sessions[i] = new FTPSession<ClientType>();
        if (sessions[i]->begin(address, port, username, password)) {
          return *sessions[i];
        } else {
          delete sessions[i];
          sessions[i] = nullptr;
        }
      } else if (sessions[i]->api().currentOperation() == NOP) {
        // Reuse existing session if it is not currently in use
        return *sessions[i];
      }
    }
    FTPLogger::writeLog(LOG_ERROR, "FTPSessionMgr", "No available sessions");
    static FTPSession<ClientType> empty_session;
    empty_session.setValid(false);
    return empty_session;  // No available session
  }
 
  bool abort(CurrentOperation op) {
    FTPLogger::writeLog(LOG_DEBUG, "FTPSessionMgr", "abort");
    for (int i = 0; i < FTP_MAX_SESSIONS; i++) {
      if (sessions[i] != nullptr &&
          sessions[i]->api().currentOperation() == op) {
        return sessions[i]->api().abort();
      }
    }
    return false;  // No session found with the specified operation
  }
 
  int count() {
    int result = 0;
    for (auto &session : sessions) {
      if (session != nullptr) {
        result++;
      }
    }
    return result;
  }
  
  int count(CurrentOperation op) {
    int result = 0;
    for (auto &session : sessions) {
      if (session != nullptr && session->api().currentOperation() == op) {
        result++;
      }
    }
    return result;
  }
 
 protected:
  FTPSession<ClientType> *sessions[FTP_MAX_SESSIONS] = {nullptr};
  IPAddress address;
  int port;
  const char *username;
  const char *password;
};
 
}  // namespace ftp_client