gost/pkg/dialer/option.go
2022-03-14 17:43:37 +08:00

67 lines
1.1 KiB
Go

package dialer
import (
"crypto/tls"
"net/url"
"github.com/go-gost/gost/v3/pkg/common/net/dialer"
"github.com/go-gost/gost/v3/pkg/logger"
)
type Options struct {
Auth *url.Userinfo
TLSConfig *tls.Config
Logger logger.Logger
}
type Option func(opts *Options)
func AuthOption(auth *url.Userinfo) Option {
return func(opts *Options) {
opts.Auth = auth
}
}
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
}
}
type DialOptions struct {
Host string
NetDialer *dialer.NetDialer
}
type DialOption func(opts *DialOptions)
func HostDialOption(host string) DialOption {
return func(opts *DialOptions) {
opts.Host = host
}
}
func NetDialerDialOption(netd *dialer.NetDialer) DialOption {
return func(opts *DialOptions) {
opts.NetDialer = netd
}
}
type HandshakeOptions struct {
Addr string
}
type HandshakeOption func(opts *HandshakeOptions)
func AddrHandshakeOption(addr string) HandshakeOption {
return func(opts *HandshakeOptions) {
opts.Addr = addr
}
}