coder/scripts/templatebuildermodulegen/write.go

package main

import (
	"encoding/json"
	"os"
	"text/template"
)

func writeModuleJSON(path string, m ModuleManifest) error {
	data, err := json.MarshalIndent(m, "", "  ")
	if err != nil {
		return err
	}
	data = append(data, '\n')
	return os.WriteFile(path, data, 0o600)
}

var tfTmplTemplate = template.Must(template.New("tf.tmpl").Funcs(template.FuncMap{
	"hclValue": func(v ModuleVariable) string {
		if v.Sensitive {
			return "var." + v.Name
		}
		return "{{ .Variables." + v.Name + " }}"
	},
}).Parse(`{{- range .SensitiveVars }}
variable "{{ .Name }}" {
  description = "{{ .Description }}"
  type        = {{ .Type }}
  sensitive   = true
  default     = ""
}
{{ end -}}
module "{{ .ID }}" {
  count    = data.coder_workspace.me.start_count
  source   = "{{"{{"}} .RegistryBase {{"}}"}}/{{ .Namespace }}/{{ .ID }}/coder"
  version  = "{{"{{"}} .PinnedVersion {{"}}"}}"
  agent_id = coder_agent.{{"{{"}} .AgentResourceName {{"}}"}}.id
{{- range .NonComputedVars }}
{{- if .Sensitive }}
  {{ .Name }} = var.{{ .Name }}
{{- else }}
  {{ .Name }} = {{"{{"}} .Variables.{{ .Name }} {{"}}"}}
{{- end }}
{{- end }}
}
`))

type tfTmplData struct {
	ID              string
	Namespace       string
	SensitiveVars   []ModuleVariable
	NonComputedVars []ModuleVariable
}

func writeTFTmpl(path string, m ModuleManifest) error {
	var sensitiveVars []ModuleVariable
	var nonComputedVars []ModuleVariable
	for _, v := range m.Variables {
		if v.Computed {
			continue
		}
		nonComputedVars = append(nonComputedVars, v)
		if v.Sensitive {
			sensitiveVars = append(sensitiveVars, v)
		}
	}

	data := tfTmplData{
		ID:              m.ID,
		Namespace:       m.Namespace,
		SensitiveVars:   sensitiveVars,
		NonComputedVars: nonComputedVars,
	}

	f, err := os.Create(path)
	if err != nil {
		return err
	}
	err = tfTmplTemplate.Execute(f, data)
	if closeErr := f.Close(); err == nil {
		err = closeErr
	}
	return err
}