implemented command-line and config-file options

main
Michael Mandl 2018-03-30 19:44:53 +02:00
parent c904e08ded
commit bd4cbb6872
4 changed files with 105 additions and 1 deletions

View File

@ -1,7 +1,28 @@
#include "settings.h"
#include <iostream> #include <iostream>
int main(int, char **) int main(int argc, char **argv)
{ {
std::cout << "BeezleBot running" << std::endl; try
{
Settings settings(argc, argv);
std::cout << "token: " << settings.token << std::endl;
std::cout << "users: " << std::flush;
for (const auto &user : settings.users)
{
std::cout << user << " ";
}
std::cout << std::endl;
}
catch (std::runtime_error &e)
{
return EXIT_FAILURE;
}
std::cout << "BeezleBot running" << std::endl;
return EXIT_SUCCESS;
} }

View File

@ -2,8 +2,16 @@ cmake_minimum_required(VERSION 3.5)
project(BeezleBot LANGUAGES CXX) project(BeezleBot LANGUAGES CXX)
find_package(Boost 1.60.0 COMPONENTS program_options REQUIRED)
add_executable(${PROJECT_NAME} add_executable(${PROJECT_NAME}
BeezleBot.cpp BeezleBot.cpp
settings.cpp
)
target_link_libraries(${PROJECT_NAME}
PRIVATE
Boost::program_options
) )
target_compile_features(${PROJECT_NAME} target_compile_features(${PROJECT_NAME}

62
BeezleBot/settings.cpp Normal file
View File

@ -0,0 +1,62 @@
#include "settings.h"
#include <boost/program_options.hpp>
#include <iostream>
#include <fstream>
#include <stdexcept>
Settings::Settings(int argc, char **argv)
{
namespace po = boost::program_options;
po::options_description commandlineOptions("Allowed options");
commandlineOptions.add_options()
("help,h", "show this help message")
("config-file,f", po::value<std::string>(), "read options from config file")
("token,t", po::value<std::string>(&token)->required(), "Telegram bot token")
("users,u", po::value<std::vector<std::string>>(&users)->multitoken()->required(), "allowed Telegram users")
;
po::variables_map configuredOptions;
po::store(po::command_line_parser(argc, argv).options(commandlineOptions).run(),
configuredOptions);
if (configuredOptions.find("help") != configuredOptions.cend())
{
std::cout << commandlineOptions << std::endl;
throw std::runtime_error("invalid options");
}
if (configuredOptions.find("config-file") != configuredOptions.cend())
{
std::ifstream configFile(configuredOptions["config-file"].as<std::string>());
po::options_description fileOptions;
fileOptions.add_options()
("token", po::value<std::string>(&token))
("users", po::value<std::vector<std::string>>(&users)->multitoken())
;
try
{
po::store(po::parse_config_file(configFile, fileOptions), configuredOptions);
}
catch (po::unknown_option &e)
{
std::cout << "Error: " << e.what() << std::endl;
throw std::runtime_error("invalid options");
}
}
try
{
po::notify(configuredOptions);
}
catch (po::required_option &e)
{
std::cout << "Error: " << e.what() << std::endl;
std::cout << commandlineOptions << std::endl;
throw std::runtime_error("invalid options");
}
}

13
BeezleBot/settings.h Normal file
View File

@ -0,0 +1,13 @@
#pragma once
#include <string>
#include <vector>
struct Settings
{
std::string token;
std::vector<std::string> users;
Settings(int argc, char **argv);
};