// SPDX-FileCopyrightText: 2020 Ethel Morgan
//
// SPDX-License-Identifier: MIT

package httputil

import (
	"net/http"
	"strings"

	"github.com/gorilla/mux"
)

// FormValues matches POST form values the same way that github.com/gorilla/mux's built-in Queries and Headers does.
func FormValues(keysAndValues ...string) mux.MatcherFunc {
	if len(keysAndValues)%2 != 0 {
		panic("an equal number of keys and values must be provided")
	}
	return func(r *http.Request, rm *mux.RouteMatch) bool {
		for i := 0; i < len(keysAndValues); i += 2 {
			key := keysAndValues[i]
			value := keysAndValues[i+1]

			formValue := r.FormValue(key)
			if formValue == "" {
				return false
			}

			if strings.HasPrefix(value, "{") && strings.HasSuffix(value, "}") {
				if rm.Vars == nil {
					rm.Vars = map[string]string{}
				}
				varKey := value[1 : len(value)-1]
				rm.Vars[varKey] = formValue
			} else {
				if formValue != value {
					return false
				}
			}
		}
		return true
	}
}