mirror of https://github.com/ory/hydra
81 lines
1.9 KiB
Go
81 lines
1.9 KiB
Go
// Copyright © 2022 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package cli
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/ory/hydra/v2/driver"
|
|
"github.com/ory/hydra/v2/driver/config"
|
|
"github.com/ory/hydra/v2/persistence/sql"
|
|
"github.com/ory/x/cmdx"
|
|
"github.com/ory/x/configx"
|
|
"github.com/ory/x/popx"
|
|
)
|
|
|
|
type MigrateHandler struct {
|
|
dOpts []driver.OptionsModifier
|
|
}
|
|
|
|
func newMigrateHandler(dOpts []driver.OptionsModifier) *MigrateHandler {
|
|
return &MigrateHandler{
|
|
dOpts: dOpts,
|
|
}
|
|
}
|
|
|
|
func (h *MigrateHandler) makeMigrationManager(cmd *cobra.Command, args []string) (*sql.MigrationManager, error) {
|
|
opts := append([]driver.OptionsModifier{
|
|
driver.WithConfigOptions(
|
|
configx.SkipValidation(),
|
|
configx.WithFlags(cmd.Flags())),
|
|
driver.DisableValidation(),
|
|
driver.DisablePreloading(),
|
|
driver.SkipNetworkInit(),
|
|
}, h.dOpts...)
|
|
if len(args) > 0 {
|
|
opts = append(opts, driver.WithConfigOptions(
|
|
configx.WithValue(config.KeyDSN, args[0]),
|
|
))
|
|
}
|
|
|
|
d, err := driver.New(
|
|
cmd.Context(),
|
|
opts...)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(d.Config().DSN()) == 0 {
|
|
_, _ = fmt.Fprintln(cmd.ErrOrStderr(), "No DSN provided. Please provide a DSN as the first argument or set the DSN environment variable.")
|
|
return nil, cmdx.FailSilently(cmd)
|
|
}
|
|
|
|
return d.Migrator(), nil
|
|
}
|
|
|
|
func (h *MigrateHandler) MigrateSQLUp(cmd *cobra.Command, args []string) (err error) {
|
|
p, err := h.makeMigrationManager(cmd, args)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return popx.MigrateSQLUp(cmd, p)
|
|
}
|
|
|
|
func (h *MigrateHandler) MigrateSQLDown(cmd *cobra.Command, args []string) (err error) {
|
|
p, err := h.makeMigrationManager(cmd, args)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return popx.MigrateSQLDown(cmd, p)
|
|
}
|
|
|
|
func (h *MigrateHandler) MigrateStatus(cmd *cobra.Command, args []string) error {
|
|
p, err := h.makeMigrationManager(cmd, args)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return popx.MigrateStatus(cmd, p)
|
|
}
|