snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/shfmt/shfmt.go

@@ -0,0 +1,38 @@
+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
+}

internal/shfmt/shfmt_test.go

@@ -0,0 +1,17 @@
+package shfmt
+
+import "testing"
+
+func TestReplace(t *testing.T) {
+	s := "POSTGRES_SQL://${PG_USER}:${PG_PASSWORD}@${PG_HOST}:${PG_PORT}/AUTHORS"
+	r := NewReplacer([]string{
+		"PG_USER=user",
+		"PG_PASSWORD=password",
+		"PG_HOST=host",
+		"PG_PORT=port",
+	})
+	e := "POSTGRES_SQL://user:password@host:port/AUTHORS"
+	if v := r.Replace(s); v != e {
+		t.Errorf("%s != %s", v, e)
+	}
+}