parser_options.cpp

Go to the documentation of this file.

#include <arkanjo/cli/parser_options.hpp>
 
#include <iostream>
#include <memory>
 
static option* build_longopts(const std::vector<CliOption>& options) {
    size_t n = options.size();
    option* opts = new option[n + 1];
 
    size_t j = 0;
    for (size_t i = 0; i < n; ++i) {
        if (options[i].has_arg == PositionalArgument) continue;
        
        opts[j].name = options[i].long_name;
        opts[j].has_arg = options[i].has_arg;
        opts[j].flag = nullptr;
        opts[j].val = options[i].short_name;
        j++;
    }
 
    opts[j] = {nullptr,0,nullptr,0};
 
    return opts;
}
 
std::string build_shortopts(const std::vector<CliOption>& options) {
    std::string shortopts;
 
    for (const auto& opt : options) {
        if (opt.has_arg == PositionalArgument) continue;
 
        if (opt.short_name > 0) {
            shortopts += static_cast<char>(opt.short_name);
 
            if (opt.has_arg == required_argument)
                shortopts += ":";
            else if (opt.has_arg == optional_argument)
                shortopts += "::";
        }
    }
 
    return shortopts;
}
 
bool parse_options(int argc, char* argv[], const std::vector<CliOption>& options, ParsedOptions& ctx_options) {
    ctx_options.extra_args.clear();
 
    int option_index = 0;
    int opt;
 
    std::unique_ptr<option[]> long_opts(build_longopts(options));
    std::string shortopts = build_shortopts(options);
 
    while ((opt = getopt_long(argc, argv, shortopts.c_str(), long_opts.get(), &option_index)) != -1) {
        if (opt == '?') {
            std::string cerr = "invalid option: ";
            if (optopt) {
                cerr += "-";
                cerr += char(optopt);
            } else {
                cerr += argv[optind - 1];
            }
            cerr += "\nTry '--help' for usage.";
            throw CLIError(cerr);
            return false;
        }
 
        std::string name;
 
        if (opt == 0) { // long option
            name = long_opts[option_index].name;
        } else { // short option
            name = std::string(1, static_cast<char>(opt));
 
            for (size_t i = 0; long_opts[i].name != nullptr; i++) {
                if (long_opts[i].val == opt) {
                    name = long_opts[i].name;
                    break;
                }
            }
        }
 
        ctx_options.args[name] = (optarg ? optarg : "");
    }
 
    for (int i = optind; i < argc; i++)
        ctx_options.extra_args.push_back(argv[i]);
 
#ifdef DEBUG
    std::cout << "[DEBUG] Parsed options:\n";
    for (auto& kv : ctx_options.args)
        std::cout << kv.first << " = " << kv.second << "\n";
    for (auto& a : ctx_options.extra_args)
        std::cout << "[DEBUG] Extra arg: " << a << "\n";
#endif
 
    return true;
}