Implement tern template support

Author: No3371

go.sum

@@ -18,6 +18,7 @@ github.com/cubicdaiya/gonp v1.0.4/go.mod h1:iWGuP/7+JVTn02OWhRemVbMmG1DOUnmrGTYY
 github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
 github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
 github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
+github.com/dlclark/regexp2 v1.11.5/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
 github.com/dustin/go-humanize v1.0.1 h1:GzkhY7T5VNhEkwH0PVJgjz+fX1rhBrR7pRT3mDkpeCY=
 github.com/dustin/go-humanize v1.0.1/go.mod h1:Mu1zIs6XwVuF/gI1OepvI0qD18qycQx+mFykh5fBlto=
 github.com/fatih/structtag v1.2.0 h1:/OdNE99OxoI/PqaW/SuSK9uxxT3f/tcSZgon/ssNSx4=

internal/compiler/compile.go

@@ -38,6 +38,11 @@ func (c *Compiler) parseCatalog(schemas []string) error {
 			continue
 		}
 		contents := migrations.RemoveRollbackStatements(string(blob))
+		contents, err = migrations.TransformStatements(filepath.Dir(filename), contents)
+		if err != nil {
+			merr.Add(filename, contents, 0, err)
+			continue
+		}
 		c.schema = append(c.schema, contents)
 		stmts, err := c.parser.Parse(strings.NewReader(contents))
 		if err != nil {

internal/migrations/migrations.go

@@ -2,6 +2,11 @@ package migrations
 
 import (
 	"bufio"
+	"errors"
+	"fmt"
+	"os"
+	"path"
+	"regexp"
 	"strings"
 )
 
@@ -37,3 +42,38 @@ func IsDown(filename string) bool {
 	// Remove golang-migrate rollback files.
 	return strings.HasSuffix(filename, ".down.sql")
 }
+
+var ternTemplateRegex *regexp.Regexp
+
+// tern: {{ template "filepath" . }}
+func TransformStatements(pwd, content string) (string, error) {
+	if !strings.Contains(content, "{{ template \"") {
+		return content, nil
+	}
+
+	var err error
+	var processed string
+	
+	if ternTemplateRegex == nil {
+		defer func() {
+			if r := recover(); r != nil {
+				fmt.Printf("failed to compile regexp: %v\n", r)
+			}
+			// It is tested, just recovering for it's technically possible to panic
+		}()
+		ternTemplateRegex = regexp.MustCompile(`\{\{ template \"(.+)\" \. \}\}`)
+	}
+
+	processed = ternTemplateRegex.ReplaceAllStringFunc(content, func(match string) string {
+		filePath := ternTemplateRegex.FindStringSubmatch(match)[1]
+		filePath = path.Join(pwd, filePath)
+		read, err := os.ReadFile(filePath)
+		if err != nil {
+			err = errors.Join(err, fmt.Errorf("error reading file %s: %w", filePath, err))
+			return match
+		}
+		return string(read)
+	})
+
+	return processed, err
+}