taskconfig.h

// Copyright (c) 2018 IIS (The Internet Foundation in Sweden)
// Written by Göran Andersson <initgoran@gmail.com>
 
#pragma once
 
#include <map>
#include <set>
#include <string>
#include <stdexcept>
#include <iostream>
#include <fstream>
 
class BadTaskConfig : public std::exception {
public:
    BadTaskConfig(const std::string &msg = "cannot read config file") :
        err_msg(msg) {
    }
    const char *what() const noexcept override;
    BadTaskConfig(const BadTaskConfig &old) : err_msg(old.err_msg) {
    }
    ~BadTaskConfig() noexcept override;
    std::string err_msg;
};
 
class TaskConfig {
public:
    TaskConfig() {}
 
    TaskConfig(std::istream &cfg_stream);
 
    TaskConfig(const std::string &cfg_text);
 
    void add(const std::string &key, const std::string &val);
 
    void set(const std::string &key, const std::string &val) {
        the_config.erase(key);
        add(key, val);
    }
 
    void erase(const std::string &key) {
        the_config.erase(key);
    }
 
    void setDefault(const std::string &key, const std::string &val) {
        if (the_config.find(key) == the_config.end())
            add(key, val);
    }
 
    void addLine(const std::string &line);
 
    std::multimap<std::string, std::string>::iterator begin() {
        return the_config.begin();
    }
 
    std::multimap<std::string, std::string>::iterator end() {
        return the_config.end();
    }
 
    std::multimap<std::string, std::string>::const_iterator begin() const {
        return the_config.begin();
    }
 
    std::multimap<std::string, std::string>::const_iterator end() const {
        return the_config.end();
    }
 
    void workerAttributes(const std::set<std::string> &attrs);
 
    static TaskConfig load(const std::string &filename);
 
    const std::multimap<std::string, std::string> &cfg() const {
        return the_config;
    }
 
    std::string value(const std::string &key) const;
 
    bool hasKey(const std::string &key) const {
        return the_config.find(key) != the_config.end();
    }
 
    std::pair<std::multimap<std::string, std::string>::const_iterator,
        std::multimap<std::string, std::string>::const_iterator>
        range(const std::string &key) const {
        return the_config.equal_range(key);
    }
 
    void openlog(std::ofstream &logger, bool append = false) const;
 
    std::set<std::string>
        parseList(const std::string &category = "whitelist") const;
 
    void parseArgs(int &argc, char **&argv);
 
    std::map<std::string, std::string>
        parseKeyVal(const std::string &category = "user") const;
 
    bool saveJsonToFile(const std::string &filename);
 
    static TaskConfig loadJsonFromFile(const std::string &filename);
 
private:
    void _load(std::istream &cfg_stream);
    std::multimap<std::string, std::string> the_config;
};
 
std::ostream &operator<<(std::ostream &out, const TaskConfig &tc);