#pragma once

/*
 *   __ _ _ __ __ _ _ __   __ _ _ __ ___  ___
 *  / _` | '__/ _` | '_ \ / _` | '__/ __|/ _ \ Argument Parser for Modern C++
 * | (_| | | | (_| | |_) | (_| | |  \__ \  __/ http://github.com/p-ranav/argparse
 *  \__,_|_|  \__, | .__/ \__,_|_|  |___/\___|
 *            |___/|_|
 * * Licensed under the MIT License <http://opensource.org/licenses/MIT>.
 * SPDX-License-Identifier: MIT
 * Copyright (c) 2019-2022 Pranav Srinivas Kumar <pranav.srinivas.kumar@gmail.com>
 * and other contributors.
 *
 * Permission is hereby  granted, free of charge, to any  person obtaining a copy
 * of this software and associated  documentation files (the "Software"), to deal
 * in the Software  without restriction, including without  limitation the rights
 * to  use, copy,  modify, merge,  publish, distribute,  sublicense, and/or  sell
 * copies  of  the Software,  and  to  permit persons  to  whom  the Software  is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in all
 * copies or substantial portions of the Software.
 *
 * THE SOFTWARE  IS PROVIDED "AS  IS", WITHOUT WARRANTY  OF ANY KIND,  EXPRESS OR
 * IMPLIED,  INCLUDING BUT  NOT  LIMITED TO  THE  WARRANTIES OF  MERCHANTABILITY,
 * FITNESS FOR  A PARTICULAR PURPOSE AND  NONINFRINGEMENT. IN NO EVENT  SHALL THE
 * AUTHORS  OR COPYRIGHT  HOLDERS  BE  LIABLE FOR  ANY  CLAIM,  DAMAGES OR  OTHER
 * LIABILITY, WHETHER IN AN ACTION OF  CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE  OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#include <cassert>
#include <cerrno>

#ifndef ARGPARSE_MODULE_USE_STD_MODULE
  #include <algorithm>
  #include <any>
  #include <array>
  #include <charconv>
  #include <cstdlib>
  #include <filesystem>
  #include <functional>
  #include <iostream>
  #include <iterator>
  #include <limits>
  #include <list>
  #include <map>
  #include <numeric>
  #include <optional>
  #include <ranges>
  #include <set>
  #include <sstream>
  #include <stdexcept>
  #include <string>
  #include <string_view>
  #include <tuple>
  #include <type_traits>
  #include <utility>
  #include <variant>
  #include <vector>
#endif

#include "src/util/defs.hpp"
#include "src/util/error.hpp"
#include "src/util/types.hpp"

#ifndef ARGPARSE_CUSTOM_STRTOF
  #define ARGPARSE_CUSTOM_STRTOF strtof
#endif

#ifndef ARGPARSE_CUSTOM_STRTOD
  #define ARGPARSE_CUSTOM_STRTOD strtod
#endif

#ifndef ARGPARSE_CUSTOM_STRTOLD
  #define ARGPARSE_CUSTOM_STRTOLD strtold
#endif

// ReSharper disable CppTemplateParameterNeverUsed, CppDFATimeOver
// NOLINTBEGIN(readability-identifier-naming, readability-identifier-length, modernize-use-nullptr)
namespace argparse {
  using namespace util::types;
  using util::error::DracError, util::error::DracErrorCode;

  namespace details {
    template <typename T, typename = void>
    struct HasContainerTraits : std::false_type {};

    template <>
    struct HasContainerTraits<String> : std::false_type {};

    template <>
    struct HasContainerTraits<StringView> : std::false_type {};

    template <typename T>
    struct HasContainerTraits<T, std::void_t<typename T::value_type, decltype(std::declval<T>().begin()), decltype(std::declval<T>().end()), decltype(std::declval<T>().size())>> : std::true_type {};

    template <typename T>
    inline constexpr bool IsContainer = HasContainerTraits<T>::value;

    template <typename T, typename = void>
    struct HasStreamableTraits : std::false_type {};

    template <typename T>
    struct HasStreamableTraits<T, std::void_t<decltype(std::declval<std::ostream&>() << std::declval<T>())>> : std::true_type {};

    template <typename T>
    inline constexpr bool IsStreamable = HasStreamableTraits<T>::value;

    constexpr usize repr_max_container_size = 5;

    template <typename T, typename CharT = char>
    concept Formattable = requires(const T& t, std::basic_format_context<CharT*, CharT> ctx) {
      std::formatter<std::remove_cvref_t<T>, CharT>().format(t, ctx);
    };

    template <typename T>
    static auto repr(const T& val) -> String {
      if constexpr (std::is_same_v<T, bool>)
        return val ? "true" : "false";
      else if constexpr (std::is_convertible_v<T, StringView>)
        return std::format("\"{}\"", String { StringView { val } });
      else if constexpr (IsContainer<T>) {
        String result = "{";

        const auto size = val.size();

        if (size > 0) {
          bool first = true;

          auto transformed_view = val | std::views::transform([](const auto& elem) { return details::repr(elem); });

          if (size <= repr_max_container_size) {
            for (const String& elem_repr : transformed_view) {
              if (!first)
                result += " ";

              result += elem_repr;
              first = false;
            }
          } else {
            for (const String& elem_repr : transformed_view | std::views::take(repr_max_container_size - 1)) {
              if (!first)
                result += " ";

              result += elem_repr;
              first = false;
            }

            result += "... ";

            result += details::repr(*std::prev(val.end()));
          }
        }

        result += "}";
        return result;
      } else if constexpr (Formattable<T>)
        return std::format("{}", val);
      else if constexpr (IsStreamable<T>) {
        std::stringstream out;
        out << val;
        return out.str();
      } else
        return "<not representable>";
    }

    constexpr i32 radix_2  = 2;
    constexpr i32 radix_8  = 8;
    constexpr i32 radix_10 = 10;
    constexpr i32 radix_16 = 16;

    template <class F, class Tuple, class Extra, usize... I>
    constexpr fn apply_plus_one_impl(F&& f, Tuple&& t, Extra&& x, std::index_sequence<I...> /*unused*/) -> decltype(auto) {
      return std::invoke(std::forward<F>(f), std::get<I>(std::forward<Tuple>(t))..., std::forward<Extra>(x));
    }

    template <class F, class Tuple, class Extra>
    constexpr fn apply_plus_one(F&& f, Tuple&& t, Extra&& x) -> decltype(auto) {
      return details::apply_plus_one_impl(
        std::forward<F>(f), std::forward<Tuple>(t), std::forward<Extra>(x), std::make_index_sequence<std::tuple_size_v<std::remove_reference_t<Tuple>>> {}
      );
    }

    constexpr fn pointer_range(const StringView s) noexcept -> std::tuple<const char*, const char*> {
      return { s.data(), s.data() + s.size() };
    }

    template <class CharT, class Traits>
    constexpr fn starts_with(std::basic_string_view<CharT, Traits> prefix, std::basic_string_view<CharT, Traits> s) noexcept -> bool {
      return s.substr(0, prefix.size()) == prefix;
    }

    enum class chars_format : u8 {
      scientific = 0xf1,
      fixed      = 0xf2,
      hex        = 0xf4,
      binary     = 0xf8,
      general    = fixed | scientific
    };

    struct ConsumeBinaryPrefixResult {
      bool       is_binary;
      StringView rest;
    };

    constexpr fn consume_binary_prefix(StringView s) -> ConsumeBinaryPrefixResult {
      if (starts_with(StringView { "0b" }, s) ||
          starts_with(StringView { "0B" }, s)) {
        s.remove_prefix(2);
        return { .is_binary = true, .rest = s };
      }

      return { .is_binary = false, .rest = s };
    }

    struct ConsumeHexPrefixResult {
      bool       is_hexadecimal;
      StringView rest;
    };

    using namespace std::literals;

    constexpr fn consume_hex_prefix(StringView s) -> ConsumeHexPrefixResult {
      if (starts_with("0x"sv, s) || starts_with("0X"sv, s)) {
        s.remove_prefix(2);
        return { .is_hexadecimal = true, .rest = s };
      }

      return { .is_hexadecimal = false, .rest = s };
    }

    template <class T, auto Param>
    fn do_from_chars(const StringView s) -> Result<T> {
      T x { 0 };
      auto [first, last] = pointer_range(s);
      auto [ptr, ec]     = std::from_chars(first, last, x, Param);

      if (ec == std::errc()) {
        if (ptr == last)
          return x;

        return Err(DracError(DracErrorCode::ParseError, std::format("pattern '{}' does not match to the end", String(s))));
      }

      if (ec == std::errc::invalid_argument)
        return Err(DracError(DracErrorCode::InvalidArgument, std::format("pattern '{}' not found", String(s))));

      if (ec == std::errc::result_out_of_range)
        return Err(DracError(DracErrorCode::ParseError, std::format("'{}' not representable", String(s))));

      return Err(DracError(DracErrorCode::InternalError, std::format("Unknown parsing error for '{}'", String(s))));
    }

    template <class T, auto Param = 0>
    struct parse_number {
      static fn operator()(const StringView s)->Result<T> {
        return do_from_chars<T, Param>(s);
      }
    };

    template <class T>
    struct parse_number<T, radix_2> {
      static fn operator()(const StringView s)->Result<T> {
        if (auto [ok, rest] = consume_binary_prefix(s); ok)
          return do_from_chars<T, radix_2>(rest);

        return Err(DracError(DracErrorCode::InvalidArgument, "pattern not found"));
      }
    };

    template <class T>
    struct parse_number<T, radix_16> {
      static fn operator()(const StringView s)->Result<T> {
        Result<T> result;

        if (starts_with("0x"sv, s) || starts_with("0X"sv, s)) {
          if (auto [ok, rest] = consume_hex_prefix(s); ok)
            result = do_from_chars<T, radix_16>(rest);
          else
            return Err(DracError(DracErrorCode::InternalError, std::format("Inconsistent hex prefix detection for '{}'", String(s))));
        } else
          result = do_from_chars<T, radix_16>(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as hexadecimal: {}", String(s), result.error().message)));

        return result;
      }
    };

    template <class T>
    struct parse_number<T> {
      static fn operator()(const StringView s)->Result<T> {
        if (auto [ok, rest] = consume_hex_prefix(s); ok) {
          Result<T> result = do_from_chars<T, radix_16>(rest);

          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as hexadecimal: {}", String(s), result.error().message)));

          return result;
        }

        if (auto [ok_binary, rest_binary] = consume_binary_prefix(s); ok_binary) {
          Result<T> result = do_from_chars<T, radix_2>(rest_binary);

          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as binary: {}", String(s), result.error().message)));

          return result;
        }

        if (starts_with("0"sv, s)) {
          Result<T> result = do_from_chars<T, radix_8>(s);

          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as octal: {}", String(s), result.error().message)));

          return result;
        }

        Result<T> result = do_from_chars<T, radix_10>(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as decimal integer: {}", String(s), result.error().message)));

        return result;
      }
    };

    template <class T>
    inline constexpr std::nullptr_t generic_strtod = nullptr;
    template <>
    inline const auto generic_strtod<float> = ARGPARSE_CUSTOM_STRTOF;
    template <>
    inline const auto generic_strtod<double> = ARGPARSE_CUSTOM_STRTOD;
    template <>
    inline const auto generic_strtod<long double> = ARGPARSE_CUSTOM_STRTOLD;

    template <class T>
    fn do_strtod(const String& s) -> Result<T> {
      if (isspace(static_cast<unsigned char>(s[0])) || s[0] == '+')
        return Err(DracError(DracErrorCode::InvalidArgument, std::format("pattern '{}' not found", s)));

      auto [first, last] = pointer_range(s);

      char* ptr = nullptr;

      errno = 0;

      auto x = generic_strtod<T>(first, &ptr);

      if (errno == 0) {
        if (ptr == last)
          return x;

        return Err(DracError(DracErrorCode::ParseError, std::format("pattern '{}' does not match to the end", s)));
      }

      if (errno == ERANGE)
        return Err(DracError(DracErrorCode::ParseError, std::format("'{}' not representable", s)));

      return Err(DracError(std::error_code(errno, std::system_category())));
    }

    template <class T>
    struct parse_number<T, chars_format::general> {
      fn operator()(const String& s)->Result<T> {
        if (auto [is_hex, rest] = consume_hex_prefix(s); is_hex)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::general does not parse hexfloat"));

        if (auto [is_bin, rest] = consume_binary_prefix(s); is_bin)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::general does not parse binfloat"));

        Result<T> result = do_strtod<T>(s);
        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as number: {}", s, result.error().message)));
        return result;
      }
    };

    template <class T>
    struct parse_number<T, chars_format::hex> {
      fn operator()(const String& s)->Result<T> {
        if (auto [is_hex, rest] = consume_hex_prefix(s); !is_hex)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::hex requires hexfloat format (e.g., 0x1.2p3)"));

        if (auto [is_bin, rest] = consume_binary_prefix(s); is_bin)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::hex does not parse binfloat"));

        Result<T> result = do_strtod<T>(s);
        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as hexadecimal float: {}", s, result.error().message)));
        return result;
      }
    };

    template <class T>
    struct parse_number<T, chars_format::binary> {
      fn operator()(const String& s)->Result<T> {
        if (auto [is_hex, rest] = consume_hex_prefix(s); is_hex)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::binary does not parse hexfloat"));

        if (auto [is_bin, rest] = consume_binary_prefix(s); !is_bin)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::binary requires binfloat format (e.g., 0b1.01p2)"));

        Result<T> result = do_strtod<T>(s);
        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as binary float: {}", s, result.error().message)));
        return result;
      }
    };

    template <class T>
    struct parse_number<T, chars_format::scientific> {
      fn operator()(const String& s)->Result<T> {
        if (const auto [is_hex, rest] = consume_hex_prefix(s); is_hex)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::scientific does not parse hexfloat"));

        if (const auto [is_bin, rest] = consume_binary_prefix(s); is_bin)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::scientific does not parse binfloat"));

        if (s.find_first_of("eE") == String::npos)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::scientific requires exponent part"));

        Result<T> result = do_strtod<T>(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as scientific notation: {}", s, result.error().message)));

        return result;
      }
    };

    template <class T>
    struct parse_number<T, chars_format::fixed> {
      fn operator()(const String& s)->Result<T> {
        if (const auto [is_hex, rest] = consume_hex_prefix(s); is_hex)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::fixed does not parse hexfloat"));

        if (const auto [is_bin, rest] = consume_binary_prefix(s); is_bin)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::fixed does not parse binfloat"));

        if (s.find_first_of("eE") != String::npos)
          return Err(DracError(DracErrorCode::InvalidArgument, "chars_format::fixed does not parse exponent part"));

        Result<T> result = do_strtod<T>(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as fixed notation: {}", s, result.error().message)));

        return result;
      }
    };

    template <typename T>
    concept ToStringConvertible = std::convertible_to<T, std::string> ||
      std::convertible_to<T, std::string_view> ||
      requires(const T& t) { std::format("{}", t); };

    template <typename StrIt>
    fn join(StrIt first, StrIt last, const String& separator) -> String {
      if (first == last)
        return "";

      std::stringstream value;
      value << *first;
      ++first;

      while (first != last) {
        value << separator << *first;
        ++first;
      }

      return value.str();
    }

    template <typename T>
    struct can_invoke_to_string {
      template <typename U>
      // ReSharper disable CppFunctionIsNotImplemented
      static fn test(int) -> decltype(std::to_string(std::declval<U>()), std::true_type {});

      template <typename U>
      static fn test(...) -> std::false_type;
      // ReSharper restore CppFunctionIsNotImplemented

      static constexpr bool value = decltype(test<T>(0))::value;
    };

    template <typename T>
    struct IsChoiceTypeSupported {
      using CleanType         = std::decay_t<T>;
      static const bool value = std::is_integral_v<CleanType> ||
        std::is_same_v<CleanType, String> ||
        std::is_same_v<CleanType, StringView> ||
        std::is_same_v<CleanType, const char*>;
    };

    template <typename StringType>
    fn get_levenshtein_distance(const StringType& s1, const StringType& s2) -> usize {
      Vec<Vec<usize>> dp(
        s1.size() + 1, Vec<usize>(s2.size() + 1, 0)
      );

      for (usize i = 0; i <= s1.size(); ++i) {
        for (usize j = 0; j <= s2.size(); ++j) {
          if (i == 0) {
            dp[i][j] = j;
          } else if (j == 0) {
            dp[i][j] = i;
          } else if (s1[i - 1] == s2[j - 1]) {
            dp[i][j] = dp[i - 1][j - 1];
          } else {
            dp[i][j] = 1 + std::min<usize>({ dp[i - 1][j], dp[i][j - 1], dp[i - 1][j - 1] });
          }
        }
      }

      return dp[s1.size()][s2.size()];
    }

    template <typename ValueType>
    fn get_most_similar_string(const Map<String, ValueType>& map, const String& input) -> String {
      String most_similar {};
      usize  min_distance = (std::numeric_limits<usize>::max)();

      for (const auto& entry : map)
        if (const usize distance = get_levenshtein_distance(entry.first, input); distance < min_distance) {
          min_distance = distance;
          most_similar = entry.first;
        }

      return most_similar;
    }

    template <typename Test, template <typename...> class Ref>
    struct is_specialization : std::false_type {};

    template <template <typename...> class Ref, typename... Args>
    struct is_specialization<Ref<Args...>, Ref> : std::true_type {};

    template <typename Test, template <typename...> class Ref>
    inline constexpr bool is_specialization_v = is_specialization<Test, Ref>::value;
  } // namespace details

  enum class nargs_pattern : u8 {
    optional,
    any,
    at_least_one,
  };

  enum class default_arguments : u8 {
    none    = 0,
    help    = 1,
    version = 2,
    all     = help | version,
  };

  inline fn operator&(const default_arguments& a, const default_arguments& b)->default_arguments {
    return static_cast<default_arguments>(
      std::to_underlying(a) &
      std::to_underlying(b)
    );
  }

  class ArgumentParser;

  class Argument {
    friend class ArgumentParser;
    friend fn operator<<(std::ostream& stream, const ArgumentParser& parser)
      ->std::ostream&;

    template <usize N, usize... I>
    explicit Argument(const StringView prefix_chars, std::array<StringView, N>&& a, std::index_sequence<I...> /*unused*/) // NOLINT(cppcoreguidelines-rvalue-reference-param-not-moved)
      : m_accepts_optional_like_value(false),
        m_is_optional((is_optional(a[I], prefix_chars) || ...)),
        m_is_required(false),
        m_is_repeatable(false),
        m_is_used(false),
        m_is_hidden(false),
        m_prefix_chars(prefix_chars) {
      ((void)m_names.emplace_back(a[I]), ...);
      std::sort(
        m_names.begin(), m_names.end(), [](const auto& lhs, const auto& rhs) {
          return lhs.size() == rhs.size() ? lhs < rhs : lhs.size() < rhs.size();
        }
      );
    }

   public:
    template <usize N>
    explicit Argument(StringView prefix_chars, std::array<StringView, N>&& a)
      : Argument(prefix_chars, std::move(a), std::make_index_sequence<N> {}) {}

    fn help(String help_text) -> Argument& {
      m_help = std::move(help_text);
      return *this;
    }

    fn metavar(String metavar) -> Argument& {
      m_metavar = std::move(metavar);
      return *this;
    }

    template <typename T>
    fn default_value(T&& value) -> Argument& {
      m_num_args_range     = NArgsRange { 0, m_num_args_range.get_max() };
      m_default_value_repr = details::repr(value);

      if constexpr (std::is_convertible_v<T, StringView>)
        m_default_value_str = String { StringView { value } };
      else if constexpr (details::can_invoke_to_string<T>::value)
        m_default_value_str = std::to_string(value);

      m_default_value = std::forward<T>(value);
      return *this;
    }

    fn default_value(const char* value) -> Argument& {
      return default_value(String(value));
    }

    fn required() -> Argument& {
      m_is_required = true;
      return *this;
    }

    fn implicit_value(std::any value) -> Argument& {
      m_implicit_value = std::move(value);
      m_num_args_range = NArgsRange { 0, 0 };
      return *this;
    }

    // This is shorthand for:
    //   program.add_argument("foo")
    //     .default_value(false)
    //     .implicit_value(true)
    fn flag() -> Argument& {
      default_value(false);
      implicit_value(true);
      return *this;
    }

    template <class F, class... Args>
    fn action(F&& callable, Args&&... bound_args)
      -> Argument&
      requires(std::is_invocable_v<F, Args..., const String>)
    {
      using RawReturnType = std::invoke_result_t<F, Args..., const String>;

      if constexpr (std::is_void_v<RawReturnType>) {
        m_actions.emplace_back<void_action>(
          [f = std::forward<F>(callable), tup = std::make_tuple(std::forward<Args>(bound_args)...)](const String& opt) mutable -> Result<void> {
            details::apply_plus_one(f, tup, opt);
            return {};
          }
        );
      } else if constexpr (argparse::details::is_specialization_v<RawReturnType, Result> && std::is_void_v<typename RawReturnType::value_type>) {
        m_actions.emplace_back<void_action>(
          [f = std::forward<F>(callable), tup = std::make_tuple(std::forward<Args>(bound_args)...)](const String& opt) mutable -> Result<void> {
            return details::apply_plus_one(f, tup, opt);
          }
        );
      } else if constexpr (argparse::details::is_specialization_v<RawReturnType, Result>) {
        m_actions.emplace_back<valued_action>(
          [f = std::forward<F>(callable), tup = std::make_tuple(std::forward<Args>(bound_args)...)](const String& opt) mutable -> Result<std::any> {
            RawReturnType result = details::apply_plus_one(f, tup, opt);
            if (result) {
              if constexpr (!std::is_void_v<typename RawReturnType::value_type>) {
                return result.value();
              } else {
                return std::any {};
              }
            } else {
              return Err(result.error());
            }
          }
        );
      } else {
        m_actions.emplace_back<valued_action>(
          [f = std::forward<F>(callable), tup = std::make_tuple(std::forward<Args>(bound_args)...)](const String& opt) mutable -> Result<std::any> {
            return details::apply_plus_one(f, tup, opt);
          }
        );
      }
      return *this;
    }

    fn store_into(bool& var)
      -> Argument& {
      if ((!m_default_value.has_value()) && (!m_implicit_value.has_value()))
        flag();

      if (m_default_value.has_value())
        var = std::any_cast<bool>(m_default_value);

      action([&var](const String& /*unused*/) -> Result<bool> {
        var = true;
        return var;
      });

      return *this;
    }

    template <typename T>
    fn store_into(T& var) -> Argument&
      requires(std::is_integral_v<T>)
    {
      if (m_default_value.has_value())
        var = std::any_cast<T>(m_default_value);

      action([&var](const auto& s) -> Result<T> {
        Result<T> result = details::parse_number<T, details::radix_10>()(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as decimal integer: {}", s, result.error().message)));

        var = *result;
        return result;
      });

      return *this;
    }

    template <typename T>
    fn store_into(T& var) -> Argument&
      requires(std::is_floating_point_v<T>)
    {
      if (m_default_value.has_value())
        var = std::any_cast<T>(m_default_value);

      action([&var](const auto& s) -> Result<T> {
        Result<T> result = details::parse_number<T, details::chars_format::general>()(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as number: {}", s, result.error().message)));

        var = *result;
        return result;
      });

      return *this;
    }

    fn store_into(String& var)
      -> Argument& {
      if (m_default_value.has_value())
        var = std::any_cast<String>(m_default_value);

      action([&var](const String& s) -> Result<String> {
        var = s;
        return s;
      });

      return *this;
    }

    fn store_into(std::filesystem::path& var) -> Argument& {
      if (m_default_value.has_value())
        var = std::any_cast<std::filesystem::path>(m_default_value);

      action([&var](const String& s) -> Result<std::filesystem::path> {
        var = s;
        return var;
      });

      return *this;
    }

    fn store_into(Vec<String>& var) -> Argument& {
      if (m_default_value.has_value())
        var = std::any_cast<Vec<String>>(m_default_value);

      action([this, &var](const String& s) -> Result<Vec<String>> {
        if (!m_is_used)
          var.clear();

        m_is_used = true;
        var.push_back(s);
        return var;
      });

      return *this;
    }

    fn store_into(Vec<int>& var) -> Argument& {
      if (m_default_value.has_value())
        var = std::any_cast<Vec<int>>(m_default_value);

      action([this, &var](const String& s) -> Result<Vec<int>> {
        if (!m_is_used)
          var.clear();

        m_is_used = true;

        Result<int> result = details::parse_number<int, details::radix_10>()(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as decimal integer for vector: {}", s, result.error().message)));

        var.push_back(*result);
        return var;
      });

      return *this;
    }

    fn store_into(std::set<String>& var) -> Argument& {
      if (m_default_value.has_value())
        var = std::any_cast<std::set<String>>(m_default_value);

      action([this, &var](const String& s) -> Result<std::set<String>> {
        if (!m_is_used)
          var.clear();

        m_is_used = true;
        var.insert(s);
        return var;
      });

      return *this;
    }

    fn store_into(std::set<int>& var) -> Argument& {
      if (m_default_value.has_value())
        var = std::any_cast<std::set<int>>(m_default_value);

      action([this, &var](const String& s) -> Result<std::set<int>> {
        if (!m_is_used)
          var.clear();

        m_is_used = true;

        Result<int> result = details::parse_number<int, details::radix_10>()(s);

        if (!result)
          return Err(DracError(result.error().code, std::format("Failed to parse '{}' as decimal integer for set: {}", s, result.error().message)));

        var.insert(*result);
        return var;
      });

      return *this;
    }

    fn append() -> Argument& {
      m_is_repeatable = true;
      return *this;
    }

    fn hidden() -> Argument& {
      m_is_hidden = true;
      return *this;
    }

    template <char Shape, typename T>
    fn scan() -> Argument&
      requires(std::is_arithmetic_v<T>)
    {
      static_assert(!(std::is_const_v<T> || std::is_volatile_v<T>), "T should not be cv-qualified");

      fn is_one_of = [](char c, auto... x) constexpr {
        return ((c == x) || ...);
      };

      if constexpr (Shape == 'd' && std::is_integral_v<T>)
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::radix_10>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as decimal integer (scan 'd'): {}", s, result.error().message)));
          return result;
        });
      else if constexpr (Shape == 'i' && std::is_integral_v<T>)
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as integer (scan 'i'): {}", s, result.error().message)));
          return result;
        });
      else if constexpr (Shape == 'u' && (std::is_integral_v<T> && std::is_unsigned_v<T>))
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::radix_10>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as unsigned decimal integer (scan 'u'): {}", s, result.error().message)));
          return result;
        });
      else if constexpr (Shape == 'b' && (std::is_integral_v<T> && std::is_unsigned_v<T>))
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::radix_2>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as binary integer (scan 'b'): {}", s, result.error().message)));
          return result;
        });
      else if constexpr (Shape == 'o' && (std::is_integral_v<T> && std::is_unsigned_v<T>))
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::radix_8>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as octal integer (scan 'o'): {}", s, result.error().message)));
          return result;
        });
      else if constexpr (is_one_of(Shape, 'x', 'X') && (std::is_integral_v<T> && std::is_unsigned_v<T>))
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::radix_16>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as hexadecimal integer (scan '{}'): {}", s, Shape, result.error().message)));
          return result;
        });
      else if constexpr (is_one_of(Shape, 'a', 'A') && std::is_floating_point_v<T>)
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::chars_format::hex>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as hexadecimal float (scan '{}'): {}", s, Shape, result.error().message)));
          return result;
        });
      else if constexpr (is_one_of(Shape, 'e', 'E') && std::is_floating_point_v<T>)
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::chars_format::scientific>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as scientific float (scan '{}'): {}", s, Shape, result.error().message)));
          return result;
        });
      else if constexpr (is_one_of(Shape, 'f', 'F') && std::is_floating_point_v<T>)
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::chars_format::fixed>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as fixed float (scan '{}'): {}", s, Shape, result.error().message)));
          return result;
        });
      else if constexpr (is_one_of(Shape, 'g', 'G') && std::is_floating_point_v<T>)
        action([](const String& s) -> Result<T> {
          Result<T> result = details::parse_number<T, details::chars_format::general>()(s);
          if (!result)
            return Err(DracError(result.error().code, std::format("Failed to parse '{}' as general float (scan '{}'): {}", s, Shape, result.error().message)));
          return result;
        });
      else
        static_assert(false, "No scan specification for T");

      return *this;
    }

    fn nargs(const usize num_args)
      -> Argument& {
      m_num_args_range = NArgsRange { num_args, num_args };
      return *this;
    }

    fn nargs(const usize num_args_min, const usize num_args_max) -> Argument& {
      m_num_args_range = NArgsRange { num_args_min, num_args_max };
      return *this;
    }

    fn nargs(const nargs_pattern pattern) -> Argument& {
      switch (pattern) {
        case nargs_pattern::optional:
          m_num_args_range = NArgsRange { 0, 1 };
          break;
        case nargs_pattern::any:
          m_num_args_range = NArgsRange { 0, (std::numeric_limits<std::size_t>::max)() };
          break;
        case nargs_pattern::at_least_one:
          m_num_args_range = NArgsRange { 1, (std::numeric_limits<std::size_t>::max)() };
          break;
      }

      return *this;
    }

    fn remaining() -> Argument& {
      m_accepts_optional_like_value = true;
      return nargs(nargs_pattern::any);
    }

    template <typename T>
    fn add_choice(T&& choice) -> void {
      static_assert(details::IsChoiceTypeSupported<T>::value, "Only string or integer type supported for choice");
      static_assert(std::is_convertible_v<T, StringView> || details::can_invoke_to_string<T>::value, "Choice is not convertible to string_type");
      if (!m_choices.has_value())
        m_choices = Vec<String> {};

      if constexpr (std::is_convertible_v<T, StringView>)
        m_choices.value().emplace_back(StringView { std::forward<T>(choice) });
      else if constexpr (details::can_invoke_to_string<T>::value)
        m_choices.value().push_back(std::to_string(std::forward<T>(choice)));
    }

    fn choices() -> Result<Argument*> {
      if (!m_choices.has_value() || m_choices.value().empty())
        return Err(DracError(DracErrorCode::InvalidArgument, "Zero choices provided"));

      return this;
    }

    template <typename T, typename... U>
    fn choices(T&& first, U&&... rest) -> Result<Argument*> {
      add_choice(std::forward<T>(first));
      if constexpr (sizeof...(rest) == 0) {
        return choices();
      } else {
        return choices(std::forward<U>(rest)...);
      }
    }

    fn find_default_value_in_choices_or_throw() const -> Result<void> {
      assert(m_choices.has_value());
      const Vec<std::string>& choices = m_choices.value();

      if (m_default_value.has_value()) {
        if (std::ranges::find(choices, m_default_value_str.value_or("")) == choices.end()) {
          const String choices_as_csv =
            std::accumulate(choices.begin(), choices.end(), String(), [](const String& a, const String& b) {
              return a + (a.empty() ? "" : ", ") + b;
            });

          return Err(DracError(DracErrorCode::InvalidArgument, String { "Invalid default value " } + m_default_value_repr + " - allowed options: {" + choices_as_csv + "}"));
        }
      }
      return {};
    }

    template <typename Iterator>
    [[nodiscard]] fn is_value_in_choices(Iterator option_it) const -> bool {
      assert(m_choices.has_value());
      const Vec<std::string>& choices = m_choices.value();

      return (std::find(choices.begin(), choices.end(), *option_it) != choices.end());
    }

    /* The dry_run parameter can be set to true to avoid running the actions,
     * and setting m_is_used. This may be used by a pre-processing step to do
     * a first iteration over arguments.
     */
    template <typename Iterator>
    fn consume(Iterator start, Iterator end, const StringView used_name = {}, const bool dry_run = false) -> Result<Iterator> {
      if (!m_is_repeatable && m_is_used)
        return Err(DracError(DracErrorCode::InvalidArgument, String("Duplicate argument ").append(used_name)));

      m_used_name = used_name;

      usize passed_options = 0;

      if (m_choices.has_value()) {
        const auto max_number_of_args = m_num_args_range.get_max();
        const auto min_number_of_args = m_num_args_range.get_min();

        for (auto it = start; it != end; ++it) {
          if (is_value_in_choices(it)) {
            passed_options += 1;
            continue;
          }

          if ((passed_options >= min_number_of_args) &&
              (passed_options <= max_number_of_args))
            break;

          const String choices_as_csv = std::accumulate(
            m_choices.value().begin(), m_choices.value().end(), String(), [](const String& option_a, const String& option_b) {
              return std::format("{}{}{}", option_a, option_a.empty() ? "" : ", ", option_b);
            }
          );

          return Err(DracError(DracErrorCode::InvalidArgument, String { "Invalid argument " } + details::repr(*it) + " - allowed options: {" + choices_as_csv + "}"));
        }
      }

      const usize num_args_max = (m_choices.has_value()) ? passed_options : m_num_args_range.get_max();
      const usize num_args_min = m_num_args_range.get_min();

      if (num_args_max == 0) {
        if (!dry_run) {
          m_values.emplace_back(m_implicit_value);

          for (usize i = 0; i < m_actions.size(); ++i) {
            auto&        action = m_actions[i];
            Result<void> action_call_result;
            std::visit([&](auto& f) {
              if constexpr (std::is_same_v<decltype(f({})), Result<std::any>>) {
                Result<std::any> valued_result = f({});
                if (!valued_result)
                  action_call_result = Err(valued_result.error());
              } else {
                action_call_result = f({});
              }
            },
                       action);
            if (!action_call_result)
              return Err(action_call_result.error());
          }

          if (m_actions.empty()) {
            Result<void> action_call_result;
            std::visit([&](auto& f) {
              if constexpr (std::is_same_v<decltype(f({})), Result<std::any>>) {
                Result<std::any> valued_result = f({});
                if (!valued_result)
                  action_call_result = Err(valued_result.error());
              } else {
                action_call_result = f({});
              }
            },
                       m_default_action);
            if (!action_call_result)
              return Err(action_call_result.error());
          }
          m_is_used = true;
        }
        return start;
      }

      if (auto dist = static_cast<usize>(std::distance(start, end)); dist >= num_args_min) {
        if (num_args_max < dist)
          end = std::next(start, static_cast<typename Iterator::difference_type>(num_args_max));

        if (!m_accepts_optional_like_value) {
          end = std::find_if(
            start, end, [this]<typename T>(T&& PH1) { return is_optional(std::forward<T>(PH1), m_prefix_chars); }
          );

          dist = static_cast<usize>(std::distance(start, end));

          if (dist < num_args_min)
            return Err(DracError(DracErrorCode::InvalidArgument, "Too few arguments for '" + String(m_used_name) + "'."));
        }

        struct ActionApply {
          ActionApply(Iterator f, Iterator l, Argument& s)
            : first(f), last(l), self(&s) {}
          Iterator  first, last;
          Argument* self;

          fn operator()(valued_action& f)->Result<void> {
            for (auto it_arg = first; it_arg != last; ++it_arg) {
              Result<std::any> res = f(*it_arg);
              if (!res)
                return Err(res.error());
              self->m_values.push_back(res.value());
            }
            return {};
          }

          fn operator()(void_action& f)->Result<void> {
            for (auto it_arg = first; it_arg != last; ++it_arg) {
              Result<void> res = f(*it_arg);
              if (!res)
                return Err(res.error());
            }
            if (!self->m_default_value.has_value())
              if (!self->m_accepts_optional_like_value)
                self->m_values.resize(
                  static_cast<usize>(std::distance(first, last))
                );
            return {};
          }
        };

        if (!dry_run) {
          for (usize i = 0; i < m_actions.size(); ++i) {
            auto&        action       = m_actions[i];
            Result<void> apply_result = std::visit(ActionApply { start, end, *this }, action);
            if (!apply_result)
              return Err(apply_result.error());
          }

          if (m_actions.empty()) {
            Result<void> apply_result = std::visit(ActionApply { start, end, *this }, m_default_action);
            if (!apply_result)
              return Err(apply_result.error());
          }
          m_is_used = true;
        }

        return end;
      }
      if (m_default_value.has_value()) {
        if (!dry_run)
          m_is_used = true;

        return start;
      }
      return Err(DracError(DracErrorCode::InvalidArgument, std::format("Too few arguments for '{}'", m_used_name)));
    }

    /*
     * @returns Result<void> indicating success or failure
     */
    [[nodiscard]] fn validate() const -> Result<void> {
      if (m_num_args_range.get_min() > m_num_args_range.get_max()) {
        return Err(DracError(DracErrorCode::InvalidArgument, std::format("Invalid nargs range for argument '{}': min ({}) > max ({}). This indicates a configuration error when defining the argument.", m_names.empty() ? "UnnamedArgument" : m_names[0], m_num_args_range.get_min(), m_num_args_range.get_max())));
      }

      if (m_is_optional) {
        // TODO: check if an implicit value was programmed for this argument
        if (!m_is_used && !m_default_value.has_value() && m_is_required)
          return Err(DracError(DracErrorCode::InvalidArgument, std::format("Required argument '{}' was not provided", m_names[0])));

        if (m_is_used && m_is_required && m_values.empty())
          return Err(DracError(DracErrorCode::InvalidArgument, std::format("Required argument '{}' requires a value, but none was provided", m_names[0])));

        if (m_is_used && m_num_args_range.get_min() > m_values.size())
          return Err(DracError(DracErrorCode::InvalidArgument, std::format("Too few arguments for optional argument '{}'. Expected at least {}, got {}.", m_names[0], m_num_args_range.get_min(), m_values.size())));
      } else {
        if (!m_num_args_range.contains(m_values.size()) && !m_default_value.has_value()) {
          String expected_str;

          if (m_num_args_range.is_exact())
            expected_str = std::to_string(m_num_args_range.get_min());
          else if (!m_num_args_range.is_right_bounded())
            expected_str = std::format("at least {}", m_num_args_range.get_min());
          else
            expected_str = std::format("{} to {}", m_num_args_range.get_min(), m_num_args_range.get_max());

          return Err(DracError(DracErrorCode::InvalidArgument, std::format("Incorrect number of arguments for positional argument '{}'. Expected {}, got {}.", (m_metavar.empty() ? m_names[0] : m_metavar), expected_str, m_values.size())));
        }

        if (m_num_args_range.get_min() > m_values.size())
          return Err(DracError(DracErrorCode::InvalidArgument, std::format("Too few arguments for positional argument '{}'. Expected at least {}, got {}.", (m_metavar.empty() ? m_names[0] : m_metavar), m_num_args_range.get_min(), m_values.size())));
      }

      if (m_num_args_range.get_max() < m_values.size()) {
        if (m_is_optional)
          return Err(DracError(DracErrorCode::InvalidArgument, std::format("Too many arguments for optional argument '{}'. Expected at most {}, got {}.", m_names[0], m_num_args_range.get_max(), m_values.size())));

        return Err(DracError(DracErrorCode::InvalidArgument, std::format("Too many arguments for positional argument '{}'. Expected at most {}, got {}.", (m_metavar.empty() ? m_names[0] : m_metavar), m_num_args_range.get_max(), m_values.size())));
      }

      if (m_choices.has_value()) {
        const Vec<String>& choices = m_choices.value();

        if (m_default_value.has_value())
          if (const String& default_val_str = m_default_value_str.value(); std::ranges::find(choices, default_val_str) == choices.end()) {
            const String choices_as_csv = std::accumulate(
              choices.begin(), choices.end(), String(), [](const String& option_a, const String& option_b) -> String { return option_a + (option_a.empty() ? "" : ", ") + option_b; }
            );
            return Err(DracError(DracErrorCode::InvalidArgument, std::format("Default value '{}' is not in the allowed choices: {{{}}}", default_val_str, choices_as_csv)));
          }

        for (const auto& value_any : m_values) {
          if (value_any.type() != typeid(String))
            return Err(DracError(DracErrorCode::InvalidArgument, std::format("Invalid argument type for choice validation - expected string, got '{}'", value_any.type().name())));

          if (const String& value = std::any_cast<const String&>(value_any); std::ranges::find(choices, value) == choices.end()) {
            const String choices_as_csv = std::accumulate(
              choices.begin(), choices.end(), String(), [](const String& option_a, const String& option_b) -> String { return std::format("{}{}{}", option_a, option_a.empty() ? "" : ", ", option_b); }
            );

            return Err(DracError(DracErrorCode::InvalidArgument, std::format("Invalid argument '{}' - allowed options: {{{}}}", value, choices_as_csv)));
          }
        }
      }

      return {};
    }

    [[nodiscard]] fn get_names_csv(const char separator = ',') const -> String {
      return std::accumulate(
        m_names.begin(), m_names.end(), String { "" }, [&](const String& result, const String& name) {
          return result.empty() ? name : result + separator + name;
        }
      );
    }

    [[nodiscard]] fn get_usage_full() const -> String {
      std::stringstream usage;

      usage << get_names_csv('/');
      const String metavar = !m_metavar.empty() ? m_metavar : "VAR";

      if (m_num_args_range.get_max() > 0) {
        usage << " " << metavar;
        if (m_num_args_range.get_max() > 1)
          usage << "...";
      }

      return usage.str();
    }

    [[nodiscard]] fn get_inline_usage() const -> String {
      std::stringstream usage;

      String longest_name = m_names.front();
      for (const String& s : m_names)
        if (s.size() > longest_name.size())
          longest_name = s;

      if (!m_is_required)
        usage << "[";

      usage << longest_name;
      const String metavar = !m_metavar.empty() ? m_metavar : "VAR";

      if (m_num_args_range.get_max() > 0) {
        usage << " " << metavar;
        if (m_num_args_range.get_max() > 1 && m_metavar.contains("> <"))
          usage << "...";
      }

      if (!m_is_required)
        usage << "]";

      if (m_is_repeatable)
        usage << "...";

      return usage.str();
    }

    [[nodiscard]] fn get_arguments_length() const -> usize {
      const usize names_size = std::accumulate(
        std::begin(m_names), std::end(m_names), static_cast<usize>(0), [](const u32& sum, const String& s) { return sum + s.size(); }
      );

      if (is_positional(m_names.front(), m_prefix_chars)) {
        if (!m_metavar.empty())
          return 2 + m_metavar.size();

        return 2 + names_size + (m_names.size() - 1);
      }

      usize size = names_size + (2 * (m_names.size() - 1));
      if (!m_metavar.empty() && m_num_args_range == NArgsRange { 1, 1 })
        size += m_metavar.size() + 1;

      return size + 2;
    }

    friend fn operator<<(std::ostream& stream, const Argument& argument)->std::ostream& {
      String name_str = "  ";

      if (argparse::Argument::is_positional(argument.m_names.front(), argument.m_prefix_chars)) {
        if (!argument.m_metavar.empty()) {
          name_str += argument.m_metavar;
        } else {
          name_str += details::join(argument.m_names.begin(), argument.m_names.end(), " ");
        }
      } else {
        name_str += details::join(argument.m_names.begin(), argument.m_names.end(), ", ");
        if (!argument.m_metavar.empty() &&
            ((argument.m_num_args_range == NArgsRange { 1, 1 }) ||
             (argument.m_num_args_range.get_min() == argument.m_num_args_range.get_max() &&
              argument.m_metavar.contains("> <")))) {
          name_str += std::format(" {}", argument.m_metavar);
        }
      }

      const std::streamsize stream_width = stream.width();
      const String          name_padding = String(name_str.size(), ' ');

      auto pos  = String::size_type {};
      auto prev = String::size_type {};

      bool        first_line = true;
      const char* hspace     = "  ";

      stream << name_str;

      const StringView help_view(argument.m_help);

      while ((pos = argument.m_help.find('\n', prev)) != String::npos) {
        const StringView line = help_view.substr(prev, pos - prev + 1);

        if (first_line) {
          stream << hspace << line;
          first_line = false;
        } else {
          stream.width(stream_width);
          stream << name_padding << hspace << line;
        }

        prev += pos - prev + 1;
      }

      if (first_line)
        stream << hspace << argument.m_help;
      else if (const StringView leftover = help_view.substr(prev, argument.m_help.size() - prev); !leftover.empty()) {
        stream.width(stream_width);
        stream << name_padding << hspace << leftover;
      }

      if (!argument.m_help.empty())
        stream << " ";

      stream << argument.m_num_args_range;

      bool add_space = false;
      if (argument.m_default_value.has_value() &&
          argument.m_num_args_range != NArgsRange { 0, 0 }) {
        stream << std::format("[default: {}]", argument.m_default_value_repr);
        add_space = true;
      } else if (argument.m_is_required) {
        stream << "[required]";
        add_space = true;
      }

      if (argument.m_is_repeatable) {
        if (add_space)
          stream << " ";

        stream << "[may be repeated]";
      }

      stream << "\n";
      return stream;
    }

    template <typename T>
    fn operator!=(const T& rhs) const->bool {
      return !(*this == rhs);
    }

    /*
     * Compare to an argument value of known type
     * @throws std::logic_error in case of incompatible types
     */
    template <typename T>
    fn operator==(const T& rhs) const->bool {
      Result<T> lhs_res = get<T>();
      if (!lhs_res) {
        return false;
      }

      const T& lhs_val = lhs_res.value();

      if constexpr (!details::IsContainer<T>) {
        return lhs_val == rhs;
      } else {
        if (lhs_val.size() != rhs.size()) {
          return false;
        }
        return std::equal(std::begin(lhs_val), std::end(lhs_val), std::begin(rhs));
      }
    }

    /*
     * positional:
     *    _empty_
     *    '-'
     *    '-' decimal-literal
     *    !'-' anything
     */
    static fn is_positional(StringView name, const StringView prefix_chars) -> bool {
      const int first = lookahead(name);

      if (first == eof)
        return true;

      if (prefix_chars.contains(static_cast<char>(first))) {
        name.remove_prefix(1);

        if (name.empty())
          return true;

        return is_decimal_literal(name);
      }

      return true;
    }

   private:
    class NArgsRange {
      usize m_min;
      usize m_max;

     public:
      NArgsRange(const usize minimum, const usize maximum)
        : m_min(minimum), m_max(maximum) {}

      [[nodiscard]] fn contains(const usize value) const -> bool {
        return value >= m_min && value <= m_max;
      }

      [[nodiscard]] fn is_exact() const -> bool {
        return m_min == m_max;
      }

      [[nodiscard]] fn is_right_bounded() const -> bool {
        return m_max < (std::numeric_limits<usize>::max)();
      }

      [[nodiscard]] fn get_min() const -> usize {
        return m_min;
      }

      [[nodiscard]] fn get_max() const -> usize {
        return m_max;
      }

      friend fn operator<<(std::ostream& stream, const NArgsRange& range)
        ->std::ostream& {
        if (range.m_min == range.m_max) {
          if (range.m_min != 0 && range.m_min != 1)
            stream << std::format("[nargs: {}] ", range.m_min);
        } else if (range.m_max == (std::numeric_limits<usize>::max)())
          stream << std::format("[nargs: {} or more] ", range.m_min);
        else
          stream << std::format("[nargs={}..{}] ", range.m_min, range.m_max);

        return stream;
      }

      fn operator==(const NArgsRange& rhs) const->bool {
        return rhs.m_min == m_min && rhs.m_max == m_max;
      }

      fn operator!=(const NArgsRange& rhs) const->bool {
        return !(*this == rhs);
      }
    };

    static constexpr int eof = std::char_traits<char>::eof();

    static fn lookahead(const StringView sview) -> int {
      if (sview.empty())
        return eof;

      return static_cast<unsigned char>(sview[0]);
    }

    /*
     * decimal-literal:
     *    '0'
     *    nonzero-digit digit-sequence_opt
     *    integer-part fractional-part
     *    fractional-part
     *    integer-part '.' exponent-part_opt
     *    integer-part exponent-part
     *
     * integer-part:
     *    digit-sequence
     *
     * fractional-part:
     *    '.' post-decimal-point
     *
     * post-decimal-point:
     *    digit-sequence exponent-part_opt
     *
     * exponent-part:
     *    'e' post-e
     *    'E' post-e
     *
     * post-e:
     *    sign_opt digit-sequence
     *
     * sign: one of
     *    '+' '-'
     */
    // NOLINTBEGIN(cppcoreguidelines-avoid-goto)
    static fn is_decimal_literal(StringView s) -> bool {
      fn is_digit = [](auto c) constexpr -> bool {
        return c >= '0' && c <= '9';
      };

      fn consume_digits = [=](StringView sd) -> StringView {
        const auto it = std::ranges::find_if_not(sd, is_digit);

        return sd.substr(static_cast<usize>(it - std::begin(sd)));
      };

      switch (lookahead(s)) {
        case '0': {
          s.remove_prefix(1);

          if (s.empty())
            return true;

          goto integer_part;
        }
        case '1':
        case '2':
        case '3':
        case '4':
        case '5':
        case '6':
        case '7':
        case '8':
        case '9': {
          s = consume_digits(s);

          if (s.empty())
            return true;

          goto integer_part_consumed;
        }
        case '.': {
          s.remove_prefix(1);
          goto post_decimal_point;
        }
        default:
          return false;
      }

    integer_part:
      s = consume_digits(s);
    integer_part_consumed:
      switch (lookahead(s)) {
        case '.': {
          s.remove_prefix(1);

          if (is_digit(lookahead(s)))
            goto post_decimal_point;

          goto exponent_part_opt;
        }
        case 'e':
        case 'E': {
          s.remove_prefix(1);
          goto post_e;
        }
        default:
          return false;
      }

    post_decimal_point:
      if (is_digit(lookahead(s))) {
        s = consume_digits(s);
        goto exponent_part_opt;
      }
      return false;

    exponent_part_opt:
      switch (lookahead(s)) {
        case eof:
          return true;
        case 'e':
        case 'E': {
          s.remove_prefix(1);
          goto post_e;
        }
        default:
          return false;
      }

    post_e:
      switch (lookahead(s)) {
        case '-':
        case '+':
          s.remove_prefix(1);
        default:
          break;
      }

      if (is_digit(lookahead(s))) {
        s = consume_digits(s);
        return s.empty();
      }

      return false;
    }
    // NOLINTEND(cppcoreguidelines-avoid-goto)

    static fn is_optional(const StringView name, const StringView prefix_chars) -> bool {
      return !is_positional(name, prefix_chars);
    }

    /*
     * Get argument value given a type
     * @throws std::logic_error in case of incompatible types
     */
    template <typename T>
    fn get() const -> Result<T> {
      if (!m_values.empty()) {
        try {
          if constexpr (details::IsContainer<T>)
            return any_cast_container<T>(m_values);
          else
            return std::any_cast<T>(m_values.front());
        } catch (const std::bad_any_cast& e) {
          return Err(DracError(DracErrorCode::InternalError, std::format("Bad any_cast for value in get(): {}", e.what())));
        }
      }

      if (m_default_value.has_value()) {
        try {
          return std::any_cast<T>(m_default_value);
        } catch (const std::bad_any_cast& e) {
          return Err(DracError(DracErrorCode::InternalError, std::format("Bad any_cast for default_value in get(): {}", e.what())));
        }
      }

      if constexpr (details::IsContainer<T>)
        if (!m_accepts_optional_like_value && m_values.empty())
          return T {};

      return Err(DracError(DracErrorCode::NotFound, std::format("No value provided for '{}'", m_names.back())));
    }

    /*
     * Get argument value given a type.
     * @pre The object has no default value.
     * @returns The stored value if any, std::nullopt otherwise.
     */
    template <typename T>
    fn present() const -> Result<Option<T>> {
      if (m_default_value.has_value())
        return Err(DracError(DracErrorCode::InvalidArgument, std::format("present() called on argument '{}' which has a default value.", m_names.back())));

      if (m_values.empty())
        return std::nullopt;

      try {
        if constexpr (details::IsContainer<T>)
          return any_cast_container<T>(m_values);
        else
          return std::any_cast<T>(m_values.front());
      } catch (const std::bad_any_cast& e) {
        return Err(DracError(DracErrorCode::InternalError, std::format("Bad any_cast in present(): {}", e.what())));
      }
    }

    template <typename T>
    static fn any_cast_container(const Vec<std::any>& operand) -> T {
      using ValueType = typename T::value_type;

      T result;

      std::transform(
        std::begin(operand), std::end(operand), std::back_inserter(result), [](const auto& value) { return std::any_cast<ValueType>(value); }
      );

      return result;
    }

    fn set_usage_newline_counter(const int i) -> void {
      m_usage_newline_counter = i;
    }

    fn set_group_idx(const usize i) -> void {
      m_group_idx = i;
    }

    Vec<String>    m_names;
    StringView     m_used_name;
    String         m_help;
    String         m_metavar;
    std::any       m_default_value;
    String         m_default_value_repr;
    Option<String> m_default_value_str;

    std::any m_implicit_value;

    Option<Vec<String>> m_choices { std::nullopt };

    using valued_action = std::function<Result<std::any>(const String&)>;
    using void_action   = std::function<Result<void>(const String&)>;

    Vec<std::variant<valued_action, void_action>> m_actions;

    std::variant<valued_action, void_action> m_default_action {
      std::in_place_type<valued_action>,
      [](const String& value) -> Result<std::any> { return value; }
    };

    Vec<std::any> m_values;
    NArgsRange    m_num_args_range { 1, 1 };

    bool       m_accepts_optional_like_value : 1;
    bool       m_is_optional : 1;
    bool       m_is_required : 1;
    bool       m_is_repeatable : 1;
    bool       m_is_used : 1;
    bool       m_is_hidden : 1;
    StringView m_prefix_chars;
    int        m_usage_newline_counter = 0;
    usize      m_group_idx             = 0;
  };

  class ArgumentParser {
   public:
    explicit ArgumentParser(String program_name = {}, String version = "1.0", const default_arguments add_args = default_arguments::all, const bool exit_on_default_arguments = true, std::ostream& os = std::cout)
      : m_program_name(std::move(program_name)), m_version(std::move(version)), m_exit_on_default_arguments(exit_on_default_arguments), m_parser_path(m_program_name) {
      if ((add_args & default_arguments::help) == default_arguments::help)
        add_argument("-h", "--help")
          .action([&](const String& /*unused*/) {
            os << help().str();

            if (m_exit_on_default_arguments)
              std::exit(0);
          })
          .default_value(false)
          .help("shows help message and exits")
          .implicit_value(true)
          .nargs(0);

      if ((add_args & default_arguments::version) == default_arguments::version)
        add_argument("-v", "--version")
          .action([&](const String& /*unused*/) {
            os << m_version << '\n';

            if (m_exit_on_default_arguments)
              std::exit(0);
          })
          .default_value(false)
          .help("prints version information and exits")
          .implicit_value(true)
          .nargs(0);
    }

    ~ArgumentParser() = default;

    ArgumentParser(const ArgumentParser& other)                = delete;
    fn operator=(const ArgumentParser& other)->ArgumentParser& = delete;
    ArgumentParser(ArgumentParser&&) noexcept                  = delete;
    fn operator=(ArgumentParser&&)->ArgumentParser&            = delete;

    explicit operator bool() const {
      const bool arg_used = std::ranges::any_of(m_argument_map, [](auto& it) { return it.second->m_is_used; });

      const bool subparser_used =
        std::ranges::any_of(m_subparser_used, [](auto& it) { return it.second; });

      return m_is_parsed && (arg_used || subparser_used);
    }

    template <typename... Targs>
    fn add_argument(Targs... f_args) -> Argument& {
      using array_of_sv = std::array<StringView, sizeof...(Targs)>;

      auto argument = m_optional_arguments.emplace(std::cend(m_optional_arguments), m_prefix_chars, array_of_sv { f_args... });

      if (!argument->m_is_optional)
        m_positional_arguments.splice(std::cend(m_positional_arguments), m_optional_arguments, argument);

      argument->set_usage_newline_counter(m_usage_newline_counter);
      argument->set_group_idx(m_group_names.size());

      index_argument(argument);
      return *argument;
    }

    class MutuallyExclusiveGroup {
      friend class ArgumentParser;

     public:
      MutuallyExclusiveGroup() = delete;

      ~MutuallyExclusiveGroup() = default;

      fn operator=(MutuallyExclusiveGroup&&)->MutuallyExclusiveGroup& = delete;
      explicit MutuallyExclusiveGroup(ArgumentParser& parent, const bool required = false)
        : m_parent(parent), m_required(required), m_elements({}) {}

      MutuallyExclusiveGroup(const MutuallyExclusiveGroup& other) = delete;

      fn operator=(const MutuallyExclusiveGroup& other)->MutuallyExclusiveGroup& = delete;

      MutuallyExclusiveGroup(MutuallyExclusiveGroup&& other) noexcept
        : m_parent(other.m_parent), m_required(other.m_required), m_elements(std::move(other.m_elements)) {
        other.m_elements.clear();
      }

      template <typename... Targs>
      fn add_argument(Targs... f_args) -> Argument& {
        Argument& argument = m_parent.add_argument(std::forward<Targs>(f_args)...);
        m_elements.push_back(&argument);
        argument.set_usage_newline_counter(m_parent.m_usage_newline_counter);
        argument.set_group_idx(m_parent.m_group_names.size());
        return argument;
      }

     private:
      ArgumentParser& m_parent; // NOLINT(cppcoreguidelines-avoid-const-or-ref-data-members)
      bool            m_required = false;
      Vec<Argument*>  m_elements;
    };

    fn add_mutually_exclusive_group(bool required = false) -> MutuallyExclusiveGroup& {
      m_mutually_exclusive_groups.emplace_back(*this, required);
      return m_mutually_exclusive_groups.back();
    }

    template <typename... Targs>
    fn add_parents(const Targs&... f_args) -> ArgumentParser& {
      for (const ArgumentParser& parent_parser : { std::ref(f_args)... }) {
        for (const Argument& argument : parent_parser.m_positional_arguments) {
          const auto it = m_positional_arguments.insert(
            std::cend(m_positional_arguments), argument
          );

          index_argument(it);
        }

        for (const Argument& argument : parent_parser.m_optional_arguments) {
          const auto it = m_optional_arguments.insert(std::cend(m_optional_arguments), argument);

          index_argument(it);
        }
      }

      return *this;
    }

    // Ask for the next optional arguments to be displayed on a separate
    // line in usage() output. Only effective if set_usage_max_line_width() is
    // also used.
    fn add_usage_newline() -> ArgumentParser& {
      ++m_usage_newline_counter;
      return *this;
    }

    // Ask for the next optional arguments to be displayed in a separate section
    // in usage() and help (<< *this) output.
    // For usage(), this is only effective if set_usage_max_line_width() is
    // also used.
    fn add_group(String group_name) -> ArgumentParser& {
      m_group_names.emplace_back(std::move(group_name));
      return *this;
    }

    fn add_description(String description) -> ArgumentParser& {
      m_description = std::move(description);
      return *this;
    }

    fn add_epilog(String epilog) -> ArgumentParser& {
      m_epilog = std::move(epilog);
      return *this;
    }

    // Add a un-documented/hidden alias for an argument.
    // Ideally we'd want this to be a method of Argument, but Argument
    // does not own its owing ArgumentParser.
    fn add_hidden_alias_for(const Argument& arg, const StringView alias) -> Result<ArgumentParser*> {
      for (auto it = m_optional_arguments.begin();
           it != m_optional_arguments.end();
           ++it)
        if (&(*it) == &arg) {
          m_argument_map.insert_or_assign(String(alias), it);
          return this;
        }

      return Err(DracError(DracErrorCode::InvalidArgument, std::format("Argument is not an optional argument of this parser")));
    }

    /* Getter for arguments and subparsers.
     * @throws std::logic_error in case of an invalid argument or subparser name
     */
    template <typename T = Argument>
    fn at(const StringView name) -> Result<T*> {
      if constexpr (std::is_same_v<T, Argument>) {
        Result<Argument*> arg_result = (*this)[name];
        if (!arg_result) {
          return Err(DracError(DracErrorCode::NotFound, std::format("Argument not found in 'at': {}", name)));
        }
        return arg_result.value();
      } else {
        static_assert(std::is_same_v<T, ArgumentParser>, "T must be Argument or ArgumentParser for at()");
        const String str_name(name);
        if (const auto subparser_it = m_subparser_map.find(str_name); subparser_it != m_subparser_map.end()) {
          return &(subparser_it->second->get());
        }
        return Err(DracError(DracErrorCode::NotFound, std::format("No such subparser: {}", str_name)));
      }
    }

    fn set_prefix_chars(String prefix_chars) -> ArgumentParser& {
      m_prefix_chars = std::move(prefix_chars);
      return *this;
    }

    fn set_assign_chars(String assign_chars) -> ArgumentParser& {
      m_assign_chars = std::move(assign_chars);
      return *this;
    }

    /* Call parse_args_internal - which does all the work
     * Then, validate the parsed arguments
     * This variant is used mainly for testing
     * @throws std::runtime_error in case of any invalid argument
     */
    // NOLINTNEXTLINE(misc-no-recursion)
    fn parse_args(const Vec<String>& arguments) -> Result<void> {
      Result<void> pres = parse_args_internal(arguments);
      if (!pres)
        return pres;

      for (const auto& argument_entry : m_argument_map) {
        if (Result<> validation_result = argument_entry.second->validate(); !validation_result) {
          return Err(validation_result.error());
        }
      }

      for (const MutuallyExclusiveGroup& group : m_mutually_exclusive_groups) {
        bool            mutex_argument_used = false;
        const Argument* mutex_argument_ptr  = nullptr;

        for (const Argument* arg : group.m_elements) {
          auto is_used_res = this->is_used(arg->m_names.front());
          if (!is_used_res)
            return Err(is_used_res.error());

          if (!mutex_argument_used && is_used_res.value()) {
            mutex_argument_used = true;
            mutex_argument_ptr  = arg;
          } else if (mutex_argument_used && is_used_res.value()) {
            return Err(DracError(DracErrorCode::InvalidArgument, std::format("Argument '{}' not allowed with '{}'", arg->get_usage_full(), mutex_argument_ptr->get_usage_full())));
          }
        }

        if (!mutex_argument_used && group.m_required) {
          String      argument_names {};
          usize       i    = 0;
          const usize size = group.m_elements.size();
          for (const Argument* arg : group.m_elements) {
            if (i + 1 == size)
              argument_names += String("'") + arg->get_usage_full() + String("' ");
            else
              argument_names += String("'") + arg->get_usage_full() + String("' or ");
            i += 1;
          }
          return Err(DracError(DracErrorCode::InvalidArgument, std::format("One of the arguments {}is required", argument_names)));
        }
      }
      return {};
    }

    /* Call parse_known_args_internal - which does all the work
     * Then, validate the parsed arguments
     * This variant is used mainly for testing
     * @throws std::runtime_error in case of any invalid argument
     */
    // NOLINTNEXTLINE(misc-no-recursion)
    fn parse_known_args_internal(const Vec<String>& raw_arguments) -> Result<Vec<String>> {
      Vec<String> arguments = preprocess_arguments(raw_arguments);
      Vec<String> unknown_arguments {};

      if (m_program_name.empty() && !arguments.empty())
        m_program_name = arguments.front();

      const auto end                    = std::end(arguments);
      auto       positional_argument_it = std::begin(m_positional_arguments);

      for (auto it = std::next(std::begin(arguments)); it != end;) {
        const String& current_argument = *it;
        if (Argument::is_positional(current_argument, m_prefix_chars)) {
          if (positional_argument_it == std::end(m_positional_arguments)) {
            if (auto subparser_it = m_subparser_map.find(current_argument); subparser_it != m_subparser_map.end()) {
              const Vec<String> unprocessed_arguments = Vec<String>(it, end);
              m_is_parsed                             = true;
              m_subparser_used[current_argument]      = true;
              return subparser_it->second->get().parse_known_args_internal(unprocessed_arguments);
            }
            unknown_arguments.push_back(current_argument);
            ++it;
          } else {
            const auto           argument       = positional_argument_it++;
            Result<decltype(it)> consume_result = argument->consume(it, end);
            if (!consume_result)
              return Err(consume_result.error());
            it = consume_result.value();
          }
          continue;
        }

        auto arg_map_it = m_argument_map.find(current_argument);
        if (arg_map_it != m_argument_map.end()) {
          const auto           argument       = arg_map_it->second;
          Result<decltype(it)> consume_result = argument->consume(std::next(it), end, arg_map_it->first);
          if (!consume_result)
            return Err(consume_result.error());
          it = consume_result.value();
        } else if (const String& compound_arg = current_argument;
                   compound_arg.size() > 1 &&
                   is_valid_prefix_char(compound_arg[0]) &&
                   !is_valid_prefix_char(compound_arg[1])) {
          ++it;
          for (usize j = 1; j < compound_arg.size(); j++) {
            const String hypothetical_arg = { '-', compound_arg[j] };
            auto         arg_map_it2      = m_argument_map.find(hypothetical_arg);
            if (arg_map_it2 != m_argument_map.end()) {
              const auto           argument       = arg_map_it2->second;
              Result<decltype(it)> consume_result = argument->consume(it, end, arg_map_it2->first);
              if (!consume_result)
                return Err(consume_result.error());
              it = consume_result.value();
            } else {
              unknown_arguments.push_back(current_argument);
              break;
            }
          }
        } else {
          unknown_arguments.push_back(current_argument);
          ++it;
        }
      }
      m_is_parsed = true;
      return unknown_arguments;
    }

    /* Main entry point for parsing command-line arguments using this
     * ArgumentParser
     * @throws std::runtime_error in case of any invalid argument
     */
    // NOLINTNEXTLINE(*-avoid-c-arrays)
    fn parse_args(const int argc, const char* const argv[]) -> Result<void> {
      return parse_args({ argv, argv + argc });
    }

    /* Main entry point for parsing command-line arguments using this
     * ArgumentParser
     * @throws std::runtime_error in case of any invalid argument
     */
    // NOLINTNEXTLINE(*-avoid-c-arrays)
    fn parse_known_args(const int argc, const char* const argv[]) -> Result<Vec<String>> {
      return parse_known_args_internal({ argv, argv + argc });
    }

    /* Getter for options with default values.
     * @throws std::logic_error if parse_args() has not been previously called
     * @throws std::logic_error if there is no such option
     * @throws std::logic_error if the option has no value
     * @throws std::bad_any_cast if the option is not of type T
     */
    template <typename T = String>
    fn get(const StringView arg_name) const -> Result<T> {
      if (!m_is_parsed)
        return Err(DracError(DracErrorCode::InternalError, "Nothing parsed, no arguments are available."));

      Result<Argument*> arg_ref_result = (*this)[arg_name];
      if (!arg_ref_result)
        return Err(arg_ref_result.error());

      return arg_ref_result.value()->get<T>();
    }

    /* Getter for options without default values.
     * @pre The option has no default value.
     * @throws std::logic_error if there is no such option
     * @throws std::bad_any_cast if the option is not of type T
     */
    template <typename T = String>
    fn present(const StringView arg_name) const -> Result<Option<T>> {
      if (!m_is_parsed)
        return Err(DracError(DracErrorCode::InternalError, "Nothing parsed, no arguments are available for present()."));

      Result<Argument*> arg_ref_result = (*this)[arg_name];
      if (!arg_ref_result)
        return Err(arg_ref_result.error());

      return arg_ref_result.value()->present<T>();
    }

    /* Getter that returns true for user-supplied options. Returns false if not
     * user-supplied, even with a default value.
     */
    [[nodiscard]] fn is_used(const StringView arg_name) const -> Result<bool> {
      if (!m_is_parsed)
        return Err(DracError(DracErrorCode::InternalError, "Nothing parsed, cannot check if argument is used."));

      Result<Argument*> arg_ref_result = (*this)[arg_name];
      if (!arg_ref_result)
        return Err(arg_ref_result.error());
      return bool { arg_ref_result.value()->m_is_used };
    }

    /* Getter that returns true if a subcommand is used.
     */
    [[nodiscard]] fn is_subcommand_used(const StringView subcommand_name) const -> Result<bool> {
      if (!m_is_parsed)
        return Err(DracError(DracErrorCode::InternalError, "Nothing parsed, cannot check if subcommand is used."));
      try {
        return m_subparser_used.at(String(subcommand_name));
      } catch (const std::out_of_range& oor) {
        return Err(DracError(DracErrorCode::NotFound, std::format("Subcommand '{}' not found for is_subcommand_used check.", subcommand_name)));
      }
    }

    /* Getter that returns true if a subcommand is used.
     */
    [[nodiscard]] fn is_subcommand_used(const ArgumentParser& subparser) const -> Result<bool> {
      return is_subcommand_used(subparser.m_program_name);
    }

    /* Indexing operator. Return a reference to an Argument object
     * Used in conjunction with Argument.operator== e.g., parser["foo"] == true
     * @throws std::logic_error in case of an invalid argument name
     */
    fn operator[](const StringView arg_name) const->Result<Argument*> {
      String name(arg_name);

      auto it = m_argument_map.find(name);

      if (it != m_argument_map.end())
        return &(*(it->second));

      if (!is_valid_prefix_char(arg_name.front())) {
        const char legal_prefix_char = get_any_valid_prefix_char();

        const String prefix = String(1, legal_prefix_char);

        name = prefix + String(arg_name);
        it   = m_argument_map.find(name);
        if (it != m_argument_map.end())
          return &(*(it->second));

        name = prefix + name;
        it   = m_argument_map.find(name);

        if (it != m_argument_map.end())
          return &(*(it->second));
      }

      return Err(DracError(DracErrorCode::NotFound, std::format("No such argument: {}", arg_name)));
    }

    // Print help message
    friend fn operator<<(std::ostream& stream, const ArgumentParser& parser)->std::ostream& {
      stream.setf(std::ios_base::left);

      const usize longest_arg_length = parser.get_length_of_longest_argument();

      stream << parser.usage() << "\n\n";

      if (!parser.m_description.empty())
        stream << parser.m_description << "\n\n";

      const bool has_visible_positional_args = std::ranges::find_if(parser.m_positional_arguments, [](const Argument& argument) { return !argument.m_is_hidden; }) !=
        parser.m_positional_arguments.end();
      if (has_visible_positional_args)
        stream << "Positional arguments:\n";

      for (const Argument& argument : parser.m_positional_arguments)
        if (!argument.m_is_hidden) {
          stream.width(static_cast<std::streamsize>(longest_arg_length));
          stream << argument;
        }

      if (!parser.m_optional_arguments.empty())
        stream << (!has_visible_positional_args ? "" : "\n")
               << "Optional arguments:\n";

      for (const Argument& argument : parser.m_optional_arguments)
        if (argument.m_group_idx == 0 && !argument.m_is_hidden) {
          stream.width(static_cast<std::streamsize>(longest_arg_length));
          stream << argument;
        }

      for (usize i_group = 0; i_group < parser.m_group_names.size(); ++i_group) {
        stream << std::format("\n{} (detailed usage):\n", parser.m_group_names[i_group]);

        for (const Argument& argument : parser.m_optional_arguments)
          if (argument.m_group_idx == i_group + 1 && !argument.m_is_hidden) {
            stream.width(static_cast<std::streamsize>(longest_arg_length));
            stream << argument;
          }
      }

      if (std::ranges::any_of(parser.m_subparser_map, [](auto& p) { return !p.second->get().m_suppress; })) {
        stream << (parser.m_positional_arguments.empty()
                     ? (parser.m_optional_arguments.empty() ? "" : "\n")
                     : "\n")
               << "Subcommands:\n";
        for (const auto& [command, subparser] : parser.m_subparser_map) {
          if (subparser->get().m_suppress)
            continue;

          stream << std::format("  {:<{}} {}", command, longest_arg_length - 2, subparser->get().m_description) << "\n";
        }
      }

      if (!parser.m_epilog.empty()) {
        stream << '\n';
        stream << parser.m_epilog << "\n\n";
      }

      return stream;
    }

    // Format help message
    [[nodiscard]] fn help() const -> std::stringstream {
      std::stringstream out;
      out << *this;
      return out;
    }

    // Sets the maximum width for a line of the Usage message
    fn set_usage_max_line_width(const usize w) -> ArgumentParser& {
      this->m_usage_max_line_width = w;
      return *this;
    }

    // Asks to display arguments of mutually exclusive group on separate lines in
    // the Usage message
    fn set_usage_break_on_mutex() -> ArgumentParser& {
      this->m_usage_break_on_mutex = true;
      return *this;
    }

    // Format usage part of help only
    [[nodiscard]] fn usage() const -> String {
      String     curline = std::format("Usage: {}", this->m_parser_path);
      const bool multiline_usage =
        this->m_usage_max_line_width < (std::numeric_limits<usize>::max)();
      const usize indent_size = curline.size();
      String      result;

      const fn deal_with_options_of_group = [&](const usize group_idx) {
        bool found_options = false;

        const MutuallyExclusiveGroup* cur_mutex             = nullptr;
        int                           usage_newline_counter = -1;

        for (const Argument& argument : this->m_optional_arguments) {
          if (argument.m_is_hidden) {
            continue;
          }
          if (multiline_usage) {
            if (argument.m_group_idx != group_idx) {
              continue;
            }
            if (usage_newline_counter != argument.m_usage_newline_counter) {
              if (usage_newline_counter >= 0) {
                if (curline.size() > indent_size) {
                  result += std::format("\n{}", curline);
                  curline = String(indent_size, ' ');
                }
              }
              usage_newline_counter = argument.m_usage_newline_counter;
            }
          }
          found_options                                  = true;
          const String                  arg_inline_usage = argument.get_inline_usage();
          const MutuallyExclusiveGroup* arg_mutex =
            get_belonging_mutex(&argument);
          if ((cur_mutex != nullptr) && (arg_mutex == nullptr)) {
            curline += ']';
            if (this->m_usage_break_on_mutex) {
              result += std::format("\n{}", curline);
              curline = String(indent_size, ' ');
            }
          } else if ((cur_mutex == nullptr) && (arg_mutex != nullptr)) {
            if ((this->m_usage_break_on_mutex && curline.size() > indent_size) ||
                curline.size() + 3 + arg_inline_usage.size() >
                  this->m_usage_max_line_width) {
              result += std::format("\n{}", curline);
              curline = String(indent_size, ' ');
            }
            curline += " [";
          } else if ((cur_mutex != nullptr) && (arg_mutex != nullptr)) {
            if (cur_mutex != arg_mutex) {
              curline += ']';
              if (this->m_usage_break_on_mutex ||
                  curline.size() + 3 + arg_inline_usage.size() >
                    this->m_usage_max_line_width) {
                result += std::format("\n{}", curline);
                curline = String(indent_size, ' ');
              }
              curline += " [";
            } else {
              curline += '|';
            }
          }
          cur_mutex = arg_mutex;
          if (curline.size() != indent_size &&
              curline.size() + 1 + arg_inline_usage.size() >
                this->m_usage_max_line_width) {
            result += std::format("\n{}", curline);
            curline = String(indent_size, ' ');
            curline += " ";
          } else if (cur_mutex == nullptr) {
            curline += " ";
          }
          curline += arg_inline_usage;
        }
        if (cur_mutex != nullptr) {
          curline += ']';
        }
        return found_options;
      };

      if (const bool found_options = deal_with_options_of_group(0); found_options && multiline_usage &&
          !this->m_positional_arguments.empty()) {
        result += std::format("\n{}", curline);
        curline = String(indent_size, ' ');
      }

      for (const Argument& argument : this->m_positional_arguments) {
        if (argument.m_is_hidden)
          continue;

        const String pos_arg = !argument.m_metavar.empty()
          ? argument.m_metavar
          : argument.m_names.front();

        if (curline.size() + 1 + pos_arg.size() > this->m_usage_max_line_width) {
          result += std::format("\n{}", curline);
          curline = String(indent_size, ' ');
        }

        curline += " ";

        if (argument.m_num_args_range.get_min() == 0 &&
            !argument.m_num_args_range.is_right_bounded()) {
          curline += "[";
          curline += pos_arg;
          curline += "]...";
        } else if (argument.m_num_args_range.get_min() == 1 &&
                   !argument.m_num_args_range.is_right_bounded()) {
          curline += pos_arg;
          curline += "...";
        } else
          curline += pos_arg;
      }

      if (multiline_usage)
        for (usize i = 0; i < m_group_names.size(); ++i) {
          result += std::format("\n\n{}:\n", m_group_names[i]);
          curline = String(indent_size, ' ');
          deal_with_options_of_group(i + 1);
        }

      result += curline;

      if (!m_subparser_map.empty()) {
        result += " {";
        usize i { 0 };
        for (const auto& [command, subparser] : m_subparser_map) {
          if (subparser->get().m_suppress)
            continue;

          if (i == 0)
            result += command;
          else
            result += std::format(",{}", command);

          ++i;
        }
        result += "}";
      }

      return result;
    }

    fn add_subparser(ArgumentParser& parser) -> void {
      parser.m_parser_path = m_program_name + " " + parser.m_program_name;

      auto it = m_subparsers.emplace(std::cend(m_subparsers), parser);

      m_subparser_map.insert_or_assign(parser.m_program_name, it);
      m_subparser_used.insert_or_assign(parser.m_program_name, false);
    }

    fn set_suppress(const bool suppress) -> void {
      m_suppress = suppress;
    }

   protected:
    fn get_belonging_mutex(const Argument* arg) const -> const MutuallyExclusiveGroup* {
      for (const MutuallyExclusiveGroup& mutex : m_mutually_exclusive_groups)
        if (std::ranges::find(mutex.m_elements, arg) !=
            mutex.m_elements.end())
          return &mutex;

      return nullptr;
    }

    [[nodiscard]] fn is_valid_prefix_char(const char c) const -> bool {
      return m_prefix_chars.contains(c);
    }

    [[nodiscard]] fn get_any_valid_prefix_char() const -> char {
      return m_prefix_chars[0];
    }

    /*
     * Pre-process this argument list. Anything starting with "--", that
     * contains an =, where the prefix before the = has an entry in the
     * options table, should be split.
     */
    [[nodiscard]] fn preprocess_arguments(const Vec<String>& raw_arguments) const -> Vec<String> {
      Vec<String> arguments {};
      for (const String& arg : raw_arguments) {
        const auto argument_starts_with_prefix_chars =
          [this](const String& a) -> bool {
          if (!a.empty()) {
            // Windows-style
            // if '/' is a legal prefix char
            // then allow single '/' followed by argument name, followed by an
            // assign char, e.g., ':' e.g., 'test.exe /A:Foo'
            if (is_valid_prefix_char('/')) {
              if (is_valid_prefix_char(a[0]))
                return true;
            } else
              // Slash '/' is not a legal prefix char
              // For all other characters, only support long arguments
              // i.e., the argument must start with 2 prefix chars, e.g,
              // '--foo' e,g, './test --foo=Bar -DARG=yes'
              if (a.size() > 1)
                return (is_valid_prefix_char(a[0]) && is_valid_prefix_char(a[1]));
          }

          return false;
        };

        // Check that:
        // - We don't have an argument named exactly this
        // - The argument starts with a prefix char, e.g., "--"
        // - The argument contains an assign char, e.g., "="

        if (const usize assign_char_pos = arg.find_first_of(m_assign_chars); !m_argument_map.contains(arg) &&
            argument_starts_with_prefix_chars(arg) &&
            assign_char_pos != String::npos)
          // Get the name of the potential option, and check it exists
          if (String opt_name = arg.substr(0, assign_char_pos); m_argument_map.contains(opt_name)) {
            // This is the name of an option! Split it into two parts
            arguments.push_back(std::move(opt_name));
            arguments.push_back(arg.substr(assign_char_pos + 1));
            continue;
          }

        // If we've fallen through to here, then it's a standard argument
        arguments.push_back(arg);
      }

      return arguments;
    }

    /*
     * @throws std::runtime_error in case of any invalid argument
     */
    // NOLINTNEXTLINE(misc-no-recursion)
    fn parse_args_internal(const Vec<String>& raw_arguments) -> Result<void> {
      Vec<String> arguments = preprocess_arguments(raw_arguments);

      if (m_program_name.empty() && !arguments.empty())
        m_program_name = arguments.front();

      auto end                    = std::end(arguments);
      auto positional_argument_it = std::begin(m_positional_arguments);

      for (auto it = std::next(std::begin(arguments)); it != end;) {
        const String& current_argument = *it;
        if (Argument::is_positional(current_argument, m_prefix_chars)) {
          if (positional_argument_it == std::end(m_positional_arguments)) {
            if (const auto subparser_it = m_subparser_map.find(current_argument); subparser_it != m_subparser_map.end()) {
              const Vec<String> unprocessed_arguments = Vec<String>(it, end);
              m_is_parsed                             = true;
              m_subparser_used[current_argument]      = true;
              // parse_args on subparser will be void, or we change it to Result<void>
              // Assuming subparser itself handles its errors internally by throwing/exiting, or needs refactor too.
              // For now, let's assume it might throw, or we adapt it later.
              // If subparser.parse_args itself returns Result<void>, we'd check it.
              // This is recursive, so sub-parsers also need this refactoring.
              Result<void> sub_parse_res = subparser_it->second->get().parse_args_internal(unprocessed_arguments);
              if (!sub_parse_res)
                return sub_parse_res;
              return {};
            }

            if (m_positional_arguments.empty()) {
              if (!m_subparser_map.empty())
                return Err(DracError(DracErrorCode::InvalidArgument, std::format("Failed to parse '{}', did you mean '{}'", current_argument, details::get_most_similar_string(m_subparser_map, current_argument))));
              if (!m_optional_arguments.empty()) {
                for (const Argument& opt : m_optional_arguments) {
                  if (!opt.m_implicit_value.has_value()) {
                    if (!opt.m_is_used) {
                      return Err(DracError(DracErrorCode::InvalidArgument, std::format("Zero positional arguments expected, did you mean '{}'", opt.get_usage_full())));
                    }
                  }
                }
                return Err(DracError(DracErrorCode::InvalidArgument, "Zero positional arguments expected"));
              }
              return Err(DracError(DracErrorCode::InvalidArgument, "Zero positional arguments expected"));
            }
            return Err(DracError(DracErrorCode::InvalidArgument, std::format("Maximum number of positional arguments exceeded, failed to parse '{}'", current_argument)));
          }

          const auto argument_ptr = positional_argument_it++;
          if (argument_ptr->m_num_args_range.get_min() == 1 &&
              argument_ptr->m_num_args_range.get_max() == (std::numeric_limits<usize>::max)() &&
              positional_argument_it != std::end(m_positional_arguments) &&
              std::next(positional_argument_it) == std::end(m_positional_arguments) &&
              positional_argument_it->m_num_args_range.get_min() == 1 &&
              positional_argument_it->m_num_args_range.get_max() == 1) {
            if (std::next(it) != end) {
              Result<decltype(end)> consume_res = positional_argument_it->consume(std::prev(end), end);
              if (!consume_res)
                return Err(consume_res.error());
              end = std::prev(end);
            } else
              return Err(DracError(DracErrorCode::InvalidArgument, std::format("Missing {}", positional_argument_it->m_names.front())));
          }

          Result<decltype(it)> consume_result = argument_ptr->consume(it, end);
          if (!consume_result)
            return Err(consume_result.error());
          it = consume_result.value();
          continue;
        }

        auto arg_map_it = m_argument_map.find(current_argument);
        if (arg_map_it != m_argument_map.end()) {
          const auto           argument_iter  = arg_map_it->second;
          Result<decltype(it)> consume_result = argument_iter->consume(std::next(it), end, arg_map_it->first);
          if (!consume_result)
            return Err(consume_result.error());
          it = consume_result.value();
        } else if (const String& compound_arg = current_argument;
                   compound_arg.size() > 1 &&
                   is_valid_prefix_char(compound_arg[0]) &&
                   !is_valid_prefix_char(compound_arg[1])) {
          ++it;
          for (usize j = 1; j < compound_arg.size(); j++) {
            const String hypothetical_arg = { '-', compound_arg[j] };
            auto         arg_map_it2      = m_argument_map.find(hypothetical_arg);
            if (arg_map_it2 != m_argument_map.end()) {
              const auto           argument_iter2 = arg_map_it2->second;
              Result<decltype(it)> consume_result = argument_iter2->consume(it, end, arg_map_it2->first);
              if (!consume_result)
                return Err(consume_result.error());
              it = consume_result.value();
            } else
              return Err(DracError(DracErrorCode::InvalidArgument, std::format("Unknown argument: {} in compound {}", hypothetical_arg, current_argument)));
          }
        } else
          return Err(DracError(DracErrorCode::InvalidArgument, std::format("Unknown argument: {}", current_argument)));
      }
      m_is_parsed = true;
      return {};
    }

    [[nodiscard]] fn get_length_of_longest_argument() const -> usize {
      if (m_argument_map.empty())
        return 0;

      usize max_size = 0;

      for (const auto& argument : m_argument_map | std::views::values)
        max_size =
          std::max<usize>(max_size, argument->get_arguments_length());

      for (const String& command : m_subparser_map | std::views::keys)
        max_size = std::max<usize>(max_size, command.size());

      return max_size;
    }

    using argument_it    = std::list<Argument>::iterator;
    using mutex_group_it = Vec<MutuallyExclusiveGroup>::iterator;
    using argument_parser_it =
      std::list<std::reference_wrapper<ArgumentParser>>::iterator;

    fn index_argument(argument_it it) -> void {
      for (const String& name : std::as_const(it->m_names))
        m_argument_map.insert_or_assign(name, it);
    }

   private:
    String                                            m_program_name;
    String                                            m_version;
    String                                            m_description;
    String                                            m_epilog;
    bool                                              m_exit_on_default_arguments = true;
    String                                            m_prefix_chars { "-" };
    String                                            m_assign_chars { "=" };
    bool                                              m_is_parsed = false;
    std::list<Argument>                               m_positional_arguments;
    std::list<Argument>                               m_optional_arguments;
    Map<String, argument_it>                          m_argument_map;
    String                                            m_parser_path;
    std::list<std::reference_wrapper<ArgumentParser>> m_subparsers;
    Map<String, argument_parser_it>                   m_subparser_map;
    Map<String, bool>                                 m_subparser_used;
    Vec<MutuallyExclusiveGroup>                       m_mutually_exclusive_groups;
    bool                                              m_suppress              = false;
    usize                                             m_usage_max_line_width  = (std::numeric_limits<usize>::max)();
    bool                                              m_usage_break_on_mutex  = false;
    int                                               m_usage_newline_counter = 0;
    Vec<String>                                       m_group_names;
  };
} // namespace argparse

// NOLINTEND(readability-identifier-naming, readability-identifier-length, modernize-use-nullptr)