gorilla.go

// Mux vars extension for github.com/gorilla/mux package.

package httpin

import (
	"mime/multipart"
	"net/http"
)

// GorillaMuxVarsFunc is mux.Vars
type GorillaMuxVarsFunc func(*http.Request) map[string]string

// UseGorillaMux registers a new directive executor which can extract values
// from `mux.Vars`, i.e. path variables.
// https://ggicci.github.io/httpin/integrations/gorilla
//
// Usage:
//
//    func init() {
//        httpin.UseGorillaMux("path", mux.Vars)
//    }
func UseGorillaMux(executor string, fnVars GorillaMuxVarsFunc) {
	RegisterDirectiveExecutor(executor, &gorillaMuxVarsExtractor{Vars: fnVars}, nil)
}

type gorillaMuxVarsExtractor struct {
	Vars GorillaMuxVarsFunc
}

func (mux *gorillaMuxVarsExtractor) Execute(ctx *DirectiveContext) error {
	var kvs = make(map[string][]string)
	for key, value := range mux.Vars(ctx.Request) {
		kvs[key] = []string{value}
	}

	extractor := &extractor{
		Form: multipart.Form{
			Value: kvs,
		},
	}
	return extractor.Execute(ctx)
}