/*
 * AIM script2txt2 (simpler version)
 * Copyright (C) 2024 lzwdgc
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <mmap.h>
#include <types.h>

#include <primitives/sw/main.h>
#include <primitives/sw/settings.h>
#include <primitives/sw/cl.h>

#include <fstream>

int main(int argc, char *argv[]) {
    cl::opt<path> p(cl::Positional, cl::desc("<script.scr or scripts dir>"), cl::Required);

    cl::ParseCommandLineOptions(argc, argv);

    auto func = [](auto filename) {
        primitives::templates2::mmap_file<uint8_t> f{filename};
        stream s{f};
        script scr = s;
        std::vector<std::string_view> lines;
        int sz{};
        for (int i = 0; i < scr.nlines; ++i) {
            sz += lines.emplace_back((const char *)f.p + sizeof(script) + sz).size() + 1;
        }

        // write script
        {
            filename += ".txt";
            if (std::ofstream ofile(filename); ofile) {
                std::string indent, space = "    "s;
                auto inc = [&]() {
                    indent += space;
                };
                auto dec = [&]() {
                    if (!indent.empty()) {
                        indent.resize(indent.size() - space.size());
                        return true;
                    }
                    return false;
                };
                int procs{};
                for (auto &&l : lines) {
                    auto proc = l.starts_with("PROC"sv);
                    auto end = l == "END"sv;
                    auto lbrace = l == "{"sv;
                    auto rbrace = l == "}"sv;

                    if (rbrace) {
                        if (!dec()) {
                            ofile << "// script2txt2 comment: unbalanced!\n";
                        }
                    }
                    if (end && procs) {
                        if (!dec()) {
                            ofile << "// script2txt2 comment: unbalanced!\n";
                        }
                    }

                    ofile << indent << l << "\n";

                    if (end && indent.empty()) {
                        //ofile << "\n";
                    }
                    if (end && procs) {
                        procs = 0;
                    }
                    if (lbrace || proc) {
                        indent += space;
                    }
                    if (proc) {
                        procs = 1;
                    }
                }
            }
        }
    };

    if (fs::is_regular_file(p)) {
        func(p.string());
    } else if (fs::is_directory(p)) {
        auto files = enumerate_files_like(p, ".*\\.scr", false);
        auto files2 = enumerate_files_like(p, ".*\\.QST", false);
        files.insert(files2.begin(), files2.end());
        for (auto &f : files) {
            std::cout << "processing: " << f << "\n";
            func(f.string());
        }
    } else {
        throw std::runtime_error("Bad fs object");
    }

    return 0;
}