X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/df9f528d8ea5b3ba19bebcaf0324c3282edeffcc..acdf6ab94fee26ebc4017bf32cf6171c7d9462e9:/projects/stargazer/plugins/configuration/sgconfig/parser_users.h

diff --git a/projects/stargazer/plugins/configuration/sgconfig/parser_users.h b/projects/stargazer/plugins/configuration/sgconfig/parser_users.h
index f5db4e7b..cdef451e 100644
--- a/projects/stargazer/plugins/configuration/sgconfig/parser_users.h
+++ b/projects/stargazer/plugins/configuration/sgconfig/parser_users.h
@@ -19,25 +19,26 @@
  *    Author : Maxim Mamontov <faust@stargazer.dp.ua>
  */
 
-#ifndef __STG_SGCONFIG_PARSER_USERS_H__
-#define __STG_SGCONFIG_PARSER_USERS_H__
+#pragma once
 
 #include "parser.h"
 
 #include "stg/user_conf.h"
 #include "stg/user_stat.h"
-#include "stg/resetable.h"
+#include "stg/common.h"
+#include "stg/optional.h"
 
 #include <string>
 
-class USERS;
-class USER;
-class TARIFFS;
-class ADMIN;
-class STORE;
-
 namespace STG
 {
+
+struct Users;
+struct User;
+struct Tariffs;
+struct Admin;
+struct Store;
+
 namespace PARSER
 {
 
@@ -47,22 +48,23 @@ class GET_USERS: public BASE_PARSER
         class FACTORY : public BASE_PARSER::FACTORY
         {
             public:
-                FACTORY(const ADMIN & admin, USERS & users)
-                    : m_admin(admin), m_users(users)
-                {}
-                virtual BASE_PARSER * create() { return new GET_USERS(m_admin, m_users); }
+                explicit FACTORY(Users & users) : m_users(users) {}
+                virtual BASE_PARSER * create(const Admin & admin) { return new GET_USERS(admin, m_users); }
+                static void Register(REGISTRY & registry, Users & users)
+                { registry[ToLower(tag)] = new FACTORY(users); }
             private:
-                const ADMIN & m_admin;
-                USERS & m_users;
+                Users & m_users;
         };
 
-        GET_USERS(const ADMIN & admin, USERS & users)
-            : BASE_PARSER(admin, "GetUsers"), m_users(users),
+        static const char * tag;
+
+        GET_USERS(const Admin & admin, Users & users)
+            : BASE_PARSER(admin, tag), m_users(users),
               m_lastUserUpdateTime(0) {}
         int Start(void * data, const char * el, const char ** attr);
 
     private:
-        USERS & m_users;
+        Users & m_users;
         time_t m_lastUserUpdateTime;
 
         void CreateAnswer();
@@ -74,21 +76,22 @@ class GET_USER: public BASE_PARSER
         class FACTORY : public BASE_PARSER::FACTORY
         {
             public:
-                FACTORY(const ADMIN & admin, const USERS & users)
-                    : m_admin(admin), m_users(users)
-                {}
-                virtual BASE_PARSER * create() { return new GET_USER(m_admin, m_users); }
+                explicit FACTORY(const Users & users) : m_users(users) {}
+                virtual BASE_PARSER * create(const Admin & admin) { return new GET_USER(admin, m_users); }
+                static void Register(REGISTRY & registry, const Users & users)
+                { registry[ToLower(tag)] = new FACTORY(users); }
             private:
-                const ADMIN & m_admin;
-                const USERS & m_users;
+                const Users & m_users;
         };
 
-        GET_USER(const ADMIN & admin, const USERS & users)
-            : BASE_PARSER(admin, "GetUser"), m_users(users) {}
+        static const char * tag;
+
+        GET_USER(const Admin & admin, const Users & users)
+            : BASE_PARSER(admin, tag), m_users(users) {}
         int Start(void * data, const char * el, const char ** attr);
 
     private:
-        const USERS & m_users;
+        const Users & m_users;
         std::string m_login;
 
         void CreateAnswer();
@@ -100,21 +103,22 @@ class ADD_USER: public BASE_PARSER
         class FACTORY : public BASE_PARSER::FACTORY
         {
             public:
-                FACTORY(const ADMIN & admin, USERS & users)
-                    : m_admin(admin), m_users(users)
-                {}
-                virtual BASE_PARSER * create() { return new ADD_USER(m_admin, m_users); }
+                explicit FACTORY(Users & users) : m_users(users) {}
+                virtual BASE_PARSER * create(const Admin & admin) { return new ADD_USER(admin, m_users); }
+                static void Register(REGISTRY & registry, Users & users)
+                { registry[ToLower(tag)] = new FACTORY(users); }
             private:
-                const ADMIN & m_admin;
-                USERS & m_users;
+                Users & m_users;
         };
 
-        ADD_USER(const ADMIN & admin, USERS & users)
-            : BASE_PARSER(admin, "AddUser"), m_users(users) {}
+        static const char * tag;
+
+        ADD_USER(const Admin & admin, Users & users)
+            : BASE_PARSER(admin, tag), m_users(users) {}
         int Start(void * data, const char * el, const char ** attr);
 
     private:
-        USERS & m_users;
+        Users & m_users;
         std::string m_login;
 
         void CreateAnswer();
@@ -126,40 +130,41 @@ class CHG_USER: public BASE_PARSER
         class FACTORY : public BASE_PARSER::FACTORY
         {
             public:
-                FACTORY(const ADMIN & admin, USERS & users, STORE & store, const TARIFFS & tariffs)
-                    : m_admin(admin), m_users(users), m_store(store), m_tariffs(tariffs)
+                FACTORY(Users & users, Store & store, const Tariffs & tariffs)
+                    : m_users(users), m_store(store), m_tariffs(tariffs)
                 {}
-                virtual BASE_PARSER * create() { return new CHG_USER(m_admin, m_users, m_store, m_tariffs); }
+                virtual BASE_PARSER * create(const Admin & admin) { return new CHG_USER(admin, m_users, m_store, m_tariffs); }
+                static void Register(REGISTRY & registry, Users & users, Store & store, const Tariffs & tariffs)
+                { registry[ToLower(tag)] = new FACTORY(users, store, tariffs); }
             private:
-                const ADMIN & m_admin;
-                USERS & m_users;
-                STORE & m_store;
-                const TARIFFS & m_tariffs;
+                Users & m_users;
+                Store & m_store;
+                const Tariffs & m_tariffs;
         };
 
-        CHG_USER(const ADMIN & admin, USERS & users,
-                 STORE & store, const TARIFFS & tariffs)
-            : BASE_PARSER(admin, "SetUser"),
+        static const char * tag;
+
+        CHG_USER(const Admin & admin, Users & users,
+                 Store & store, const Tariffs & tariffs)
+            : BASE_PARSER(admin, tag),
               m_users(users),
               m_store(store),
               m_tariffs(tariffs),
-              m_cashMustBeAdded(false),
-              m_res(0) {}
+              m_cashMustBeAdded(false) {}
 
         int Start(void * data, const char * el, const char ** attr);
 
     private:
-        USERS & m_users;
-        STORE & m_store;
-        const TARIFFS & m_tariffs;
-        USER_STAT_RES m_usr;
-        USER_CONF_RES m_ucr;
-        RESETABLE<uint64_t> m_upr[DIR_NUM];
-        RESETABLE<uint64_t> m_downr[DIR_NUM];
+        Users & m_users;
+        Store & m_store;
+        const Tariffs & m_tariffs;
+        UserStatOpt m_usr;
+        UserConfOpt m_ucr;
+        Optional<uint64_t> m_upr[DIR_NUM];
+        Optional<uint64_t> m_downr[DIR_NUM];
         std::string m_cashMsg;
         std::string m_login;
         bool m_cashMustBeAdded;
-        int m_res;
 
         int ApplyChanges();
         void CreateAnswer();
@@ -171,24 +176,25 @@ class DEL_USER: public BASE_PARSER
         class FACTORY : public BASE_PARSER::FACTORY
         {
             public:
-                FACTORY(const ADMIN & admin, USERS & users)
-                    : m_admin(admin), m_users(users)
-                {}
-                virtual BASE_PARSER * create() { return new DEL_USER(m_admin, m_users); }
+                explicit FACTORY(Users & users) : m_users(users) {}
+                virtual BASE_PARSER * create(const Admin & admin) { return new DEL_USER(admin, m_users); }
+                static void Register(REGISTRY & registry, Users & users)
+                { registry[ToLower(tag)] = new FACTORY(users); }
             private:
-                const ADMIN & m_admin;
-                USERS & m_users;
+                Users & m_users;
         };
 
-        DEL_USER(const ADMIN & admin, USERS & users)
-            : BASE_PARSER(admin, "DelUser"), m_users(users), res(0), u(NULL) {}
+        static const char * tag;
+
+        DEL_USER(const Admin & admin, Users & users)
+            : BASE_PARSER(admin, tag), m_users(users), res(0), u(NULL) {}
         int Start(void * data, const char * el, const char ** attr);
         int End(void * data, const char * el);
 
     private:
-        USERS & m_users;
+        Users & m_users;
         int res;
-        USER * u;
+        User * u;
 
         void CreateAnswer();
 };
@@ -199,22 +205,23 @@ class CHECK_USER: public BASE_PARSER
         class FACTORY : public BASE_PARSER::FACTORY
         {
             public:
-                FACTORY(const ADMIN & admin, const USERS & users)
-                    : m_admin(admin), m_users(users)
-                {}
-                virtual BASE_PARSER * create() { return new CHECK_USER(m_admin, m_users); }
+                explicit FACTORY(const Users & users) : m_users(users) {}
+                virtual BASE_PARSER * create(const Admin & admin) { return new CHECK_USER(admin, m_users); }
+                static void Register(REGISTRY & registry, const Users & users)
+                { registry[ToLower(tag)] = new FACTORY(users); }
             private:
-                const ADMIN & m_admin;
-                const USERS & m_users;
+                const Users & m_users;
         };
 
-        CHECK_USER(const ADMIN & admin, const USERS & users)
-            : BASE_PARSER(admin, "CheckUser"), m_users(users) {}
+        static const char * tag;
+
+        CHECK_USER(const Admin & admin, const Users & users)
+            : BASE_PARSER(admin, tag), m_users(users) {}
         int Start(void * data, const char * el, const char ** attr);
         int End(void * data, const char * el);
 
     private:
-        const USERS & m_users;
+        const Users & m_users;
 
         void CreateAnswer(const char * error);
         void CreateAnswer() {} // dummy
@@ -222,5 +229,3 @@ class CHECK_USER: public BASE_PARSER
 
 } // namespace PARSER
 } // namespace STG
-
-#endif