mirror of
https://github.com/hasura/graphql-engine.git
synced 2024-12-15 09:22:43 +03:00
90 lines
2.1 KiB
Go
90 lines
2.1 KiB
Go
package commands
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"os"
|
|
"text/tabwriter"
|
|
|
|
"github.com/hasura/graphql-engine/cli/util"
|
|
|
|
"github.com/hasura/graphql-engine/cli"
|
|
"github.com/hasura/graphql-engine/cli/migrate"
|
|
"github.com/pkg/errors"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
func newMigrateStatusCmd(ec *cli.ExecutionContext) *cobra.Command {
|
|
opts := &MigrateStatusOptions{
|
|
EC: ec,
|
|
}
|
|
migrateStatusCmd := &cobra.Command{
|
|
Use: "status",
|
|
Short: "Display current status of migrations on a database",
|
|
Example: ` # Use with admin secret:
|
|
hasura migrate status --admin-secret "<your-admin-secret>"
|
|
|
|
# Check status on a different server:
|
|
hasura migrate status --endpoint "<endpoint>"`,
|
|
SilenceUsage: true,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
opts.EC.Spin("Fetching migration status...")
|
|
opts.Database = ec.Database
|
|
status, err := opts.Run()
|
|
opts.EC.Spinner.Stop()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
buf := printStatus(status)
|
|
fmt.Fprintf(os.Stdout, "%s", buf)
|
|
return nil
|
|
},
|
|
}
|
|
return migrateStatusCmd
|
|
}
|
|
|
|
type MigrateStatusOptions struct {
|
|
EC *cli.ExecutionContext
|
|
Database string
|
|
}
|
|
|
|
func (o *MigrateStatusOptions) Run() (*migrate.Status, error) {
|
|
migrateDrv, err := migrate.NewMigrate(o.EC, true, o.Database)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
status, err := executeStatus(migrateDrv)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "cannot fetch migrate status")
|
|
}
|
|
return status, nil
|
|
}
|
|
|
|
func printStatus(status *migrate.Status) *bytes.Buffer {
|
|
out := new(tabwriter.Writer)
|
|
buf := &bytes.Buffer{}
|
|
out.Init(buf, 0, 8, 2, ' ', 0)
|
|
w := util.NewPrefixWriter(out)
|
|
w.Write(util.LEVEL_0, "VERSION\tNAME\tSOURCE STATUS\tDATABASE STATUS\n")
|
|
for _, version := range status.Index {
|
|
w.Write(util.LEVEL_0, "%d\t%s\t%s\t%s\n",
|
|
version,
|
|
status.Migrations[version].Name,
|
|
convertBool(status.Migrations[version].IsPresent),
|
|
convertBool(status.Migrations[version].IsApplied),
|
|
)
|
|
}
|
|
out.Flush()
|
|
return buf
|
|
}
|
|
|
|
func convertBool(ok bool) string {
|
|
switch ok {
|
|
case true:
|
|
return "Present"
|
|
case false:
|
|
return "Not Present"
|
|
}
|
|
return ""
|
|
}
|