mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-29 02:16:28 +02:00
08623ef346
* add tests/benchmarks for http1/http2 tcp tunnels and http1 websockets testenv: - add new TCP upstream - add websocket functions to HTTP upstream - add https support to mock idp (default on) - add new debug flags -env.bind-address and -env.use-trace-environ to allow changing the default bind address, and enabling otel environment based trace config, respectively * linter pass --------- Co-authored-by: Denis Mishin <dmishin@pomerium.com>
93 lines
2 KiB
Go
93 lines
2 KiB
Go
package testenv
|
|
|
|
import (
|
|
"fmt"
|
|
"net/url"
|
|
"strings"
|
|
|
|
"github.com/pomerium/pomerium/config"
|
|
"github.com/pomerium/pomerium/internal/testenv/values"
|
|
"github.com/pomerium/pomerium/pkg/policy/parser"
|
|
)
|
|
|
|
// PolicyRoute is a [Route] implementation suitable for most common use cases
|
|
// that can be used in implementations of [Upstream].
|
|
type PolicyRoute struct {
|
|
DefaultAttach
|
|
from values.Value[string]
|
|
to values.List[string]
|
|
edits []func(*config.Policy)
|
|
}
|
|
|
|
// Modify implements Route.
|
|
func (b *PolicyRoute) Modify(cfg *config.Config) {
|
|
to := make(config.WeightedURLs, 0, len(b.to))
|
|
for _, u := range b.to {
|
|
u, err := url.Parse(u.Value())
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
to = append(to, config.WeightedURL{URL: *u})
|
|
}
|
|
p := config.Policy{
|
|
From: b.from.Value(),
|
|
To: to,
|
|
}
|
|
for _, edit := range b.edits {
|
|
edit(&p)
|
|
}
|
|
cfg.Options.Policies = append(cfg.Options.Policies, p)
|
|
}
|
|
|
|
// From implements Route.
|
|
func (b *PolicyRoute) From(fromURL values.Value[string]) Route {
|
|
b.from = fromURL
|
|
return b
|
|
}
|
|
|
|
// To implements Route.
|
|
func (b *PolicyRoute) To(toURL values.Value[string]) Route {
|
|
b.to = append(b.to, toURL)
|
|
return b
|
|
}
|
|
|
|
// To implements Route.
|
|
func (b *PolicyRoute) Policy(edit func(*config.Policy)) Route {
|
|
b.edits = append(b.edits, edit)
|
|
return b
|
|
}
|
|
|
|
// PPL implements Route.
|
|
func (b *PolicyRoute) PPL(ppl string) Route {
|
|
pplPolicy, err := parser.ParseYAML(strings.NewReader(ppl))
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
b.edits = append(b.edits, func(p *config.Policy) {
|
|
p.Policy = &config.PPLPolicy{
|
|
Policy: pplPolicy,
|
|
}
|
|
})
|
|
return b
|
|
}
|
|
|
|
// To implements Route.
|
|
func (b *PolicyRoute) URL() values.Value[string] {
|
|
return b.from
|
|
}
|
|
|
|
type TCPRoute struct {
|
|
PolicyRoute
|
|
}
|
|
|
|
func (b *TCPRoute) From(fromURL values.Value[string]) Route {
|
|
b.from = values.Bind(fromURL, func(urlStr string) string {
|
|
from, _ := url.Parse(urlStr)
|
|
from.Scheme = "tcp+https"
|
|
from.Host = fmt.Sprintf("%s:%s", from.Hostname(), from.Port())
|
|
return from.String()
|
|
})
|
|
return b
|
|
}
|
|
|
|
var _ Route = (*TCPRoute)(nil)
|