X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/1cabba873c9409b7c01280d9839499e955d3bf58..05e29980ab6ee71b0436f7ca58d109a79a512b0f:/projects/sgconf/users.cpp?ds=inline diff --git a/projects/sgconf/users.cpp b/projects/sgconf/users.cpp index 45603199..fe74e677 100644 --- a/projects/sgconf/users.cpp +++ b/projects/sgconf/users.cpp @@ -3,12 +3,17 @@ #include "api_action.h" #include "options.h" #include "config.h" +#include "utils.h" #include "stg/servconf.h" #include "stg/servconf_types.h" +#include "stg/user_conf.h" +#include "stg/user_stat.h" +#include "stg/user_ips.h" #include "stg/common.h" #include +#include #include #include @@ -46,6 +51,7 @@ std::cout << Indent(level, true) << "login: " << info.login << "\n" << Indent(level) << "name: " << info.name << "\n" << Indent(level) << "address: " << info.address << "\n" << Indent(level) << "phone: " << info.phone << "\n" + << Indent(level) << "corporation: " << info.corp << "\n" << Indent(level) << "last ping time: " << TimeToString(info.pingTime) << "\n" << Indent(level) << "last activity time: " << TimeToString(info.lastActivityTime) << "\n" << Indent(level) << "traffic:\n"; @@ -60,6 +66,12 @@ for (size_t i = 0; i < DIR_NUM; ++i) std::cout << Indent(level) << "user data:\n"; for (size_t i = 0; i < USERDATA_NUM; ++i) std::cout << Indent(level + 1, true) << "user data " << i << ": " << info.userData[i] << "\n"; +if (!info.services.empty()) + { + std::cout << Indent(level) << "services:\n"; + for (size_t i = 0; i < info.services.size(); ++i) + std::cout << Indent(level + 1, true) << info.services[i] << "\n"; + } if (!info.authBy.empty()) { std::cout << Indent(level) << "auth by:\n"; @@ -71,45 +83,151 @@ if (!info.authBy.empty()) std::vector GetUserParams() { std::vector params; -params.push_back({"password", "", "\tuser's password"}); -params.push_back({"cash", "", "\t\tuser's cash"}); -params.push_back({"credit", "", "\tuser's credit"}); -params.push_back({"credit-expire", "", "\tcredit expiration"}); -params.push_back({"free", "", "\tprepaid traffic"}); -params.push_back({"disabled", "", "\tdisable user (y|n)"}); -params.push_back({"passive", "", "\tmake user passive (y|n)"}); -params.push_back({"disable-detail-stat", "", "disable detail stat (y|n)"}); -params.push_back({"always-online", "", "\tmake user always online (y|n)"}); -params.push_back({"ips", "", "\t\tcoma-separated list of ips"}); -params.push_back({"tariff", "", "\tcurrent tariff"}); -params.push_back({"next-tariff", "", "tariff starting from the next month"}); -params.push_back({"group", "", "\t\tuser's group"}); -params.push_back({"note", "", "\t\tuser's note"}); -params.push_back({"email", "", "\t\tuser's email"}); -params.push_back({"name", "", "\tuser's real name"}); -params.push_back({"address", "
", "\tuser's postal address"}); -params.push_back({"phone", "", "\t\tuser's phone number"}); -params.push_back({"session-traffic", "", "coma-separated session upload and download"}); -params.push_back({"month-traffic", "", "coma-separated month upload and download"}); -params.push_back({"user-data", "", "coma-separated user data values"}); +params.push_back(SGCONF::API_ACTION::PARAM("password", "", "\tuser's password")); +params.push_back(SGCONF::API_ACTION::PARAM("cash-add", "", "cash to add (with optional comment)")); +params.push_back(SGCONF::API_ACTION::PARAM("cash-set", "", "cash to set (with optional comment)")); +params.push_back(SGCONF::API_ACTION::PARAM("credit", "", "\tuser's credit")); +params.push_back(SGCONF::API_ACTION::PARAM("credit-expire", "", "\tcredit expiration")); +params.push_back(SGCONF::API_ACTION::PARAM("free", "", "\tprepaid traffic")); +params.push_back(SGCONF::API_ACTION::PARAM("disabled", "", "\tdisable user (y|n)")); +params.push_back(SGCONF::API_ACTION::PARAM("passive", "", "\tmake user passive (y|n)")); +params.push_back(SGCONF::API_ACTION::PARAM("disable-detail-stat", "", "disable detail stat (y|n)")); +params.push_back(SGCONF::API_ACTION::PARAM("always-online", "", "\tmake user always online (y|n)")); +params.push_back(SGCONF::API_ACTION::PARAM("ips", "", "\t\tcoma-separated list of ips")); +params.push_back(SGCONF::API_ACTION::PARAM("tariff", "", "\tcurrent tariff")); +params.push_back(SGCONF::API_ACTION::PARAM("next-tariff", "", "tariff starting from the next month")); +params.push_back(SGCONF::API_ACTION::PARAM("group", "", "\t\tuser's group")); +params.push_back(SGCONF::API_ACTION::PARAM("note", "", "\t\tuser's note")); +params.push_back(SGCONF::API_ACTION::PARAM("email", "", "\t\tuser's email")); +params.push_back(SGCONF::API_ACTION::PARAM("name", "", "\tuser's real name")); +params.push_back(SGCONF::API_ACTION::PARAM("address", "
", "\tuser's postal address")); +params.push_back(SGCONF::API_ACTION::PARAM("phone", "", "\t\tuser's phone number")); +params.push_back(SGCONF::API_ACTION::PARAM("corp", "", "\tcorporation name")); +params.push_back(SGCONF::API_ACTION::PARAM("session-traffic", "", "coma-separated session upload and download")); +params.push_back(SGCONF::API_ACTION::PARAM("month-traffic", "", "coma-separated month upload and download")); +params.push_back(SGCONF::API_ACTION::PARAM("user-data", "", "coma-separated user data values")); return params; } std::vector GetCheckParams() { std::vector params; -params.push_back({"password", "", "\tuser's password"}); +params.push_back(SGCONF::API_ACTION::PARAM("password", "", "\tuser's password")); return params; } std::vector GetMessageParams() { std::vector params; -params.push_back({"logins", "", "\tlist of logins to send a message"}); -params.push_back({"text", "", "\t\tmessage text"}); +params.push_back(SGCONF::API_ACTION::PARAM("logins", "", "\tlist of logins to send a message")); +params.push_back(SGCONF::API_ACTION::PARAM("text", "", "\t\tmessage text")); return params; } +void ConvBool(const std::string & value, RESETABLE & res) +{ +res = !value.empty() && value[0] == 'y'; +} + +void Splice(std::vector > & lhs, const std::vector > & rhs) +{ +for (size_t i = 0; i < lhs.size() && i < rhs.size(); ++i) + lhs[i].splice(rhs[i]); +} + +RESETABLE ConvString(const std::string & value) +{ +return value; +} + +void ConvStringList(std::string value, std::vector > & res) +{ +Splice(res, Split > >(value, ',', ConvString)); +} + +void ConvServices(std::string value, RESETABLE > & res) +{ +value.erase(std::remove(value.begin(), value.end(), ' '), value.end()); +res = Split >(value, ','); +} + +void ConvCreditExpire(const std::string & value, RESETABLE & res) +{ +struct tm brokenTime; +if (stg_strptime(value.c_str(), "%Y-%m-%d %H:%M:%S", &brokenTime) == NULL) + throw SGCONF::ACTION::ERROR("Credit expiration should be in format 'YYYY-MM-DD HH:MM:SS'. Got: '" + value + "'"); +res = stg_timegm(&brokenTime); +} + +void ConvIPs(const std::string & value, RESETABLE & res) +{ +res = StrToIPS(value); +} + +struct TRAFF +{ + uint64_t up; + uint64_t down; +}; + +TRAFF ConvTraff(const std::string & value) +{ +TRAFF res; +size_t slashPos = value.find_first_of('/'); +if (slashPos == std::string::npos) + throw SGCONF::ACTION::ERROR("Traffic record should be in format 'upload/download'. Got: '" + value + "'"); + +if (str2x(value.substr(0, slashPos), res.up) < 0) + throw SGCONF::ACTION::ERROR("Traffic value should be an integer. Got: '" + value.substr(0, slashPos) + "'"); +if (str2x(value.substr(slashPos + 1, value.length() - slashPos), res.down) < 0) + throw SGCONF::ACTION::ERROR("Traffic value should be an integer. Got: '" + value.substr(slashPos + 1, value.length() - slashPos) + "'"); +return res; +} + +void ConvSessionTraff(std::string value, USER_STAT_RES & res) +{ +value.erase(std::remove(value.begin(), value.end(), ' '), value.end()); +std::vector traff(Split >(value, ',', ConvTraff)); +if (traff.size() != DIR_NUM) + throw SGCONF::ACTION::ERROR("There should be prcisely " + x2str(DIR_NUM) + " records of session traffic."); +for (size_t i = 0; i < DIR_NUM; ++i) + { + res.sessionUp[i] = traff[i].up; + res.sessionDown[i] = traff[i].down; + } +} + +void ConvMonthTraff(std::string value, USER_STAT_RES & res) +{ +value.erase(std::remove(value.begin(), value.end(), ' '), value.end()); +std::vector traff(Split >(value, ',', ConvTraff)); +if (traff.size() != DIR_NUM) + throw SGCONF::ACTION::ERROR("There should be prcisely " + x2str(DIR_NUM) + " records of month traffic."); +for (size_t i = 0; i < DIR_NUM; ++i) + { + res.monthUp[i] = traff[i].up; + res.monthDown[i] = traff[i].down; + } +} + +void ConvCashInfo(const std::string & value, RESETABLE & res) +{ +CASH_INFO info; +size_t pos = value.find_first_of(':'); +if (pos == std::string::npos) + { + if (str2x(value, info.first) < 0) + throw SGCONF::ACTION::ERROR("Cash should be a double value. Got: '" + value + "'"); + } +else + { + if (str2x(value.substr(0, pos), info.first) < 0) + throw SGCONF::ACTION::ERROR("Cash should be a double value. Got: '" + value + "'"); + info.second = value.substr(pos + 1); + } +res = info; +} + void SimpleCallback(bool result, const std::string & reason, void * /*data*/) @@ -185,38 +303,111 @@ return proto.DelUser(arg, SimpleCallback, NULL) == STG::st_ok; bool AddUserFunction(const SGCONF::CONFIG & config, const std::string & arg, - const std::map & /*options*/) + const std::map & options) { -// TODO -std::cerr << "Unimplemented.\n"; -return false; +USER_CONF_RES conf; +SGCONF::MaybeSet(options, "password", conf.password); +SGCONF::MaybeSet(options, "passive", conf.passive, ConvBool); +SGCONF::MaybeSet(options, "disabled", conf.disabled, ConvBool); +SGCONF::MaybeSet(options, "disable-detail-stat", conf.disabledDetailStat, ConvBool); +SGCONF::MaybeSet(options, "always-online", conf.alwaysOnline, ConvBool); +SGCONF::MaybeSet(options, "tariff", conf.tariffName); +SGCONF::MaybeSet(options, "address", conf.address); +SGCONF::MaybeSet(options, "phone", conf.phone); +SGCONF::MaybeSet(options, "email", conf.email); +SGCONF::MaybeSet(options, "note", conf.note); +SGCONF::MaybeSet(options, "name", conf.realName); +SGCONF::MaybeSet(options, "corp", conf.corp); +SGCONF::MaybeSet(options, "services", conf.services, ConvServices); +SGCONF::MaybeSet(options, "group", conf.group); +SGCONF::MaybeSet(options, "credit", conf.credit); +SGCONF::MaybeSet(options, "next-tariff", conf.nextTariff); +SGCONF::MaybeSet(options, "user-data", conf.userdata, ConvStringList); +SGCONF::MaybeSet(options, "credit-expire", conf.creditExpire, ConvCreditExpire); +SGCONF::MaybeSet(options, "ips", conf.ips, ConvIPs); +USER_STAT_RES stat; +SGCONF::MaybeSet(options, "cash-set", stat.cashSet, ConvCashInfo); +SGCONF::MaybeSet(options, "free", stat.freeMb); +SGCONF::MaybeSet(options, "session-traffic", stat, ConvSessionTraff); +SGCONF::MaybeSet(options, "month-traffic", stat, ConvMonthTraff); +STG::SERVCONF proto(config.server.data(), + config.port.data(), + config.userName.data(), + config.userPass.data()); +return proto.AddUser(arg, conf, stat, SimpleCallback, NULL) == STG::st_ok; } bool ChgUserFunction(const SGCONF::CONFIG & config, const std::string & arg, const std::map & options) { -// TODO -std::cerr << "Unimplemented.\n"; -return false; +USER_CONF_RES conf; +SGCONF::MaybeSet(options, "password", conf.password); +SGCONF::MaybeSet(options, "passive", conf.passive, ConvBool); +SGCONF::MaybeSet(options, "disabled", conf.disabled, ConvBool); +SGCONF::MaybeSet(options, "disable-detail-stat", conf.disabledDetailStat, ConvBool); +SGCONF::MaybeSet(options, "always-online", conf.alwaysOnline, ConvBool); +SGCONF::MaybeSet(options, "tariff", conf.tariffName); +SGCONF::MaybeSet(options, "address", conf.address); +SGCONF::MaybeSet(options, "phone", conf.phone); +SGCONF::MaybeSet(options, "email", conf.email); +SGCONF::MaybeSet(options, "note", conf.note); +SGCONF::MaybeSet(options, "name", conf.realName); +SGCONF::MaybeSet(options, "corp", conf.corp); +SGCONF::MaybeSet(options, "services", conf.services, ConvServices); +SGCONF::MaybeSet(options, "group", conf.group); +SGCONF::MaybeSet(options, "credit", conf.credit); +SGCONF::MaybeSet(options, "next-tariff", conf.nextTariff); +SGCONF::MaybeSet(options, "user-data", conf.userdata, ConvStringList); +SGCONF::MaybeSet(options, "credit-expire", conf.creditExpire, ConvCreditExpire); +SGCONF::MaybeSet(options, "ips", conf.ips, ConvIPs); +USER_STAT_RES stat; +SGCONF::MaybeSet(options, "cash-add", stat.cashAdd, ConvCashInfo); +SGCONF::MaybeSet(options, "cash-set", stat.cashSet, ConvCashInfo); +SGCONF::MaybeSet(options, "free", stat.freeMb); +SGCONF::MaybeSet(options, "session-traffic", stat, ConvSessionTraff); +SGCONF::MaybeSet(options, "month-traffic", stat, ConvMonthTraff); +STG::SERVCONF proto(config.server.data(), + config.port.data(), + config.userName.data(), + config.userPass.data()); +return proto.ChgUser(arg, conf, stat, SimpleCallback, NULL) == STG::st_ok; } bool CheckUserFunction(const SGCONF::CONFIG & config, const std::string & arg, const std::map & options) { -// TODO -std::cerr << "Unimplemented.\n"; -return false; +std::map::const_iterator it(options.find("password")); +if (it == options.end()) + throw SGCONF::ACTION::ERROR("Password is not specified."); +STG::SERVCONF proto(config.server.data(), + config.port.data(), + config.userName.data(), + config.userPass.data()); +return proto.CheckUser(arg, it->second, SimpleCallback, NULL) == STG::st_ok; } bool SendMessageFunction(const SGCONF::CONFIG & config, - const std::string & arg, + const std::string & /*arg*/, const std::map & options) { -// TODO -std::cerr << "Unimplemented.\n"; -return false; +std::map::const_iterator it(options.find("logins")); +if (it == options.end()) + throw SGCONF::ACTION::ERROR("Logins are not specified."); +std::string logins = it->second; +for (size_t i = 0; i < logins.length(); ++i) + if (logins[i] == ',') + logins[i] = ':'; +it = options.find("text"); +if (it == options.end()) + throw SGCONF::ACTION::ERROR("Message text is not specified."); +std::string text = it->second; +STG::SERVCONF proto(config.server.data(), + config.port.data(), + config.userName.data(), + config.userPass.data()); +return proto.SendMessage(logins, text, SimpleCallback, NULL) == STG::st_ok; } } // namespace anonymous