Source/Tools/TextBaker.cpp - master - Mr.Stalin/FOnline-Engine

Source/Tools/TextBaker.cpp
187 строк7 KB
cvet
Non const locals (#190)
24 июл 2026, 10:46
24 июл 2026, 10:464883d25
//      __________        ___               ______            _
//     / ____/ __ \____  / (_)___  ___     / ____/___  ____ _(_)___  ___
//    / /_  / / / / __ \/ / / __ \/ _ \   / __/ / __ \/ __ `/ / __ \/ _ `
//   / __/ / /_/ / / / / / / / / /  __/  / /___/ / / / /_/ / / / / /  __/
//  /_/    \____/_/ /_/_/_/_/ /_/\___/  /_____/_/ /_/\__, /_/_/ /_/\___/
//                                                  /____/
// FOnline Engine
// https://fonline.ru
// https://github.com/cvet/fonline
//
// MIT License
//
// Copyright (c) 2006 - 2026, Anton Tsvetinskiy aka cvet <cvet@tut.by>
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
//

#include "TextBaker.h"
#include "ConfigFile.h"
#include "TextPack.h"

FO_BEGIN_NAMESPACE

TextBaker::TextBaker(shared_ptr<BakingContext> ctx) :
    BaseBaker(std::move(ctx), NAME)
{
    FO_STACK_TRACE_ENTRY();

    if (_context->Settings->BakeLanguages.empty()) {
        throw TextBakerException("No bake languages specified");
    }
}

TextBaker::~TextBaker()
{
    FO_STACK_TRACE_ENTRY();
}

void TextBaker::BakeFiles(const FileCollection& files, string_view target_path) const
{
    FO_STACK_TRACE_ENTRY();

    if (!target_path.empty() && !strex(target_path).get_file_extension().starts_with("fotxt")) {
        return;
    }

    // Collect files
    vector<File> filtered_files;
    set<string> filtered_paths;
    set<string> packs_to_complete;

    for (const auto& file_header : files) {
        string ext = strex(file_header.GetPath()).get_file_extension();

        if (ext != "fotxt") {
            continue;
        }

        auto name_pair = strex(file_header.GetNameNoExt()).split('.');
        FO_VERIFY_AND_THROW(name_pair.size() == 2, "Text filename must contain pack name and language suffix", file_header.GetPath(), file_header.GetNameNoExt(), name_pair.size());
        const auto& text_pack_name = name_pair[0];
        const auto& lang_name = name_pair[1];

        if (_context->BakeChecker && !_context->BakeChecker(strex("{}.{}.{}.fotxt-bin", _context->PackName, text_pack_name, lang_name), file_header.GetWriteTime())) {
            continue;
        }

        filtered_files.emplace_back(File::Load(file_header));
        filtered_paths.emplace(string(file_header.GetPath()));
        packs_to_complete.emplace(text_pack_name);
    }

    // Ensure full language set for changed text packs
    for (const auto& file_header : files) {
        string ext = strex(file_header.GetPath()).get_file_extension();

        if (ext != "fotxt") {
            continue;
        }

        auto name_pair = strex(file_header.GetNameNoExt()).split('.');
        FO_VERIFY_AND_THROW(name_pair.size() == 2, "Text filename must contain pack name and language suffix", file_header.GetPath(), file_header.GetNameNoExt(), name_pair.size());
        const auto& text_pack_name = name_pair[0];
        const auto& lang_name = name_pair[1];

        if (!packs_to_complete.contains(text_pack_name)) {
            continue;
        }
        if (std::ranges::find(_context->Settings->BakeLanguages, lang_name) == _context->Settings->BakeLanguages.end()) {
            continue;
        }
        if (filtered_paths.contains(string(file_header.GetPath()))) {
            continue;
        }

        filtered_files.emplace_back(File::Load(file_header));
        filtered_paths.emplace(string(file_header.GetPath()));
    }

    if (filtered_files.empty()) {
        return;
    }

    // Find languages
    const auto& default_lang = _context->Settings->BakeLanguages.front();
    set<string> languages;
    set<string> all_languages;

    for (const auto& file : filtered_files) {
        auto name_pair = strex(file.GetNameNoExt()).split('.');
        FO_VERIFY_AND_THROW(name_pair.size() == 2, "Text filename must contain pack name and language suffix", file.GetPath(), file.GetNameNoExt(), name_pair.size());
        const auto& lang_name = name_pair[1];

        if (all_languages.emplace(lang_name).second) {
            if (std::ranges::find(_context->Settings->BakeLanguages, lang_name) == _context->Settings->BakeLanguages.end()) {
                WriteLog(LogType::Warning, "Unsupported language: {}. Skip", lang_name);
            }
            else {
                languages.emplace(lang_name);
            }
        }
    }

    if (languages.count(default_lang) == 0) {
        throw TextBakerException("Default language not found");
    }

    // Parse texts
    vector<pair<string, map<string, TextPack>>> lang_packs;
    HashStorage hashes {};

    for (const auto& target_lang : languages) {
        map<string, TextPack> lang_pack;

        for (const auto& file : filtered_files) {
            auto name_pair = strex(file.GetNameNoExt()).split('.');
            FO_VERIFY_AND_THROW(name_pair.size() == 2, "Text filename must contain pack name and language suffix", file.GetPath(), file.GetNameNoExt(), name_pair.size());
            const auto& text_pack_name = name_pair[0];
            const auto& lang_name = name_pair[1];

            if (lang_name == target_lang) {
                TextPack text_pack {&hashes};

                if (!text_pack.LoadFromString(file.GetStr(), text_pack_name)) {
                    throw TextPackException("Invalid text file", file.GetPath());
                }

                lang_pack.emplace(text_pack_name, std::move(text_pack));
            }
        }

        if (target_lang == default_lang) {
            // Default language should be first
            lang_packs.insert(lang_packs.begin(), std::make_pair(target_lang, std::move(lang_pack)));
        }
        else {
            lang_packs.emplace_back(std::make_pair(target_lang, std::move(lang_pack)));
        }
    }

    TextPack::FixPacks(_context->Settings->BakeLanguages, lang_packs);

    // Save parsed packs
    for (auto&& [lang_name, lang_pack] : lang_packs) {
        for (auto&& [pack_name, text_pack] : lang_pack) {
            _context->WriteData(strex("{}.{}.{}.fotxt-bin", _context->PackName, pack_name, lang_name), text_pack.GetBinaryData());
        }
    }
}

FO_END_NAMESPACE