0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
39 lines
626 B
Go
39 lines
626 B
Go
package shfmt
|
|
|
|
import (
|
|
"os"
|
|
"regexp"
|
|
"strings"
|
|
)
|
|
|
|
var pat = regexp.MustCompile(`\$\{[A-Z_]+\}`)
|
|
|
|
type Replacer struct {
|
|
envmap map[string]string
|
|
}
|
|
|
|
func (r *Replacer) Replace(f string) string {
|
|
return pat.ReplaceAllStringFunc(f, func(s string) string {
|
|
s = strings.TrimPrefix(s, "${")
|
|
s = strings.TrimSuffix(s, "}")
|
|
return r.envmap[s]
|
|
})
|
|
}
|
|
|
|
func NewReplacer(env []string) *Replacer {
|
|
r := Replacer{
|
|
envmap: map[string]string{},
|
|
}
|
|
if env == nil {
|
|
env = os.Environ()
|
|
}
|
|
for _, e := range env {
|
|
k, v, _ := strings.Cut(e, "=")
|
|
if k == "SQLC_AUTH_TOKEN" {
|
|
continue
|
|
}
|
|
r.envmap[k] = v
|
|
}
|
|
return &r
|
|
}
|