iris/core/handlerconv/from_std.go

package handlerconv

import (
	"fmt"
	"net/http"

	"github.com/kataras/iris/v12/context"
)

// FromStd converts native http.Handler & http.HandlerFunc to context.Handler.
//
// Supported form types:
//
//	.FromStd(h http.Handler)
//	.FromStd(func(w http.ResponseWriter, r *http.Request))
//	.FromStd(func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc))
func FromStd(handler interface{}) context.Handler {
	switch h := handler.(type) {
	case context.Handler:
		return h
	// case func(*context.Context):
	// 	return h
	case http.Handler:
		// handlerFunc.ServeHTTP(w,r)
		return func(ctx *context.Context) {
			h.ServeHTTP(ctx.ResponseWriter(), ctx.Request())
		}
	case func(http.ResponseWriter, *http.Request):
		// handlerFunc(w,r)
		return FromStd(http.HandlerFunc(h))
	case func(http.ResponseWriter, *http.Request, http.HandlerFunc):
		// handlerFunc(w,r, http.HandlerFunc)
		//
		return FromStdWithNext(h)
	case func(http.Handler) http.Handler:
		panic(fmt.Errorf(`
			Passed handler cannot be converted directly:
			- http.Handler(http.Handler)
			---------------------------------------------------------------------
			Please use the Application.WrapRouter method instead, example code:
			app := iris.New()
			// ...
			app.WrapRouter(func(w http.ResponseWriter, r *http.Request, router http.HandlerFunc) {
			    httpThirdPartyHandler(router).ServeHTTP(w, r)
			})`))
	default:
		// No valid handler passed
		panic(fmt.Errorf(`
			Passed argument is not a func(iris.Context) neither one of these types:
			- http.Handler
			- func(w http.ResponseWriter, r *http.Request)
			- func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc)
			---------------------------------------------------------------------
			It seems to be a %T points to: %v`, handler, handler))
	}
}

// FromStdWithNext receives a standar handler - middleware form - and returns a
// compatible context.Handler wrapper.
func FromStdWithNext(h func(w http.ResponseWriter, r *http.Request, next http.HandlerFunc)) context.Handler {
	return func(ctx *context.Context) {
		next := func(w http.ResponseWriter, r *http.Request) {
			ctx.ResetRequest(r)
			ctx.Next()
		}

		h(ctx.ResponseWriter(), ctx.Request(), next)
	}
}