feat: initial commit

internal/command/database/migrate.go

@@ -0,0 +1,105 @@
+package database
+
+import (
+	"forge.cadoles.com/cadoles/bouncer/internal/command/common"
+	"forge.cadoles.com/cadoles/bouncer/internal/migrate"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+	"gitlab.com/wpetit/goweb/logger"
+)
+
+const (
+	MigrateVersionUp     = "up"
+	MigrateVersionLatest = "latest"
+	MigrateVersionDown   = "down"
+)
+
+func MigrateCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "migrate",
+		Usage: "Migrate database schema to latest version",
+		Flags: []cli.Flag{
+			&cli.StringFlag{
+				Name:  "target",
+				Usage: "Migration target, default to latest",
+				Value: "latest",
+			},
+			&cli.IntFlag{
+				Name:  "force",
+				Usage: "Force migration to version",
+				Value: -1,
+			},
+		},
+		Action: func(ctx *cli.Context) error {
+			conf, err := common.LoadConfig(ctx)
+			if err != nil {
+				return errors.Wrap(err, "Could not load configuration")
+			}
+
+			driver := string(conf.Database.Driver)
+			dsn := string(conf.Database.DSN)
+
+			migr, err := migrate.New("migrations", driver, dsn)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			version, dirty, err := migr.Version()
+			if err != nil && !errors.Is(err, migrate.ErrNilVersion) {
+				return errors.WithStack(err)
+			}
+
+			logger.Info(
+				ctx.Context, "current database shema",
+				logger.F("version", version),
+				logger.F("dirty", dirty),
+			)
+
+			target := ctx.String("target")
+			force := ctx.Int("force")
+
+			if force != -1 {
+				logger.Info(ctx.Context, "forcing database schema version", logger.F("version", force))
+
+				if err := migr.Force(force); err != nil {
+					return errors.WithStack(err)
+				}
+
+				return nil
+			}
+
+			switch target {
+			case "":
+				fallthrough
+			case MigrateVersionLatest:
+				err = migr.Up()
+			case MigrateVersionDown:
+				err = migr.Steps(-1)
+			case MigrateVersionUp:
+				err = migr.Steps(1)
+			default:
+				return errors.Errorf(
+					"unknown migration target: '%s', available: '%s' (default), '%s' or '%s'",
+					target, MigrateVersionLatest, MigrateVersionUp, MigrateVersionDown,
+				)
+			}
+
+			if err != nil && !errors.Is(err, migrate.ErrNoChange) {
+				return errors.Wrap(err, "could not apply migration")
+			}
+
+			version, dirty, err = migr.Version()
+			if err != nil && !errors.Is(err, migrate.ErrNilVersion) {
+				return errors.WithStack(err)
+			}
+
+			logger.Info(
+				ctx.Context, "database shema after migration",
+				logger.F("version", version),
+				logger.F("dirty", dirty),
+			)
+
+			return nil
+		},
+	}
+}

internal/command/database/ping.go

@@ -0,0 +1,48 @@
+package database
+
+import (
+	"database/sql"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/command/common"
+
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+	"gitlab.com/wpetit/goweb/logger"
+)
+
+func PingCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "ping",
+		Usage: "Test database connectivity",
+		Action: func(ctx *cli.Context) error {
+			conf, err := common.LoadConfig(ctx)
+			if err != nil {
+				return errors.Wrap(err, "Could not load configuration")
+			}
+
+			logger.Info(ctx.Context, "connecting to database", logger.F("dsn", conf.Database.DSN))
+
+			driver := string(conf.Database.Driver)
+			dsn := string(conf.Database.DSN)
+
+			db, err := sql.Open(driver, dsn)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			defer func() {
+				if err := db.Close(); err != nil {
+					logger.Error(ctx.Context, "error while closing database connection", logger.E(errors.WithStack(err)))
+				}
+			}()
+
+			if err := db.PingContext(ctx.Context); err != nil {
+				return errors.WithStack(err)
+			}
+
+			logger.Info(ctx.Context, "connection succeeded", logger.F("dsn", conf.Database.DSN))
+
+			return nil
+		},
+	}
+}

internal/command/database/reset.go

@@ -0,0 +1,38 @@
+package database
+
+import (
+	"forge.cadoles.com/cadoles/bouncer/internal/command/common"
+	"forge.cadoles.com/cadoles/bouncer/internal/migrate"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+	"gitlab.com/wpetit/goweb/logger"
+)
+
+func ResetCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "reset",
+		Usage: "Reset database",
+		Action: func(ctx *cli.Context) error {
+			conf, err := common.LoadConfig(ctx)
+			if err != nil {
+				return errors.Wrap(err, "Could not load configuration")
+			}
+
+			driver := string(conf.Database.Driver)
+			dsn := string(conf.Database.DSN)
+
+			migr, err := migrate.New("migrations", driver, dsn)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			if err := migr.Drop(); err != nil {
+				return errors.Wrap(err, "could not drop tables")
+			}
+
+			logger.Info(ctx.Context, "database schema reinitialized")
+
+			return nil
+		},
+	}
+}

internal/command/database/root.go

@@ -0,0 +1,15 @@
+package database
+
+import "github.com/urfave/cli/v2"
+
+func Root() *cli.Command {
+	return &cli.Command{
+		Name:  "database",
+		Usage: "Database related commands",
+		Subcommands: []*cli.Command{
+			MigrateCommand(),
+			PingCommand(),
+			ResetCommand(),
+		},
+	}
+}