Serve compliance pages via domain routing

Rename the compliancepage HTTP package to complianceportal and drop the slug-based id middleware. Compliance pages are now served exclusively over SNI-based custom-domain HTTPS, with the SNI middleware resolving the page by TLS server name and redirecting secondary domains to the canonical host. Signed-off-by: Bryan Frimin <bryan@probo.com>
2026-07-10 15:14:41 +02:00
parent a9a126899e
commit 860dafeaf2
6 changed files with 46 additions and 128 deletions
--- a/pkg/server/api/complianceportal/compliance_page_presence_middleware.go
+++ b/pkg/server/api/complianceportal/compliance_page_presence_middleware.go
@@ -0,0 +1,59 @@
+// Copyright (c) 2026 Probo Inc <hello@probo.com>.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+package complianceportal
+
+import (
+	"net/http"
+
+	"github.com/99designs/gqlgen/graphql"
+	"github.com/vektah/gqlparser/v2/gqlerror"
+	"go.gearno.de/kit/httpserver"
+	"go.probo.inc/probo/pkg/server/gqlutils"
+)
+
+func NewCompliancePagePresenceMiddleware() func(next http.Handler) http.Handler {
+	return func(next http.Handler) http.Handler {
+		return http.HandlerFunc(
+			func(w http.ResponseWriter, r *http.Request) {
+				compliancePage := CompliancePageFromContext(r.Context())
+
+				if compliancePage == nil {
+					httpserver.RenderJSON(
+						w,
+						http.StatusNotFound,
+						&graphql.Response{
+							Errors: gqlerror.List{
+								gqlutils.NotFoundf(
+									r.Context(),
+									"compliance page not found",
+								),
+							},
+						},
+					)
+
+					return
+				}
+
+				next.ServeHTTP(w, r)
+			},
+		)
+	}
+}
--- a/pkg/server/api/complianceportal/context.go
+++ b/pkg/server/api/complianceportal/context.go
@@ -0,0 +1,44 @@
+// Copyright (c) 2026 Probo Inc <hello@probo.com>.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+package complianceportal
+
+import (
+	"context"
+
+	"go.probo.inc/probo/pkg/coredata"
+)
+
+type ctxKey struct{ name string }
+
+var (
+	compliancePageKey        = &ctxKey{name: "compliance_page"}
+	compliancePageBaseURLKey = &ctxKey{name: "compliance_page_base_url"}
+)
+
+func CompliancePageFromContext(ctx context.Context) *coredata.TrustCenter {
+	page, _ := ctx.Value(compliancePageKey).(*coredata.TrustCenter)
+	return page
+}
+
+func CompliancePageBaseURLFromContext(ctx context.Context) *string {
+	page, _ := ctx.Value(compliancePageBaseURLKey).(*string)
+	return page
+}
--- a/pkg/server/api/complianceportal/markdown_handler.go
+++ b/pkg/server/api/complianceportal/markdown_handler.go
@@ -0,0 +1,94 @@
+// Copyright (c) 2026 Probo Inc <hello@probo.com>.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+package complianceportal
+
+import (
+	"net/http"
+
+	trust "go.probo.inc/probo/pkg/complianceportal/visitor"
+	"go.probo.inc/probo/pkg/coredata"
+)
+
+type Handler struct {
+	trustService *trust.Service
+}
+
+func NewHandler(trustService *trust.Service) *Handler {
+	return &Handler{trustService: trustService}
+}
+
+func (h *Handler) HandleLLMsTxt(w http.ResponseWriter, r *http.Request) {
+	tc := CompliancePageFromContext(r.Context())
+	if tc == nil {
+		http.Error(w, "not found", http.StatusNotFound)
+		return
+	}
+
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+
+	scope := coredata.NewScopeFromObjectID(tc.ID)
+
+	if err := h.trustService.RenderCompliancePageMarkdown(r.Context(), w, tc.ID, scope); err != nil {
+		http.Error(w, "internal server error", http.StatusInternalServerError)
+	}
+}
+
+func (h *Handler) HandleRobotsTxt(w http.ResponseWriter, r *http.Request) {
+	tc := CompliancePageFromContext(r.Context())
+	if tc == nil {
+		http.Error(w, "not found", http.StatusNotFound)
+		return
+	}
+
+	baseURL := CompliancePageBaseURLFromContext(r.Context())
+	if baseURL == nil {
+		http.Error(w, "not found", http.StatusNotFound)
+		return
+	}
+
+	w.Header().Set("Content-Type", "text/plain; charset=utf-8")
+
+	if err := h.trustService.RenderRobotsTxt(r.Context(), w, tc.SearchEngineIndexing, *baseURL); err != nil {
+		http.Error(w, "internal server error", http.StatusInternalServerError)
+	}
+}
+
+func (h *Handler) HandleSitemap(w http.ResponseWriter, r *http.Request) {
+	tc := CompliancePageFromContext(r.Context())
+	if tc == nil {
+		http.Error(w, "not found", http.StatusNotFound)
+		return
+	}
+
+	baseURL := CompliancePageBaseURLFromContext(r.Context())
+	if baseURL == nil {
+		http.Error(w, "not found", http.StatusNotFound)
+		return
+	}
+
+	w.Header().Set("Content-Type", "application/xml; charset=utf-8")
+
+	scope := coredata.NewScopeFromObjectID(tc.ID)
+
+	if err := h.trustService.RenderSitemap(r.Context(), w, tc.ID, scope, *baseURL); err != nil {
+		http.Error(w, "internal server error", http.StatusInternalServerError)
+	}
+}
--- a/pkg/server/api/complianceportal/member_provisioning_middleware.go
+++ b/pkg/server/api/complianceportal/member_provisioning_middleware.go
@@ -0,0 +1,68 @@
+// Copyright (c) 2026 Probo Inc <hello@probo.com>.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+package complianceportal
+
+import (
+	"net/http"
+
+	"github.com/99designs/gqlgen/graphql"
+	"github.com/vektah/gqlparser/v2/gqlerror"
+	"go.gearno.de/kit/httpserver"
+	"go.gearno.de/kit/log"
+	trust "go.probo.inc/probo/pkg/complianceportal/visitor"
+	"go.probo.inc/probo/pkg/server/api/authn"
+	"go.probo.inc/probo/pkg/server/gqlutils"
+)
+
+func NewMemberProvisioningMiddleware(trustSvc *trust.Service, logger *log.Logger) 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()
+
+				identity := authn.IdentityFromContext(ctx)
+				if identity == nil {
+					next.ServeHTTP(w, r)
+					return
+				}
+
+				compliancePage := CompliancePageFromContext(r.Context())
+
+				if _, err := trustSvc.ProvisionPortalMember(ctx, compliancePage.ID, identity.ID); err != nil {
+					logger.ErrorCtx(ctx, "cannot provision member", log.Error(err))
+					httpserver.RenderJSON(
+						w,
+						http.StatusInternalServerError,
+						&graphql.Response{
+							Errors: gqlerror.List{
+								gqlutils.Internal(ctx),
+							},
+						},
+					)
+
+					return
+				}
+
+				next.ServeHTTP(w, r)
+			},
+		)
+	}
+}
--- a/pkg/server/api/complianceportal/sni_middleware.go
+++ b/pkg/server/api/complianceportal/sni_middleware.go
@@ -0,0 +1,124 @@
+// Copyright (c) 2026 Probo Inc <hello@probo.com>.
+//
+// Permission is hereby granted, free of charge, to any person obtaining a copy
+// of this software and associated documentation files (the "Software"), to deal
+// in the Software without restriction, including without limitation the rights
+// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+// copies of the Software, and to permit persons to whom the Software is
+// furnished to do so, subject to the following conditions:
+//
+// The above copyright notice and this permission notice shall be included in
+// all copies or substantial portions of the Software.
+//
+// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+// SOFTWARE.
+
+package complianceportal
+
+import (
+	"context"
+	"errors"
+	"net/http"
+	"net/url"
+	"strings"
+
+	"github.com/99designs/gqlgen/graphql"
+	"github.com/vektah/gqlparser/v2/gqlerror"
+	"go.gearno.de/kit/httpserver"
+	trust "go.probo.inc/probo/pkg/complianceportal/visitor"
+	"go.probo.inc/probo/pkg/server/gqlutils"
+)
+
+func NewSNIMiddleware(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()
+
+			if r.TLS == nil {
+				next.ServeHTTP(w, r)
+				return
+			}
+
+			compliancePage, err := trustSvc.GetPortalByDomainName(ctx, r.TLS.ServerName)
+			if err != nil {
+				if errors.Is(err, trust.ErrPageNotFound) {
+					next.ServeHTTP(w, r)
+					return
+				}
+
+				httpserver.RenderJSON(
+					w,
+					http.StatusInternalServerError,
+					&graphql.Response{
+						Errors: gqlerror.List{
+							gqlutils.Internal(ctx),
+						},
+					},
+				)
+
+				return
+			}
+
+			// Redirect secondary domains to the canonical host so a compliance
+			// page is only ever served under a single origin. ACME HTTP-01
+			// challenges are handled upstream and never reach this middleware.
+			if !strings.HasPrefix(r.URL.Path, "/.well-known/") {
+				canonicalHost, err := trustSvc.GetPortalEffectiveCanonicalHost(ctx, compliancePage.ID)
+				if err != nil {
+					httpserver.RenderJSON(
+						w,
+						http.StatusInternalServerError,
+						&graphql.Response{
+							Errors: gqlerror.List{
+								gqlutils.Internal(ctx),
+							},
+						},
+					)
+
+					return
+				}
+
+				if canonicalHost != "" && canonicalHost != r.Host {
+					target := &url.URL{
+						Scheme:   "https",
+						Host:     canonicalHost,
+						Path:     r.URL.Path,
+						RawQuery: r.URL.RawQuery,
+					}
+
+					http.Redirect(w, r, target.String(), http.StatusMovedPermanently)
+
+					return
+				}
+			}
+
+			baseURL := &url.URL{
+				Host:   r.Host,
+				Path:   r.URL.Path,
+				Scheme: "https",
+			}
+			baseURLString := baseURL.String()
+
+			ctx = context.WithValue(
+				ctx,
+				compliancePageBaseURLKey,
+				&baseURLString,
+			)
+			r = r.WithContext(ctx)
+
+			if compliancePage.Active {
+				ctx = context.WithValue(ctx, compliancePageKey, compliancePage)
+				next.ServeHTTP(w, r.WithContext(ctx))
+
+				return
+			}
+
+			next.ServeHTTP(w, r)
+		})
+	}
+}