32 lines
781 B
Go
32 lines
781 B
Go
package request
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
)
|
|
|
|
// DontFollow is a redirect policy that does not follow redirects
|
|
func DontFollow(req *http.Request, via []*http.Request) error {
|
|
return http.ErrUseLastResponse
|
|
}
|
|
|
|
type InterceptResult = func() (string, error)
|
|
|
|
// InterceptRedirect captures a redirect url parameter
|
|
func InterceptRedirect(param string, stop bool) (func(req *http.Request, via []*http.Request) error, InterceptResult) {
|
|
var val string
|
|
return func(req *http.Request, via []*http.Request) error {
|
|
if val == "" {
|
|
if val = req.URL.Query().Get(param); val != "" && stop {
|
|
return http.ErrUseLastResponse
|
|
}
|
|
}
|
|
return nil
|
|
},
|
|
func() (string, error) {
|
|
if val == "" {
|
|
return "", fmt.Errorf("%s not found", param)
|
|
}
|
|
return val, nil
|
|
}
|
|
}
|