#pragma once

#include "ISave.h"

#include <unordered_map>
#include <memory>
#include <string>
#include <vector>
#include <list>

class SaveMap final
{
    std::string class_name;
    std::shared_ptr<SaveMap> parent_;

    // Индивидуальные значения
    std::unordered_map<std::string, long long> save_long;
    std::unordered_map<std::string, double> save_double;
    std::unordered_map<std::string, std::string> save_string;
    std::unordered_map<std::string, std::shared_ptr<ISave>> save_objects;
    
    // Массивы
    std::unordered_map<std::string, std::vector<long long>> save_vector_integer;
    std::unordered_map<std::string, std::vector<double>> save_vector_double;
    std::unordered_map<std::string, std::vector<std::shared_ptr<ISave>>> save_vector_objects;
    std::unordered_map<std::string, std::vector<std::string>> save_vector_strings;

    // Связаные списки
    std::unordered_map<std::string, std::list<long long>> save_list_integer;
    std::unordered_map<std::string, std::list<double>> save_list_double;
    std::unordered_map<std::string, std::list<std::shared_ptr<ISave>>> save_list_objects;
    std::unordered_map<std::string, std::list<std::string>> save_list_strings;

    static ISave* MakeObjectByName(const std::string& name);

public:
    // Создаёт пустой объект для заполнения
    SaveMap(const char* class_name);
    // Токенезирут JSON строку
    SaveMap(const std::string& json_data);
    // Коприрование
    SaveMap(SaveMap& save_map);
    // Перемещение
    SaveMap(SaveMap&& save_map) noexcept;

    // Методы созранения значений
    std::shared_ptr<SaveMap> SaveInteger(const char* name, int value) noexcept;
    std::shared_ptr<SaveMap> SaveDouble(const char* name, double value) noexcept;
    std::shared_ptr<SaveMap> SaveString(const char* name, const std::string& value) noexcept;
    std::shared_ptr<SaveMap> SaveObject(const char* name, std::shared_ptr<ISave> object) noexcept;
    std::shared_ptr<SaveMap> SaveVectorInteger(const char* name, std::vector<long long>&& value) noexcept;
    std::shared_ptr<SaveMap> SaveVectorDouble(const char* name, std::vector<double>&& value) noexcept;
    std::shared_ptr<SaveMap> SaveVectorStrings(const char* name, std::vector<std::string>&& value) noexcept;
    std::shared_ptr<SaveMap> SaveVectorObject(const char* name, std::vector<std::shared_ptr<ISave>>&& value) noexcept;
    std::shared_ptr<SaveMap> SaveListInteger(const char* name, std::list<long long>&& value) noexcept;
    std::shared_ptr<SaveMap> SaveListDouble(const char* name, std::list<double>&& value) noexcept;
    std::shared_ptr<SaveMap> SaveListStrings(const char* name, std::list<std::string>&& value) noexcept;
    std::shared_ptr<SaveMap> SaveListObject(const char* name, std::list<std::shared_ptr<ISave>>&& value) noexcept;

    // Методы получения значенй
    long long GetInteger(const char* name);
    double GetDouble(const char* name);
    std::string GetString(const char* name);
    std::shared_ptr<ISave> GetObject(const char* name);
    std::vector<long long>& GetVectorInteger(const char* name);
    std::vector<double>& GetVectorDouble(const char* name);
    std::vector<std::string> GetVectorString(const char* name);
    std::vector<std::shared_ptr<ISave>>& GetVectorObject(const char* name);
    std::list<long long>& GetListInteger(const char* name);
    std::list<double>& GetListDouble(const char* name);
    std::list<std::string>& GetListString(const char* name);
    std::list<std::shared_ptr<ISave>>& GetListObject(const char* name);
    
    std::shared_ptr<SaveMap> getParent() const { return parent_; }
    const std::string& getClassName() const { return class_name; }

    // Собирает все токены в JSON объект
    std::string serialize() noexcept;

    std::shared_ptr<SaveMap> connect_to(const std::shared_ptr<SaveMap>& parent);

    static std::string CleaningJSON(const std::string& json_data);
};