
Thanks to @Atemo and @aleos89 for their help. Co-authored-by: Atemo <capucrath@gmail.com> Co-authored-by: Aleos <aleos89@users.noreply.github.com>
251 lines
7.7 KiB
C++
251 lines
7.7 KiB
C++
// Copyright (c) rAthena Dev Teams - Licensed under GNU GPL
|
|
// For more information, see LICENCE in the main folder
|
|
|
|
#include "yamlupgrade.hpp"
|
|
|
|
static bool upgrade_achievement_db(std::string file, const uint32 source_version);
|
|
static bool upgrade_item_db(std::string file, const uint32 source_version);
|
|
|
|
template<typename Func>
|
|
bool process(const std::string &type, uint32 version, const std::vector<std::string> &paths, const std::string &name, Func lambda) {
|
|
for (const std::string &path : paths) {
|
|
const std::string name_ext = name + ".yml";
|
|
const std::string from = path + "/" + name_ext;
|
|
const std::string to = path + "/" + name + "-upgrade.yml";
|
|
|
|
inNode.reset();
|
|
|
|
if (fileExists(from)) {
|
|
inNode = YAML::LoadFile(from);
|
|
uint32 source_version = getHeaderVersion(inNode);
|
|
|
|
if (source_version >= version) {
|
|
continue;
|
|
}
|
|
|
|
if (!askConfirmation("Found the file \"%s\", which requires an upgrade.\nDo you want to convert it now? (Y/N)\n", from.c_str())) {
|
|
continue;
|
|
}
|
|
|
|
if (fileExists(to)) {
|
|
if (!askConfirmation("The file \"%s\" already exists.\nDo you want to replace it? (Y/N)\n", to.c_str())) {
|
|
continue;
|
|
}
|
|
}
|
|
|
|
ShowNotice("Upgrade process has begun.\n");
|
|
|
|
std::ofstream out;
|
|
|
|
body.~Emitter();
|
|
new (&body) YAML::Emitter();
|
|
out.open(to);
|
|
|
|
if (!out.is_open()) {
|
|
ShowError("Can not open file \"%s\" for writing.\n", to.c_str());
|
|
return false;
|
|
}
|
|
|
|
prepareHeader(out, type, version, name);
|
|
if (inNode["Body"].IsDefined()) {
|
|
prepareBody();
|
|
|
|
if (!lambda(path, name_ext, source_version)) {
|
|
out.close();
|
|
return false;
|
|
}
|
|
|
|
finalizeBody();
|
|
out << body.c_str();
|
|
}
|
|
prepareFooter(out);
|
|
// Make sure there is an empty line at the end of the file for git
|
|
out << "\n";
|
|
out.close();
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
int do_init(int argc, char** argv) {
|
|
const std::string path_db = std::string(db_path);
|
|
const std::string path_db_mode = path_db + "/" + DBPATH;
|
|
const std::string path_db_import = path_db + "/" + DBIMPORT;
|
|
|
|
// Loads required conversion constants
|
|
if (fileExists(item_db.getDefaultLocation())) {
|
|
item_db.load();
|
|
} else {
|
|
parse_item_constants_txt((path_db_mode + "item_db.txt").c_str());
|
|
parse_item_constants_txt((path_db_import + "item_db.txt").c_str());
|
|
}
|
|
if (fileExists(mob_db.getDefaultLocation())) {
|
|
mob_db.load();
|
|
} else {
|
|
sv_readdb(path_db_mode.c_str(), "mob_db.txt", ',', 31 + 2 * MAX_MVP_DROP + 2 * MAX_MOB_DROP, 31 + 2 * MAX_MVP_DROP + 2 * MAX_MOB_DROP, -1, &parse_mob_constants_txt, false);
|
|
sv_readdb(path_db_import.c_str(), "mob_db.txt", ',', 31 + 2 * MAX_MVP_DROP + 2 * MAX_MOB_DROP, 31 + 2 * MAX_MVP_DROP + 2 * MAX_MOB_DROP, -1, &parse_mob_constants_txt, false);
|
|
}
|
|
if (fileExists(skill_db.getDefaultLocation())) {
|
|
skill_db.load();
|
|
} else {
|
|
sv_readdb(path_db_mode.c_str(), "skill_db.txt", ',', 18, 18, -1, parse_skill_constants_txt, false);
|
|
sv_readdb(path_db_import.c_str(), "skill_db.txt", ',', 18, 18, -1, parse_skill_constants_txt, false);
|
|
}
|
|
|
|
// Load constants
|
|
#define export_constant_npc(a) export_constant(a)
|
|
#include "../map/script_constants.hpp"
|
|
|
|
std::vector<std::string> root_paths = {
|
|
path_db,
|
|
path_db_mode,
|
|
path_db_import
|
|
};
|
|
|
|
if (!process("ACHIEVEMENT_DB", 2, root_paths, "achievement_db", [](const std::string &path, const std::string &name_ext, uint32 source_version) -> bool {
|
|
return upgrade_achievement_db(path + name_ext, source_version);
|
|
})) {
|
|
return 0;
|
|
}
|
|
|
|
if (!process("ITEM_DB", 2, root_paths, "item_db", [](const std::string& path, const std::string& name_ext, uint32 source_version) -> bool {
|
|
return upgrade_item_db(path + name_ext, source_version);
|
|
})) {
|
|
return 0;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
void do_final(void) {
|
|
}
|
|
|
|
// Implementation of the upgrade functions
|
|
static bool upgrade_achievement_db(std::string file, const uint32 source_version) {
|
|
size_t entries = 0;
|
|
|
|
for (const auto &input : inNode["Body"]) {
|
|
body << YAML::BeginMap;
|
|
body << YAML::Key << "Id" << YAML::Value << input["ID"];
|
|
|
|
std::string constant = input["Group"].as<std::string>();
|
|
|
|
constant.erase(0, 3); // Remove "AG_"
|
|
if (constant.compare("Chat") == 0) // Chat -> Chatting
|
|
constant.insert(4, "ting");
|
|
else if (constant.compare("Hear") == 0 || constant.compare("See") == 0)
|
|
constant = "Chatting"; // Aegis treats these as general "Talk to NPC" achievements.
|
|
else if (constant.compare("Refine") == 0) { // Refine -> Enchant
|
|
constant.erase(0, 6);
|
|
constant = "Enchant" + constant;
|
|
}
|
|
body << YAML::Key << "Group" << YAML::Value << name2Upper(constant);
|
|
body << YAML::Key << "Name" << YAML::Value << input["Name"];
|
|
|
|
if (input["Target"].IsDefined()) {
|
|
body << YAML::Key << "Targets";
|
|
body << YAML::BeginSeq;
|
|
|
|
for (const auto &it : input["Target"]) {
|
|
body << YAML::BeginMap;
|
|
body << YAML::Key << "Id" << YAML::Value << it["Id"];
|
|
if (it["MobID"].IsDefined()) {
|
|
uint16 mob_id = it["MobID"].as<uint16>();
|
|
std::string *mob_name = util::umap_find(aegis_mobnames, mob_id);
|
|
|
|
if (mob_name == nullptr) {
|
|
ShowWarning("mob_avail reading: Invalid mob-class %hu, Mob not read.\n", mob_id);
|
|
return false;
|
|
}
|
|
|
|
body << YAML::Key << "Mob" << YAML::Value << *mob_name;
|
|
}
|
|
if (it["Count"].IsDefined() && it["Count"].as<int32>() > 1)
|
|
body << YAML::Key << "Count" << YAML::Value << it["Count"];
|
|
body << YAML::EndMap;
|
|
}
|
|
|
|
body << YAML::EndSeq;
|
|
}
|
|
|
|
if (input["Condition"].IsDefined())
|
|
body << YAML::Key << "Condition" << YAML::Value << input["Condition"];
|
|
|
|
if (input["Map"].IsDefined())
|
|
body << YAML::Key << "Map" << YAML::Value << input["Map"];
|
|
|
|
if (input["Dependent"].IsDefined()) {
|
|
body << YAML::Key << "Dependents";
|
|
body << YAML::BeginMap;
|
|
|
|
for (const auto &it : input["Dependent"]) {
|
|
body << YAML::Key << it["Id"] << YAML::Value << true;
|
|
}
|
|
|
|
body << YAML::EndMap;
|
|
}
|
|
|
|
/**
|
|
* Example usage for adding label at specific version.
|
|
if (source_version < ?) {
|
|
body << YAML::Key << "CustomLabel" << YAML::Value << "Unique";
|
|
}
|
|
*/
|
|
|
|
if (input["Reward"].IsDefined()) {
|
|
body << YAML::Key << "Rewards";
|
|
body << YAML::BeginMap;
|
|
if (input["Reward"]["ItemID"].IsDefined()) {
|
|
t_itemid item_id = input["Reward"]["ItemID"].as<t_itemid>();
|
|
std::string *item_name = util::umap_find(aegis_itemnames, item_id);
|
|
|
|
if (item_name == nullptr) {
|
|
ShowError("Reward item name for item ID %u is not known.\n", item_id);
|
|
return false;
|
|
}
|
|
|
|
body << YAML::Key << "Item" << YAML::Value << *item_name;
|
|
}
|
|
if (input["Reward"]["Amount"].IsDefined() && input["Reward"]["Amount"].as<uint16>() > 1)
|
|
body << YAML::Key << "Amount" << YAML::Value << input["Reward"]["Amount"];
|
|
if (input["Reward"]["Script"].IsDefined())
|
|
body << YAML::Key << "Script" << YAML::Value << input["Reward"]["Script"];
|
|
if (input["Reward"]["TitleID"].IsDefined())
|
|
body << YAML::Key << "TitleId" << YAML::Value << input["Reward"]["TitleID"];
|
|
body << YAML::EndMap;
|
|
}
|
|
|
|
body << YAML::Key << "Score" << YAML::Value << input["Score"];
|
|
|
|
body << YAML::EndMap;
|
|
entries++;
|
|
}
|
|
|
|
ShowStatus("Done converting/upgrading '" CL_WHITE "%d" CL_RESET "' achievements in '" CL_WHITE "%s" CL_RESET "'.\n", entries, file.c_str());
|
|
|
|
return true;
|
|
}
|
|
|
|
static bool upgrade_item_db(std::string file, const uint32 source_version) {
|
|
size_t entries = 0;
|
|
|
|
for( auto &input : inNode["Body"] ){
|
|
// If under version 2
|
|
if( source_version < 2 ){
|
|
// Add armor level to all equipments
|
|
if( input["Type"].IsDefined() && input["Type"].as<std::string>() == "Armor" ){
|
|
input["ArmorLevel"] = 1;
|
|
}
|
|
}
|
|
|
|
// TODO: this currently converts all scripts using Literal syntax to normal double quote strings
|
|
body << input;
|
|
entries++;
|
|
}
|
|
|
|
ShowStatus("Done converting/upgrading '" CL_WHITE "%d" CL_RESET "' items in '" CL_WHITE "%s" CL_RESET "'.\n", entries, file.c_str());
|
|
|
|
return true;
|
|
}
|