85 lines
2 KiB
Go
85 lines
2 KiB
Go
|
package commands
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/manifoldco/promptui"
|
||
|
|
"github.com/sirupsen/logrus"
|
||
|
|
"github.com/urfave/cli"
|
||
|
|
|
||
|
|
"github.com/photoprism/photoprism/internal/config"
|
||
|
|
"github.com/photoprism/photoprism/internal/entity"
|
||
|
|
)
|
||
|
|
|
||
|
|
// UsersResetCommand configures the command name, flags, and action.
|
||
|
|
var UsersResetCommand = cli.Command{
|
||
|
|
Name: "reset",
|
||
|
|
Usage: "Resets the user database to a clean state",
|
||
|
|
Flags: []cli.Flag{
|
||
|
|
cli.BoolFlag{
|
||
|
|
Name: "trace, t",
|
||
|
|
Usage: "show trace logs for debugging",
|
||
|
|
},
|
||
|
|
cli.BoolFlag{
|
||
|
|
Name: "yes, y",
|
||
|
|
Usage: "assume \"yes\" and run non-interactively",
|
||
|
|
},
|
||
|
|
},
|
||
|
|
Action: usersResetAction,
|
||
|
|
}
|
||
|
|
|
||
|
|
// usersResetAction deletes recreates the user management database tables.
|
||
|
|
func usersResetAction(ctx *cli.Context) error {
|
||
|
|
return CallWithDependencies(ctx, func(conf *config.Config) error {
|
||
|
|
confirmed := ctx.Bool("yes")
|
||
|
|
|
||
|
|
// Show prompt?
|
||
|
|
if !confirmed {
|
||
|
|
actionPrompt := promptui.Prompt{
|
||
|
|
Label: fmt.Sprintf("Reset the user database to a clean state?"),
|
||
|
|
IsConfirm: true,
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := actionPrompt.Run(); err != nil {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
if ctx.Bool("trace") {
|
||
|
|
log.SetLevel(logrus.TraceLevel)
|
||
|
|
log.Infoln("reset: enabled trace mode")
|
||
|
|
}
|
||
|
|
|
||
|
|
db := conf.Db()
|
||
|
|
|
||
|
|
// Drop existing user management tables.
|
||
|
|
if err := db.DropTableIfExists(entity.User{}, entity.UserDetails{}, entity.UserSettings{}, entity.UserShare{}).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Re-create auth_users.
|
||
|
|
if err := db.CreateTable(entity.User{}).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Re-create auth_users_details.
|
||
|
|
if err := db.CreateTable(entity.UserDetails{}).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Re-create auth_users_settings.
|
||
|
|
if err := db.CreateTable(entity.UserSettings{}).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Re-create auth_users_shares.
|
||
|
|
if err := db.CreateTable(entity.UserShare{}).Error; err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
log.Infof("the user database has been recreated and is now in a clean state")
|
||
|
|
|
||
|
|
return nil
|
||
|
|
})
|
||
|
|
}
|