storage.h

#ifndef STORAGE_H
#define STORAGE_H

#include <filesystem>
#include <fstream>
#include <string>
#include <unordered_map>
#include <random>
#include <chrono>
#include <mutex>

#include <nlohmann/json.hpp>

nlohmann::json reduceJsonObject(nlohmann::json&, std::map<std::string, bool>&);

class group_storage_t {
private:
	std::filesystem::path storagePath;
	std::fstream storageFile;
	std::mutex fileLock;
	std::unordered_map<size_t, size_t> idPositions; // 1. id of doc 2. row index
	std::unordered_map<size_t, nlohmann::json> newDocuments; // 1. id of doc 2. full document
	std::unordered_map<size_t, nlohmann::json> editedDocuments; // 1. row index 2. update operation
	std::set<size_t> removedDocuments; // row index

public:
	group_storage_t(std::string storagePath);
	bool savedHere(size_t);
	void getDocuments(std::vector<size_t>* ids, std::vector<nlohmann::json>& documents, bool allDocuments = false, std::map<std::string, bool> projection = {});
	
	size_t countDocuments();
	void insertDocument(const nlohmann::json& document);
	void editDocument(const size_t id, const nlohmann::json& update);
	bool removeDocument(const size_t id);

	std::vector<size_t> getAllIds();
	void getAllIds(std::vector<size_t>& container);
	void doFuncOnAllDocuments(std::function<void(const nlohmann::json&)> func);
	void save();
};


#endif // !STORAGE_H