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/optional.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, STG::Optional<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, STG::Optional<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<TariffData> 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, &DirPriceData::priceDayA, pt_mega) + "\"/>" +
 106                   "<PriceDayB value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DirPriceData::priceDayB, pt_mega) + "\"/>" +
 107                   "<PriceNightA value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DirPriceData::priceNightA, pt_mega) + "\"/>" +
 108                   "<PriceNightB value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DirPriceData::priceNightB, pt_mega) + "\"/>" +
 109                   "<Threshold value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DirPriceData::threshold, 1) + "\"/>" +
 110                   "<SinglePrice value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DirPriceData::singlePrice, 1) + "\"/>" +
 111                   "<NoDiscount value=\"" + AOS2String(it->dirPrice, DIR_NUM, &DirPriceData::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::toString(it->tariffConf.traffType) + "\"/>" +
 116                   "<Period value=\"" + Tariff::toString(it->tariffConf.period) + "\"/>" +
 117                   "<ChangePolicy value=\"" + Tariff::toString(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 auto 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, &DirPriceDataOpt::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, &DirPriceDataOpt::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, &DirPriceDataOpt::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, &DirPriceDataOpt::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, &DirPriceDataOpt::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, &DirPriceDataOpt::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, &DirPriceDataOpt::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::parseTraffType(attr[1]);
 289             return 0;
 290         }
 291
 292         if (strcasecmp(el, "Period") == 0)
 293         {
 294             td.tariffConf.period = Tariff::parsePeriod(attr[1]);
 295             return 0;
 296         }
 297
 298         if (strcasecmp(el, "ChangePolicy") == 0)
 299         {
 300             td.tariffConf.changePolicy = Tariff::parseChangePolicy(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         auto tariffData = td.get({});
 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 }