]> git.stg.codes - stg.git/blob - stglibs/srvconf.lib/parsers/chg_user.cpp
3f7effd9e32e22e6f53fb6643a9e252eeb42deed
[stg.git] / stglibs / srvconf.lib / parsers / chg_user.cpp
1 /*
2  *    This program is free software; you can redistribute it and/or modify
3  *    it under the terms of the GNU General Public License as published by
4  *    the Free Software Foundation; either version 2 of the License, or
5  *    (at your option) any later version.
6  *
7  *    This program is distributed in the hope that it will be useful,
8  *    but WITHOUT ANY WARRANTY; without even the implied warranty of
9  *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
10  *    GNU General Public License for more details.
11  *
12  *    You should have received a copy of the GNU General Public License
13  *    along with this program; if not, write to the Free Software
14  *    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
15  */
16
17 /*
18  *    Author : Boris Mikhailenko <stg34@stargazer.dp.ua>
19  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
20  */
21
22 #include "chg_user.h"
23
24 #include "resetable_utils.h"
25
26 #include "stg/user_conf.h"
27 #include "stg/user_stat.h"
28 #include "stg/common.h"
29
30 #include <sstream>
31 #include <iostream>
32
33 #include <strings.h>
34
35 using namespace STG;
36
37 namespace
38 {
39
40 RESETABLE<std::string> MaybeEncode(const RESETABLE<std::string> & value)
41 {
42 RESETABLE<std::string> res;
43 if (!value.empty())
44     res = Encode12str(value.data());
45 return res;
46 }
47
48 RESETABLE<std::string> MaybeIconv(const RESETABLE<std::string> & value, const std::string & fromEncoding, const std::string & toEncoding)
49 {
50 RESETABLE<std::string> res;
51 if (!value.empty())
52     res = IconvString(value.data(), fromEncoding, toEncoding);
53 return res;
54 }
55
56 }
57
58 CHG_USER::PARSER::PARSER(SIMPLE::CALLBACK f, void * d)
59     : callback(f),
60       data(d),
61       depth(0)
62 {
63 }
64 //-----------------------------------------------------------------------------
65 int CHG_USER::PARSER::ParseStart(const char *el, const char **attr)
66 {
67 depth++;
68 if (depth == 1)
69     {
70     if (strcasecmp(el, "SetUser") == 0)
71         ParseAnswer(el, attr);
72     else if (strcasecmp(el, "DelUser") == 0)
73         ParseAnswer(el, attr);
74     else if (strcasecmp(el, "AddUser") == 0)
75         ParseAnswer(el, attr);
76     }
77 return 0;
78 }
79 //-----------------------------------------------------------------------------
80 void CHG_USER::PARSER::ParseEnd(const char *)
81 {
82 depth--;
83 }
84 //-----------------------------------------------------------------------------
85 void CHG_USER::PARSER::ParseAnswer(const char * /*el*/, const char ** attr)
86 {
87 if (!callback)
88     return;
89 if (attr && attr[0] && attr[1])
90     callback(strcasecmp(attr[1], "ok") == 0, attr[2] && attr[3] ? attr[3] : "", data);
91 else
92     callback(false, "Invalid response.", data);
93 }
94
95 std::string CHG_USER::Serialize(const USER_CONF_RES & conf, const USER_STAT_RES & stat, const std::string & encoding)
96 {
97 std::ostringstream stream;
98
99 // Conf
100
101 appendResetable(stream, "credit", conf.credit);
102 appendResetable(stream, "creditExpire", conf.creditExpire);
103 appendResetable(stream, "password", conf.password);
104 appendResetable(stream, "down", conf.disabled); // TODO: down -> disabled
105 appendResetable(stream, "passive", conf.passive);
106 appendResetable(stream, "disableDetailStat", conf.disabledDetailStat); // TODO: disable -> disabled
107 appendResetable(stream, "aonline", conf.alwaysOnline); // TODO: aonline -> alwaysOnline
108 appendResetable(stream, "ip", conf.ips); // TODO: ip -> ips
109
110 if (!conf.nextTariff.empty())
111     stream << "<tariff delayed=\"" << conf.nextTariff.data() << "\"/>";
112 else if (!conf.tariffName.empty())
113     stream << "<tariff now=\"" << conf.tariffName.data() << "\"/>";
114
115 appendResetable(stream, "note", MaybeIconv(MaybeEncode(conf.note), "koi8-ru", encoding));
116 appendResetable(stream, "name", MaybeIconv(MaybeEncode(conf.realName), "koi8-ru", encoding)); // TODO: name -> realName
117 appendResetable(stream, "address", MaybeIconv(MaybeEncode(conf.address), "koi8-ru", encoding));
118 appendResetable(stream, "email", MaybeIconv(MaybeEncode(conf.email), "koi8-ru", encoding));
119 appendResetable(stream, "phone", MaybeIconv(MaybeEncode(conf.phone), "koi8-ru", encoding));
120 appendResetable(stream, "group", MaybeIconv(MaybeEncode(conf.group), "koi8-ru", encoding));
121 appendResetable(stream, "corp", conf.corp);
122
123 for (size_t i = 0; i < conf.userdata.size(); ++i)
124     appendResetable(stream, "userdata", i, MaybeIconv(MaybeEncode(conf.userdata[i]), "koi8-ru", encoding));
125
126 if (!conf.services.empty())
127     {
128     stream << "<services>";
129     for (size_t i = 0; i < conf.services.data().size(); ++i)
130         stream << "<service name=\"" << conf.services.data()[i] << "\"/>";
131     stream << "</services>";
132     }
133
134 // Stat
135
136 if (!stat.cashAdd.empty())
137     stream << "<cash add=\"" << stat.cashAdd.data().first << "\" msg=\"" << IconvString(Encode12str(stat.cashAdd.data().second), "koi8-ru", encoding) << "\"/>";
138 else if (!stat.cashSet.empty())
139     stream << "<cash set=\"" << stat.cashSet.data().first << "\" msg=\"" << IconvString(Encode12str(stat.cashSet.data().second), "koi8-ru", encoding) << "\"/>";
140
141 appendResetable(stream, "freeMb", stat.freeMb);
142
143 std::ostringstream traff;
144 for (size_t i = 0; i < stat.sessionUp.size(); ++i)
145     if (!stat.sessionUp[i].empty())
146         traff << " SU" << i << "=\"" << stat.sessionUp[i].data() << "\"";
147 for (size_t i = 0; i < stat.sessionDown.size(); ++i)
148     if (!stat.sessionDown[i].empty())
149         traff << " SD" << i << "=\"" << stat.sessionDown[i].data() << "\"";
150 for (size_t i = 0; i < stat.monthUp.size(); ++i)
151     if (!stat.monthUp[i].empty())
152         traff << " MU" << i << "=\"" << stat.monthUp[i].data() << "\"";
153 for (size_t i = 0; i < stat.monthDown.size(); ++i)
154     if (!stat.monthDown[i].empty())
155         traff << " MD" << i << "=\"" << stat.monthDown[i].data() << "\"";
156
157 std::string traffData = traff.str();
158 if (!traffData.empty())
159     stream << "<traff" << traffData << "/>";
160
161 std::cerr << stream.str() << "\n";
162 return stream.str();
163 }