#include "ModuleEncryption.h"
#include "DataIngestionLayer.h"
#include "DataOutputLayer.h"
#include "KeyManager.h"
#include <iostream>
#include <GCrypt/GCipher.h>

using namespace Module;
using namespace Leonetienne::GCrypt;

void Encryption::Run() {

  // Initialize the data ingestion layer
  IO::DataIngestionLayer::Init();

  // Initialize the data output layer
  IO::DataOutputLayer::Init();

  // Initialize a cipher
  GCipher cipher(
    KeyManager::GetKey(),
    GCipher::DIRECTION::ENCIPHER
  );

  while (!IO::DataOutputLayer::IsFinished()) {
    // Read in new blocks, if not reached eof
    if (!IO::DataIngestionLayer::ReachedEOF()) {
      IO::DataIngestionLayer::ReadBlock();
    }

    // Process a block, if one is ready
    if (IO::DataIngestionLayer::IsBlockReady()) {
      const Block cleartext = IO::DataIngestionLayer::GetNextBlock();
      const Block ciphertext = cipher.Digest(cleartext);

      // Enqueue the block for output
      IO::DataOutputLayer::Enqueue(ciphertext);

      // Tell the data output layer that it just received the
      // last block, if it did
      if (IO::DataIngestionLayer::IsFinished()) {
        IO::DataOutputLayer::ReachedEOF();
      }
    }

    // Attempt to write a block
    IO::DataOutputLayer::WriteBlock();
  }

  // Destruct the data ingestion layer
  IO::DataIngestionLayer::Destruct();

  // Destruct the data output layer
  IO::DataOutputLayer::Destruct();

  return;
}