wehub-resource-sync
91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
package config
|
|
|
|
import (
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
// varRef matches ${VAR} and ${VAR:-default}: a shell-style reference with an
|
|
// optional ":-default" fallback used when the variable is unset or empty.
|
|
var varRef = regexp.MustCompile(`\$\{([A-Za-z_][A-Za-z0-9_]*)(:-([^}]*))?\}`)
|
|
|
|
type envLookup func(string) (string, bool)
|
|
|
|
// ExpandVars substitutes ${VAR} / ${VAR:-default} references from the process
|
|
// environment. An unset variable with no default expands to "" (matching the
|
|
// MCP / Claude Code convention), so a missing secret yields an empty header
|
|
// rather than a literal "${TOKEN}" leaking onto the wire.
|
|
func ExpandVars(s string) string {
|
|
return expandVarsWithLookup(s, func(name string) (string, bool) {
|
|
v, ok := os.LookupEnv(name)
|
|
return v, ok && v != ""
|
|
})
|
|
}
|
|
|
|
func expandVarsWithLookup(s string, lookup envLookup) string {
|
|
if !strings.Contains(s, "${") {
|
|
return s
|
|
}
|
|
return varRef.ReplaceAllStringFunc(s, func(m string) string {
|
|
g := varRef.FindStringSubmatch(m)
|
|
name, hasDefault, def := g[1], g[2] != "", g[3]
|
|
if v, ok := lookup(name); ok {
|
|
return v
|
|
}
|
|
if hasDefault {
|
|
return def
|
|
}
|
|
return ""
|
|
})
|
|
}
|
|
|
|
func scopedEnvLookup(scoped map[string]string) envLookup {
|
|
return func(name string) (string, bool) {
|
|
if v, ok := os.LookupEnv(name); ok {
|
|
return v, v != ""
|
|
}
|
|
if v, ok := scoped[name]; ok && v != "" {
|
|
return v, true
|
|
}
|
|
return "", false
|
|
}
|
|
}
|
|
|
|
func (c *Config) expandVars(s string) string {
|
|
if c == nil {
|
|
return ExpandVars(s)
|
|
}
|
|
return expandVarsWithLookup(s, scopedEnvLookup(c.expansionEnv))
|
|
}
|
|
|
|
// ExpandedPlugin returns a copy of e with ${VAR} references expanded across the
|
|
// command, args, env values, url, and header values — the fields Claude Code
|
|
// also expands. The entry itself is left untouched.
|
|
func (e PluginEntry) ExpandedPlugin() PluginEntry {
|
|
lookup := scopedEnvLookup(e.expansionEnv)
|
|
out := e
|
|
out.Command = expandVarsWithLookup(e.Command, lookup)
|
|
out.URL = expandVarsWithLookup(e.URL, lookup)
|
|
if len(e.Args) > 0 {
|
|
out.Args = make([]string, len(e.Args))
|
|
for i, a := range e.Args {
|
|
out.Args[i] = expandVarsWithLookup(a, lookup)
|
|
}
|
|
}
|
|
out.Env = expandMap(e.Env, lookup)
|
|
out.Headers = expandMap(e.Headers, lookup)
|
|
return out
|
|
}
|
|
|
|
func expandMap(m map[string]string, lookup envLookup) map[string]string {
|
|
if len(m) == 0 {
|
|
return m
|
|
}
|
|
out := make(map[string]string, len(m))
|
|
for k, v := range m {
|
|
out[k] = expandVarsWithLookup(v, lookup)
|
|
}
|
|
return out
|
|
}
|