c60cc7ca75
Some checks failed
CI / Run tests and lints (push) Failing after 29s
CI / Build CLI docs (push) Has been skipped
CI / Run Playwright end-to-end tests (mysql) (push) Failing after 3s
CI / Run Playwright end-to-end tests (postgres) (push) Failing after 2s
CI / Run Playwright end-to-end tests (sqlite) (push) Failing after 3s
CodeQL / Analyze (go) (push) Failing after 4s
CodeQL / Analyze (javascript) (push) Failing after 3s
Docker Image Scanners / scanners (push) Failing after 4s
Synchronize Issue Labels / Synchronize Issue Labels (push) Has been skipped
CI / Run end-to-end tests (postgres) (push) Failing after 19s
CI / Run end-to-end tests (sqlite) (push) Failing after 2s
CI / Run Playwright end-to-end tests (cockroach) (push) Failing after 2s
CI / Draft newsletter (push) Has been skipped
CI / Generate release (push) Has been skipped
CI / Pending approval Slack notification (push) Has been skipped
CI / Send newsletter (push) Has been skipped
106 lines
2.6 KiB
Go
106 lines
2.6 KiB
Go
// Copyright © 2023 Ory Corp
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package migrate
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/ory/x/popx"
|
|
|
|
"github.com/ory/x/contextx"
|
|
|
|
"github.com/ory/x/configx"
|
|
|
|
"github.com/spf13/cobra"
|
|
|
|
"github.com/ory/kratos/driver"
|
|
"github.com/ory/kratos/driver/config"
|
|
"github.com/ory/x/cmdx"
|
|
"github.com/ory/x/flagx"
|
|
)
|
|
|
|
type MigrateHandler struct{}
|
|
|
|
func NewMigrateHandler() *MigrateHandler {
|
|
return &MigrateHandler{}
|
|
}
|
|
|
|
func (h *MigrateHandler) getPersister(cmd *cobra.Command, args []string, opts []driver.RegistryOption) (driver.Registry, error) {
|
|
return getPersister(cmd, args, opts)
|
|
}
|
|
|
|
func getPersister(cmd *cobra.Command, args []string, opts []driver.RegistryOption) (d driver.Registry, err error) {
|
|
if flagx.MustGetBool(cmd, "read-from-env") {
|
|
d, err = driver.NewWithoutInit(
|
|
cmd.Context(),
|
|
cmd.ErrOrStderr(),
|
|
driver.WithConfigOptions(
|
|
configx.WithFlags(cmd.Flags()),
|
|
configx.SkipValidation(),
|
|
))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if len(d.Config().DSN(cmd.Context())) == 0 {
|
|
fmt.Println(cmd.UsageString())
|
|
fmt.Println("")
|
|
fmt.Println("When using flag -e, environment variable DSN must be set")
|
|
return nil, cmdx.FailSilently(cmd)
|
|
}
|
|
} else {
|
|
if len(args) != 1 {
|
|
fmt.Println(cmd.UsageString())
|
|
return nil, cmdx.FailSilently(cmd)
|
|
}
|
|
d, err = driver.NewWithoutInit(
|
|
cmd.Context(),
|
|
cmd.ErrOrStderr(),
|
|
driver.WithConfigOptions(
|
|
configx.WithFlags(cmd.Flags()),
|
|
configx.SkipValidation(),
|
|
configx.WithValue(config.ViperKeyDSN, args[0]),
|
|
))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
err = d.Init(cmd.Context(), &contextx.Default{}, append(opts, driver.SkipNetworkInit)...)
|
|
if err != nil {
|
|
return nil, errors.Wrap(err, "an error occurred initializing migrations")
|
|
}
|
|
|
|
if err := popx.VerifyDialect(cmd.Context(), d.Persister().GetConnection(cmd.Context())); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return d, nil
|
|
}
|
|
|
|
func (h *MigrateHandler) MigrateSQLDown(cmd *cobra.Command, args []string, opts ...driver.RegistryOption) error {
|
|
p, err := h.getPersister(cmd, args, opts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return popx.MigrateSQLDown(cmd, p.Persister())
|
|
}
|
|
|
|
func (h *MigrateHandler) MigrateSQLStatus(cmd *cobra.Command, args []string, opts ...driver.RegistryOption) error {
|
|
p, err := h.getPersister(cmd, args, opts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return popx.MigrateStatus(cmd, p.Persister())
|
|
}
|
|
|
|
func (h *MigrateHandler) MigrateSQLUp(cmd *cobra.Command, args []string, opts ...driver.RegistryOption) error {
|
|
p, err := h.getPersister(cmd, args, opts)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return popx.MigrateSQLUp(cmd, p.Persister())
|
|
}
|