#pragma once

#include "common.hpp"

POLYBAR_NS

namespace command_line {
  DEFINE_ERROR(argument_error);
  DEFINE_ERROR(value_error);

  class option;
  using choices = vector<string>;
  using options = vector<option>;
  using values = map<string, string>;

  // class definition : option {{{

  class option {
   public:
    string flag;
    string flag_long;
    string desc;
    string token;
    const choices values;

    explicit option(
        string flag, string flag_long, string desc, string token = "", const choices c = {})
        : flag(flag), flag_long(flag_long), desc(desc), token(token), values(c) {}
  };

  // }}}
  // class definition : parser {{{

  class parser {
   public:
    explicit parser(const string& synopsis, const options& opts)
        : m_synopsis(synopsis), m_opts(opts) {}

    void usage() const;

    void process_input(const vector<string>& values);

    bool has(const string& option) const;
    string get(string opt) const;
    bool compare(string opt, string val) const;

   protected:
    auto is_short(string option, string opt_short) const;
    auto is_long(string option, string opt_long) const;
    auto is(string option, string opt_short, string opt_long) const;

    auto parse_value(string input, string input_next, choices values) const;
    void parse(string input, string input_next = "");

   private:
    string m_synopsis;
    options m_opts;
    values m_optvalues;
    bool m_skipnext = false;
  };

  // }}}
}

using cliparser = command_line::parser;
using clioption = command_line::option;
using clioptions = command_line::options;

namespace {
  /**
   * Configure injection module
   */
  template <class T = cliparser>
  di::injector<T> configure_cliparser(string scriptname, const clioptions& opts) {
    // clang-format off
    return di::make_injector(
        di::bind<>().to("Usage: " + scriptname + " bar_name [OPTION...]"),
        di::bind<>().to(opts));
    // clang-format on
  }
}

POLYBAR_NS_END