From 05a088fcf0679d362095036635b081c14b9b800d Mon Sep 17 00:00:00 2001
From: ionutboangiu <ionut.boangiu@itsyscom.com>
Date: Tue, 27 Apr 2021 10:25:57 +0300
Subject: [PATCH] Implement timings on master

---
 apier/v1/timing.go | 91 ++++++++++++++++++++++++++++++++++++++++++++++
 utils/apitpdata.go |  5 +++
 utils/consts.go    |  3 ++
 3 files changed, 99 insertions(+)
 create mode 100644 apier/v1/timing.go

diff --git a/apier/v1/timing.go b/apier/v1/timing.go
new file mode 100644
index 000000000..eac708206
--- /dev/null
+++ b/apier/v1/timing.go
@@ -0,0 +1,91 @@
+/*
+Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
+Copyright (C) ITsysCOM GmbH
+
+This program is free software: you can redistribute it and/or modify
+it under the terms of the GNU General Public License as published by
+the Free Software Foundation, either version 3 of the License, or
+(at your option) any later version.
+
+This program is distributed in the hope that it will be useful,
+but WITHOUT ANY WARRANTY; without even the implied warranty of
+MERCHANTABILITY or FITNEtS FOR A PARTICULAR PURPOSE.  See the
+GNU General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program.  If not, see <http://www.gnu.org/licenses/>
+*/
+
+package v1
+
+import (
+	"time"
+
+	"github.com/cgrates/cgrates/utils"
+)
+
+// GetTiming returns a TPTiming object
+func (apierSv1 *APIerSv1) GetTiming(arg *utils.ArgsGetTimingID, reply *utils.TPTiming) (err error) {
+	if missing := utils.MissingStructFields(arg, []string{utils.ID}); len(missing) != 0 { //Params missing
+		return utils.NewErrMandatoryIeMissing(missing...)
+	}
+
+	tm, err := apierSv1.DataManager.GetTiming(arg.ID, false, utils.NonTransactional)
+	if err != nil {
+		return utils.APIErrorHandler(err)
+	}
+	*reply = *tm
+	return
+}
+
+// SetTiming alters/creates a TPTimingWithAPIOpts
+func (apierSv1 *APIerSv1) SetTiming(args *utils.TPTimingWithAPIOpts, reply *string) error {
+	if missing := utils.MissingStructFields(args.TPTiming, []string{utils.ID}); len(missing) != 0 {
+		return utils.NewErrMandatoryIeMissing(missing...)
+	}
+
+	if err := apierSv1.DataManager.SetTiming(args.TPTiming); err != nil {
+		return utils.APIErrorHandler(err)
+	}
+	//generate a loadID for CacheTimings and store it in database
+	loadID := time.Now().UnixNano()
+	if err := apierSv1.DataManager.SetLoadIDs(map[string]int64{utils.CacheTimings: loadID}); err != nil {
+		return utils.APIErrorHandler(err)
+	}
+	//handle caching for Timings
+	if err := apierSv1.CallCache(utils.IfaceAsString(args.APIOpts[utils.CacheOpt]), args.Tenant, utils.CacheTimings,
+		args.ID, nil, nil, args.APIOpts); err != nil {
+		return utils.APIErrorHandler(err)
+	}
+
+	*reply = utils.OK
+	return nil
+}
+
+// RemoveTiming removes a specific TPTimingWithAPIOpts instance
+func (apierSv1 *APIerSv1) RemoveTiming(args *utils.TPTimingWithAPIOpts, reply *string) error {
+	if missing := utils.MissingStructFields(args.TPTiming, []string{utils.ID}); len(missing) != 0 { //Params missing
+		return utils.NewErrMandatoryIeMissing(missing...)
+	}
+	tnt := args.Tenant
+	if tnt == utils.EmptyString {
+		tnt = apierSv1.Config.GeneralCfg().DefaultTenant
+	}
+	if err := apierSv1.DataManager.RemoveTiming(args.ID, utils.NonTransactional); err != nil {
+		return utils.APIErrorHandler(err)
+	}
+	//handle caching for Timings
+	if err := apierSv1.CallCache(utils.IfaceAsString(args.APIOpts[utils.CacheOpt]), tnt, utils.CacheTimings,
+		args.ID, nil, nil, args.APIOpts); err != nil {
+		return utils.APIErrorHandler(err)
+	}
+
+	//generate a loadID for CacheTimings and store it in database
+	loadID := time.Now().UnixNano()
+	if err := apierSv1.DataManager.SetLoadIDs(map[string]int64{utils.CacheTimings: loadID}); err != nil {
+		return utils.APIErrorHandler(err)
+	}
+
+	*reply = utils.OK
+	return nil
+}
diff --git a/utils/apitpdata.go b/utils/apitpdata.go
index 95c9b8dd8..b07237452 100644
--- a/utils/apitpdata.go
+++ b/utils/apitpdata.go
@@ -197,6 +197,11 @@ type TPTimingWithAPIOpts struct {
 	APIOpts map[string]interface{}
 }
 
+// ArgsGetTimingID is used by GetTiming API
+type ArgsGetTimingID struct {
+	ID string
+}
+
 func NewTiming(ID, years, mounths, mounthdays, weekdays, time string) (rt *TPTiming) {
 	rt = &TPTiming{}
 	rt.ID = ID
diff --git a/utils/consts.go b/utils/consts.go
index 78fdf73ee..2c50e07b8 100644
--- a/utils/consts.go
+++ b/utils/consts.go
@@ -1431,6 +1431,9 @@ const (
 	APIerSv1GetAccount                  = "APIerSv1.GetAccount"
 	APIerSv1GetAttributeProfileIDsCount = "APIerSv1.GetAttributeProfileIDsCount"
 	APIerSv1GetMaxUsage                 = "APIerSv1.GetMaxUsage"
+	APIerSv1GetTiming                   = "APIerSv1.GetTiming"
+	APIerSv1SetTiming                   = "APIerSv1.SetTiming"
+	APIerSv1RemoveTiming                = "APIerSv1.RemoveTiming"
 )
 
 // APIerSv1 TP APIs