mirror of
https://github.com/yusing/godoxy.git
synced 2026-03-14 06:05:32 +01:00
53f3397b7a
* Add comprehensive post-request rules support for response phase * Enable response body, status, and header manipulation via set commands * Refactor command handlers to support both request and response phases * Implement response modifier system for post-request template execution * Support response-based rule matching with status and header checks * Add comprehensive benchmarks for matcher performance * Refactor authentication and proxying commands for unified error handling * Support negated conditions with ! * Enhance error handling, error formatting and validation * Routes: add `rule_file` field with rule preset support * Environment variable substitution: now supports variables without `GODOXY_` prefix * new conditions: * `on resp_header <key> [<value>]` * `on status <status>` * new commands: * `require_auth` * `set resp_header <key> <template>` * `set resp_body <template>` * `set status <code>` * `log <level> <path> <template>` * `notify <level> <provider> <title_template> <body_template>`
109 lines
2.3 KiB
Go
109 lines
2.3 KiB
Go
package rules
|
|
|
|
import (
|
|
"net"
|
|
"net/http"
|
|
"net/url"
|
|
"sync"
|
|
)
|
|
|
|
// Cache is a map of cached values for a request.
|
|
// It prevents the same value from being parsed multiple times.
|
|
type (
|
|
Cache map[string]any
|
|
UpdateFunc[T any] func(T) T
|
|
)
|
|
|
|
const (
|
|
cacheKeyQueries = "queries"
|
|
cacheKeyCookies = "cookies"
|
|
cacheKeyRemoteIP = "remote_ip"
|
|
cacheKeyBasicAuth = "basic_auth"
|
|
)
|
|
|
|
var cachePool = sync.Pool{
|
|
New: func() any {
|
|
return make(Cache)
|
|
},
|
|
}
|
|
|
|
// NewCache returns a new Cached.
|
|
func NewCache() Cache {
|
|
return cachePool.Get().(Cache)
|
|
}
|
|
|
|
// Release clear the contents of the Cached and returns it to the pool.
|
|
func (c Cache) Release() {
|
|
clear(c)
|
|
cachePool.Put(c)
|
|
}
|
|
|
|
// GetQueries returns the queries.
|
|
// If r does not have queries, an empty map is returned.
|
|
func (c Cache) GetQueries(r *http.Request) url.Values {
|
|
v, ok := c[cacheKeyQueries]
|
|
if !ok {
|
|
v = r.URL.Query()
|
|
c[cacheKeyQueries] = v
|
|
}
|
|
return v.(url.Values)
|
|
}
|
|
|
|
func (c Cache) UpdateQueries(r *http.Request, update func(url.Values)) {
|
|
queries := c.GetQueries(r)
|
|
update(queries)
|
|
r.URL.RawQuery = queries.Encode()
|
|
}
|
|
|
|
// GetCookies returns the cookies.
|
|
// If r does not have cookies, an empty slice is returned.
|
|
func (c Cache) GetCookies(r *http.Request) []*http.Cookie {
|
|
v, ok := c[cacheKeyCookies]
|
|
if !ok {
|
|
v = r.Cookies()
|
|
c[cacheKeyCookies] = v
|
|
}
|
|
return v.([]*http.Cookie)
|
|
}
|
|
|
|
func (c Cache) UpdateCookies(r *http.Request, update UpdateFunc[[]*http.Cookie]) {
|
|
cookies := update(c.GetCookies(r))
|
|
c[cacheKeyCookies] = cookies
|
|
r.Header.Del("Cookie")
|
|
for _, cookie := range cookies {
|
|
r.AddCookie(cookie)
|
|
}
|
|
}
|
|
|
|
// GetRemoteIP returns the remote ip address.
|
|
// If r.RemoteAddr is not a valid ip address, nil is returned.
|
|
func (c Cache) GetRemoteIP(r *http.Request) net.IP {
|
|
v, ok := c[cacheKeyRemoteIP]
|
|
if !ok {
|
|
host, _, err := net.SplitHostPort(r.RemoteAddr)
|
|
if err != nil {
|
|
host = r.RemoteAddr
|
|
}
|
|
v = net.ParseIP(host)
|
|
c[cacheKeyRemoteIP] = v
|
|
}
|
|
return v.(net.IP)
|
|
}
|
|
|
|
// GetBasicAuth returns *Credentials the basic auth username and password.
|
|
// If r does not have basic auth, nil is returned.
|
|
func (c Cache) GetBasicAuth(r *http.Request) *Credentials {
|
|
v, ok := c[cacheKeyBasicAuth]
|
|
if !ok {
|
|
u, p, ok := r.BasicAuth()
|
|
if ok {
|
|
v = &Credentials{u, []byte(p)}
|
|
c[cacheKeyBasicAuth] = v
|
|
} else {
|
|
c[cacheKeyBasicAuth] = nil
|
|
return nil
|
|
}
|
|
}
|
|
return v.(*Credentials)
|
|
}
|