77 lines
1.5 KiB
C++
77 lines
1.5 KiB
C++
|
#include "datastore.h"
|
||
|
|
#include "pch.h"
|
||
|
|
|
||
|
|
DataStore::DataStore(const char* fileName, bool isConfig) noexcept
|
||
|
|
{
|
||
|
|
// Output config file in the same directory as the asi
|
||
|
|
path = PLUGIN_PATH((char*)(isConfig ? "/" : "/CheatMenu/data/")) + std::string(fileName) + fileExt;
|
||
|
|
|
||
|
|
if (std::filesystem::exists(path))
|
||
|
|
{
|
||
|
|
toml::parse_result result = toml::parse_file(path);
|
||
|
|
|
||
|
|
if (result)
|
||
|
|
{
|
||
|
|
pTable = std::make_unique<toml::table>(std::move(result));
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if (pTable == nullptr)
|
||
|
|
{
|
||
|
|
pTable = std::make_unique<toml::table>();
|
||
|
|
|
||
|
|
if (isConfig)
|
||
|
|
{
|
||
|
|
Log::Print<eLogLevel::Info>("Creating {}{}", fileName, fileExt);
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
Log::Print<eLogLevel::Warn>("Error parsing {}{}", fileName, fileExt);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
bool DataStore::Contains(const char* ele) noexcept
|
||
|
|
{
|
||
|
|
if (pTable)
|
||
|
|
{
|
||
|
|
return pTable->contains(ele);
|
||
|
|
}
|
||
|
|
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
DataStore::Table& DataStore::Items() noexcept
|
||
|
|
{
|
||
|
|
return pTable->as_table()->ref<DataStore::Table>();
|
||
|
|
}
|
||
|
|
|
||
|
|
void DataStore::RemoveTable(const char* key) noexcept
|
||
|
|
{
|
||
|
|
if (pTable)
|
||
|
|
{
|
||
|
|
pTable->erase(key);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void DataStore::RemoveKey(const char* key, const char* entry) noexcept
|
||
|
|
{
|
||
|
|
if (pTable)
|
||
|
|
{
|
||
|
|
(*pTable)[key].as_table()->erase(entry);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
void DataStore::Save() noexcept
|
||
|
|
{
|
||
|
|
if (pTable)
|
||
|
|
{
|
||
|
|
std::ofstream file(path);
|
||
|
|
if (file.good())
|
||
|
|
{
|
||
|
|
file << *pTable << std::endl;
|
||
|
|
file.close();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|