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