Merge pull request #1 from wallester/feature/initial-implementation

Initial implementation

Author: dlogvinenko

.gitignore

@@ -6,6 +6,7 @@
 # Folders
 _obj
 _test
+.idea
 
 # Architecture specific extensions/prefixes
 *.[568vq]
@@ -22,3 +23,10 @@ _testmain.go
 *.exe
 *.test
 *.prof
+
+# Built binary
+migrate
+
+# Coverage output
+coverage.cov
+coverage.html

Makefile

@@ -0,0 +1,19 @@
+
+build: fmt
+	@go build
+
+fmt:
+	@go fmt ./...
+
+lint:
+	@gometalinter ./... --deadline=5m --vendor --enable misspell --enable goimports --disable aligncheck --concurrency 4
+
+test:
+	@go list ./... | grep -v vendor | xargs go test
+
+install:
+	@go install
+
+cov:
+	@go test -test.covermode=count -test.coverprofile coverage.cov
+	@go tool cover -html=coverage.cov -o coverage.html

README.md

@@ -1,2 +1,18 @@
 # migrate
 Command line tool for PostgreSQL migrations 
+
+## Features
+
+* Runs migrations in transactions
+* Stores migration version details in auto-generated table ``schema_migrations``.
+
+## Usage
+
+```bash
+migrate -url postgres://user@host:port/database -path ./db/migrations create add_field_to_table
+migrate -url postgres://user@host:port/database -path ./db/migrations up
+migrate -url postgres://user@host:port/database -path ./db/migrations up 1
+migrate -url postgres://user@host:port/database -path ./db/migrations down
+migrate -url postgres://user@host:port/database -path ./db/migrations down 1
+migrate help # for more info
+```

app/Makefile

@@ -0,0 +1 @@
+../Makefile

app/app.go

@@ -0,0 +1,56 @@
+package app
+
+import (
+	"github.com/urfave/cli"
+	"github.com/wallester/migrate/commander"
+	"github.com/wallester/migrate/driver/postgres"
+	"github.com/wallester/migrate/flag"
+	"github.com/wallester/migrate/migrator"
+	"github.com/wallester/migrate/printer"
+)
+
+// New returns new cli.App instance
+func New() *cli.App {
+	cmd := commander.New(migrator.New(postgres.New(), printer.New()))
+
+	app := cli.NewApp()
+	app.Name = "migrate"
+	app.Usage = "Command line tool for PostgreSQL migrations"
+	app.Version = "1.0.0"
+	app.Commands = []cli.Command{
+		cli.Command{
+			Name:      "create",
+			Usage:     "Create a new migration",
+			ArgsUsage: "<name>",
+			Action:    cmd.Create,
+			Flags: []cli.Flag{
+				flag.Path,
+			},
+		},
+		cli.Command{
+			Name:      "up",
+			Usage:     "Apply <n> or all up migrations",
+			Action:    cmd.Up,
+			ArgsUsage: "<n>",
+			Flags: []cli.Flag{
+				flag.Path,
+				flag.URL,
+			},
+		},
+		cli.Command{
+			Name:      "down",
+			Usage:     "Apply <n> or all down migrations",
+			Action:    cmd.Down,
+			ArgsUsage: "<n>",
+			Flags: []cli.Flag{
+				flag.Path,
+				flag.URL,
+			},
+		},
+	}
+	app.Flags = []cli.Flag{
+		flag.Path,
+		flag.URL,
+	}
+	return app
+}

app/app_test.go

@@ -0,0 +1,45 @@
+package app
+
+import (
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/urfave/cli"
+)
+
+func Test_New_ReturnsInstance_InCaseOfSuccess(t *testing.T) {
+	// Act
+	app := New()
+
+	// Assert
+	assert.NotNil(t, app)
+	assert.Equal(t, "migrate", app.Name)
+	assert.Equal(t, "Command line tool for PostgreSQL migrations", app.Usage)
+	assert.NotNil(t, app.Commands)
+	assert.True(t, hasCommand("create", app.Commands))
+	assert.True(t, hasCommand("up", app.Commands))
+	assert.True(t, hasCommand("down", app.Commands))
+	assert.NotNil(t, app.Flags)
+	assert.True(t, hasFlag("path", app.Flags))
+	assert.True(t, hasFlag("url", app.Flags))
+}
+
+func hasCommand(name string, commands []cli.Command) bool {
+	for _, command := range commands {
+		if command.Name == name {
+			return true
+		}
+	}
+
+	return false
+}
+
+func hasFlag(name string, flags []cli.Flag) bool {
+	for _, flag := range flags {
+		if flag.GetName() == name {
+			return true
+		}
+	}
+
+	return false
+}

commander/Makefile

@@ -0,0 +1 @@
+../Makefile

commander/commander.go

@@ -0,0 +1,109 @@
+package commander
+
+import (
+	"strconv"
+
+	"github.com/juju/errors"
+	"github.com/urfave/cli"
+	"github.com/wallester/migrate/direction"
+	"github.com/wallester/migrate/flag"
+	"github.com/wallester/migrate/migrator"
+)
+
+// Commander represents app commands
+type Commander interface {
+	Create(c *cli.Context) error
+	Up(c *cli.Context) error
+	Down(c *cli.Context) error
+}
+
+type commander struct {
+	m migrator.Migrator
+}
+
+// New returns new instance
+func New(m migrator.Migrator) Commander {
+	return &commander{m}
+}
+
+// Create creates new migration files
+func (cmd *commander) Create(c *cli.Context) error {
+	name := c.Args().First()
+	if name == "" {
+		return errors.New("please specify migration name")
+	}
+
+	path := flag.Get(c, flag.FlagPath)
+	if path == "" {
+		return flag.NewRequiredFlagError(flag.FlagPath)
+	}
+
+	if _, err := cmd.m.Create(name, path); err != nil {
+		return errors.Annotate(err, "creating migration failed")
+	}
+
+	return nil
+}
+
+// Up migrates up
+func (cmd *commander) Up(c *cli.Context) error {
+	args, err := parseMigrateArguments(c)
+	if err != nil {
+		return errors.Annotate(err, "parsing parameters failed")
+	}
+
+	if err := cmd.m.Migrate(args.path, args.url, direction.Up, args.steps); err != nil {
+		return errors.Annotate(err, "migrating up failed")
+	}
+
+	return nil
+}
+
+// Down migrates down
+func (cmd *commander) Down(c *cli.Context) error {
+	args, err := parseMigrateArguments(c)
+	if err != nil {
+		return errors.Annotate(err, "parsing parameters failed")
+	}
+
+	if err := cmd.m.Migrate(args.path, args.url, direction.Down, args.steps); err != nil {
+		return errors.Annotate(err, "migrating down failed")
+	}
+
+	return nil
+}
+
+type migrateArguments struct {
+	path  string
+	url   string
+	steps int
+}
+
+func parseMigrateArguments(c *cli.Context) (*migrateArguments, error) {
+	path := flag.Get(c, flag.FlagPath)
+	if path == "" {
+		return nil, flag.NewRequiredFlagError(flag.FlagPath)
+	}
+
+	url := flag.Get(c, flag.FlagURL)
+	if url == "" {
+		return nil, flag.NewRequiredFlagError(flag.FlagURL)
+	}
+
+	var steps int
+	s := c.Args().First()
+	if s != "" {
+		n, err := strconv.Atoi(s)
+		if err != nil {
+			return nil, errors.Annotate(err, "parsing <n> failed")
+		}
+
+		steps = n
+	}
+
+	return &migrateArguments{
+		path:  path,
+		url:   url,
+		steps: steps,
+	}, nil
+}