projects/stargazer/plugins/configuration/sgconfig/parser_users.h

   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 #pragma once
  23
  24 #include "parser.h"
  25
  26 #include "stg/user_conf.h"
  27 #include "stg/user_stat.h"
  28 #include "stg/common.h"
  29
  30 #include <string>
  31 #include <optional>
  32
  33 namespace STG
  34 {
  35
  36 struct Users;
  37 struct User;
  38 struct Tariffs;
  39 struct Admin;
  40 struct Store;
  41
  42 namespace PARSER
  43 {
  44
  45 class GET_USERS: public BASE_PARSER
  46 {
  47     public:
  48         class FACTORY : public BASE_PARSER::FACTORY
  49         {
  50             public:
  51                 explicit FACTORY(Users & users) : m_users(users) {}
  52                 BASE_PARSER * create(const Admin & admin) override { return new GET_USERS(admin, m_users); }
  53                 static void Register(REGISTRY & registry, Users & users)
  54                 { registry[ToLower(tag)] = new FACTORY(users); }
  55             private:
  56                 Users & m_users;
  57         };
  58
  59         static const char * tag;
  60
  61         GET_USERS(const Admin & admin, Users & users)
  62             : BASE_PARSER(admin, tag), m_users(users),
  63               m_lastUserUpdateTime(0) {}
  64         int Start(void * data, const char * el, const char ** attr) override;
  65
  66     private:
  67         Users & m_users;
  68         time_t m_lastUserUpdateTime;
  69
  70         void CreateAnswer() override;
  71 };
  72
  73 class GET_USER: public BASE_PARSER
  74 {
  75     public:
  76         class FACTORY : public BASE_PARSER::FACTORY
  77         {
  78             public:
  79                 explicit FACTORY(const Users & users) : m_users(users) {}
  80                 BASE_PARSER * create(const Admin & admin) override { return new GET_USER(admin, m_users); }
  81                 static void Register(REGISTRY & registry, const Users & users)
  82                 { registry[ToLower(tag)] = new FACTORY(users); }
  83             private:
  84                 const Users & m_users;
  85         };
  86
  87         static const char * tag;
  88
  89         GET_USER(const Admin & admin, const Users & users)
  90             : BASE_PARSER(admin, tag), m_users(users) {}
  91         int Start(void * data, const char * el, const char ** attr) override;
  92
  93     private:
  94         const Users & m_users;
  95         std::string m_login;
  96
  97         void CreateAnswer() override;
  98 };
  99
 100 class ADD_USER: public BASE_PARSER
 101 {
 102     public:
 103         class FACTORY : public BASE_PARSER::FACTORY
 104         {
 105             public:
 106                 explicit FACTORY(Users & users) : m_users(users) {}
 107                 BASE_PARSER * create(const Admin & admin) override { return new ADD_USER(admin, m_users); }
 108                 static void Register(REGISTRY & registry, Users & users)
 109                 { registry[ToLower(tag)] = new FACTORY(users); }
 110             private:
 111                 Users & m_users;
 112         };
 113
 114         static const char * tag;
 115
 116         ADD_USER(const Admin & admin, Users & users)
 117             : BASE_PARSER(admin, tag), m_users(users) {}
 118         int Start(void * data, const char * el, const char ** attr) override;
 119
 120     private:
 121         Users & m_users;
 122         std::string m_login;
 123
 124         void CreateAnswer() override;
 125 };
 126
 127 class CHG_USER: public BASE_PARSER
 128 {
 129     public:
 130         class FACTORY : public BASE_PARSER::FACTORY
 131         {
 132             public:
 133                 FACTORY(Users & users, Store & store, const Tariffs & tariffs)
 134                     : m_users(users), m_store(store), m_tariffs(tariffs)
 135                 {}
 136                 BASE_PARSER * create(const Admin & admin) override { return new CHG_USER(admin, m_users, m_store, m_tariffs); }
 137                 static void Register(REGISTRY & registry, Users & users, Store & store, const Tariffs & tariffs)
 138                 { registry[ToLower(tag)] = new FACTORY(users, store, tariffs); }
 139             private:
 140                 Users & m_users;
 141                 Store & m_store;
 142                 const Tariffs & m_tariffs;
 143         };
 144
 145         static const char * tag;
 146
 147         CHG_USER(const Admin & admin, Users & users,
 148                  Store & store, const Tariffs & tariffs)
 149             : BASE_PARSER(admin, tag),
 150               m_users(users),
 151               m_store(store),
 152               m_tariffs(tariffs),
 153               m_cashMustBeAdded(false) {}
 154
 155         int Start(void * data, const char * el, const char ** attr) override;
 156
 157     private:
 158         Users & m_users;
 159         Store & m_store;
 160         const Tariffs & m_tariffs;
 161         UserStatOpt m_usr;
 162         UserConfOpt m_ucr;
 163         std::optional<uint64_t> m_upr[DIR_NUM];
 164         std::optional<uint64_t> m_downr[DIR_NUM];
 165         std::string m_cashMsg;
 166         std::string m_login;
 167         bool m_cashMustBeAdded;
 168
 169         int ApplyChanges();
 170         void CreateAnswer() override;
 171 };
 172
 173 class DEL_USER: public BASE_PARSER
 174 {
 175     public:
 176         class FACTORY : public BASE_PARSER::FACTORY
 177         {
 178             public:
 179                 explicit FACTORY(Users & users) : m_users(users) {}
 180                 BASE_PARSER * create(const Admin & admin) override { return new DEL_USER(admin, m_users); }
 181                 static void Register(REGISTRY & registry, Users & users)
 182                 { registry[ToLower(tag)] = new FACTORY(users); }
 183             private:
 184                 Users & m_users;
 185         };
 186
 187         static const char * tag;
 188
 189         DEL_USER(const Admin & admin, Users & users)
 190             : BASE_PARSER(admin, tag), m_users(users), res(0), u(NULL) {}
 191         int Start(void * data, const char * el, const char ** attr) override;
 192         int End(void * data, const char * el) override;
 193
 194     private:
 195         Users & m_users;
 196         int res;
 197         User * u;
 198
 199         void CreateAnswer() override;
 200 };
 201
 202 class CHECK_USER: public BASE_PARSER
 203 {
 204     public:
 205         class FACTORY : public BASE_PARSER::FACTORY
 206         {
 207             public:
 208                 explicit FACTORY(const Users & users) : m_users(users) {}
 209                 BASE_PARSER * create(const Admin & admin) override { return new CHECK_USER(admin, m_users); }
 210                 static void Register(REGISTRY & registry, const Users & users)
 211                 { registry[ToLower(tag)] = new FACTORY(users); }
 212             private:
 213                 const Users & m_users;
 214         };
 215
 216         static const char * tag;
 217
 218         CHECK_USER(const Admin & admin, const Users & users)
 219             : BASE_PARSER(admin, tag), m_users(users) {}
 220         int Start(void * data, const char * el, const char ** attr) override;
 221         int End(void * data, const char * el) override;
 222
 223     private:
 224         const Users & m_users;
 225
 226         void CreateAnswer(const char * error);
 227         void CreateAnswer() override {} // dummy
 228 };
 229
 230 } // namespace PARSER
 231 } // namespace STG