-
Notifications
You must be signed in to change notification settings - Fork 0
/
pflag.go
57 lines (48 loc) · 1.27 KB
/
pflag.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
50
51
52
53
54
55
56
57
// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0
package configx
import (
"strings"
"github.com/clinia/x/jsonschemax"
"github.com/knadh/koanf"
"github.com/knadh/koanf/providers/posflag"
"github.com/pkg/errors"
"github.com/spf13/pflag"
"github.com/ory/jsonschema/v3"
)
type PFlagProvider struct {
p *posflag.Posflag
paths []jsonschemax.Path
}
func NewPFlagProvider(rawSchema []byte, schema *jsonschema.Schema, f *pflag.FlagSet, k *koanf.Koanf) (*PFlagProvider, error) {
paths, err := getSchemaPaths(rawSchema, schema)
if err != nil {
return nil, err
}
return &PFlagProvider{
p: posflag.Provider(f, ".", k),
paths: paths,
}, nil
}
func (p *PFlagProvider) ReadBytes() ([]byte, error) {
return nil, errors.New("pflag provider does not support this method")
}
func (p *PFlagProvider) Read() (map[string]interface{}, error) {
all, err := p.p.Read()
if err != nil {
return nil, errors.WithStack(err)
}
knownFlags := make(map[string]interface{}, len(all))
for k, v := range all {
k = strings.ReplaceAll(k, ".", "-")
for _, path := range p.paths {
normalized := strings.ReplaceAll(path.Name, ".", "-")
if k == normalized {
knownFlags[k] = v
break
}
}
}
return knownFlags, nil
}
var _ koanf.Provider = (*PFlagProvider)(nil)