#pragma once
#include <TestProject/RNG/model/IModel.h>
#include <nlohmann/json.hpp>
#include <functional>
#include <map>
#include <memory>
#include <string>
#include <vector>
using json = nlohmann::json;

class ModelRegistry {
public:
  using Ctor = std::function<std::unique_ptr<IModel>(const json& params, float defaultVal)>;

  static ModelRegistry& instance();

  void loadModels(const std::string& jsonPath);

  IModel* getOrCreate(const std::string& tables1Spec, float defaultVal,
                      const std::string& instanceKey = "");

  std::vector<IModel*> findByModelName(const std::string& modelName);

  void registerMode(const std::string& mode, Ctor ctor);

private:
  ModelRegistry();
  std::unique_ptr<IModel> createModel(const std::string& modelName, float defaultVal);

  struct ModelDef {
    std::string mode;
    json params;
  };
  std::map<std::string, ModelDef> modelTemplates;
  std::map<std::string, std::unique_ptr<IModel>> instances;
  std::map<std::string, std::vector<IModel*>> byModelName;
  std::map<std::string, Ctor> factory;
};