go/sqlgen/template.go

package sqlgen

import (
	_ "embed"
	"os"
	"os/exec"
	"path/filepath"
	"text/template"
)

//go:embed gen.go.tmpl
var fileTemplate string

func render(driver, schemaPath, outputPath string) error {
	sch, err := parsePath(driver, schemaPath)
	if err != nil {
		return err
	}

	tmpl := template.Must(template.New("").Parse(fileTemplate))
	fOut, err := os.Create(outputPath)
	if err != nil {
		return err
	}
	defer fOut.Close()

	err = tmpl.Execute(fOut, struct {
		PackageName string
		Schema      *schema
	}{filepath.Base(filepath.Dir(outputPath)), sch})
	if err != nil {
		return err
	}

	return exec.Command("gofmt", "-w", outputPath).Run()
}