|
| 1 | +// Package config provides [cli.ConfigResolver] implementations for common |
| 2 | +// configuration sources. |
| 3 | +// |
| 4 | +// A [cli.ConfigResolver] is a function that maps flag names to string values. |
| 5 | +// The cli framework handles type conversion, validation, required checks, and |
| 6 | +// enum enforcement — resolvers only return strings. This package ships |
| 7 | +// format-agnostic building blocks; any source that can produce a |
| 8 | +// map[string]string works out of the box via [FromMap]. |
| 9 | +// |
| 10 | +// Priority chain: explicit CLI flag > env var > config > default > zero value. |
| 11 | +// |
| 12 | +// # Provided resolvers |
| 13 | +// |
| 14 | +// - [FromMap] — backed by a string map (useful for testing, in-memory config, or as the building block for custom formats) |
| 15 | +// - [FromJSON] — decodes a flat JSON object from an [io.Reader] |
| 16 | +// - [Chain] — tries multiple resolvers in order, returning the first match |
| 17 | +// |
| 18 | +// # Custom format adapters |
| 19 | +// |
| 20 | +// Because [FromMap] accepts any map[string]string, adding support for a new |
| 21 | +// configuration format is a matter of decoding into a flat map. The examples |
| 22 | +// below are complete, copy-paste-ready adapters. |
| 23 | +// |
| 24 | +// YAML (using gopkg.in/yaml.v3): |
| 25 | +// |
| 26 | +// func FromYAML(r io.Reader) (cli.ConfigResolver, error) { |
| 27 | +// var m map[string]string |
| 28 | +// if err := yaml.NewDecoder(r).Decode(&m); err != nil { |
| 29 | +// return nil, err |
| 30 | +// } |
| 31 | +// return config.FromMap(m), nil |
| 32 | +// } |
| 33 | +// |
| 34 | +// TOML (using github.com/BurntSushi/toml): |
| 35 | +// |
| 36 | +// func FromTOML(r io.Reader) (cli.ConfigResolver, error) { |
| 37 | +// var m map[string]string |
| 38 | +// if _, err := toml.NewDecoder(r).Decode(&m); err != nil { |
| 39 | +// return nil, err |
| 40 | +// } |
| 41 | +// return config.FromMap(m), nil |
| 42 | +// } |
| 43 | +// |
| 44 | +// HCL (using github.com/hashicorp/hcl/v2): |
| 45 | +// |
| 46 | +// func FromHCL(r io.Reader) (cli.ConfigResolver, error) { |
| 47 | +// data, err := io.ReadAll(r) |
| 48 | +// if err != nil { return nil, err } |
| 49 | +// var m map[string]string |
| 50 | +// if err := hcl.Unmarshal(data, &m); err != nil { |
| 51 | +// return nil, err |
| 52 | +// } |
| 53 | +// return config.FromMap(m), nil |
| 54 | +// } |
| 55 | +// |
| 56 | +// .env files (using github.com/joho/godotenv): |
| 57 | +// |
| 58 | +// func FromDotenv(r io.Reader) (cli.ConfigResolver, error) { |
| 59 | +// m, err := godotenv.Parse(r) |
| 60 | +// if err != nil { return nil, err } |
| 61 | +// return config.FromMap(m), nil |
| 62 | +// } |
| 63 | +// |
| 64 | +// # Direct resolver functions |
| 65 | +// |
| 66 | +// For sources that don't map cleanly to a flat file (remote stores, |
| 67 | +// structured configs, or computed values), implement [cli.ConfigResolver] |
| 68 | +// directly: |
| 69 | +// |
| 70 | +// // Consul KV adapter |
| 71 | +// func FromConsul(client *consul.Client, prefix string) cli.ConfigResolver { |
| 72 | +// return func(flagName string) (string, bool) { |
| 73 | +// pair, _, err := client.KV().Get(prefix+"/"+flagName, nil) |
| 74 | +// if err != nil || pair == nil { return "", false } |
| 75 | +// return string(pair.Value), true |
| 76 | +// } |
| 77 | +// } |
| 78 | +// |
| 79 | +// # Layered configuration |
| 80 | +// |
| 81 | +// Use [Chain] to try multiple sources in priority order. The first resolver |
| 82 | +// that finds a value wins: |
| 83 | +// |
| 84 | +// resolver := config.Chain( |
| 85 | +// localOverrides, // highest priority config source |
| 86 | +// projectConfig, // project-level config |
| 87 | +// globalConfig, // user-level defaults |
| 88 | +// ) |
| 89 | +package config |
| 90 | + |
| 91 | +import ( |
| 92 | + "encoding/json" |
| 93 | + "io" |
| 94 | + |
| 95 | + "github.com/bjaus/cli" |
| 96 | +) |
| 97 | + |
| 98 | +// FromMap returns a [cli.ConfigResolver] backed by a string map. |
| 99 | +func FromMap(m map[string]string) cli.ConfigResolver { |
| 100 | + return func(flagName string) (string, bool) { |
| 101 | + v, ok := m[flagName] |
| 102 | + return v, ok |
| 103 | + } |
| 104 | +} |
| 105 | + |
| 106 | +// FromJSON decodes a flat JSON object from r and returns a [cli.ConfigResolver]. |
| 107 | +// The JSON must be a flat object with string values: {"port": "8080"}. |
| 108 | +func FromJSON(r io.Reader) (cli.ConfigResolver, error) { |
| 109 | + var m map[string]string |
| 110 | + if err := json.NewDecoder(r).Decode(&m); err != nil { |
| 111 | + return nil, err |
| 112 | + } |
| 113 | + return FromMap(m), nil |
| 114 | +} |
| 115 | + |
| 116 | +// Chain returns a [cli.ConfigResolver] that tries each resolver in order, |
| 117 | +// returning the value from the first resolver that reports found. |
| 118 | +func Chain(resolvers ...cli.ConfigResolver) cli.ConfigResolver { |
| 119 | + return func(flagName string) (string, bool) { |
| 120 | + for _, r := range resolvers { |
| 121 | + if v, ok := r(flagName); ok { |
| 122 | + return v, true |
| 123 | + } |
| 124 | + } |
| 125 | + return "", false |
| 126 | + } |
| 127 | +} |
0 commit comments