lego/internal/releaser/generator.go

package main

import (
	"bytes"
	"embed"
	"fmt"
	"go/format"
	"os"
	"path/filepath"
	"text/template"
)

const (
	dnsTemplate   = "templates/dns.go.tmpl"
	dnsTargetFile = "./providers/dns/internal/useragent/useragent.go"
)

const (
	senderTemplate   = "templates/sender.go.tmpl"
	senderTargetFile = "./acme/api/internal/sender/useragent.go"
)

const (
	versionTemplate   = "templates/version.go.tmpl"
	versionTargetFile = "./cmd/lego/zz_gen_version.go"
)

//go:embed templates
var templateFS embed.FS

type Generator struct {
	templatePath string
	targetFile   string
}

func NewGenerator(templatePath string, targetFile string) *Generator {
	return &Generator{templatePath: templatePath, targetFile: targetFile}
}

func (g *Generator) Generate(version, comment string) error {
	tmpl, err := template.New(filepath.Base(g.templatePath)).ParseFS(templateFS, g.templatePath)
	if err != nil {
		return fmt.Errorf("parsing template (%s): %w", g.templatePath, err)
	}

	b := &bytes.Buffer{}

	err = tmpl.Execute(b, map[string]string{
		"version": version,
		"comment": comment,
	})
	if err != nil {
		return fmt.Errorf("execute template (%s): %w", g.templatePath, err)
	}

	source, err := format.Source(b.Bytes())
	if err != nil {
		return fmt.Errorf("format generated content (%s): %w", g.targetFile, err)
	}

	err = os.WriteFile(g.targetFile, source, 0o644)
	if err != nil {
		return fmt.Errorf("write file (%s): %w", g.targetFile, err)
	}

	return nil
}

func generate(targetVersion, comment string) error {
	generators := []*Generator{
		NewGenerator(dnsTemplate, dnsTargetFile),
		NewGenerator(senderTemplate, senderTargetFile),
		NewGenerator(versionTemplate, versionTargetFile),
	}

	for _, generator := range generators {
		err := generator.Generate(targetVersion, comment)
		if err != nil {
			return fmt.Errorf("generate file(s): %w", err)
		}
	}

	return nil
}