X-Git-Url: https://git.stg.codes/stg.git/blobdiff_plain/309ad3660f15b57dc29f386d2cdc97d53cc93204..e483fbc3e51aca90ff4abc5347862173366c40ef:/include/stg/tariff.h

diff --git a/include/stg/tariff.h b/include/stg/tariff.h
index 1371190b..6f9d8767 100644
--- a/include/stg/tariff.h
+++ b/include/stg/tariff.h
@@ -21,12 +21,11 @@
 #ifndef TARIFF_H
 #define TARIFF_H
 
-#include "os_int.h"
-
 #include <string>
+#include <istream>
 #include <cstring>
 #include <ctime>
-#include <istream>
+#include <cstdint>
 
 struct TARIFF_DATA;
 
@@ -38,7 +37,7 @@ public:
 
     enum TRAFF_TYPE { TRAFF_UP = 0, TRAFF_DOWN, TRAFF_UP_DOWN, TRAFF_MAX };
 
-    static std::string ChangePolicyToString(CHANGE_POLICY change_policy);
+    static std::string ChangePolicyToString(CHANGE_POLICY changePolicy);
     static CHANGE_POLICY StringToChangePolicy(const std::string& value);
 
     static std::string PeriodToString(PERIOD period);
@@ -59,6 +58,7 @@ public:
     virtual double  GetFree() const = 0;
     virtual PERIOD  GetPeriod() const = 0;
     virtual CHANGE_POLICY GetChangePolicy() const = 0;
+    virtual time_t  GetChangePolicyTimeout() const = 0;
 
     virtual const   std::string & GetName() const = 0;
     virtual void    SetName(const std::string & name) = 0;
@@ -67,12 +67,13 @@ public:
     virtual int64_t GetTraffByType(uint64_t up, uint64_t down) const = 0;
     virtual int     GetThreshold(int dir) const = 0;
     virtual const TARIFF_DATA & GetTariffData() const = 0;
+    virtual std::string TariffChangeIsAllowed(const TARIFF & to, time_t currentTime) const = 0;
 };
 
 inline
-std::string TARIFF::ChangePolicyToString(TARIFF::CHANGE_POLICY change_policy)
+std::string TARIFF::ChangePolicyToString(TARIFF::CHANGE_POLICY changePolicy)
 {
-switch (change_policy)
+switch (changePolicy)
     {
     case ALLOW: return "allow";
     case TO_CHEAP: return "to_cheap";