From 662693c2c9e95ea44cae5cc6af1991f79a47ea7c Mon Sep 17 00:00:00 2001
From: Michael Quigley <michael@quigley.com>
Date: Mon, 27 Mar 2023 11:34:29 -0400
Subject: [PATCH] share action skeletons (#276)

---
 controller/limits/agent.go              | 26 ++++++++++++++++---------
 controller/limits/shareLimitAction.go   | 21 ++++++++++++++++++++
 controller/limits/shareRelaxAction.go   | 21 ++++++++++++++++++++
 controller/limits/shareWarningAction.go | 21 ++++++++++++++++++++
 4 files changed, 80 insertions(+), 9 deletions(-)
 create mode 100644 controller/limits/shareLimitAction.go
 create mode 100644 controller/limits/shareRelaxAction.go
 create mode 100644 controller/limits/shareWarningAction.go

diff --git a/controller/limits/agent.go b/controller/limits/agent.go
index ec2186e5..0417ba47 100644
--- a/controller/limits/agent.go
+++ b/controller/limits/agent.go
@@ -32,15 +32,23 @@ type Agent struct {
 }
 
 func NewAgent(cfg *Config, ifxCfg *metrics.InfluxConfig, zCfg *zrokEdgeSdk.Config, str *store.Store) (*Agent, error) {
-	return &Agent{
-		cfg:   cfg,
-		ifx:   newInfluxReader(ifxCfg),
-		zCfg:  zCfg,
-		str:   str,
-		queue: make(chan *metrics.Usage, 1024),
-		close: make(chan struct{}),
-		join:  make(chan struct{}),
-	}, nil
+	edge, err := zrokEdgeSdk.Client(zCfg)
+	if err != nil {
+		return nil, err
+	}
+	a := &Agent{
+		cfg:               cfg,
+		ifx:               newInfluxReader(ifxCfg),
+		zCfg:              zCfg,
+		str:               str,
+		queue:             make(chan *metrics.Usage, 1024),
+		shrWarningEnforce: []ShareAction{newShareWarningAction(str, edge)},
+		shrLimitEnforce:   []ShareAction{newShareLimitAction(str, edge)},
+		shrLimitRelax:     []ShareAction{newShareRelaxAction(str, edge)},
+		close:             make(chan struct{}),
+		join:              make(chan struct{}),
+	}
+	return a, nil
 }
 
 func (a *Agent) Start() {
diff --git a/controller/limits/shareLimitAction.go b/controller/limits/shareLimitAction.go
new file mode 100644
index 00000000..61655ef4
--- /dev/null
+++ b/controller/limits/shareLimitAction.go
@@ -0,0 +1,21 @@
+package limits
+
+import (
+	"github.com/openziti/edge/rest_management_api_client"
+	"github.com/openziti/zrok/controller/store"
+	"github.com/sirupsen/logrus"
+)
+
+type shareLimitAction struct {
+	str  *store.Store
+	edge *rest_management_api_client.ZitiEdgeManagement
+}
+
+func newShareLimitAction(str *store.Store, edge *rest_management_api_client.ZitiEdgeManagement) *shareLimitAction {
+	return &shareLimitAction{str, edge}
+}
+
+func (a *shareLimitAction) HandleShare(s *store.Share, rxBytes, txBytes int64, limit *BandwidthPerPeriod) error {
+	logrus.Infof("limiting '%v'", s.Token)
+	return nil
+}
diff --git a/controller/limits/shareRelaxAction.go b/controller/limits/shareRelaxAction.go
new file mode 100644
index 00000000..3c805dc7
--- /dev/null
+++ b/controller/limits/shareRelaxAction.go
@@ -0,0 +1,21 @@
+package limits
+
+import (
+	"github.com/openziti/edge/rest_management_api_client"
+	"github.com/openziti/zrok/controller/store"
+	"github.com/sirupsen/logrus"
+)
+
+type shareRelaxAction struct {
+	str  *store.Store
+	edge *rest_management_api_client.ZitiEdgeManagement
+}
+
+func newShareRelaxAction(str *store.Store, edge *rest_management_api_client.ZitiEdgeManagement) *shareRelaxAction {
+	return &shareRelaxAction{str, edge}
+}
+
+func (a *shareRelaxAction) HandleShare(s *store.Share, rxBytes, txBytes int64, limit *BandwidthPerPeriod) error {
+	logrus.Infof("relaxing '%v'", s.Token)
+	return nil
+}
diff --git a/controller/limits/shareWarningAction.go b/controller/limits/shareWarningAction.go
new file mode 100644
index 00000000..05085c74
--- /dev/null
+++ b/controller/limits/shareWarningAction.go
@@ -0,0 +1,21 @@
+package limits
+
+import (
+	"github.com/openziti/edge/rest_management_api_client"
+	"github.com/openziti/zrok/controller/store"
+	"github.com/sirupsen/logrus"
+)
+
+type shareWarningAction struct {
+	str  *store.Store
+	edge *rest_management_api_client.ZitiEdgeManagement
+}
+
+func newShareWarningAction(str *store.Store, edge *rest_management_api_client.ZitiEdgeManagement) *shareWarningAction {
+	return &shareWarningAction{str, edge}
+}
+
+func (a *shareWarningAction) HandleShare(s *store.Share, rxBytes, txBytes int64, limit *BandwidthPerPeriod) error {
+	logrus.Infof("warning '%v'", s.Token)
+	return nil
+}