#include <eosio/trace_api/cmd_registration.hpp>

#include <iostream>

#include <boost/program_options.hpp>

using namespace eosio::trace_api;
namespace bpo = boost::program_options;

command_registration* command_registration::_list = nullptr;

namespace {
   auto create_command_map() {
      auto result = std::map<std::string, command_registration*>();
      command_registration* cur = command_registration::_list;
      while (cur != nullptr) {
         if (result.count(cur->name) > 0) {
            std::cerr << "Illformed Program, duplicate subcommand: " << cur->name << "\n";
            exit(1);
         }
         result[cur->name] = cur;
         cur = cur->_next;
      }

      return result;
   }
}

int main(int argc, char** argv) {
   auto command_map = create_command_map();

   bpo::options_description vis_desc("Options");
   auto vis_opts = vis_desc.add_options();
   vis_opts("help,h", "show usage help message");

   bpo::options_description hidden_desc;
   auto hidden_opts = hidden_desc.add_options();
   hidden_opts("subargs", bpo::value<std::vector<std::string>>(), "args");

   bpo::positional_options_description pos_desc;
   pos_desc.add("subargs", -1);

   bpo::options_description cmdline_options;
   cmdline_options.add(vis_desc).add(hidden_desc);

   bpo::variables_map vm;
   auto parsed_args = bpo::command_line_parser(argc, argv).options(cmdline_options).positional(pos_desc).allow_unregistered().run();
   bpo::store(parsed_args, vm);

   std::vector<std::string> args = bpo::collect_unrecognized(parsed_args.options, bpo::include_positional);

   auto show_help = [&](std::ostream& os) {
      os <<
         "Usage: trace_api_util <options> command ...\n"
         "\n"
         "Commands:\n";

      for (const auto& e: command_map) {
         os << "  " << e.second->name << "              " << e.second->slug << "\n";
      }

      os << "\n" << vis_desc << "\n";
   };

   if (args.size() < 1) {
      if (vm.count("help") > 0) {
         show_help(std::cout);
         return 0;
      }

      std::cerr << "Error: No command provided\n\n";
      show_help(std::cerr);
      return 1;
   } else if (command_map.count(args.at(0)) == 0) {
      std::cerr << "Error: unknown command \"" << args.at(0) << "\"\n\n";
      show_help(std::cerr);
      return 1;
   }

   // trim the command name and pass the rest of the args to the defined command
   return command_map.at(args.at(0))->func(vm, std::vector<std::string>(args.begin() + 1, args.end()));
}