/
factory.go
49 lines (40 loc) · 1.38 KB
/
factory.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
package googleidinfo
import (
"flag"
"github.com/grepplabs/kafka-proxy/pkg/apis"
"github.com/grepplabs/kafka-proxy/pkg/libs/util"
"github.com/grepplabs/kafka-proxy/pkg/registry"
)
func init() {
registry.NewComponentInterface(new(apis.TokenInfoFactory))
registry.Register(new(Factory), "google-id-info")
}
func (f *pluginMeta) flagSet() *flag.FlagSet {
fs := flag.NewFlagSet("google-id info settings", flag.ContinueOnError)
return fs
}
type pluginMeta struct {
timeout int
certsRefreshInterval int
audience util.ArrayFlags
emailsRegex util.ArrayFlags
}
type Factory struct {
}
// New implements apis.TokenInfoFactory
func (t *Factory) New(params []string) (apis.TokenInfo, error) {
pluginMeta := &pluginMeta{}
fs := pluginMeta.flagSet()
fs.IntVar(&pluginMeta.timeout, "timeout", 10, "Request timeout in seconds")
fs.IntVar(&pluginMeta.certsRefreshInterval, "certs-refresh-interval", 60*60, "Certificates refresh interval in seconds")
fs.Var(&pluginMeta.audience, "audience", "The audience of a token")
fs.Var(&pluginMeta.emailsRegex, "email-regex", "Regex of the email claim")
fs.Parse(params)
opts := TokenInfoOptions{
Timeout: pluginMeta.timeout,
CertsRefreshInterval: pluginMeta.certsRefreshInterval,
Audience: pluginMeta.audience,
EmailsRegex: pluginMeta.emailsRegex,
}
return NewTokenInfo(opts)
}