CCats/cli/include/fileman.h

#ifndef FILEMAN_H
#define FILEMAN_H

#include <fstream>
#include <string>
#include <vector>

#define BLOCKSIZE 8

/**
 * @class FileMan
 *
 * Provides File I/O abstraction
 */
class FileMan {
private:
  /**
   * Internal state
   *
   * Filestreams for put and get
   * Vector for holding received filenames from listing
   * Names for put and get
   * Size for internal read
   * Total and Remaining chunks for put, get and list
   * Boolean replacement for filestreams being open for list
   *
   */
  std::ifstream putfile;
  std::ofstream getfile;
  std::vector<std::string> listdata;
  std::string getname, putname;
  const unsigned int max_read_len = 8;
  int putsize;
  int putchunks;
  int putchunksRemaining;
  int getchunks;
  int getchunksRemaining;
  int listchunks;
  int listchunksRemaining;

  bool islisting;

public:
  /**
   * Constructor and Destructor
   */
  FileMan();
  ~FileMan();

  /**
   * Query internal state
   *
   * Return true if the corresponding action is being performed, false otherwise
   */
  bool isGetting();
  bool isPutting();
  bool isListing();

  /**
   * Check for and prepare state and streams for reading/writing
   *
   * Return true if successful, false otherwise
   */
  bool openPut(const std::string &name);
  bool openGet(const std::string &name);
  bool openList();

  /**
   * Close the respective filestream
   */
  void closePut();
  void closeGet();
  void closeList();

  /**
   * Query the names of the file currently being put or get
   */
  std::string getPutName();
  std::string getGetName();

  /**
   * Cancel a put, get or list, depreparing internal state (closing streams if
   * required)
   */
  void cancelPut();
  void cancelGet();
  void cancelList();

  /**
   * Read max_rea_len bytes from the current file opened for put
   */
  std::vector<char> readPut();
  /**
   * Write the provided vector to the current file opened for get
   */
  void writeGet(std::vector<char> data);

  /**
   * Wrapper methods for reading and writing base64 encoded data instead of raw
   * bytes
   */
  std::string readBase64();
  void writeBase64(std::string data);

  /**
   * read and write emulating methods for list
   */
  void putListData(std::vector<std::string> names);
  std::vector<std::string> getListData();

  /**
   * Query internal state, requesting the corresponding size
   */
  int getPutChunks();
  int getGetChunks();
  int getListChunks();
  int getPutRemainingChunks();
  int getGetRemainingChunks();
  int getListRemainingChunks();
  int getPutSize();

  /**
   * Set internal state, adjusting the chunks as well as chunks remaining for
   * get and list
   */
  void setGetChunks(int chunks);
  void setListChunks(int chunks);
};

#endif