#pragma once

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

class ToolSpec final : public ToolBase {
  enum Mode { MLIST = 0, MENABLE, MDISABLE } mode = MLIST;

public:
  ToolSpec(const ToolPassInfo& info) : ToolBase(info) {
    if (info.args.empty())
      return;

    if (!info.project)
      LogModule.report(logvisor::Fatal, fmt("hecl spec must be ran within a project directory"));

    const auto& specs = info.project->getDataSpecs();
    hecl::SystemString firstArg = info.args.front();
    hecl::ToLower(firstArg);

    if (!firstArg.compare(_SYS_STR("enable")))
      mode = MENABLE;
    else if (!firstArg.compare(_SYS_STR("disable")))
      mode = MDISABLE;
    else
      return;

    if (info.args.size() < 2)
      LogModule.report(logvisor::Fatal, fmt("Speclist argument required"));

    auto it = info.args.begin();
    ++it;
    for (; it != info.args.end(); ++it) {

      bool found = false;
      for (auto& spec : specs) {
        if (!it->compare(spec.spec.m_name)) {
          found = true;
          break;
        }
      }
      if (!found)
        LogModule.report(logvisor::Fatal, fmt(_SYS_STR("'{}' is not found in the dataspec registry")), *it);
    }
  }

  static void Help(HelpOutput& help) {
    help.secHead(_SYS_STR("NAME"));
    help.beginWrap();
    help.wrap(_SYS_STR("hecl-spec - Configure target data options\n"));
    help.endWrap();

    help.secHead(_SYS_STR("SYNOPSIS"));
    help.beginWrap();
    help.wrap(_SYS_STR("hecl spec [enable|disable] [<specname>...]\n"));
    help.endWrap();

    help.secHead(_SYS_STR("DESCRIPTION"));
    help.beginWrap();
    help.wrap(
        _SYS_STR("This command configures the HECL project with the user's preferred target DataSpecs.\n\n")
            _SYS_STR("Providing enable/disable argument will bulk-set the enable status of the provided spec(s)")
                _SYS_STR("list. If enable/disable is not provided, a list of supported DataSpecs is printed.\n\n"));
    help.endWrap();

    help.secHead(_SYS_STR("OPTIONS"));
    help.optionHead(_SYS_STR("<specname>..."), _SYS_STR("DataSpec name(s)"));
    help.beginWrap();
    help.wrap(_SYS_STR("Specifies platform-names to enable/disable"));
    help.endWrap();
  }

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

  int run() {
    if (!m_info.project) {
      for (const hecl::Database::DataSpecEntry* spec : hecl::Database::DATA_SPEC_REGISTRY) {
        if (XTERM_COLOR)
          fmt::print(fmt(_SYS_STR("" BOLD CYAN "{}" NORMAL "\n")), spec->m_name);
        else
          fmt::print(fmt(_SYS_STR("{}\n")), spec->m_name);
        fmt::print(fmt(_SYS_STR("  {}\n")), spec->m_desc);
      }
      return 0;
    }

    const auto& specs = m_info.project->getDataSpecs();
    if (mode == MLIST) {
      for (auto& spec : specs) {
        if (XTERM_COLOR)
          fmt::print(fmt(_SYS_STR("" BOLD CYAN "{}" NORMAL "")), spec.spec.m_name);
        else
          fmt::print(fmt(_SYS_STR("{}")), spec.spec.m_name);
        if (spec.active) {
          if (XTERM_COLOR)
            fmt::print(fmt(_SYS_STR(" " BOLD GREEN "[ENABLED]" NORMAL "")));
          else
            fmt::print(fmt(_SYS_STR(" [ENABLED]")));
        }
        fmt::print(fmt(_SYS_STR("\n  {}\n")), spec.spec.m_desc);
      }
      return 0;
    }

    std::vector<hecl::SystemString> opSpecs;
    auto it = m_info.args.begin();
    ++it;
    for (; it != m_info.args.end(); ++it) {
      hecl::SystemString itName = *it;
      hecl::ToLower(itName);
      for (auto& spec : specs) {
        hecl::SystemString compName(spec.spec.m_name);
        hecl::ToLower(compName);
        if (!itName.compare(compName)) {
          opSpecs.emplace_back(spec.spec.m_name);
          break;
        }
      }
    }

    if (opSpecs.size()) {
      if (mode == MENABLE)
        m_info.project->enableDataSpecs(opSpecs);
      else if (mode == MDISABLE)
        m_info.project->disableDataSpecs(opSpecs);
    }

    return 0;
  }
};