]> git.stg.codes - stg.git/blob - stargazer/plugins/configuration/sgconfig/parser_tariffs.cpp
ae22cf044d656fbe6c276371752323be10a9836f
[stg.git] / stargazer / plugins / configuration / sgconfig / parser_tariffs.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 "parser_tariffs.h"
23
24 #include "stg/tariffs.h"
25 #include "stg/users.h"
26 #include "stg/resetable.h"
27
28 #include <cstdio> // snprintf
29 #include <cstring>
30
31 using STG::PARSER::GET_TARIFFS;
32 using STG::PARSER::ADD_TARIFF;
33 using STG::PARSER::DEL_TARIFF;
34 using STG::PARSER::CHG_TARIFF;
35
36 const char * GET_TARIFFS::tag = "GetTariffs";
37 const char * ADD_TARIFF::tag  = "AddTariff";
38 const char * DEL_TARIFF::tag  = "DelTariff";
39 const char * CHG_TARIFF::tag  = "SetTariff";
40
41 namespace
42 {
43
44 const double pt_mega = 1024 * 1024;
45
46 template <typename A, typename C, typename F>
47 std::string AOS2String(const A & array, size_t size, const F C::* field, F multiplier)
48 {
49     std::string res;
50     for (size_t i = 0; i < size; ++i)
51     {
52         if (!res.empty())
53             res += "/";
54         res += std::to_string((array[i].*field) * multiplier);
55     }
56     return res;
57 }
58
59 template <typename T>
60 bool str2res(const std::string& source, RESETABLE<T>& dest, T divisor)
61 {
62     T value = 0;
63     if (str2x(source, value))
64         return false;
65     dest = value / divisor;
66     return true;
67 }
68
69 template <typename A, typename C, typename F>
70 bool String2AOS(const std::string & source, A & array, size_t size, RESETABLE<F> C::* field, F divisor)
71 {
72     size_t index = 0;
73     std::string::size_type from = 0;
74     std::string::size_type pos = 0;
75     while (index < size && (pos = source.find('/', from)) != std::string::npos)
76     {
77         if (!str2res(source.substr(from, pos - from), array[index].*field, divisor))
78             return false;
79         from = pos + 1;
80         ++index;
81     }
82     if (str2res(source.substr(from), array[index].*field, divisor))
83         return false;
84     return true;
85 }
86
87 }
88
89 void GET_TARIFFS::CreateAnswer()
90 {
91     m_answer = "<Tariffs>";
92
93     std::vector<TARIFF_DATA> dataList;
94     m_tariffs.GetTariffsData(&dataList);
95     auto it = dataList.begin();
96     for (; it != dataList.end(); ++it)
97         {
98         m_answer += "<tariff name=\"" + it->tariffConf.name + "\">";
99
100         for (size_t i = 0; i < DIR_NUM; i++)
101             m_answer += "<Time" + std::to_string(i) + " value=\"" +
102                 std::to_string(it->dirPrice[i].hDay)   + ":" + std::to_string(it->dirPrice[i].mDay)   + "-" +
103                 std::to_string(it->dirPrice[i].hNight) + ":" + std::to_string(it->dirPrice[i].mNight) + "\"/>";
104
105         m_answer += "<PriceDayA value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DIRPRICE_DATA::priceDayA, pt_mega) + "\"/>" +
106                   "<PriceDayB value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DIRPRICE_DATA::priceDayB, pt_mega) + "\"/>" +
107                   "<PriceNightA value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DIRPRICE_DATA::priceNightA, pt_mega) + "\"/>" +
108                   "<PriceNightB value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DIRPRICE_DATA::priceNightB, pt_mega) + "\"/>" +
109                   "<Threshold value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DIRPRICE_DATA::threshold, 1) + "\"/>" +
110                   "<SinglePrice value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DIRPRICE_DATA::singlePrice, 1) + "\"/>" +
111                   "<NoDiscount value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DIRPRICE_DATA::noDiscount, 1) + "\"/>" +
112                   "<Fee value=\"" + std::to_string(it->tariffConf.fee) + "\"/>" +
113                   "<PassiveCost value=\"" + std::to_string(it->tariffConf.passiveCost) + "\"/>" +
114                   "<Free value=\"" + std::to_string(it->tariffConf.free) + "\"/>" +
115                   "<TraffType value=\"" + TARIFF::TraffTypeToString(it->tariffConf.traffType) + "\"/>" +
116                   "<Period value=\"" + TARIFF::PeriodToString(it->tariffConf.period) + "\"/>" +
117                   "<ChangePolicy value=\"" + TARIFF::ChangePolicyToString(it->tariffConf.changePolicy) + "\"/>" +
118                   "<ChangePolicyTimeout value=\"" + std::to_string(it->tariffConf.changePolicyTimeout) + "\"/>" +
119                   "</tariff>";
120         }
121
122     m_answer += "</Tariffs>";
123 }
124
125 int ADD_TARIFF::Start(void *, const char * el, const char ** attr)
126 {
127     if (strcasecmp(el, m_tag.c_str()) != 0)
128         return -1;
129
130     if (attr[1] == NULL)
131         return -1;
132
133     tariff = attr[1];
134     return 0;
135 }
136
137 void ADD_TARIFF::CreateAnswer()
138 {
139     if (m_tariffs.Add(tariff, &m_currAdmin) == 0)
140         m_answer = "<" + m_tag + " Result=\"Ok\"/>";
141     else
142         m_answer = "<" + m_tag + " Result=\"Error. " + m_tariffs.GetStrError() + "\"/>";
143 }
144
145 int DEL_TARIFF::Start(void *, const char * el, const char ** attr)
146 {
147     if (strcasecmp(el, m_tag.c_str()) != 0)
148         return -1;
149
150     if (attr[1] == NULL)
151         return -1;
152
153     tariff = attr[1];
154     return 0;
155 }
156
157 void DEL_TARIFF::CreateAnswer()
158 {
159     if (m_users.TariffInUse(tariff))
160         m_answer = "<" + m_tag + " Result=\"Error. Tariff \'" + tariff + "\' cannot be deleted, it is in use.\"/>";
161     else if (m_tariffs.Del(tariff, &m_currAdmin) == 0)
162         m_answer = "<" + m_tag + " Result=\"Ok\"/>";
163     else
164         m_answer = "<" + m_tag + " Result=\"Error. " + m_tariffs.GetStrError() + "\"/>";
165 }
166
167 int CHG_TARIFF::Start(void *, const char * el, const char ** attr)
168 {
169     m_depth++;
170
171     if (m_depth == 1)
172     {
173         if (strcasecmp(el, m_tag.c_str()) == 0)
174         {
175             const TARIFF * tariff = m_tariffs.FindByName(attr[1]);
176             if (tariff != NULL)
177                 td = tariff->GetTariffData();
178             else
179                 return -1;
180             return 0;
181         }
182     }
183     else
184     {
185         if (strcasecmp(el, "PriceDayA") == 0)
186         {
187             if (!String2AOS(attr[1], td.dirPrice, DIR_NUM, &DIRPRICE_DATA_RES::priceDayA, pt_mega))
188                 return -1; // TODO: log it
189             else
190                 return 0;
191         }
192
193         if (strcasecmp(el, "PriceDayB") == 0)
194         {
195             if (!String2AOS(attr[1], td.dirPrice, DIR_NUM, &DIRPRICE_DATA_RES::priceDayB, pt_mega))
196                 return -1; // TODO: log it
197             else
198                 return 0;
199         }
200
201         if (strcasecmp(el, "PriceNightA") == 0)
202         {
203             if (!String2AOS(attr[1], td.dirPrice, DIR_NUM, &DIRPRICE_DATA_RES::priceNightA, pt_mega))
204                 return -1; // TODO: log it
205             else
206                 return 0;
207         }
208
209         if (strcasecmp(el, "PriceNightB") == 0)
210         {
211             if (!String2AOS(attr[1], td.dirPrice, DIR_NUM, &DIRPRICE_DATA_RES::priceNightB, pt_mega))
212                 return -1; // TODO: log it
213             else
214                 return 0;
215         }
216
217         if (strcasecmp(el, "Threshold") == 0)
218         {
219             if (!String2AOS(attr[1], td.dirPrice, DIR_NUM, &DIRPRICE_DATA_RES::threshold, 1))
220                 return -1; // TODO: log it
221             else
222                 return 0;
223         }
224
225         if (strcasecmp(el, "SinglePrice") == 0)
226         {
227             if (!String2AOS(attr[1], td.dirPrice, DIR_NUM, &DIRPRICE_DATA_RES::singlePrice, 1))
228                 return -1; // TODO: log it
229             else
230                 return 0;
231         }
232
233         if (strcasecmp(el, "NoDiscount") == 0)
234         {
235             if (!String2AOS(attr[1], td.dirPrice, DIR_NUM, &DIRPRICE_DATA_RES::noDiscount, 1))
236                 return -1; // TODO: log it
237             else
238                 return 0;
239         }
240
241         for (int j = 0; j < DIR_NUM; j++)
242         {
243             char st[50];
244             snprintf(st, 50, "Time%d", j);
245             if (strcasecmp(el, st) == 0)
246             {
247                 int h1 = 0;
248                 int m1 = 0;
249                 int h2 = 0;
250                 int m2 = 0;
251                 if (ParseTariffTimeStr(attr[1], h1, m1, h2, m2) == 0)
252                     {
253                     td.dirPrice[j].hDay = h1;
254                     td.dirPrice[j].mDay = m1;
255                     td.dirPrice[j].hNight = h2;
256                     td.dirPrice[j].mNight = m2;
257                     }
258                 return 0;
259             }
260         }
261
262         if (strcasecmp(el, "Fee") == 0)
263         {
264             double fee;
265             if (strtodouble2(attr[1], fee) == 0)
266                 td.tariffConf.fee = fee;
267             return 0;
268         }
269
270         if (strcasecmp(el, "PassiveCost") == 0)
271         {
272             double pc;
273             if (strtodouble2(attr[1], pc) == 0)
274                 td.tariffConf.passiveCost = pc;
275             return 0;
276         }
277
278         if (strcasecmp(el, "Free") == 0)
279         {
280             double free;
281             if (strtodouble2(attr[1], free) == 0)
282                 td.tariffConf.free = free;
283             return 0;
284         }
285
286         if (strcasecmp(el, "TraffType") == 0)
287         {
288             td.tariffConf.traffType = TARIFF::StringToTraffType(attr[1]);
289             return 0;
290         }
291
292         if (strcasecmp(el, "Period") == 0)
293         {
294             td.tariffConf.period = TARIFF::StringToPeriod(attr[1]);
295             return 0;
296         }
297
298         if (strcasecmp(el, "ChangePolicy") == 0)
299         {
300             td.tariffConf.changePolicy = TARIFF::StringToChangePolicy(attr[1]);
301             return 0;
302         }
303
304         if (strcasecmp(el, "ChangePolicyTimeout") == 0)
305         {
306             int64_t policyTime = 0;
307             if (str2x(attr[1], policyTime) == 0)
308                 td.tariffConf.changePolicyTimeout = (time_t)policyTime;
309             return 0;
310         }
311     }
312     return -1;
313 }
314
315 void CHG_TARIFF::CreateAnswer()
316 {
317     if (!td.tariffConf.name.data().empty())
318     {
319         TARIFF_DATA tariffData = td.GetData();
320         if (m_tariffs.Chg(tariffData, &m_currAdmin) == 0)
321             m_answer = "<" + m_tag + " Result=\"ok\"/>";
322         else
323             m_answer = "<" + m_tag + " Result=\"Change tariff error! " + m_tariffs.GetStrError() + "\"/>";
324     }
325     else
326         m_answer = "<" + m_tag + " Result=\"Change tariff error!\"/>";
327 }