forked from Wirlaburla/P3NG0
119 lines
3.1 KiB
C++
119 lines
3.1 KiB
C++
#ifndef CONFIG_H
|
|
#define CONFIG_H
|
|
#include <fstream>
|
|
#include <vector>
|
|
#include <stdlib.h>
|
|
|
|
class PengoConfig {
|
|
public:
|
|
PengoConfig() {
|
|
std::ifstream mconf("conf/pengobot.conf");
|
|
|
|
std::string line;
|
|
while (std::getline(mconf, line)) {
|
|
// Check if line is a comment line.
|
|
if (line.rfind("#", 0) != 0) {
|
|
std::string name = line.substr(0, line.find("="));
|
|
std::string value = line.substr(line.find("=")+1, line.length());
|
|
std::cout << name << ": \"" << value << "\"" << std::endl;
|
|
conf[name] = value;
|
|
}
|
|
}
|
|
|
|
for (auto const& mc : conf) {
|
|
std::string msgName;
|
|
if ((mc.first.substr(0, mc.first.find("_"))) == "msg") {
|
|
msgName = mc.first.substr(mc.first.find("_")+1, mc.first.length());
|
|
std::ifstream msglines(mc.second);
|
|
std::string line;
|
|
std::vector<std::string> lines;
|
|
while (std::getline(msglines, line)) {
|
|
if (line.rfind("#", 0) != 0) {
|
|
lines.push_back(line);
|
|
std::cout << msgName << "=" << line << std::endl;
|
|
}
|
|
}
|
|
messages[msgName] = lines;
|
|
}
|
|
}
|
|
|
|
if ((conf["worldfile"]).length() > 0) {
|
|
std::ifstream wrlds(conf["worldfile"]);
|
|
std::string wline;
|
|
while (std::getline(wrlds, wline)) {
|
|
if (wline.rfind("#", 0) != 0) {
|
|
std::string name = wline.substr(0, wline.find("="));
|
|
std::string value = wline.substr(wline.find("=")+1, wline.length());
|
|
std::cout << name << ": \"" << value << "\"" << std::endl;
|
|
worlds[name] = value;
|
|
}
|
|
}
|
|
}
|
|
|
|
if ((conf["responsefile"]).length() > 0) {
|
|
std::ifstream rspn(conf["responsefile"]);
|
|
std::string rline;
|
|
while (std::getline(rspn, rline)) {
|
|
if (rline.rfind("#", 0) != 0) {
|
|
std::string name = rline.substr(0, rline.find("="));
|
|
std::string value = rline.substr(rline.find("=")+1, rline.length());
|
|
std::cout << name << ": \"" << value << "\"" << std::endl;
|
|
responses[name] = value;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
PengoConfig operator = (PengoConfig *pc) { return *pc; };
|
|
|
|
std::string getValue(std::string name, std::string def) {
|
|
if (((std::string)conf[name]).length() > 0) return conf[name];
|
|
else return def;
|
|
}
|
|
|
|
int getInt(std::string name, int def) {
|
|
if (((std::string)conf[name]).length() > 0) return std::stoi(conf[name]);
|
|
else return def;
|
|
}
|
|
|
|
void setValue(std::string name, std::string value) {
|
|
conf[name] = value;
|
|
}
|
|
|
|
void setInt(std::string name, int value) {
|
|
conf[name] = std::to_string(value);
|
|
}
|
|
|
|
std::string getMessage(std::string type) {
|
|
return (messages[type])[(int)(rand() % ((messages[type]).size()))];
|
|
}
|
|
|
|
std::vector<std::string> getMessages(std::string type) {
|
|
return messages[type];
|
|
}
|
|
|
|
std::string getWorld(std::string name) {
|
|
return worlds[name];
|
|
}
|
|
|
|
std::map<std::string, std::string> getWorlds() {
|
|
return worlds;
|
|
}
|
|
|
|
std::string getReply(std::string name) {
|
|
return responses[name];
|
|
}
|
|
|
|
std::map<std::string, std::string> getResponses() {
|
|
return responses;
|
|
}
|
|
|
|
private:
|
|
std::map<std::string, std::string> conf;
|
|
std::map<std::string, std::string> worlds;
|
|
std::map<std::string, std::string> responses;
|
|
std::map<std::string, std::vector<std::string>> messages;
|
|
};
|
|
|
|
#endif
|
|
|