Add #182 Single tenant and multi tenant configuration

src/config/config.go

@@ -79,11 +79,12 @@ type Configuration struct {
 		UserName string
 		Password string
 	}
-	Iam      Iam
-	Twilio   Twilio
-	Firebase Firebase
-	Smtp     SmtpConfig
-	Webhooks WebhooksConfig
+	ApplicationMode string
+	Iam             Iam
+	Twilio          Twilio
+	Firebase        Firebase
+	Smtp            SmtpConfig
+	Webhooks        WebhooksConfig
 }
 
 // Load the config file

src/config/constants.go

@@ -7,3 +7,9 @@ const (
 	ContentTypeImage          = "image/jpeg"
 	ContentTypeFormURLEncoded = "application/x-www-form-urlencoded"
 )
+
+// Application mode
+const (
+	SingleTenat = "single-tenant"
+	MultiTenant = "multi-tenant"
+)

src/handler/organization_handler.go

@@ -134,6 +134,16 @@ func GetOrganizationByID(w http.ResponseWriter, r *http.Request) {
 	w.Write(response)
 }
 
+// GetOrganizationId Gets an organization Id.
+func GetOrganizationId() (string, error) {
+	org, err := org.GetOrganization()
+	if err != nil {
+		log.Printf("Failed to get organization")
+		return "", err
+	}
+	return org.ID.Hex(), err
+}
+
 type orgUpdateReq struct {
 	Name        string
 	Location    string

src/main/main.go

@@ -13,6 +13,7 @@ import (
 	"github.com/bb-consent/api/src/firebaseUtils"
 	"github.com/bb-consent/api/src/handler"
 	"github.com/bb-consent/api/src/kafkaUtils"
+	"github.com/bb-consent/api/src/middleware"
 	"github.com/bb-consent/api/src/notifications"
 	"github.com/bb-consent/api/src/token"
 	"github.com/bb-consent/api/src/webhookdispatcher"
@@ -82,6 +83,9 @@ func main() {
 			firebaseUtils.Init(config)
 			log.Println("Firebase initialized")
 
+			middleware.ApplicationModeInit(config)
+			log.Println("Application mode initialized")
+
 			// setup casbin auth rules
 			authEnforcer, err := casbin.NewEnforcer("/opt/bb-consent/api/config/auth_model.conf", "/opt/bb-consent/api/config/rbac_policy.csv")
 			if err != nil {

src/middleware/middleware.go

@@ -7,12 +7,13 @@ import (
 	"time"
 
 	"github.com/bb-consent/api/src/apikey"
+	"github.com/bb-consent/api/src/handler"
 	"github.com/bb-consent/api/src/rbac"
 	"github.com/casbin/casbin/v2"
 	"github.com/gorilla/mux"
 
 	"github.com/bb-consent/api/src/common"
-	"github.com/bb-consent/api/src/handler"
+	"github.com/bb-consent/api/src/config"
 	"github.com/bb-consent/api/src/token"
 	"github.com/bb-consent/api/src/user"
 )
@@ -201,3 +202,33 @@ func Authorize(e *casbin.Enforcer) Middleware {
 		}
 	}
 }
+
+var ApplicationMode string
+
+func ApplicationModeInit(config *config.Configuration) {
+	ApplicationMode = config.ApplicationMode
+}
+
+// SetApplicationMode sets application modes for routes to either single tenant or multi tenant
+func SetApplicationMode() Middleware {
+	// Create a new Middleware
+	return func(f http.HandlerFunc) http.HandlerFunc {
+
+		// Define the http.HandlerFunc
+		return func(w http.ResponseWriter, r *http.Request) {
+
+			if ApplicationMode == config.SingleTenat {
+				organizationId, err := handler.GetOrganizationId()
+				if err != nil {
+					m := "failed to find organization"
+					common.HandleError(w, http.StatusBadRequest, m, err)
+					return
+				}
+				r.Header.Set("OrganizationID", organizationId)
+			}
+
+			// Call the next middleware/handler in chain
+			f(w, r)
+		}
+	}
+}

src/org/organizations.go

@@ -259,6 +259,15 @@ func Get(organizationID string) (Organization, error) {
 	return result, err
 }
 
+// Get Gets a single organization
+func GetOrganization() (Organization, error) {
+
+	var result Organization
+	err := collection().FindOne(context.TODO(), bson.M{}).Decode(&result)
+
+	return result, err
+}
+
 // Update Updates the organization
 func Update(org Organization) (Organization, error) {