photoprism/internal/commands/auth_reset.go

package commands

import (
	"github.com/sirupsen/logrus"

	"github.com/manifoldco/promptui"
	"github.com/urfave/cli/v2"

	"github.com/photoprism/photoprism/internal/config"
	"github.com/photoprism/photoprism/internal/entity"
)

const AuthResetDescription = "This command recreates the auth_sessions database table so that it is compatible with the current version. As a result, all users and clients must re-authenticate. Note that any client access tokens and app passwords that users may have created are also deleted and must be recreated."

// AuthResetCommand configures the command name, flags, and action.
var AuthResetCommand = &cli.Command{
	Name:        "reset",
	Usage:       "Resets the authentication of all users and clients",
	Description: AuthResetDescription,
	Flags: []cli.Flag{
		&cli.BoolFlag{
			Name:    "trace",
			Aliases: []string{"t"},
			Usage:   "shows trace logs for debugging",
		},
		&cli.BoolFlag{
			Name:    "yes",
			Aliases: []string{"y"},
			Usage:   "runs the command non-interactively",
		},
	},
	Action: authResetAction,
}

// authResetAction removes all sessions and resets the related database table to a clean state.
func authResetAction(ctx *cli.Context) error {
	return CallWithDependencies(ctx, func(conf *config.Config) error {
		confirmed := RunNonInteractively(ctx.Bool("yes"))

		// Show prompt?
		if !confirmed {
			actionPrompt := promptui.Prompt{
				Label:     "Remove all sessions and reset the database table to a clean state?",
				IsConfirm: true,
			}

			if _, err := actionPrompt.Run(); err != nil {
				return nil
			}
		}

		if ctx.Bool("trace") {
			log.SetLevel(logrus.TraceLevel)
			log.Infoln("clear: enabled trace mode")
		}

		db := conf.Db()

		// Drop existing sessions table.
		if err := db.DropTableIfExists(entity.Session{}).Error; err != nil {
			return err
		}

		// Re-create auth_sessions.
		if err := db.CreateTable(entity.Session{}).Error; err != nil {
			return err
		}

		log.Infof("all sessions have been removed")

		return nil
	})
}