add callback to migration cli

Author: mekramy

migration/cmd.go

@@ -19,6 +19,6 @@ func NewMigrationCLI(m Migration, options ...CLIOptions) *cobra.Command {
 cmd.AddCommand(cmdUp(m, option))
 cmd.AddCommand(cmdDown(m, option))
 cmd.AddCommand(cmdRefresh(m, option))
-cmd.AddCommand(cmdSummary(m))
+cmd.AddCommand(cmdSummary(m, option))
 return cmd
 }

migration/cmd_down.go

@@ -14,6 +14,10 @@ func cmdDown(m Migration, option *cliOption) *cobra.Command {
 downCmd.Short = "rollback migrations"
 downCmd.Flags().StringP("name", "n", "", "migration name")
 downCmd.Run = func(cmd *cobra.Command, args []string) {
+if option.callback != nil {
+defer option.callback()
+}
+
 stages := append([]string{}, args...)
 if len(stages) == 0 {
 stages = option.stages.Elements()

migration/cmd_new.go

@@ -14,6 +14,10 @@ func cmdNew(m Migration, option *cliOption) *cobra.Command {
 Short: "Create a new migration file with default stages in the output path",
 Args: cobra.MinimumNArgs(1),
 Run: func(cmd *cobra.Command, args []string) {
+if option.callback != nil {
+defer option.callback()
+}
+
 if option.root == "" {
 console.Message().
 Red("Create").Italic().

migration/cmd_option.go

@@ -7,6 +7,7 @@ type cliOption struct {
 refreshes *optionSet
 only *optionSet
 exclude *optionSet
+callback func()
 }
 
 func newCLIOption() *cliOption {
@@ -64,3 +65,10 @@ func WithNewCMD(enabled bool) CLIOptions {
 o.create = enabled
 }
 }
+
+// WithCallback register a callback function to call after command finished.
+func WithCallback(cb func()) CLIOptions {
+return func(o *cliOption) {
+o.callback = cb
+}
+}

migration/cmd_refresh.go

@@ -14,6 +14,10 @@ func cmdRefresh(m Migration, option *cliOption) *cobra.Command {
 reCmd.Short = "refresh migrations"
 reCmd.Flags().StringP("name", "n", "", "migration name")
 reCmd.Run = func(cmd *cobra.Command, args []string) {
+if option.callback != nil {
+defer option.callback()
+}
+
 stages := append([]string{}, args...)
 if len(stages) == 0 {
 stages = option.refreshes.Elements()

migration/cmd_summary.go

@@ -9,11 +9,15 @@ import (
 "github.com/spf13/cobra"
 )
 
-func cmdSummary(m Migration) *cobra.Command {
+func cmdSummary(m Migration, option *cliOption) *cobra.Command {
 return &cobra.Command{
 Use: "summary",
 Short: "show migration summary",
 Run: func(cmd *cobra.Command, args []string) {
+if option.callback != nil {
+defer option.callback()
+}
+
 summary, err := m.Summary()
 if err != nil {
 console.Message().Red("Summary").Italic().Print(err.Error())

migration/cmd_up.go

@@ -14,6 +14,10 @@ func cmdUp(m Migration, option *cliOption) *cobra.Command {
 upCmd.Short = "applies migrations"
 upCmd.Flags().StringP("name", "n", "", "migration name")
 upCmd.Run = func(cmd *cobra.Command, args []string) {
+if option.callback != nil {
+defer option.callback()
+}
+
 stages := append([]string{}, args...)
 if len(stages) == 0 {
 stages = option.stages.Elements()