stglibs/srvconf.lib/parser_get_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 "stg/parser_get_user.h"
  23
  24 #include "stg/common.h"
  25
  26 #include <cstddef>
  27
  28 #include <strings.h>
  29
  30 namespace
  31 {
  32
  33 bool checkValue(const char ** attr)
  34 {
  35 return attr && attr[0] && attr[1] && strcasecmp(attr[0], "value") == 0;
  36 }
  37
  38 template <typename T>
  39 T getValue(const char ** attr)
  40 {
  41 T value = 0;
  42 if (checkValue(attr))
  43     if (str2x(attr[1], value) < 0)
  44         return 0;
  45 return value;
  46 }
  47
  48 template <>
  49 std::string getValue<std::string>(const char ** attr)
  50 {
  51 if (checkValue(attr))
  52     return attr[1];
  53 return "";
  54 }
  55
  56 template <>
  57 double getValue<double>(const char ** attr)
  58 {
  59 double value = 0;
  60 if (checkValue(attr))
  61     if (strtodouble2(attr[1], value) == EINVAL)
  62         return 0;
  63 return value;
  64 }
  65
  66 template <>
  67 STAT getValue<STAT>(const char ** attr)
  68 {
  69 STAT value;
  70 if (!attr)
  71     return value;
  72 std::map<std::string, long long &> props;
  73 for (size_t i = 0; i < DIR_NUM; ++i)
  74     {
  75     props.insert("su" + x2str(i), value.su[i]);
  76     props.insert("sd" + x2str(i), value.sd[i]);
  77     props.insert("mu" + x2str(i), value.mu[i]);
  78     props.insert("md" + x2str(i), value.md[i]);
  79     }
  80 size_t pos = 0;
  81 while (attr[pos])
  82     {
  83         std::string name(ToLower(attr[pos++]));
  84         std::map<std::string, long long &>::iterator it(props.find(name));
  85         if (it != props.end())
  86             str2x(attr[pos++], it->second);
  87     }
  88 return value;
  89 }
  90
  91 std::string getEncodedValue(const char ** attr)
  92 {
  93 if (checkValue(attr))
  94     return Decode21str(attr[1]);
  95 return "";
  96 }
  97
  98 template <typename T>
  99 void addParser(PROPERTY_PARSERS & parsers, const std::string & name, T & value, bool /*encoded*/)
 100 {
 101     parsers.insert(ToLower(name), new PROPERTY_PARSER(value, getValue<T>));
 102 }
 103
 104 template <>
 105 void addParser<std::string>(PROPERTY_PARSERS & parsers, const std::string & name, std::string & value, bool encoded)
 106 {
 107     if (encoded)
 108         parsers.insert(ToLower(name), new PROPERTY_PARSER(value, getEncodedValue));
 109     else
 110         parsers.insert(ToLower(name), new PROPERTY_PARSER(value, getValue<T>));
 111 }
 112
 113 void tryParse(PROPERTY_PARSERS & parsers, const std::string & name, const char ** attr)
 114 {
 115     PROPERTY_PARSERS::iterator it(parsers.find(name));
 116     if (it != parsers.end())
 117         it->second->Parse(attr);
 118 }
 119
 120 } // namespace anonymous
 121
 122 PARSER_GET_USER::PARSER_GET_USER()
 123     : callback(NULL),
 124       data(NULL),
 125       depth(0)
 126 {
 127     addParser(propertyParsers, "login", info.login);
 128     addParser(propertyParsers, "password", info.password);
 129     addParser(propertyParsers, "cash", info.cash);
 130     addParser(propertyParsers, "credit", info.credit);
 131     addParser(propertyParsers, "creditExpire", info.creditExpire);
 132     addParser(propertyParsers, "lastCash", info.lastCash);
 133     addParser(propertyParsers, "prepaidTraff", info.prepaidTraff);
 134     addParser(propertyParsers, "down", info.down);
 135     addParser(propertyParsers, "passive", info.passive);
 136     addParser(propertyParsers, "disableDetailStat", info.disableDetailStat);
 137     addParser(propertyParsers, "connected", info.connected);
 138     addParser(propertyParsers, "alwaysOnline", info.alwaysOnline);
 139     addParser(propertyParsers, "ip", info.ip);
 140     addParser(propertyParsers, "ips", info.ips);
 141     addParser(propertyParsers, "tariff", info.tariff);
 142     addParser(propertyParsers, "group", info.group, true);
 143     addParser(propertyParsers, "note", info.note, true);
 144     addParser(propertyParsers, "email", info.email, true);
 145     addParser(propertyParsers, "name", info.name, true);
 146     addParser(propertyParsers, "address", info.address, true);
 147     addParser(propertyParsers, "phone", info.phone, true);
 148     addParser(propertyParsers, "traff", info.stat);
 149
 150     for (size_t i = 0; i < USERDATA_NUM; ++i)
 151         addParser(propertyParsers, "userData" + x2str(i), info.userData[i], true);
 152 }
 153 //-----------------------------------------------------------------------------
 154 PARSER_GET_USER::~PARSER_GET_USER()
 155 {
 156     PROPERTY_PARSERS::iterator it(propertyParsers.begin());
 157     while (it != propertyParsers.end())
 158         delete (it++)->second;
 159 }
 160 //-----------------------------------------------------------------------------
 161 int PARSER_GET_USER::ParseStart(const char *el, const char **attr)
 162 {
 163 depth++;
 164 if (depth == 1)
 165     ParseUser(el, attr);
 166
 167 if (depth == 2)
 168     ParseUserParams(el, attr);
 169
 170 return 0;
 171 }
 172 //-----------------------------------------------------------------------------
 173 void PARSER_GET_USER::ParseEnd(const char *)
 174 {
 175 depth--;
 176 if (depth == 0)
 177     if (callback)
 178         callback(info, data);
 179 }
 180 //-----------------------------------------------------------------------------
 181 void PARSER_GET_USER::ParseUser(const char * el, const char ** attr)
 182 {
 183 if (strcasecmp(el, "user") == 0)
 184     if (strcasecmp(attr[1], "error") == 0)
 185         user.login = "";
 186 }
 187 //-----------------------------------------------------------------------------
 188 void PARSER_GET_USER::ParseUserParams(const char * el, const char ** attr)
 189 {
 190 tryParse(propertyParsers, ToLower(el), attr);
 191 }
 192 //-----------------------------------------------------------------------------
 193 void PARSER_GET_USER::SetCallback(CALLBACK f, void * d)
 194 {
 195 callback = f;
 196 data = d;
 197 }