config_param.h

#ifndef CONFIG_PARAM_H
#define CONFIG_PARAM_H


#include <string>
#include <list>
#include <map>

#include "simpleini/SimpleIni.h"
#include "sniff.h"
#include "tools.h"


using namespace std;


class cConfigItem {
public:
	struct sMapValue {
		sMapValue(string str, int value) {
			this->str = str;
			this->value = value;
		}
		string str;
		int value;
	};
	enum eTypeLevel {
		levelNormal,
		levelAdvanced,
		levelExpert,
		levelObsolete
	};
public:
	cConfigItem(const char *name);
	virtual ~cConfigItem() {}
	cConfigItem *addAlias(const char *name_alias);
	cConfigItem *setDefaultValueStr(const char *defaultValueStr);
	cConfigItem *setNaDefaultValueStr();
	cConfigItem *setMinor();
	cConfigItem *setMinorGroupIfNotSet();
	cConfigItem *setReadOnly();
	cConfigItem *setDisableIf(const char *disableIf);
	cConfigItem *setAlwaysShow();
	void setConfigFileSection(const char *config_file_section);
	cConfigItem *addValue(const char *str, int value);
	cConfigItem *addValues(const char *str_values);
	cConfigItem *addStringItem(const char *str);
	cConfigItem *addStringItems(const char *str_values);
	cConfigItem *setSubtype(const char *subtype);
	cConfigItem *setDescription(const char *description);
	cConfigItem *setHelp(const char *help);
	virtual string getValueStr(bool /*configFile*/ = false) { return(""); }
	virtual int64_t getValueInt() { return(0); }
	virtual list<string> getValueListStr() { list<string> l; l.push_back(getValueStr()); return(l); }
	virtual string normalizeStringValueForCmp(string value) { return(value); }
	virtual bool enableMultiValues() { return(false); }
protected:
	virtual bool setParamFromConfigFile(CSimpleIniA *ini) = 0;
	virtual bool setParamFromValueStr(string value_str) = 0;
	virtual bool setParamFromValuesStr(vector<string> /*list_value_str*/) { return(false); }
	string getValueFromConfigFile(CSimpleIniA *ini);
	vector<string> getValuesFromConfigFile(CSimpleIniA *ini);
	bool getValueFromMapValues(const char *str_value, int *rslt_value);
	string getStringFromMapValues(int value);
	void init();
	virtual void initParamPointers() {}
	virtual void initOther() {}
	virtual void initVirtParam() {}
	virtual void initBeforeSet() {}
	virtual string getTypeName() = 0;
	virtual list<sMapValue> getMenuItems();
	void addItemToMenuItems(list<sMapValue> *menu, sMapValue menuItem);
	string getJson();
	void setDefaultValue();
	void clearToDefaultValue();
protected:
	string config_name;
	list<string> config_name_alias;
	string config_file_section;
	eTypeLevel level;
	string group_name;
	string subgroup_name;
	string subtype;
	string description;
	string help;
	class cConfig *config;
	list<sMapValue> mapValues;
	bool set;
	string defaultValueStr;
	bool defaultValueStr_set;
	bool naDefaultValueStr;
	bool minor;
	bool minorGroupIfNotSet;
	bool readOnly;
	string disableIf;
	bool alwaysShow;
friend class cConfig;
};

class cConfigItem_yesno : public cConfigItem {
public:
	cConfigItem_yesno(const char *name, bool *param);
	cConfigItem_yesno(const char *name, int *param = NULL);
	cConfigItem_yesno *disableYes() {
		disable_yes = true;
		return(this);
	}
	cConfigItem_yesno *disableNo() {
		disable_no = true;
		return(this);
	}
	cConfigItem_yesno *setNeg() {
		neg = true;
		return(this);
	}
	int getValue();
	string getValueStr(bool configFile = false);
	int64_t getValueInt() { return(getValue()); }
	string normalizeStringValueForCmp(string value);
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	void initParamPointers() {
		param_bool = NULL;
		param_int = NULL;
	}
	void initOther() {
		disable_yes = false;
		disable_no =  false;
		neg = false;
	}
	void initVirtParam() {
		param_virt = 0;
	}
	string getTypeName() {
		return("yesno");
	}
	list<sMapValue> getMenuItems();
protected:
	bool *param_bool;
	int *param_int;
	int param_virt;
	bool disable_yes;
	bool disable_no;
	bool neg;
};

class cConfigItem_integer : public cConfigItem {
public:
	cConfigItem_integer(const char *name, int *param);
	cConfigItem_integer(const char *name, unsigned int *param);
	cConfigItem_integer(const char *name, uint64_t *param);
	cConfigItem_integer(const char *name, int64_t *param = NULL);
	cConfigItem_integer *setIp() {
		ip = true;
		return(this);
	}
	cConfigItem_integer *setMaximum(int maximum) {
		this->maximum = maximum;
		return(this);
	}
	cConfigItem_integer *setMinimum(int minimum) {
		this->minimum = minimum;
		return(this);
	}
	cConfigItem_integer *setIfZeroOrNegative(int ifZeroOrNegative) {
		this->ifZeroOrNegative = ifZeroOrNegative;
		return(this);
	}
	cConfigItem_integer *setMultiple(double multiple) {
		this->multiple = multiple;
		return(this);
	}
	cConfigItem_integer *setYes(int yesValue = 1) {
		this->yesValue = yesValue;
		return(this);
	}
	cConfigItem_integer *setMenuValue() {
		this->menuValue = true;
		return(this);
	}
	cConfigItem_integer *setOnlyMenu() {
		this->onlyMenu = true;
		return(this);
	}
	int64_t getValue();
	string getValueStr(bool configFile = false);
	int64_t getValueInt() { return(getValue()); }
	string normalizeStringValueForCmp(string value);
	int getMaximum() {
		return(maximum);
	}
	int getMinimum() {
		return(minimum);
	}
	bool isMenuValue() {
		return(menuValue);
	}
	bool isOnlyMenu() {
		return(onlyMenu);
	}
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	void initParamPointers() {
		param_int = NULL;
		param_uint = NULL;
		param_int64 = NULL;
		param_uint64 = NULL;
	}
	void initOther() {
		ip = false;
		maximum = 0;
		minimum = 0;
		ifZeroOrNegative = 0;
		multiple = 0;
		yesValue = 0;
		menuValue = false;
		onlyMenu = false;
	}
	void initVirtParam() {
		param_virt = 0;
	}
	string getTypeName() {
		return(ip ? "ip" : "integer");
	}
protected:
	int *param_int;
	unsigned int *param_uint;
	int64_t *param_int64;
	uint64_t *param_uint64;
	int64_t param_virt;
	bool ip;
	int maximum;
	int minimum;
	int ifZeroOrNegative;
	double multiple;
	int yesValue;
	bool menuValue;
	bool onlyMenu;
};

class cConfigItem_float : public cConfigItem {
public:
	cConfigItem_float(const char *name, float *param);
	cConfigItem_float(const char *name, double *param);
	double getValue();
	string getValueStr(bool configFile = false);
	string normalizeStringValueForCmp(string value);
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	void initParamPointers() {
		param_float = NULL;
		param_double = NULL;
	}
	string getTypeName() {
		return("float");
	}
protected:
	float *param_float;
	double *param_double;
};

class cConfigItem_string : public cConfigItem {
public:
	cConfigItem_string(const char *name, string *param = NULL);
	cConfigItem_string(const char *name, char *param, int length);
	cConfigItem_string(const char *name, vector<string> *param);
	cConfigItem_string *setPrefix(const char *prefix) {
		this->prefix = prefix;
		return(this);
	}
	cConfigItem_string *setSuffix(const char *suffix) {
		this->suffix = suffix;
		return(this);
	}
	cConfigItem_string *setPrefixSuffix(const char *prefix, const char *suffix) {
		this->prefix = prefix;
		this->suffix = suffix;
		return(this);
	}
	cConfigItem_string *setExplodeSeparator(const char *explodeSeparator) {
		this->explodeSeparator = explodeSeparator;
		return(this);
	}
	cConfigItem_string *setPassword() {
		this->password = true;
		return(this);
	}
	string getValue();
	string getValueStr(bool configFile = false);
	list<string> getValueListStr();
	string normalizeStringValueForCmp(string value);
	bool enableMultiValues();
	bool isPassword() {
	       return(password);
	}
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	bool setParamFromValuesStr(vector<string> list_values_str);
	void initBeforeSet();
	void initParamPointers() {
		param_str = NULL;
		param_strchar = NULL;
		param_strchar_length = 0;
		param_vect_str = NULL;
	}
	void initOther() {
		prefix = "";
		suffix = "";
		explodeSeparator = ";";
		password = false;
	}
	void initVirtParam() {
		param_virt = "";
	}
	string getTypeName() {
		return(param_vect_str ? "string_list" : "string");
	}
protected:
	string *param_str;
	char *param_strchar;
	int param_strchar_length;
	string param_virt;
	vector<string> *param_vect_str;
	string prefix;
	string suffix;
	string explodeSeparator;
	bool password;
};

class cConfigItem_hour_interval : public cConfigItem {
public:
	cConfigItem_hour_interval(const char *name, int *from, int *to);
	string getValueStr(bool configFile = false);
	string normalizeStringValueForCmp(string value);
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	void initParamPointers() {
		param_from = NULL;
		param_to = NULL;
	}
	string getTypeName() {
		return("hour_interval");
	}
protected:
	int *param_from;
	int *param_to;
};

class cConfigItem_ports : public cConfigItem {
public:
	cConfigItem_ports(const char* name, char *port_matrix);
	string getValueStr(bool configFile = false);
	list<string> getValueListStr();
	bool enableMultiValues() { return(true); }
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	bool setParamFromValuesStr(vector<string> list_value_str);
	void initBeforeSet();
	void initParamPointers() {
		param_port_matrix = NULL;
	}
	string getTypeName() {
		return("ports");
	}
protected:
	char *param_port_matrix;
};

class cConfigItem_hosts : public cConfigItem {
public:
	cConfigItem_hosts(const char* name, vector<u_int32_t> *adresses, vector<d_u_int32_t> *nets);
	string getValueStr(bool configFile = false);
	list<string> getValueListStr();
	bool enableMultiValues() { return(true); }
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	bool setParamFromValuesStr(vector<string> list_value_str);
	void initBeforeSet();
	void initParamPointers() {
		param_adresses = NULL;
		param_nets = NULL;
	}
	string getTypeName() {
		return("hosts");
	}
protected:
	vector<u_int32_t> *param_adresses;
	vector<d_u_int32_t> *param_nets;
};

class cConfigItem_ip_port : public cConfigItem {
public:
	cConfigItem_ip_port(const char* name, ip_port *param);
	ip_port getValue();
	string getValueStr(bool configFile = false);
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	void initParamPointers() {
		param_ip_port = NULL;
	}
	string getTypeName() {
		return("ip_port");
	}
protected:
	ip_port *param_ip_port;
};

class cConfigItem_ip_ports : public cConfigItem {
public:
	cConfigItem_ip_ports(const char* name, vector<ip_port> *param);
	cConfigItem_ip_ports(const char* name, vector<ipn_port> *param);
	string getValueStr(bool configFile = false);
	list<string> getValueListStr();
	bool enableMultiValues() { return(true); }
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	bool setParamFromValuesStr(vector<string> list_value_str);
	void initBeforeSet();
	void initParamPointers() {
		param_ip_ports = NULL;
		param_ipn_ports = NULL;
	}
	string getTypeName() {
		return("ip_port_list");
	}
protected:
	vector<ip_port> *param_ip_ports;
	vector<ipn_port> *param_ipn_ports;
};

class cConfigItem_ip_port_str_map : public cConfigItem {
public:
	cConfigItem_ip_port_str_map(const char* name, map<d_u_int32_t, string> *ip_port_string_map);
	string getValueStr(bool configFile = false);
	list<string> getValueListStr();
	string normalizeStringValueForCmp(string value);
	bool enableMultiValues() { return(true); }
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	bool setParamFromValuesStr(vector<string> list_value_str);
	void initBeforeSet();
	void initParamPointers() {
		param_ip_port_string_map = NULL;
	}
	string getTypeName() {
		return("ip_port_str_list");
	}
protected:
	map<d_u_int32_t, string> *param_ip_port_string_map;
};

class cConfigItem_nat_aliases : public cConfigItem {
public:
	cConfigItem_nat_aliases(const char* name, nat_aliases_t *nat_aliases);
	string getValueStr(bool configFile = false);
	list<string> getValueListStr();
	string normalizeStringValueForCmp(string value);
	bool enableMultiValues() { return(true); }
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	bool setParamFromValuesStr(vector<string> list_value_str);
	void initBeforeSet();
	void initParamPointers() {
		param_nat_aliases = NULL;
	}
	string getTypeName() {
		return("nat_aliases_list");
	}
protected:
	nat_aliases_t *param_nat_aliases;
};

class cConfigItem_custom_headers : public cConfigItem {
public:
	cConfigItem_custom_headers(const char* name, vector<dstring> *custom_headers);
	string getValueStr(bool configFile = false);
	string normalizeStringValueForCmp(string value);
protected:
	bool setParamFromConfigFile(CSimpleIniA *ini);
	bool setParamFromValueStr(string value_str);
	void initBeforeSet();
	void initParamPointers() {
		param_custom_headers = NULL;
	}
	string getTypeName() {
		return("custom_headers_list");
	}
protected:
	vector<dstring> *param_custom_headers;
};

class cConfigItem_type_compress : public cConfigItem_yesno {
public:
	cConfigItem_type_compress(const char* name, CompressStream::eTypeCompress *type_compress = NULL);
	cConfigItem_type_compress(const char* name, FileZipHandler::eTypeCompress *type_compress = NULL);
};


class cConfigMap {
public:
	struct cItem {
		list<string> values;
		void add(const char *value) {
			values.push_back(value);
		}
		string valuesToStr();
		bool operator == (cItem& other) { 
			if(this->values.size() != other.values.size()) {
				return(false);
			}
			this->values.sort();
			other.values.sort();
			list<string>::iterator iter1 = this->values.begin();
			list<string>::iterator iter2 = other.values.begin();
			while(iter1 != this->values.end() && iter2 != other.values.end()) {
				if(*iter1 != *iter2) {
					return(false);
				}
				++iter1;
				++iter2;
			}
			return(true);
		}
	};
public:
	void addItem(const char *name, const char *value);
	bool existsItem(const char *name);
	string getFirstItem(const char *name, bool toLower = false);
	string getItems(const char *name, const char *separator = ";", bool toLower = false);
	string comp(cConfigMap *other, class cConfig *config);
public:
	map<string, cItem> config_map;
};


class cConfig {
public:
	cConfig();
	~cConfig();
	void addConfigItems();
	void addConfigItem(cConfigItem *configItem);
	void group(const char *groupName);
	void subgroup(const char *subgroupName);
	void normal();
	void advanced();
	void expert();
	void obsolete();
	void minorBegin();
	void minorEnd();
	void minorGroupIfNotSetBegin();
	void minorGroupIfNotSetEnd();
	void setDisableIfBegin(string disableIf);
	void setDisableIfEnd();
	bool loadFromConfigFileOrDirectory(const char *filename, bool silent = false);
	bool loadFromConfigFile(const char *filename, string *error = NULL, bool silent = false);
	bool loadConfigMapConfigFileOrDirectory(cConfigMap *configMap, const char *filename);
	bool loadConfigMapFromConfigFile(cConfigMap *configMap, const char *filename);
	void evSetConfigItem(cConfigItem *configItem);
	cConfigMap getConfigMap();
	string getContentConfig(bool configFile = false, bool putDefaultValues = false);
	string getJson(bool onlyIfSet = false);
	void setFromJson(const char *jsonStr, bool onlyIfSet = false);
	void setFromMysql(bool checkConnect = false);
	void putToMysql();
	void setDefaultValues();
	void clearToDefaultValues();
	void setDescription(const char *itemName, const char *description);
	void setHelp(const char *itemName, const char *help);
	string getMainItemName(const char *name);
	bool testEqValues(const char *itemName, const char *value1, const char *value2);
	bool testEqValues(string itemName, list<string> values1, list<string> values2);
private:
	void loadFromConfigFileError(const char *errorString, const char *filename, string *error = NULL);
private:
	list<string> config_list;
	map<string, cConfigItem*> config_map;
	cConfigItem::eTypeLevel defaultLevel;
	string defaultGroup;
	string defaultSubgroup;
	bool defaultMinor;
	bool defaultMinorGroupIfNotSet;
	string defaultDisableIf;
};


#endif