modify proxy handler

internal/proxy/hooks/hooks.go

@@ -0,0 +1,60 @@
+package hooks
+
+import (
+	"net/http"
+	"net/url"
+	"strings"
+	"sync"
+)
+
+// CachePolicy mirrors the proxy cache policy structure.
+type CachePolicy struct {
+	AllowCache        bool
+	AllowStore        bool
+	RequireRevalidate bool
+}
+
+// RequestContext exposes route/request details without importing server internals.
+type RequestContext struct {
+	HubName      string
+	Domain       string
+	HubType      string
+	ModuleKey    string
+	RolloutFlag  string
+	UpstreamHost string
+	Method       string
+}
+
+// Hooks describes customization points for module-specific behavior.
+type Hooks struct {
+	NormalizePath   func(ctx *RequestContext, cleanPath string) string
+	ResolveUpstream func(ctx *RequestContext, base *url.URL, cleanPath string, rawQuery []byte) *url.URL
+	RewriteResponse func(ctx *RequestContext, resp *http.Response, cleanPath string) (*http.Response, error)
+	CachePolicy     func(ctx *RequestContext, locatorPath string, current CachePolicy) CachePolicy
+	ContentType     func(ctx *RequestContext, locatorPath string) string
+}
+
+var registry sync.Map
+
+// Register stores hooks for the given module key.
+func Register(moduleKey string, hooks Hooks) {
+	key := strings.ToLower(strings.TrimSpace(moduleKey))
+	if key == "" {
+		return
+	}
+	registry.Store(key, hooks)
+}
+
+// For retrieves hooks associated with a module key.
+func For(moduleKey string) (Hooks, bool) {
+	key := strings.ToLower(strings.TrimSpace(moduleKey))
+	if key == "" {
+		return Hooks{}, false
+	}
+	if value, ok := registry.Load(key); ok {
+		if hooks, ok := value.(Hooks); ok {
+			return hooks, true
+		}
+	}
+	return Hooks{}, false
+}