#pragma once

#include <mutex>

#include <string>

#include "utils/engine_constants.h"

#include "utils/logging_utils.h"

#include <vector>

#include "utils/engine_constants.h"

#include "utils/result.hpp"

namespace config_yaml_utils {

const std::string kDefaultHost{"127.0.0.1"};

const std::string kDefaultPort{"39281"};

const int kDefaultMaxLines{100000};

constexpr const uint64_t kDefaultCheckedForUpdateAt = 0u;

constexpr const uint64_t kDefaultCheckedForLlamacppUpdateAt = 0u;

constexpr const auto kDefaultLatestRelease = "default_version";

constexpr const auto kDefaultLatestLlamacppRelease = "";

constexpr const auto kDefaultCorsEnabled = true;

const std::vector<std::string> kDefaultEnabledOrigins{

"http://localhost:39281", "http://127.0.0.1:39281", "http://0.0.0.0:39281"};

constexpr const auto kDefaultNoProxy = "example.com,::1,localhost,127.0.0.1";

const std::vector<std::string> kDefaultSupportedEngines{kLlamaEngine};

struct CortexConfig {

std::string logFolderPath;

std::string logLlamaCppPath;

std::string logOnnxPath;

std::string dataFolderPath;

int maxLogLines;

std::string apiServerHost;

std::string apiServerPort;

uint64_t checkedForUpdateAt;

uint64_t checkedForLlamacppUpdateAt;

std::string latestRelease;

std::string latestLlamacppRelease;

std::string huggingFaceToken;

/**

* Github's API requires a user-agent string.

*/

std::string gitHubUserAgent;

std::string gitHubToken;

std::string llamacppVariant;

std::string llamacppVersion;

bool enableCors;

std::vector<std::string> allowedOrigins;

std::string proxyUrl;

bool verifyProxySsl;

bool verifyProxyHostSsl;

std::string proxyUsername;

std::string proxyPassword;

std::string noProxy;

bool verifyPeerSsl;

bool verifyHostSsl;

std::string sslCertPath;

std::string sslKeyPath;

std::vector<std::string> supportedEngines;

uint64_t checkedForSyncHubAt;

std::vector<std::string> apiKeys;

};

class CortexConfigMgr {

private:

CortexConfigMgr() {}

std::mutex mtx_;

public:

CortexConfigMgr(CortexConfigMgr const&) = delete;

CortexConfigMgr& operator=(CortexConfigMgr const&) = delete;

~CortexConfigMgr() {}

static CortexConfigMgr& GetInstance() {

static CortexConfigMgr ccm;

return ccm;

}

cpp::result<void, std::string> DumpYamlConfig(const CortexConfig& config,

const std::string& path);

CortexConfig FromYaml(const std::string& path,

const CortexConfig& default_cfg);

};

} // namespace config_yaml_utils