mirror of
https://github.com/caddyserver/caddy.git
synced 2024-12-16 17:43:37 +08:00
fb63e2e40c
* feat: add generic response interceptors * fix: cs * rename intercept * add some docs * @francislavoie review (first round) * Update modules/caddyhttp/intercept/intercept.go Co-authored-by: Francis Lavoie <lavofr@gmail.com> * shorthands: ir to resp * mark exported symbols as experimental --------- Co-authored-by: Francis Lavoie <lavofr@gmail.com>
95 lines
3.6 KiB
Go
95 lines
3.6 KiB
Go
package httpcaddyfile
|
|
|
|
import (
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/caddyserver/caddy/v2/caddyconfig/caddyfile"
|
|
)
|
|
|
|
type ComplexShorthandReplacer struct {
|
|
search *regexp.Regexp
|
|
replace string
|
|
}
|
|
|
|
type ShorthandReplacer struct {
|
|
complex []ComplexShorthandReplacer
|
|
simple *strings.Replacer
|
|
}
|
|
|
|
func NewShorthandReplacer() ShorthandReplacer {
|
|
// replace shorthand placeholders (which are convenient
|
|
// when writing a Caddyfile) with their actual placeholder
|
|
// identifiers or variable names
|
|
replacer := strings.NewReplacer(placeholderShorthands()...)
|
|
|
|
// these are placeholders that allow a user-defined final
|
|
// parameters, but we still want to provide a shorthand
|
|
// for those, so we use a regexp to replace
|
|
regexpReplacements := []ComplexShorthandReplacer{
|
|
{regexp.MustCompile(`{header\.([\w-]*)}`), "{http.request.header.$1}"},
|
|
{regexp.MustCompile(`{cookie\.([\w-]*)}`), "{http.request.cookie.$1}"},
|
|
{regexp.MustCompile(`{labels\.([\w-]*)}`), "{http.request.host.labels.$1}"},
|
|
{regexp.MustCompile(`{path\.([\w-]*)}`), "{http.request.uri.path.$1}"},
|
|
{regexp.MustCompile(`{file\.([\w-]*)}`), "{http.request.uri.path.file.$1}"},
|
|
{regexp.MustCompile(`{query\.([\w-]*)}`), "{http.request.uri.query.$1}"},
|
|
{regexp.MustCompile(`{re\.([\w-\.]*)}`), "{http.regexp.$1}"},
|
|
{regexp.MustCompile(`{vars\.([\w-]*)}`), "{http.vars.$1}"},
|
|
{regexp.MustCompile(`{rp\.([\w-\.]*)}`), "{http.reverse_proxy.$1}"},
|
|
{regexp.MustCompile(`{resp\.([\w-\.]*)}`), "{http.intercept.$1}"},
|
|
{regexp.MustCompile(`{err\.([\w-\.]*)}`), "{http.error.$1}"},
|
|
{regexp.MustCompile(`{file_match\.([\w-]*)}`), "{http.matchers.file.$1}"},
|
|
}
|
|
|
|
return ShorthandReplacer{
|
|
complex: regexpReplacements,
|
|
simple: replacer,
|
|
}
|
|
}
|
|
|
|
// placeholderShorthands returns a slice of old-new string pairs,
|
|
// where the left of the pair is a placeholder shorthand that may
|
|
// be used in the Caddyfile, and the right is the replacement.
|
|
func placeholderShorthands() []string {
|
|
return []string{
|
|
"{dir}", "{http.request.uri.path.dir}",
|
|
"{file}", "{http.request.uri.path.file}",
|
|
"{host}", "{http.request.host}",
|
|
"{hostport}", "{http.request.hostport}",
|
|
"{port}", "{http.request.port}",
|
|
"{method}", "{http.request.method}",
|
|
"{path}", "{http.request.uri.path}",
|
|
"{query}", "{http.request.uri.query}",
|
|
"{remote}", "{http.request.remote}",
|
|
"{remote_host}", "{http.request.remote.host}",
|
|
"{remote_port}", "{http.request.remote.port}",
|
|
"{scheme}", "{http.request.scheme}",
|
|
"{uri}", "{http.request.uri}",
|
|
"{uuid}", "{http.request.uuid}",
|
|
"{tls_cipher}", "{http.request.tls.cipher_suite}",
|
|
"{tls_version}", "{http.request.tls.version}",
|
|
"{tls_client_fingerprint}", "{http.request.tls.client.fingerprint}",
|
|
"{tls_client_issuer}", "{http.request.tls.client.issuer}",
|
|
"{tls_client_serial}", "{http.request.tls.client.serial}",
|
|
"{tls_client_subject}", "{http.request.tls.client.subject}",
|
|
"{tls_client_certificate_pem}", "{http.request.tls.client.certificate_pem}",
|
|
"{tls_client_certificate_der_base64}", "{http.request.tls.client.certificate_der_base64}",
|
|
"{upstream_hostport}", "{http.reverse_proxy.upstream.hostport}",
|
|
"{client_ip}", "{http.vars.client_ip}",
|
|
}
|
|
}
|
|
|
|
// ApplyToSegment replaces shorthand placeholder to its full placeholder, understandable by Caddy.
|
|
func (s ShorthandReplacer) ApplyToSegment(segment *caddyfile.Segment) {
|
|
if segment != nil {
|
|
for i := 0; i < len(*segment); i++ {
|
|
// simple string replacements
|
|
(*segment)[i].Text = s.simple.Replace((*segment)[i].Text)
|
|
// complex regexp replacements
|
|
for _, r := range s.complex {
|
|
(*segment)[i].Text = r.search.ReplaceAllString((*segment)[i].Text, r.replace)
|
|
}
|
|
}
|
|
}
|
|
}
|