diff options
Diffstat (limited to 'internal/web/web.go')
| -rw-r--r-- | internal/web/web.go | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/internal/web/web.go b/internal/web/web.go index e42dc16c3..4494f8ff9 100644 --- a/internal/web/web.go +++ b/internal/web/web.go @@ -75,13 +75,15 @@ const ( type Module struct { processor *processing.Processor eTagCache cache.Cache[string, eTagCacheEntry] + cookiePolicy apiutil.CookiePolicy isURIBlocked func(context.Context, *url.URL) (bool, error) } -func New(db db.DB, processor *processing.Processor) *Module { +func New(db db.DB, processor *processing.Processor, cookiePolicy apiutil.CookiePolicy) *Module { return &Module{ processor: processor, eTagCache: newETagCache(), + cookiePolicy: cookiePolicy, isURIBlocked: db.IsURIBlocked, } } @@ -107,7 +109,7 @@ func (m *Module) Route(r *router.Router, mi ...gin.HandlerFunc) { profileGroup.Use(middleware.SignatureCheck(m.isURIBlocked), middleware.CacheControl(middleware.CacheControlConfig{ Directives: []string{"no-store"}, })) - nollamas := middleware.NoLLaMas(m.processor.InstanceGetV1) + nollamas := middleware.NoLLaMas(m.cookiePolicy, m.processor.InstanceGetV1) profileGroup.Use(nollamas) profileGroup.Handle(http.MethodGet, "", m.profileGETHandler) // use empty path here since it's the base of the group profileGroup.Handle(http.MethodGet, statusPath, m.threadGETHandler) |