#pragma once
#include "cmd/args/arg_generic.hpp"
#include "cmd/args/arg_input.hpp"
#include "util/input.hpp"

namespace fsh{

class ArgManager {
    friend class ArgFactory;

    using PosArgs = std::vector<std::shared_ptr<_Argument> >;
    using FlagOpts = std::unordered_map< std::string ,std::shared_ptr<_Argument> >;

public:
    ArgManager() {}


    std::istream& get_input(const unsigned int id) {
        if(id < pos_argument.size()) return ArgInput::get(pos_argument[id]);
        return util::cin;
    }

    template<typename T>
    std::optional<T> get(const unsigned int id) {
        if((unsigned int) id < pos_argument.size()) return Argument<T>::get(pos_argument[id]);
        return std::make_optional<T>();
    }
    template<typename T>
    std::optional<T> get(const std::string& id) {
        if(flags.find(id) != flags.end()) return Argument<T>::get(flags[id]);
        return std::make_optional<T>();
    }
    bool get(const std::string& id) {
        return flags.find(id) != flags.end();
    }

    void push_arg(std::shared_ptr<_Argument> arg) {
        pos_argument.push_back(arg);
    }
    void push_flag(const std::string &id,std::shared_ptr<_Argument> arg = nullptr) {
        flags[id] = arg;
    }
    
    private:
    PosArgs pos_argument;
    FlagOpts flags;
};
}