X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/c016156ebb870d9d81219b8c4165aa75703f1e85..2196a3a0cdc5384c082febb7f4aa5994cc7d80db:/projects/stargazer/plugins/other/smux/tables.h

diff --git a/projects/stargazer/plugins/other/smux/tables.h b/projects/stargazer/plugins/other/smux/tables.h
index ad3b55a3..17875cec 100644
--- a/projects/stargazer/plugins/other/smux/tables.h
+++ b/projects/stargazer/plugins/other/smux/tables.h
@@ -1,14 +1,18 @@
-#ifndef __TABLES_H__
-#define __TABLES_H__
+#pragma once
+
+#include "sensors.h"
 
 #include <string>
 #include <map>
 
-#include "sensors.h"
+namespace STG
+{
+class Tariffs;
+class Users;
 
 class TableSensor {
     public:
-        TableSensor(const std::string & p) : prefix(p) {}
+        explicit TableSensor(const std::string & p) : prefix(p) {}
         virtual ~TableSensor() {}
 
         const std::string & GetPrefix() const { return prefix; }
@@ -21,18 +25,19 @@ class TableSensor {
 class TariffUsersTable : public TableSensor {
     public:
         TariffUsersTable(const std::string & p,
-                         USERS & u)
+                         STG::Tariffs & t,
+                         STG::Users & u)
             : TableSensor(p),
+              tariffs(t),
               users(u)
         {}
-        virtual ~TariffUsersTable() {}
 
-        void UpdateSensors(Sensors & sensors) const;
+        void UpdateSensors(Sensors & sensors) const override;
 
     private:
-        USERS & users;
+        STG::Tariffs & tariffs;
+        STG::Users & users;
 };
 
 typedef std::map<std::string, TableSensor *> Tables;
-
-#endif
+}