67 lines
1.3 KiB
Go
67 lines
1.3 KiB
Go
|
package rebound
|
||
|
|
|
||
|
|
import (
|
||
|
|
"log"
|
||
|
|
|
||
|
|
"forge.cadoles.com/wpetit/rebound/http"
|
||
|
|
"forge.cadoles.com/wpetit/rebound/ssh"
|
||
|
|
"github.com/caarlos0/env/v9"
|
||
|
|
"github.com/pkg/errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Options struct {
|
||
|
|
Address string `env:"REBOUND_ADDRESS"`
|
||
|
|
Logger func(message string, args ...any)
|
||
|
|
SSH *ssh.Options `envPrefix:"REBOUND_SSH_"`
|
||
|
|
HTTP *http.Options `envPrefix:"REBOUND_HTTP_"`
|
||
|
|
}
|
||
|
|
|
||
|
|
func (o *Options) ParseEnv() error {
|
||
|
|
if err := env.Parse(o); err != nil {
|
||
|
|
return errors.WithStack(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
type OptionFunc func(*Options)
|
||
|
|
|
||
|
|
func DefaultOptions() *Options {
|
||
|
|
return &Options{
|
||
|
|
Address: "127.0.0.1:2222",
|
||
|
|
Logger: log.Printf,
|
||
|
|
SSH: ssh.DefaultOptions(),
|
||
|
|
HTTP: http.DefaultOptions(),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithAddress(addr string) func(*Options) {
|
||
|
|
return func(opts *Options) {
|
||
|
|
opts.Address = addr
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithLogger(logger func(message string, args ...any)) func(*Options) {
|
||
|
|
return func(opts *Options) {
|
||
|
|
opts.Logger = logger
|
||
|
|
opts.SSH.Logger = logger
|
||
|
|
opts.HTTP.Logger = logger
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WithSSHOption(funcs ...ssh.OptionFunc) func(*Options) {
|
||
|
|
return func(o *Options) {
|
||
|
|
for _, fn := range funcs {
|
||
|
|
fn(o.SSH)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func WitHTTPOption(funcs ...http.OptionFunc) func(*Options) {
|
||
|
|
return func(o *Options) {
|
||
|
|
for _, fn := range funcs {
|
||
|
|
fn(o.HTTP)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|