CliComponent.hpp

Go to the documentation of this file.

#pragma once
#ifndef _CONSOLIX_CLI_COMPONENT_HPP_INCLUDED
#define _CONSOLIX_CLI_COMPONENT_HPP_INCLUDED

 
#if CONSOLIX_USE_CXXOPTS == 1
 
#include <cxxopts.hpp>
#include <string>
#include <memory>
 
namespace consolix {

    class CliComponent : public IAppComponent {
    public:

        CliComponent(
                const std::string& name,
                const std::string& description,
                std::function<void(CliOptions&)> creator)
            : m_creator(std::move(creator)) {
            register_options_service(name, description);
        }

        CliComponent(
                const std::string& name,
                const std::string& description,
                std::function<void(CliOptions&)> creator,
                int argc,
                const char* argv[])
            : m_creator(std::move(creator)) {
            register_options_service(name, description);
            store_argv(argc, argv);
        }

        CliComponent(
                const std::string& name,
                const std::string& description,
                std::function<void(CliOptions&)> creator,
                int argc,
                char* argv[])
            : m_creator(std::move(creator)) {
            register_options_service(name, description);
            store_argv(argc, argv);
        }

        template <typename T>
        void add_option(const std::string& key, const std::string& description) {
            get_service<CliOptions>().add_options()(key, description, cxxopts::value<T>());
        }

        template <typename T>
        void add_option(const std::string& key, const std::string& description, T default_value) {
            get_service<CliOptions>().add_options()(key, description, cxxopts::value<T>()->default_value(std::to_string(default_value)));
        }

        template <typename T>
        void add_option_implicit(const std::string& key, const std::string& description, T implicit_value) {
            get_service<CliOptions>().add_options()(key, description, cxxopts::value<T>()->implicit_value(std::to_string(implicit_value)));
        }

        void parse(int argc, const char* argv[]) {
            try {
                auto result = get_service<CliOptions>().parse(argc, argv);
                register_service<CliArguments>([&result](){
                    return std::make_shared<CliArguments>(result);
                });
            } catch (const std::exception& e) {
                CONSOLIX_STREAM() << "Error parsing command-line arguments: " << e.what() << std::endl;
#               if CONSOLIX_USE_LOGIT == 1
                LOGIT_PRINT_ERROR("Error parsing command-line arguments: ", e.what());
#               endif
                throw;
            }
        }

        template <typename T>
        T get(const std::string& key) const {
            if (!get_service<CliArguments>().count(key)) {
#               if CONSOLIX_USE_LOGIT == 1
                LOGIT_PRINT_ERROR("Option not found: ", key);
#               endif
                throw std::runtime_error("Option not found: " + key);
            }
            return get_service<CliArguments>()[key].as<T>();
        }
 
    protected:

        bool initialize() override {
            try {
                auto &options = get_service<CliOptions>();
                if (!m_creator) {
                    throw std::runtime_error("CLI options creator function is null. Ensure you provide a valid function to initialize CLI options.");
                }
                m_creator(options);
#               if defined(_WIN32) || defined(_WIN64)
                parse_windows();
#               else
                if (m_argc) {
                    parse(m_argc, m_argv_c.data());
                }
#               endif
            } catch (const std::exception& e) {
#               if CONSOLIX_USE_LOGIT == 1
                LOGIT_PRINT_ERROR("CLI component initialization failed: ", e.what());
                throw;
#               else
                CONSOLIX_STREAM() << "Error initializing CLI component: " << e.what() << std::endl;
                throw;
#               endif
            }
            m_is_init = true;
            return true;
        }

        bool is_initialized() const override {
            return m_is_init;
        }

        void process() override {}
 
    private:
        int m_argc = 0;                             
        std::vector<const char*> m_argv_c;          
        std::function<void(CliOptions&)> m_creator; 
        std::atomic<bool> m_is_init{false};         

        void register_options_service(const std::string& name, const std::string& description) {
            register_service<CliOptions>([&name, &description]() {
                return std::make_shared<CliOptions>(name, description);
            });
        }

        template <typename ArgvEntry>
        void store_argv(int argc, ArgvEntry argv[]) {
            m_argc = argc;
            for (int i = 0; i < argc; ++i) {
                m_argv_c.push_back(argv[i]);
            }
        }
 
#       if defined(_WIN32) || defined(_WIN64)
        void parse_windows() {
            int argc = 0;
            LPWSTR* argv_w = CommandLineToArgvW(GetCommandLineW(), &argc);
 
            std::vector<std::string> args;
            for (int i = 0; i < argc; ++i) {
                args.push_back(utf16_to_utf8(argv_w[i]));
            }
 
            // Convert to standard `char*` for cxxopts
            std::vector<const char*> argv_c;
            for (const auto& arg : args) {
                argv_c.push_back(arg.c_str());
            }
 
            parse(argc, argv_c.data());
        }
#       endif
 
    }; // CliComponent
 
}; // namespace consolix
 
#endif // #if CONSOLIX_USE_CXXOPTS == 1
 
#endif // _CONSOLIX_CLI_COMPONENT_HPP_INCLUDED