secretsgen.go

package secretsgen

import (
	"bytes"
	"fmt"
	"strconv"

	"encr.dev/pkg/option"
	"encr.dev/v2/app"
	"encr.dev/v2/codegen"
	"encr.dev/v2/internals/pkginfo"
	"encr.dev/v2/parser/infra/secrets"
)

func Gen(gen *codegen.Generator, svc option.Option[*app.Service], pkg *pkginfo.Package, secrets []*secrets.Secrets) {
	addedImport := make(map[*pkginfo.File]bool)
	for _, secret := range secrets {
		file := secret.File
		rw := gen.Rewrite(file)

		if !addedImport[file] {
			// Add an import of the runtime package to be able to load secrets.
			insertPos := file.AST().Name.End()
			ln := gen.FS.Position(insertPos)

			rw.Insert(insertPos, []byte(fmt.Sprintf("\nimport __encore_secrets %s;/*line :%d:%d*/",
				strconv.Quote("encore.dev/appruntime/infrasdk/secrets"),
				ln.Line, ln.Column)))
			addedImport[secret.File] = true
		}

		getName := func(svc *app.Service) string { return svc.Name }
		svcName := strconv.Quote(option.Map(svc, getName).GetOrElse(""))

		// Rewrite the value spec to load the secrets.
		spec := secret.Spec
		var buf bytes.Buffer
		buf.WriteString("{\n")
		for _, key := range secret.Keys {
			fmt.Fprintf(&buf, "\t%s: __encore_secrets.Load(%s, %s),\n", key, strconv.Quote(key), svcName)
		}
		ep := gen.FS.Position(spec.End())
		fmt.Fprintf(&buf, "}/*line :%d:%d*/", ep.Line, ep.Column)
		rw.Insert(spec.Type.Pos(), []byte("= "))
		rw.Insert(spec.End(), buf.Bytes())

	}
}