Generate a `map[string]string` from sql files using `go generate`

generate.go

// +build ignore

package main

import (
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"strconv"
	"strings"
	"text/template"
)

const fileTemplate = `//DO NOT EDIT - GENERATED USING go generate
package db

sqlFiles := map[string]string{
{{range $key, $val := .Files}}
	"{{$key}}": {{$val}},
{{end}}
}
`

func main() {
	files, err := loadFiles("db/sql")
	handleError(err)

	ctx := struct{ Files map[string]string }{Files: files}
	tmpl, err := template.New("tmpl").Parse(fileTemplate)
	handleError(err)

	f, err := os.Create("db/sql_files.go")
	handleError(err)

	err = tmpl.Execute(f, ctx)
	handleError(err)
}

func loadFiles(path string) (map[string]string, error) {
	pattern := filepath.Join(path, "./*.sql")
	files, err := filepath.Glob(pattern)
	if err != nil {
		return nil, err
	}

	stmts := make(map[string]string)

	for _, file := range files {
		b, err := ioutil.ReadFile(file)
		if err != nil {
			return nil, err
		}

		name := strings.TrimSuffix(filepath.Base(file), filepath.Ext(file))
		sql := string(b)

		stmts[name] = strconv.Quote(sql)
	}

	return stmts, nil
}

func handleError(err error) {
	if err != nil {
		log.Fatalf("%v", err)
	}
}