Merge pull request #51 from stack11/feature/generate-migration-post-hook

Add generate-migration-post hook

cmd/generate.go

@@ -5,6 +5,7 @@ import (
 	"errors"
 	"fmt"
 	"io/fs"
+	"io/ioutil"
 	"log"
 	"os"
 	"os/exec"
@@ -190,6 +191,42 @@ func runWithStdout(
 		if err != nil && !errors.Is(err, ErrInvalidModel) {
 			return fmt.Errorf("failed to generate migration statements: %w", err)
 		}
+
+		file, err := ioutil.TempFile("", "migration")
+		if err != nil {
+			return fmt.Errorf("failed get temporary migration file: %w", err)
+		}
+
+		err = os.WriteFile(
+			file.Name(),
+			[]byte(statements),
+			0o600,
+		)
+		if err != nil {
+			return fmt.Errorf("failed to write temporary migration file: %w", err)
+		}
+
+		wd, err := os.Getwd()
+		if err != nil {
+			return fmt.Errorf("failed to get working directory: %w", err)
+		}
+
+		err = internal.RunHook(wd, "generate-migration-post", file.Name())
+		if err != nil {
+			return fmt.Errorf("failed to run hook: %w", err)
+		}
+
+		statementBytes, err := os.ReadFile(file.Name())
+		if err != nil {
+			return fmt.Errorf("failed to read temporary migration file: %w", err)
+		}
+		statements = string(statementBytes)
+
+		err = os.Remove(file.Name())
+		if err != nil {
+			return fmt.Errorf("failed to delete temporary migration file: %w", err)
+		}
+
 		fmt.Println("")
 		fmt.Println("--")
 		fmt.Println(statements)
@@ -250,6 +287,16 @@ func runWithFile(
 		}
 		log.Println("Wrote migration file")
 
+		wd, err := os.Getwd()
+		if err != nil {
+			return fmt.Errorf("failed to get working directory: %w", err)
+		}
+
+		err = internal.RunHook(wd, "generate-migration-post", newMigrationFilePath)
+		if err != nil {
+			return fmt.Errorf("failed to run hook: %w", err)
+		}
+
 		err = writeTemplateFiles(config, migrationNumber)
 		if err != nil {
 			return fmt.Errorf("failed to write template files: %w", err)

cmd/init.go

@@ -149,6 +149,10 @@ func NewInitCommand() *cobra.Command {
 			if err != nil {
 				log.Fatalln(err)
 			}
+			err = writeSampleHook("generate-migration-post", "echo \"Running on migration file $1\"")
+			if err != nil {
+				log.Fatalln(err)
+			}
 
 			log.Println("New project created!")
 
@@ -218,13 +222,19 @@ func writeTemplateFile(ts, filename string, templateData map[string]interface{})
 	return nil
 }
 
-func writeSampleHook(name string) error {
-	//nolint:gosec,wrapcheck
-	return os.WriteFile(fmt.Sprintf("hooks/%s.sample", name), []byte(strings.Join([]string{
+func writeSampleHook(name string, extraLines ...string) error {
+	lines := []string{
 		"#!/bin/bash",
 		"set -euxo pipefail",
 		"",
 		fmt.Sprintf("echo \"This is %s\"", name),
-		"",
-	}, "\n")), 0o755)
+	}
+	if len(extraLines) > 0 {
+		lines = append(lines, extraLines...)
+	}
+
+	lines = append(lines, "")
+
+	//nolint:gosec,wrapcheck
+	return os.WriteFile(fmt.Sprintf("hooks/%s.sample", name), []byte(strings.Join(lines, "\n")), 0o755)
 }

example/hooks/generate-migration-post.sample

@@ -0,0 +1,5 @@
+#!/bin/bash
+set -euxo pipefail
+
+echo "This is generate-migration-post"
+echo "Running on migration file $1"

internal/hooks.go

@@ -8,7 +8,7 @@ import (
 	"path/filepath"
 )
 
-func RunHook(wd, hookName string) error {
+func RunHook(wd, hookName string, args ...string) error {
 	hooksDir := filepath.Join(wd, "hooks")
 	filePath := filepath.Join(hooksDir, hookName)
 	if _, err := os.Stat(filePath); errors.Is(err, os.ErrNotExist) {
@@ -19,7 +19,7 @@ func RunHook(wd, hookName string) error {
 
 	log.Printf("Running hook %s", hookName)
 
-	cmd := exec.Command(filePath)
+	cmd := exec.Command(filePath, args...)
 	cmd.Dir = hooksDir
 	cmd.Stdout = os.Stdout
 	cmd.Stderr = os.Stderr