/**

 * This file is part of the CernVM File System.

 */

#ifndef CVMFS_RECEIVER_PAYLOAD_PROCESSOR_H_

#define CVMFS_RECEIVER_PAYLOAD_PROCESSOR_H_

#include <stdint.h>

#include <map>

#include <string>

#include <vector>

#include "pack.h"

#include "upload.h"

#include "util/raii_temp_dir.h"

namespace receiver {

  FileInfo();

  explicit FileInfo(const ObjectPackBuild::Event& event);

  FileInfo(const FileInfo& other);

  FileInfo& operator=(const FileInfo& other);

  std::string temp_path;

  size_t total_size;

  size_t current_size;

  shash::ContextPtr hash_context;

  std::vector<unsigned char> hash_buffer;

  bool skip;

};

/**

 * This class is used in the `cvmfs_receiver` tool, on repository gateway

 * machines. The receiver::Reactor class, implementing the event loop of the

 * `cvmfs_receiver` tool, dispatches the handling of the kSubmitPayload events

 * to this class.

 *

 * Its responsibility is reading the payload - containing a serialized

 * ObjectPack - from a file descriptor, and unpacking it into the repository.

 */

class PayloadProcessor {

 public:

  enum Result { kSuccess, kPathViolation, kSpoolerError, kOtherError };

  PayloadProcessor();

  virtual ~PayloadProcessor();

  Result Process(int fdin, const std::string& header_digest,

                 const std::string& path, uint64_t header_size);

  virtual void ConsumerEventCallback(const ObjectPackBuild::Event& event);

 protected:

  // NOTE: These methods are made virtual such that they can be mocked for

  //       the purpose of unit testing

  virtual Result Initialize();

  virtual Result Finalize();

  virtual void Upload(const std::string& source,

                      const std::string& dest);

  virtual bool WriteFile(int fd, const void* const buf, size_t buf_size);

 private:

  typedef std::map<shash::Any, FileInfo>::iterator FileIterator;

  std::map<shash::Any, FileInfo> pending_files_;

  std::string current_repo_;

  UniquePtr<upload::Spooler> spooler_;

  UniquePtr<RaiiTempDir> temp_dir_;

  int num_errors_;

};

}  // namespace receiver

#endif  // CVMFS_RECEIVER_PAYLOAD_PROCESSOR_H_