forked from gitea/gitea
117 lines
3.1 KiB
Go
117 lines
3.1 KiB
Go
// Copyright 2021 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package web
|
|
|
|
import (
|
|
goctx "context"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"code.gitea.io/gitea/modules/context"
|
|
"code.gitea.io/gitea/modules/web/routing"
|
|
)
|
|
|
|
// Wrap converts all kinds of routes to standard library one
|
|
func Wrap(handlers ...interface{}) http.HandlerFunc {
|
|
if len(handlers) == 0 {
|
|
panic("No handlers found")
|
|
}
|
|
|
|
ourHandlers := make([]wrappedHandlerFunc, 0, len(handlers))
|
|
|
|
for _, handler := range handlers {
|
|
ourHandlers = append(ourHandlers, convertHandler(handler))
|
|
}
|
|
return wrapInternal(ourHandlers)
|
|
}
|
|
|
|
func wrapInternal(handlers []wrappedHandlerFunc) http.HandlerFunc {
|
|
return func(resp http.ResponseWriter, req *http.Request) {
|
|
var defers []func()
|
|
defer func() {
|
|
for i := len(defers) - 1; i >= 0; i-- {
|
|
defers[i]()
|
|
}
|
|
}()
|
|
for i := 0; i < len(handlers); i++ {
|
|
handler := handlers[i]
|
|
others := handlers[i+1:]
|
|
done, deferrable := handler(resp, req, others...)
|
|
if deferrable != nil {
|
|
defers = append(defers, deferrable)
|
|
}
|
|
if done {
|
|
return
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Middle wrap a context function as a chi middleware
|
|
func Middle(f func(ctx *context.Context)) func(next http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(f)
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
ctx := context.GetContext(req)
|
|
f(ctx)
|
|
if ctx.Written() {
|
|
return
|
|
}
|
|
next.ServeHTTP(ctx.Resp, ctx.Req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// MiddleCancel wrap a context function as a chi middleware
|
|
func MiddleCancel(f func(ctx *context.Context) goctx.CancelFunc) func(netx http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(f)
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
ctx := context.GetContext(req)
|
|
cancel := f(ctx)
|
|
if cancel != nil {
|
|
defer cancel()
|
|
}
|
|
if ctx.Written() {
|
|
return
|
|
}
|
|
next.ServeHTTP(ctx.Resp, ctx.Req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// MiddleAPI wrap a context function as a chi middleware
|
|
func MiddleAPI(f func(ctx *context.APIContext)) func(next http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(f)
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
ctx := context.GetAPIContext(req)
|
|
f(ctx)
|
|
if ctx.Written() {
|
|
return
|
|
}
|
|
next.ServeHTTP(ctx.Resp, ctx.Req)
|
|
})
|
|
}
|
|
}
|
|
|
|
// WrapWithPrefix wraps a provided handler function at a prefix
|
|
func WrapWithPrefix(pathPrefix string, handler http.HandlerFunc, friendlyName ...string) func(next http.Handler) http.Handler {
|
|
funcInfo := routing.GetFuncInfo(handler, friendlyName...)
|
|
|
|
return func(next http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(resp http.ResponseWriter, req *http.Request) {
|
|
if !strings.HasPrefix(req.URL.Path, pathPrefix) {
|
|
next.ServeHTTP(resp, req)
|
|
return
|
|
}
|
|
routing.UpdateFuncInfo(req.Context(), funcInfo)
|
|
handler(resp, req)
|
|
})
|
|
}
|
|
}
|