123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109 |
- // Copyright 2021 The Gitea Authors. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package web
-
- import (
- goctx "context"
- "fmt"
- "net/http"
-
- "code.gitea.io/gitea/modules/context"
- "code.gitea.io/gitea/modules/web/routing"
- )
-
- type wrappedHandlerFunc func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func())
-
- func convertHandler(handler interface{}) wrappedHandlerFunc {
- funcInfo := routing.GetFuncInfo(handler)
- switch t := handler.(type) {
- case http.HandlerFunc:
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- if _, ok := resp.(context.ResponseWriter); !ok {
- resp = context.NewResponse(resp)
- }
- t(resp, req)
- if r, ok := resp.(context.ResponseWriter); ok && r.Status() > 0 {
- done = true
- }
- return done, deferrable
- }
- case func(http.ResponseWriter, *http.Request):
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- t(resp, req)
- if r, ok := resp.(context.ResponseWriter); ok && r.Status() > 0 {
- done = true
- }
- return done, deferrable
- }
-
- case func(ctx *context.Context):
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- ctx := context.GetContext(req)
- t(ctx)
- done = ctx.Written()
- return done, deferrable
- }
- case func(ctx *context.Context) goctx.CancelFunc:
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- ctx := context.GetContext(req)
- deferrable = t(ctx)
- done = ctx.Written()
- return done, deferrable
- }
- case func(*context.APIContext):
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- ctx := context.GetAPIContext(req)
- t(ctx)
- done = ctx.Written()
- return done, deferrable
- }
- case func(*context.APIContext) goctx.CancelFunc:
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- ctx := context.GetAPIContext(req)
- deferrable = t(ctx)
- done = ctx.Written()
- return done, deferrable
- }
- case func(*context.PrivateContext):
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- ctx := context.GetPrivateContext(req)
- t(ctx)
- done = ctx.Written()
- return done, deferrable
- }
- case func(*context.PrivateContext) goctx.CancelFunc:
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- ctx := context.GetPrivateContext(req)
- deferrable = t(ctx)
- done = ctx.Written()
- return done, deferrable
- }
- case func(http.Handler) http.Handler:
- return func(resp http.ResponseWriter, req *http.Request, others ...wrappedHandlerFunc) (done bool, deferrable func()) {
- next := http.HandlerFunc(func(http.ResponseWriter, *http.Request) {})
- if len(others) > 0 {
- next = wrapInternal(others)
- }
- routing.UpdateFuncInfo(req.Context(), funcInfo)
- if _, ok := resp.(context.ResponseWriter); !ok {
- resp = context.NewResponse(resp)
- }
- t(next).ServeHTTP(resp, req)
- if r, ok := resp.(context.ResponseWriter); ok && r.Status() > 0 {
- done = true
- }
- return done, deferrable
- }
- default:
- panic(fmt.Sprintf("Unsupported handler type: %#v", t))
- }
- }
|