You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.3 KiB
71 lines
1.3 KiB
package toml
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/pelletier/go-toml"
|
|
"gitoa.ru/go-4devs/config"
|
|
"gitoa.ru/go-4devs/config/value"
|
|
)
|
|
|
|
const (
|
|
Name = "toml"
|
|
Separator = "."
|
|
)
|
|
|
|
var _ config.Provider = (*Provider)(nil)
|
|
|
|
func NewFile(file string, opts ...Option) (*Provider, error) {
|
|
tree, err := toml.LoadFile(file)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("toml: failed load file: %w", err)
|
|
}
|
|
|
|
return configure(tree, opts...), nil
|
|
}
|
|
|
|
type Option func(*Provider)
|
|
|
|
func configure(tree *toml.Tree, opts ...Option) *Provider {
|
|
prov := &Provider{
|
|
tree: tree,
|
|
key: func(s []string) string {
|
|
return strings.Join(s, Separator)
|
|
},
|
|
}
|
|
|
|
for _, opt := range opts {
|
|
opt(prov)
|
|
}
|
|
|
|
return prov
|
|
}
|
|
|
|
func New(data []byte, opts ...Option) (*Provider, error) {
|
|
tree, err := toml.LoadBytes(data)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("toml failed load data: %w", err)
|
|
}
|
|
|
|
return configure(tree, opts...), nil
|
|
}
|
|
|
|
type Provider struct {
|
|
tree *toml.Tree
|
|
key func([]string) string
|
|
name string
|
|
}
|
|
|
|
func (p *Provider) Name() string {
|
|
return p.name
|
|
}
|
|
|
|
func (p *Provider) Value(_ context.Context, path ...string) (config.Value, error) {
|
|
if k := p.key(path); p.tree.Has(k) {
|
|
return Value{Value: value.Value{Val: p.tree.Get(k)}}, nil
|
|
}
|
|
|
|
return nil, config.ErrValueNotFound
|
|
}
|
|
|