add revision generation with command

Author: SennovE

cmd/qrafter-migrations-init/main.go

@@ -0,0 +1,14 @@
+// Command qrafter-migrations runs qrafter migration tools.
+package main
+
+import (
+	"os"
+
+	qmig "github.com/SennovE/qrafter/migrations"
+)
+
+func main() {
+	if err := qmig.RunMigrationsCommand(os.Args[1:]); err != nil {
+		panic(err)
+	}
+}

cmd/qrafter-migrations/main.go

@@ -3,19 +3,26 @@ package migrations
 import (
 	"flag"
 	"fmt"
+	"time"
 )
 
-func revisionCommandOptionsFromArgs(args []string) (string, *configOptions, error) {
-	fs := flag.NewFlagSet("revision-command", flag.ContinueOnError)
+const (
+	defaultMigrationDirectory = "./migrations"
+	defaultMigrationComment   = "migration"
+	defaultRevisionTimeout    = 30 * time.Second
+)
+
+func configOptionsFromArgs(args []string) (string, *configOptions, error) {
+	fs := flag.NewFlagSet("migrations-init", flag.ContinueOnError)
 
 	var path string
 	var options configOptions
 
 	fs.StringVar(
 		&path,
-		"path",
-		"",
-		"Path for generated cmd file",
+		"dir",
+		defaultMigrationDirectory,
+		"directory for generated qrafter config file",
 	)
 
 	fs.StringVar(
@@ -72,3 +79,85 @@ func revisionCommandOptionsFromArgs(args []string) (string, *configOptions, erro
 
 	return path, &options, nil
 }
+
+type revisionOptions struct {
+	WorkDir          string
+	MigrationDir     string
+	ConfigImportPath string
+	Comment          string
+	GoBinary         string
+	Timeout          time.Duration
+}
+
+func revisionOptionsFromArgs(args []string) (*revisionOptions, error) {
+	fs := flag.NewFlagSet("migrations-revision", flag.ContinueOnError)
+
+	options := revisionOptions{
+		WorkDir:      ".",
+		MigrationDir: defaultMigrationDirectory,
+		Comment:      defaultMigrationComment,
+		GoBinary:     "go",
+		Timeout:      defaultRevisionTimeout,
+	}
+
+	fs.StringVar(
+		&options.WorkDir,
+		"workdir",
+		options.WorkDir,
+		"project directory containing go.mod",
+	)
+
+	fs.StringVar(
+		&options.MigrationDir,
+		"dir",
+		options.MigrationDir,
+		"directory containing qrafter_config.go and generated migrations",
+	)
+
+	fs.StringVar(
+		&options.ConfigImportPath,
+		"config-import",
+		"",
+		"Go import path for package containing MigrationConfig",
+	)
+
+	fs.StringVar(
+		&options.Comment,
+		"comment",
+		options.Comment,
+		"migration comment",
+	)
+
+	fs.StringVar(
+		&options.GoBinary,
+		"go",
+		options.GoBinary,
+		"go binary used for the temporary build",
+	)
+
+	fs.DurationVar(
+		&options.Timeout,
+		"timeout",
+		options.Timeout,
+		"timeout for the temporary build and schema diff",
+	)
+
+	if err := fs.Parse(args); err != nil {
+		return nil, err
+	}
+
+	if options.WorkDir == "" {
+		return nil, fmt.Errorf("missing required flag: --workdir")
+	}
+	if options.MigrationDir == "" {
+		return nil, fmt.Errorf("missing required flag: --dir")
+	}
+	if options.GoBinary == "" {
+		return nil, fmt.Errorf("missing required flag: --go")
+	}
+	if options.Timeout <= 0 {
+		return nil, fmt.Errorf("timeout must be positive")
+	}
+
+	return &options, nil
+}

migrations/cli_params.go

@@ -0,0 +1,19 @@
+package migrations
+
+import "fmt"
+
+// RunMigrationsCommand dispatches qrafter migration CLI subcommands.
+func RunMigrationsCommand(args []string) error {
+	if len(args) == 0 {
+		return fmt.Errorf("usage: qrafter-migrations <init|revision> [flags]")
+	}
+
+	switch args[0] {
+	case "init":
+		return generateMigrationsConfig(args[1:])
+	case "revision":
+		return generateMigrationRevision(args[1:])
+	default:
+		return fmt.Errorf("unknown migrations command %q", args[0])
+	}
+}