Use trust center from context

Signed-off-by: Émile Ré <emile@getprobo.com>

pkg/server/api/compliancepage/id_middleware.go

@@ -0,0 +1,56 @@
+// Copyright (c) 2025 Probo Inc <hello@getprobo.com>.
+//
+// Permission to use, copy, modify, and/or distribute this software for any
+// purpose with or without fee is hereby granted, provided that the above
+// copyright notice and this permission notice appear in all copies.
+//
+// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
+// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
+// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
+// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
+// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
+// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
+// PERFORMANCE OF THIS SOFTWARE.
+
+package compliancepage
+
+import (
+	"context"
+	"net/http"
+
+	"github.com/go-chi/chi/v5"
+	"go.probo.inc/probo/pkg/gid"
+	"go.probo.inc/probo/pkg/trust"
+)
+
+func NewIDMiddleware(trustSvc *trust.Service) func(next http.Handler) http.Handler {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(
+			func(w http.ResponseWriter, r *http.Request) {
+				ctx := r.Context()
+				// TODO: remove slug support
+				value := chi.URLParam(r, "slugOrId")
+
+				if id, err := gid.ParseGID(value); err == nil {
+					compliancePage, err := trustSvc.Get(ctx, id)
+					if err != nil || !compliancePage.Active {
+						next.ServeHTTP(w, r)
+						return
+					}
+
+					ctx = context.WithValue(ctx, compliancePageKey, compliancePage)
+					next.ServeHTTP(w, r.WithContext(ctx))
+					return
+				}
+
+				if compliancePage, err := trustSvc.GetBySlug(ctx, value); err == nil && compliancePage.Active {
+					ctx = context.WithValue(ctx, compliancePageKey, compliancePage)
+					next.ServeHTTP(w, r.WithContext(ctx))
+					return
+				}
+
+				next.ServeHTTP(w, r)
+			},
+		)
+	}
+}