/*
 * libdatachannel streamer example
 * Copyright (c) 2020 Filip Klembara (in2core)
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; If not, see <http://www.gnu.org/licenses/>.
 */

#include "ArgParser.hpp"
#include <iostream>

ArgParser::ArgParser(std::vector<std::pair<std::string, std::string>> options, std::vector<std::pair<std::string, std::string>> flags) {
    for(auto option: options) {
        this->options.insert(option.first);
        this->options.insert(option.second);
        shortToLongMap.emplace(option.first, option.second);
        shortToLongMap.emplace(option.second, option.second);
    }
    for(auto flag: flags) {
        this->flags.insert(flag.first);
        this->flags.insert(flag.second);
        shortToLongMap.emplace(flag.first, flag.second);
        shortToLongMap.emplace(flag.second, flag.second);
    }
}

std::optional<std::string> ArgParser::toKey(std::string prefixedKey) {
    if (prefixedKey.find("--") == 0) {
        return prefixedKey.substr(2, prefixedKey.length());
    } else if (prefixedKey.find("-") == 0) {
        return prefixedKey.substr(1, prefixedKey.length());
    } else {
        return std::nullopt;
    }
}

bool ArgParser::parse(int argc, char **argv, std::function<bool (std::string, std::string)> onOption, std::function<bool (std::string)> onFlag) {
    std::optional<std::string> currentOption = std::nullopt;
    for(int i = 1; i < argc; i++) {
        std::string current = argv[i];
        auto optKey = toKey(current);
        if (!currentOption.has_value() && optKey.has_value() && flags.find(optKey.value()) != flags.end()) {
            auto check = onFlag(shortToLongMap.at(optKey.value()));
            if (!check) {
                return false;
            }
        } else if (!currentOption.has_value() && optKey.has_value() && options.find(optKey.value()) != options.end()) {
            currentOption = optKey.value();
        } else if (currentOption.has_value()) {
            auto check = onOption(shortToLongMap.at(currentOption.value()), current);
            if (!check) {
                return false;
            }
            currentOption = std::nullopt;
        } else {
            std::cerr << "Unrecognized option " << current << std::endl;
            return false;
        }
    }
    if (currentOption.has_value()) {
        std::cerr << "Missing value for " << currentOption.value() << std::endl;
        return false;
    }
    return true;
}