67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
|
package options
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
flag "github.com/spf13/pflag"
|
||
|
|
"github.com/spf13/viper"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Options contains configuration options for the OAuth2 Proxy
|
||
|
|
type Options struct {
|
||
|
|
Cookie *CookieOptions
|
||
|
|
}
|
||
|
|
|
||
|
|
// New creates a new deafulted copy of the Options struct
|
||
|
|
func New() *Options {
|
||
|
|
options := &Options{}
|
||
|
|
err := defaultStruct(options)
|
||
|
|
if err != nil {
|
||
|
|
// If we get an error here, there must be a code error
|
||
|
|
panic(err)
|
||
|
|
}
|
||
|
|
return options
|
||
|
|
}
|
||
|
|
|
||
|
|
// Load reads a config file, flag arguments and the environment to set the
|
||
|
|
// correct configuration options
|
||
|
|
func Load(config io.Reader, configType string, args []string) (*Options, error) {
|
||
|
|
flagSet := flag.NewFlagSet("oauth2-proxy", flag.ExitOnError)
|
||
|
|
|
||
|
|
// Add FlagSets to main flagSet
|
||
|
|
flagSet.AddFlagSet(cookieFlagSet)
|
||
|
|
|
||
|
|
flagSet.Parse(args)
|
||
|
|
|
||
|
|
// Create a viper for binding config
|
||
|
|
v := viper.New()
|
||
|
|
|
||
|
|
// Bind flags to viper
|
||
|
|
err := v.BindPFlags(flagSet)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Configure loading of environment variables
|
||
|
|
// All flag options are prefixed by the EnvPrefix
|
||
|
|
v.SetEnvPrefix("OAUTH2_PROXY_")
|
||
|
|
// Substitute "-" for "_" so `FOO_BAR` matches the flag `foo-bar`
|
||
|
|
v.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
|
||
|
|
v.AutomaticEnv()
|
||
|
|
|
||
|
|
// Read the configuration file
|
||
|
|
if config != nil {
|
||
|
|
v.ReadConfig(config)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Unmarhsal the config into the Options struct
|
||
|
|
options := New()
|
||
|
|
err = v.Unmarshal(&options)
|
||
|
|
if err != nil {
|
||
|
|
return nil, fmt.Errorf("error unmarshalling config: %v", err)
|
||
|
|
}
|
||
|
|
return options, nil
|
||
|
|
}
|