dbshift/main.go at master · limoli/dbshift · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
package main

import (
	"errors"
	"fmt"
	"github.com/abiosoft/ishell"
	"github.com/limoli/dbshift/command"
	"github.com/limoli/dbshift/lib"
	"github.com/limoli/dbshift/lib/mysql"
	"github.com/limoli/dbshift/util"
	"os"
)

func main() {
	// Get database instance according to database type
	db, err := getDatabase(lib.Setting.GetDatabaseType())
	if err != nil {
		util.Exit(err)
	}

	// Try connection to database
	err = db.Connect(lib.Setting.GetDatabaseConnection())
	if err != nil {
		util.Exit(lib.NewError("impossible to connect to db : '%s'", err.Error()))
	}

	// Run shell
	shell := ishell.New()

	commands := getShellCommands(db)
	for k := range commands {
		shell.AddCmd(commands[k])
	}

	if len(os.Args) > 1 {
		err = shell.Process(os.Args[1:]...)
		if err != nil {
			util.Exit(err)
		}
	} else {
		shell.Run()
	}
}

func getDatabase(dbType lib.DatabaseType) (lib.IDb, error) {
	var db lib.IDb

	switch dbType {
	case lib.DatabaseTypeMySQL:
		db = new(mysql.Db)
		break
	default:
		errMsg := fmt.Sprintf("db type \"%s\" not available %v\n", dbType, lib.AvailableDatabases)
		return nil, errors.New(errMsg)
	}

	return db, nil
}

func getShellCommands(db lib.IDb) []*ishell.Cmd {
	return []*ishell.Cmd{
		{
			Name:     "init",
			Help:     "init",
			LongHelp: "Install dbshift using your database.",
			Func: func(c *ishell.Context) {
				util.ExitIfInitialised(db)
				command.Init(db)
			},
		}, {
			Name:     "refresh",
			Help:     "refresh",
			LongHelp: "It discovers and imports new migrations.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				command.Refresh(db)
			},
		}, {
			Name:     "info",
			Help:     "info",
			LongHelp: "It returns information about dbshift.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				command.Info()
			},
		}, {
			Name:     "status",
			Help:     "status",
			LongHelp: "It returns the current status of database along migrations.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				command.Status(db)
			},
		}, {
			Name:     "create",
			Help:     "create <description>",
			LongHelp: "It creates a migration with description.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				if len(c.Args) != 1 {
					util.Exit(lib.NewError("missing migration description"))
				}
				description := c.Args[0]
				command.Create(db, description)
			},
		}, {
			Name:     "migrations-upgrade",
			Help:     "migrations-upgrade",
			LongHelp: "It returns the list of migrations eligible to upgrade.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				command.Migrations(db, lib.Upgrade)
			},
		}, {
			Name:     "migrations-downgrade",
			Help:     "migrations-downgrade",
			LongHelp: "It returns the list of migrations eligible to downgrade.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				command.Migrations(db, lib.Downgrade)
			},
		}, {
			Name:     "upgrade",
			Help:     "upgrade",
			LongHelp: "It upgrades all the migrations eligible to upgrade.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				command.Execute(db, lib.Upgrade)
			},
		}, {
			Name:     "downgrade",
			Help:     "downgrade",
			LongHelp: "It downgrades all the migrations eligible to downgrade.",
			Func: func(c *ishell.Context) {
				util.ExitIfNotInitialised(db)
				command.Execute(db, lib.Downgrade)
			},
		},
	}
}