#pragma once
#ifndef PROCESSOR_H
#define PROCESSOR_H

#include <cstdlib>
#include <string>
#include <fstream>
#include <mutex>
#include <vector>
#include <iostream>
#include <thread>

#include "cryptlib.h"
#include "rsa.h"
#include "osrng.h"
#include "files.h"
#include "base64.h"

using namespace std;
using namespace CryptoPP;

namespace Cryptography {
    template<class E, class D, class Priv, class Pub>
    class Processor
    {
    public:
        Processor() {};
        virtual ~Processor() {};

        AutoSeededRandomPool rnd;

        void EncryptFile(const string& sourceFile, const string& destFile, const Pub& publicKey)
        {
            ifstream sourceStream(sourceFile.c_str());
            ofstream destStream(destFile.c_str());

            encryptor = E(publicKey);

            FileSource fs(sourceStream, true,
                new PK_EncryptorFilter(rnd, encryptor,
                    new Base64Encoder( // если эту строку убрать, то файл будет бинарным и не расшифруется
                        new FileSink(destStream)
                    ) // и эту
                )
            );

            //string sourceData(std::istreambuf_iterator<char>(sourceStream), {})
            //    , destData;

            //StringSource ss(sourceData, true,
            //    new PK_EncryptorFilter(rnd, encryptor,
            //        new StringSink(destData)
            //    )
            //);

            //destStream << destData;
        }

        void DecryptFile(const string& sourceFile, const string& destFile, const Priv& privateKey)
        {
            ifstream sourceStream(sourceFile.c_str());
            ofstream destStream(destFile.c_str());

            decryptor = D(privateKey);

            FileSource fs(sourceStream, true,
                new Base64Decoder(
                    new PK_DecryptorFilter(rnd, decryptor,
                        new FileSink(destStream)
                    )
                )
            );
        }
    }
}