110 lines
2.1 KiB
Go
110 lines
2.1 KiB
Go
package handler
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"net/url"
|
|
|
|
"github.com/go-gost/core/auth"
|
|
"github.com/go-gost/core/bypass"
|
|
"github.com/go-gost/core/chain"
|
|
"github.com/go-gost/core/limiter/rate"
|
|
"github.com/go-gost/core/limiter/traffic"
|
|
"github.com/go-gost/core/logger"
|
|
"github.com/go-gost/core/metadata"
|
|
"github.com/go-gost/core/observer"
|
|
)
|
|
|
|
type Options struct {
|
|
Bypass bypass.Bypass
|
|
Router chain.Router
|
|
Auth *url.Userinfo
|
|
Auther auth.Authenticator
|
|
RateLimiter rate.RateLimiter
|
|
Limiter traffic.TrafficLimiter
|
|
TLSConfig *tls.Config
|
|
Logger logger.Logger
|
|
Observer observer.Observer
|
|
Service string
|
|
Netns string
|
|
}
|
|
|
|
type Option func(opts *Options)
|
|
|
|
func BypassOption(bypass bypass.Bypass) Option {
|
|
return func(opts *Options) {
|
|
opts.Bypass = bypass
|
|
}
|
|
}
|
|
|
|
func RouterOption(router chain.Router) Option {
|
|
return func(opts *Options) {
|
|
opts.Router = router
|
|
}
|
|
}
|
|
|
|
func AuthOption(auth *url.Userinfo) Option {
|
|
return func(opts *Options) {
|
|
opts.Auth = auth
|
|
}
|
|
}
|
|
|
|
func AutherOption(auther auth.Authenticator) Option {
|
|
return func(opts *Options) {
|
|
opts.Auther = auther
|
|
}
|
|
}
|
|
|
|
func RateLimiterOption(limiter rate.RateLimiter) Option {
|
|
return func(opts *Options) {
|
|
opts.RateLimiter = limiter
|
|
}
|
|
}
|
|
|
|
func TrafficLimiterOption(limiter traffic.TrafficLimiter) Option {
|
|
return func(opts *Options) {
|
|
opts.Limiter = limiter
|
|
}
|
|
}
|
|
|
|
func TLSConfigOption(tlsConfig *tls.Config) Option {
|
|
return func(opts *Options) {
|
|
opts.TLSConfig = tlsConfig
|
|
}
|
|
}
|
|
|
|
func LoggerOption(logger logger.Logger) Option {
|
|
return func(opts *Options) {
|
|
opts.Logger = logger
|
|
}
|
|
}
|
|
|
|
func ObserverOption(observer observer.Observer) Option {
|
|
return func(opts *Options) {
|
|
opts.Observer = observer
|
|
}
|
|
}
|
|
|
|
func ServiceOption(service string) Option {
|
|
return func(opts *Options) {
|
|
opts.Service = service
|
|
}
|
|
}
|
|
|
|
func NetnsOption(netns string) Option {
|
|
return func(opts *Options) {
|
|
opts.Netns = netns
|
|
}
|
|
}
|
|
|
|
type HandleOptions struct {
|
|
Metadata metadata.Metadata
|
|
}
|
|
|
|
type HandleOption func(opts *HandleOptions)
|
|
|
|
func MetadataHandleOption(md metadata.Metadata) HandleOption {
|
|
return func(opts *HandleOptions) {
|
|
opts.Metadata = md
|
|
}
|
|
}
|