reone/tools/program.cpp

/*
 * Copyright (c) 2020-2021 The reone project contributors
 *
 * 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 <https://www.gnu.org/licenses/>.
 */

#include "program.h"

#include <iostream>

#include <boost/filesystem.hpp>
#include <boost/program_options.hpp>

#include "../src/common/pathutil.h"

#include "moduleprobe.h"

using namespace std;

using namespace reone::resource;

namespace fs = boost::filesystem;
namespace po = boost::program_options;

namespace reone {

namespace tools {

Program::Program(int argc, char **argv) : _argc(argc), _argv(argv) {
}

int Program::run() {
    initOptions();
    loadOptions();
    initGameVersion();

    switch (_command) {
        case Command::List:
        case Command::Extract:
        case Command::Convert:
            initFileTool();
            switch (_command) {
                case Command::List:
                    _tool->list(_target, _keyPath);
                    break;
                case Command::Extract:
                    _tool->extract(_target, _keyPath, _destPath);
                    break;
                case Command::Convert:
                    _tool->convert(_target, _destPath);
                    break;
                default:
                    break;
            }
            break;
        case Command::ModuleProbe:
            ModuleProbe().probe(_target, _gamePath, _destPath);
            break;
        default:
            cout << _cmdLineOpts << endl;
            break;
    }

    return 0;
}

void Program::initOptions() {
    _cmdLineOpts.add_options()
        ("help", "print this message")
        ("list", "list file contents")
        ("extract", "extract file contents")
        ("convert", "convert 2DA or GFF file to JSON")
        ("modprobe", "probe module and produce a JSON file describing it")
        ("game", po::value<string>(), "path to game directory")
        ("dest", po::value<string>(), "path to destination directory")
        ("target", po::value<string>(), "target name or path to input file");
}

static fs::path getDestination(const po::variables_map &vars) {
    fs::path result;
    if (vars.count("dest") > 0) {
        result = vars["dest"].as<string>();
    } else if (vars.count("target") > 0) {
        result = fs::path(vars["target"].as<string>()).parent_path();
    } else {
        result = fs::current_path();
    }
    return move(result);
}

void Program::loadOptions() {
    po::positional_options_description positional;
    positional.add("target", 1);

    po::parsed_options parsedCmdLineOpts = po::command_line_parser(_argc, _argv)
        .options(_cmdLineOpts)
        .positional(positional)
        .run();

    po::variables_map vars;
    po::store(parsedCmdLineOpts, vars);
    po::notify(vars);

    _gamePath = vars.count("game") > 0 ? vars["game"].as<string>() : fs::current_path();
    _destPath = getDestination(vars);
    _target = vars.count("target") > 0 ? vars["target"].as<string>() : "";
    _keyPath = getPathIgnoreCase(_gamePath, "chitin.key");

    if (vars.count("help")) {
        _command = Command::Help;
    } else if (vars.count("list")) {
        _command = Command::List;
    } else if (vars.count("extract")) {
        _command = Command::Extract;
    } else if (vars.count("convert")) {
        _command = Command::Convert;
    } else if (vars.count("modprobe")) {
        _command = Command::ModuleProbe;
    }
}

void Program::initGameVersion() {
    fs::path exePath = getPathIgnoreCase(_gamePath, "swkotor2.exe");
    _version = exePath.empty() ? GameVersion::KotOR : GameVersion::TheSithLords;
}

void Program::initFileTool() {
    switch (_command) {
        case Command::List:
        case Command::Extract:
        case Command::Convert:
            if (!fs::exists(_target)) {
                throw runtime_error("Input file does not exist: " + _target);
            }
            _tool = getFileToolByPath(_version, _target);
            break;
        default:
            throw logic_error("Unsupported file tool command: " + to_string(static_cast<int>(_command)));
    }
}

} // namespace tools

} // namespace reone