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 <map>
  27 #include <utility>
  28 #include <cstddef>
  29
  30 #include <strings.h>
  31
  32 using namespace STG;
  33
  34 template <>
  35 bool GetValue<PARSER_GET_USER::STAT>(const char ** attr, PARSER_GET_USER::STAT & value)
  36 {
  37 if (!attr)
  38     return false;
  39 std::map<std::string, long long *> props;
  40 for (size_t i = 0; i < DIR_NUM; ++i)
  41     {
  42     props.insert(std::pair<std::string, long long *>("su" + x2str(i), &value.su[i]));
  43     props.insert(std::pair<std::string, long long *>("sd" + x2str(i), &value.sd[i]));
  44     props.insert(std::pair<std::string, long long *>("mu" + x2str(i), &value.mu[i]));
  45     props.insert(std::pair<std::string, long long *>("md" + x2str(i), &value.md[i]));
  46     }
  47 size_t pos = 0;
  48 while (attr[pos])
  49     {
  50         std::string name(ToLower(attr[pos++]));
  51         std::map<std::string, long long *>::iterator it(props.find(name));
  52         if (it != props.end())
  53             if (str2x(attr[pos++], *it->second) < 0)
  54                 return false;
  55     }
  56 return true;
  57 }
  58
  59 PARSER_GET_USER::PARSER_GET_USER()
  60     : callback(NULL),
  61       data(NULL),
  62       depth(0),
  63       parsingAnswer(false)
  64 {
  65     AddParser(propertyParsers, "login", info.login);
  66     AddParser(propertyParsers, "password", info.password);
  67     AddParser(propertyParsers, "cash", info.cash);
  68     AddParser(propertyParsers, "credit", info.credit);
  69     AddParser(propertyParsers, "creditExpire", info.creditExpire);
  70     AddParser(propertyParsers, "lastCash", info.lastCash);
  71     AddParser(propertyParsers, "prepaidTraff", info.prepaidTraff);
  72     AddParser(propertyParsers, "down", info.down);
  73     AddParser(propertyParsers, "passive", info.passive);
  74     AddParser(propertyParsers, "disableDetailStat", info.disableDetailStat);
  75     AddParser(propertyParsers, "connected", info.connected);
  76     AddParser(propertyParsers, "aonline", info.alwaysOnline);
  77     AddParser(propertyParsers, "currIP", info.ip, GetIPValue);
  78     AddParser(propertyParsers, "ip", info.ips);
  79     AddParser(propertyParsers, "tariff", info.tariff);
  80     AddParser(propertyParsers, "group", info.group, GetEncodedValue);
  81     AddParser(propertyParsers, "note", info.note, GetEncodedValue);
  82     AddParser(propertyParsers, "email", info.email, GetEncodedValue);
  83     AddParser(propertyParsers, "name", info.name, GetEncodedValue);
  84     AddParser(propertyParsers, "address", info.address, GetEncodedValue);
  85     AddParser(propertyParsers, "phone", info.phone, GetEncodedValue);
  86     AddParser(propertyParsers, "traff", info.stat);
  87
  88     for (size_t i = 0; i < USERDATA_NUM; ++i)
  89         AddParser(propertyParsers, "userData" + x2str(i), info.userData[i], GetEncodedValue);
  90 }
  91 //-----------------------------------------------------------------------------
  92 PARSER_GET_USER::~PARSER_GET_USER()
  93 {
  94     PROPERTY_PARSERS::iterator it(propertyParsers.begin());
  95     while (it != propertyParsers.end())
  96         delete (it++)->second;
  97 }
  98 //-----------------------------------------------------------------------------
  99 int PARSER_GET_USER::ParseStart(const char * el, const char ** attr)
 100 {
 101 depth++;
 102 if (depth == 1)
 103     ParseUser(el, attr);
 104
 105 if (depth == 2 && parsingAnswer)
 106     ParseUserParams(el, attr);
 107
 108 return 0;
 109 }
 110 //-----------------------------------------------------------------------------
 111 void PARSER_GET_USER::ParseEnd(const char * /*el*/)
 112 {
 113 depth--;
 114 if (depth == 0 && parsingAnswer)
 115     {
 116     if (callback)
 117         callback(error.empty(), error, info, data);
 118     error.clear();
 119     parsingAnswer = false;
 120     }
 121 }
 122 //-----------------------------------------------------------------------------
 123 void PARSER_GET_USER::ParseUser(const char * el, const char ** attr)
 124 {
 125 if (strcasecmp(el, "user") == 0)
 126     if (attr && attr[0] && attr[1])
 127         {
 128         if (strcasecmp(attr[1], "error") == 0)
 129             {
 130             if (attr[2] && attr[3])
 131                 error = attr[3];
 132             else
 133                 error = "User not found.";
 134             }
 135         else
 136             parsingAnswer = true;
 137         }
 138     else
 139         parsingAnswer = true;
 140 }
 141 //-----------------------------------------------------------------------------
 142 void PARSER_GET_USER::ParseUserParams(const char * el, const char ** attr)
 143 {
 144 if (!TryParse(propertyParsers, ToLower(el), attr))
 145     error = "Invalid parameter.";
 146 }
 147 //-----------------------------------------------------------------------------
 148 void PARSER_GET_USER::SetCallback(CALLBACK f, void * d)
 149 {
 150 callback = f;
 151 data = d;
 152 }