#pragma once

#include <map>
#include <string>
#include <vector>

namespace program {

template <typename Impl> class Arguments {
private:
  struct Argument;
  struct Option;
  template <typename, typename> struct WithDefault;

public:
  Arguments() = default;
  Arguments(int argc, char const * const * const argv);

protected:
  auto option(std::string const & name, std::string const & description = "");
  auto option(std::string const & name, char abbrev,
              std::string const & description = "");
  auto argument(size_t index, std::string const & name,
                std::string const & description = "");

private:
  void usage() const;

private:
  // Metadata variables
  bool primed_{false};
  std::map<std::string, std::string> argument_descriptions;
  std::map<size_t, std::string> argument_names;
  std::map<std::string, std::string> option_descriptions;
  std::map<std::string, std::string> option_name_mapping;

  // Data/Output variables
  std::string program;
  size_t optional_from{std::numeric_limits<size_t>::max()};
  std::vector<std::string> arguments;
  std::map<std::string, std::vector<std::string>> options;
};

template <typename Impl> struct Arguments<Impl>::Argument {
  Arguments<Impl> * self;
  size_t index;
  bool is_optional;
  std::string name;
  std::string description;

  template <typename T> auto operator=(T && value);
  template <typename T> operator T() const;
  bool primed() const;
  explicit operator bool() const;
};

template <typename Impl> struct Arguments<Impl>::Option {
  Arguments<Impl> * self;
  std::string name;
  char abbrev;
  std::string description;

  template <typename T> auto operator=(T && value);
  template <typename T> operator T() const;
  bool primed() const;
  explicit operator bool() const;
};

template <typename Impl>
template <typename B, typename V>
struct Arguments<Impl>::WithDefault {
  B impl;
  V default_value;
  template <typename T> operator T() const { return impl ?: T(default_value); }
};

}

#include "arguments_impl.hpp"

#define TYPED_MAIN(tname)                                                      \
  int typed_main(tname const & args);                                          \
  int main(int argc, char const * const * const argv) try {                    \
    return typed_main(tname(argc, argv));                                      \
  } catch (program::ProgramArgumentsError const & pae) {                       \
    std::cerr << "Error in program argument handling: " << pae.what() << "\n"; \
  }                                                                            \
  int typed_main(tname const & args)