projects/sgconf/admins.cpp

   1 #include "admins.h"
   2
   3 #include "api_action.h"
   4 #include "options.h"
   5 #include "makeproto.h"
   6 #include "config.h"
   7 #include "utils.h"
   8
   9 #include "stg/servconf.h"
  10 #include "stg/servconf_types.h"
  11 #include "stg/admin_conf.h"
  12
  13 #include <iostream>
  14 #include <string>
  15 #include <map>
  16 #include <cstdint>
  17 #include <cassert>
  18
  19 namespace
  20 {
  21
  22 std::string Indent(size_t level, bool dash = false)
  23 {
  24 if (level == 0)
  25     return "";
  26 return dash ? std::string(level * 4 - 2, ' ') + "- " : std::string(level * 4, ' ');
  27 }
  28
  29 std::string PrivToString(const STG::Priv& priv)
  30 {
  31 return std::string("") +
  32        (priv.corpChg ? "1" : "0") +
  33        (priv.serviceChg ? "1" : "0") +
  34        (priv.tariffChg ? "1" : "0") +
  35        (priv.adminChg ? "1" : "0") +
  36        (priv.userAddDel ? "1" : "0") +
  37        (priv.userPasswd ? "1" : "0") +
  38        (priv.userCash ? "1" : "0") +
  39        (priv.userConf ? "1" : "0") +
  40        (priv.userStat ? "1" : "0");
  41 }
  42
  43 void PrintAdmin(const STG::GetAdmin::Info & info, size_t level = 0)
  44 {
  45 std::cout << Indent(level, true) << "login: " << info.login << "\n"
  46           << Indent(level)       << "priviledges: " << PrivToString(info.priv) << "\n";
  47 }
  48
  49 std::vector<SGCONF::API_ACTION::PARAM> GetAdminParams()
  50 {
  51 std::vector<SGCONF::API_ACTION::PARAM> params;
  52 params.push_back(SGCONF::API_ACTION::PARAM("password", "<password>", "password"));
  53 params.push_back(SGCONF::API_ACTION::PARAM("priv", "<priv>", "priviledges"));
  54 return params;
  55 }
  56
  57 void ConvPriv(const std::string & value, std::optional<STG::Priv> & res)
  58 {
  59 if (value.length() != 9)
  60     throw SGCONF::ACTION::ERROR("Priviledges value should be a 9-digits length binary number.");
  61 STG::Priv priv;
  62 priv.corpChg = (value[0] == '0' ? 0 : 1);
  63 priv.serviceChg = (value[1] == '0' ? 0 : 1);
  64 priv.tariffChg = (value[2] == '0' ? 0 : 1);
  65 priv.adminChg = (value[3] == '0' ? 0 : 1);
  66 priv.userAddDel = (value[4] == '0' ? 0 : 1);
  67 priv.userPasswd = (value[5] == '0' ? 0 : 1);
  68 priv.userCash = (value[6] == '0' ? 0 : 1);
  69 priv.userConf = (value[7] == '0' ? 0 : 1);
  70 priv.userStat = (value[8] == '0' ? 0 : 1);
  71 res = priv;
  72 }
  73
  74 void SimpleCallback(bool result,
  75                     const std::string & reason,
  76                     void * /*data*/)
  77 {
  78 if (!result)
  79     {
  80     std::cerr << "Operation failed. Reason: '" << reason << "'." << std::endl;
  81     return;
  82     }
  83 std::cout << "Success.\n";
  84 }
  85
  86 void GetAdminsCallback(bool result,
  87                        const std::string & reason,
  88                        const std::vector<STG::GetAdmin::Info> & info,
  89                        void * /*data*/)
  90 {
  91 if (!result)
  92     {
  93     std::cerr << "Failed to get admin list. Reason: '" << reason << "'." << std::endl;
  94     return;
  95     }
  96 std::cout << "Admins:\n";
  97 for (size_t i = 0; i < info.size(); ++i)
  98     PrintAdmin(info[i], 1);
  99 }
 100
 101 void GetAdminCallback(bool result,
 102                       const std::string & reason,
 103                       const std::vector<STG::GetAdmin::Info> & info,
 104                       void * data)
 105 {
 106 assert(data != NULL && "Expecting pointer to std::string with the admin's login.");
 107 const std::string & login = *static_cast<const std::string *>(data);
 108 if (!result)
 109     {
 110     std::cerr << "Failed to get admin. Reason: '" << reason << "'." << std::endl;
 111     return;
 112     }
 113 for (size_t i = 0; i < info.size(); ++i)
 114     if (info[i].login == login)
 115         PrintAdmin(info[i]);
 116 }
 117
 118
 119 bool GetAdminsFunction(const SGCONF::CONFIG & config,
 120                        const std::string & /*arg*/,
 121                        const std::map<std::string, std::string> & /*options*/)
 122 {
 123 return makeProto(config).GetAdmins(GetAdminsCallback, NULL) == STG::st_ok;
 124 }
 125
 126 bool GetAdminFunction(const SGCONF::CONFIG & config,
 127                       const std::string & arg,
 128                       const std::map<std::string, std::string> & /*options*/)
 129 {
 130 // STG currently doesn't support <GetAdmin login="..."/>.
 131 // So get a list of admins and filter it. 'data' param holds a pointer to 'login'.
 132 std::string login(arg);
 133 return makeProto(config).GetAdmins(GetAdminCallback, &login) == STG::st_ok;
 134 }
 135
 136 bool DelAdminFunction(const SGCONF::CONFIG & config,
 137                       const std::string & arg,
 138                       const std::map<std::string, std::string> & /*options*/)
 139 {
 140 return makeProto(config).DelAdmin(arg, SimpleCallback, NULL) == STG::st_ok;
 141 }
 142
 143 bool AddAdminFunction(const SGCONF::CONFIG & config,
 144                       const std::string & arg,
 145                       const std::map<std::string, std::string> & options)
 146 {
 147 STG::AdminConfOpt conf;
 148 conf.login = arg;
 149 SGCONF::MaybeSet(options, "priv", conf.priv, ConvPriv);
 150 SGCONF::MaybeSet(options, "password", conf.password);
 151 return makeProto(config).AddAdmin(arg, conf, SimpleCallback, NULL) == STG::st_ok;
 152 }
 153
 154 bool ChgAdminFunction(const SGCONF::CONFIG & config,
 155                       const std::string & arg,
 156                       const std::map<std::string, std::string> & options)
 157 {
 158 STG::AdminConfOpt conf;
 159 conf.login = arg;
 160 SGCONF::MaybeSet(options, "priv", conf.priv, ConvPriv);
 161 SGCONF::MaybeSet(options, "password", conf.password);
 162 return makeProto(config).ChgAdmin(conf, SimpleCallback, NULL) == STG::st_ok;
 163 }
 164
 165 } // namespace anonymous
 166
 167 void SGCONF::AppendAdminsOptionBlock(COMMANDS & commands, OPTION_BLOCKS & blocks)
 168 {
 169 std::vector<API_ACTION::PARAM> params(GetAdminParams());
 170 blocks.Add("Admin management options")
 171       .Add("get-admins", SGCONF::MakeAPIAction(commands, GetAdminsFunction), "\tget admin list")
 172       .Add("get-admin", SGCONF::MakeAPIAction(commands, "<login>", GetAdminFunction), "get admin")
 173       .Add("add-admin", SGCONF::MakeAPIAction(commands, "<login>", params, AddAdminFunction), "add admin")
 174       .Add("del-admin", SGCONF::MakeAPIAction(commands, "<login>", DelAdminFunction), "del admin")
 175       .Add("chg-admin", SGCONF::MakeAPIAction(commands, "<login>", params, ChgAdminFunction), "change admin");
 176 }