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