From c4b31313a684cc9202dc0f51d6577e8681eba19f Mon Sep 17 00:00:00 2001
From: Jim <jlambert@hashicorp.com>
Date: Mon, 17 Apr 2023 08:48:42 -0400
Subject: [PATCH] feat (controller): define a ControllerExtension

---
 internal/daemon/controller/controller.go | 17 ++++++++++++++++-
 internal/globals/globals.go              | 16 ++++++++++++++++
 2 files changed, 32 insertions(+), 1 deletion(-)
 create mode 100644 internal/globals/globals.go

diff --git a/internal/daemon/controller/controller.go b/internal/daemon/controller/controller.go
index d58808892..e7a40286a 100644
--- a/internal/daemon/controller/controller.go
+++ b/internal/daemon/controller/controller.go
@@ -26,6 +26,7 @@ import (
 	"github.com/hashicorp/boundary/internal/daemon/controller/internal/metric"
 	"github.com/hashicorp/boundary/internal/db"
 	"github.com/hashicorp/boundary/internal/errors"
+	intglobals "github.com/hashicorp/boundary/internal/globals"
 	pluginhost "github.com/hashicorp/boundary/internal/host/plugin"
 	"github.com/hashicorp/boundary/internal/host/static"
 	"github.com/hashicorp/boundary/internal/iam"
@@ -80,6 +81,7 @@ var (
 	downstreamersFactory           func(context.Context, string, string) (common.Downstreamers, error)
 	downstreamWorkersTickerFactory func(context.Context, string, string, common.Downstreamers, downstreamReceiver) (downstreamWorkersTicker, error)
 	commandClientFactory           func(context.Context, *Controller) error
+	extControllerFactory           func(ctx context.Context, c *Controller, r db.Reader, w db.Writer) (intglobals.ControllerExtension, error)
 )
 
 type Controller struct {
@@ -142,6 +144,9 @@ type Controller struct {
 	HealthService *health.Service
 
 	pkiConnManager *cluster.DownstreamManager
+
+	// ControllerExtension defines a std way to extend the controller
+	ControllerExtension intglobals.ControllerExtension
 }
 
 func New(ctx context.Context, conf *Config) (*Controller, error) {
@@ -415,6 +420,12 @@ func New(ctx context.Context, conf *Config) (*Controller, error) {
 		}
 	}
 
+	if extControllerFactory != nil {
+		if c.ControllerExtension, err = extControllerFactory(ctx, c, dbase, dbase); err != nil {
+			return nil, fmt.Errorf("unable to extend controller: %w", err)
+		}
+	}
+
 	return c, nil
 }
 
@@ -507,7 +518,11 @@ func (c *Controller) Start() error {
 			}
 		}()
 	}
-
+	if c.ControllerExtension != nil {
+		if err := c.ControllerExtension.Start(c.baseContext); err != nil {
+			return fmt.Errorf("error starting controller extension: %w", err)
+		}
+	}
 	return nil
 }
 
diff --git a/internal/globals/globals.go b/internal/globals/globals.go
new file mode 100644
index 000000000..e360c1dc2
--- /dev/null
+++ b/internal/globals/globals.go
@@ -0,0 +1,16 @@
+// Copyright (c) HashiCorp, Inc.
+// SPDX-License-Identifier: MPL-2.0
+
+package globals
+
+import (
+	"context"
+)
+
+// ControllerExtension defines the interface implemented
+// by the enterprise controller extension. This type
+// can be used to pass the controller extension into
+// handlers and repositories.
+type ControllerExtension interface {
+	Start(context.Context) error
+}