add tenant id into contest

vinceanalytics · Feb 20, 2024 · 1f4e9e4 · 1f4e9e4
1 parent a25eb3b
commit 1f4e9e4
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/internal/api/api.go b/internal/api/api.go
@@ -58,6 +58,9 @@ func New(ctx context.Context, o *v1.Config, tenants *tenant.Tenants) (*API, erro
 		tenants: tenants,
 	}
 	base := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
+		if strings.HasPrefix(r.URL.Path, "/api/v1/") {
+			r.WithContext(tenants.Load(ctx, r.URL.Query()))
+		}
 		w.Header().Add(vary, acceptEncoding)
 		code := &statsWriter{ResponseWriter: w, compress: acceptsGzip(r)}
 		defer func() {

diff --git a/internal/tenant/tenants.go b/internal/tenant/tenants.go
@@ -1,6 +1,11 @@
 package tenant
 
-import v1 "github.com/vinceanalytics/vince/gen/go/staples/v1"
+import (
+	"context"
+	"net/url"
+
+	v1 "github.com/vinceanalytics/vince/gen/go/staples/v1"
+)
 
 const Default = "staples"
 
@@ -60,6 +65,30 @@ func (t *Tenants) AllDomains() (o []*v1.Domain) {
 	return
 }
 
+func (t *Tenants) Load(ctx context.Context, q url.Values) context.Context {
+	v := q.Get("tenant_id")
+	if v == "" {
+		site := q.Get("site_id")
+		if site != "" {
+			s := t.Get(site)
+			if s != nil {
+				v = s.Id
+			}
+		}
+	}
+	return With(ctx, v)
+}
+
 func (t *Tenants) All() []*v1.Tenant {
 	return t.all
 }
+
+type tenantId struct{}
+
+func With(ctx context.Context, id string) context.Context {
+	return context.WithValue(ctx, tenantId{}, id)
+}
+
+func Get(ctx context.Context) string {
+	return ctx.Value(tenantId{}).(string)
+}