metaforce/hecl/driver/ToolExtract.hpp

#pragma once

#include "ToolBase.hpp"
#include <cstdio>

#if _WIN32
#include <conio.h>
#else
#include <termios.h>
#endif

#include "hecl/MultiProgressPrinter.hpp"

class ToolExtract final : public ToolBase {
  hecl::Database::IDataSpec::ExtractPassInfo m_einfo;
  struct SpecExtractPass {
    const hecl::Database::DataSpecEntry* m_entry;
    std::unique_ptr<hecl::Database::IDataSpec> m_instance;
    SpecExtractPass(const hecl::Database::DataSpecEntry* entry, std::unique_ptr<hecl::Database::IDataSpec>&& instance)
    : m_entry(entry), m_instance(std::move(instance)) {}
    SpecExtractPass(const SpecExtractPass& other) = delete;
    SpecExtractPass(SpecExtractPass&& other) = default;
  };
  std::vector<SpecExtractPass> m_specPasses;
  std::vector<hecl::Database::IDataSpec::ExtractReport> m_reps;
  std::unique_ptr<hecl::Database::Project> m_fallbackProj;
  hecl::Database::Project* m_useProj = nullptr;

public:
  ToolExtract(const ToolPassInfo& info) : ToolBase(info) {
    if (!m_info.args.size())
      LogModule.report(logvisor::Fatal, fmt("hecl extract needs a source path as its first argument"));

    if (!info.project) {
      hecl::SystemString rootDir;

      if (info.output.empty()) {
        /* Get name from input file and init project there */
        hecl::SystemString baseFile = info.args.front();
        size_t slashPos = baseFile.rfind(_SYS_STR('/'));
        if (slashPos == hecl::SystemString::npos)
          slashPos = baseFile.rfind(_SYS_STR('\\'));
        if (slashPos != hecl::SystemString::npos)
          baseFile.assign(baseFile.begin() + slashPos + 1, baseFile.end());
        size_t dotPos = baseFile.rfind(_SYS_STR('.'));
        if (dotPos != hecl::SystemString::npos)
          baseFile.assign(baseFile.begin(), baseFile.begin() + dotPos);

        if (baseFile.empty())
          LogModule.report(logvisor::Fatal, fmt("hecl extract must be ran within a project directory"));

        rootDir = info.cwd + baseFile;
      } else {
        if (hecl::PathRelative(info.output.c_str()))
          rootDir = info.cwd + info.output;
        else
          rootDir = info.output;
      }

      size_t ErrorRef = logvisor::ErrorCount;
      hecl::ProjectRootPath newProjRoot(rootDir);
      newProjRoot.makeDir();
      m_fallbackProj.reset(new hecl::Database::Project(newProjRoot));
      if (logvisor::ErrorCount > ErrorRef)
        LogModule.report(logvisor::Fatal, fmt(_SYS_STR("unable to init project at '{}'")), rootDir);
      LogModule.report(logvisor::Info, fmt(_SYS_STR("initialized project at '{}/.hecl'")), rootDir);
      m_useProj = m_fallbackProj.get();
    } else
      m_useProj = info.project;

    m_einfo.srcpath = m_info.args.front();
    m_einfo.force = info.force;
    m_einfo.extractArgs.reserve(info.args.size());
    auto it = info.args.cbegin();
    ++it;
    for (; it != info.args.cend(); ++it)
      m_einfo.extractArgs.push_back(*it);

    m_specPasses.reserve(hecl::Database::DATA_SPEC_REGISTRY.size());
    for (const hecl::Database::DataSpecEntry* entry : hecl::Database::DATA_SPEC_REGISTRY) {
      if (entry->m_factory) {
        auto ds = entry->m_factory(*m_useProj, hecl::Database::DataSpecTool::Extract);
        if (ds && ds->canExtract(m_einfo, m_reps))
          m_specPasses.emplace_back(entry, std::move(ds));
      }
    }
  }

  static void Help(HelpOutput& help) {
    help.secHead(_SYS_STR("NAME"));
    help.beginWrap();
    help.wrap(_SYS_STR("hecl-extract - Extract objects from supported package/image formats\n"));
    help.endWrap();

    help.secHead(_SYS_STR("SYNOPSIS"));
    help.beginWrap();
    help.wrap(_SYS_STR("hecl extract <packagefile> [<subnode>...]\n"));
    help.endWrap();

    help.secHead(_SYS_STR("DESCRIPTION"));
    help.beginWrap();
    help.wrap(_SYS_STR("This command recursively extracts all or part of a dataspec-supported ")
                  _SYS_STR("package format. Each object is decoded to a working format and added to the project.\n\n"));
    help.endWrap();

    help.secHead(_SYS_STR("OPTIONS"));
    help.optionHead(_SYS_STR("<packagefile>[/<subnode>...]"), _SYS_STR("input file"));
    help.beginWrap();
    help.wrap(_SYS_STR("Specifies the package file or disc image to source data from. ")
                  _SYS_STR("An optional subnode specifies a named hierarchical-node specific ")
                      _SYS_STR("to the game architecture (levels/areas)."));
    help.endWrap();
  }

  hecl::SystemString toolName() const override { return _SYS_STR("extract"); }

  static void _recursivePrint(int level, hecl::Database::IDataSpec::ExtractReport& rep) {
    for (int l = 0; l < level; ++l)
      fmt::print(fmt(_SYS_STR("  ")));
    if (XTERM_COLOR)
      fmt::print(fmt(_SYS_STR("" BOLD "{}" NORMAL "")), rep.name);
    else
      fmt::print(fmt(_SYS_STR("{}")), rep.name);

    if (rep.desc.size())
      fmt::print(fmt(_SYS_STR(" [{}]")), rep.desc);
    fmt::print(fmt(_SYS_STR("\n")));
    for (hecl::Database::IDataSpec::ExtractReport& child : rep.childOpts)
      _recursivePrint(level + 1, child);
  }

  int run() override {
    if (m_specPasses.empty()) {
      if (XTERM_COLOR)
        fmt::print(fmt(_SYS_STR("" RED BOLD "NOTHING TO EXTRACT" NORMAL "\n")));
      else
        fmt::print(fmt(_SYS_STR("NOTHING TO EXTRACT\n")));
      return 1;
    }

    if (XTERM_COLOR)
      fmt::print(fmt(_SYS_STR("" GREEN BOLD "ABOUT TO EXTRACT:" NORMAL "\n")));
    else
      fmt::print(fmt(_SYS_STR("ABOUT TO EXTRACT:\n")));

    for (hecl::Database::IDataSpec::ExtractReport& rep : m_reps) {
      _recursivePrint(0, rep);
      fmt::print(fmt(_SYS_STR("\n")));
    }
    fflush(stdout);

    if (continuePrompt()) {
      for (SpecExtractPass& ds : m_specPasses) {
        if (XTERM_COLOR)
          fmt::print(fmt(_SYS_STR("" MAGENTA BOLD "Using DataSpec {}:" NORMAL "\n")), ds.m_entry->m_name);
        else
          fmt::print(fmt(_SYS_STR("Using DataSpec {}:\n")), ds.m_entry->m_name);

        ds.m_instance->doExtract(m_einfo, {true});
        fmt::print(fmt(_SYS_STR("\n\n")));
      }
    }

    return 0;
  }
};