filebrowser/cmd/users.go

package cmd

import (
	"errors"
	"fmt"
	"os"
	"strconv"
	"text/tabwriter"

	"github.com/spf13/cobra"
	"github.com/spf13/pflag"

	"github.com/filebrowser/filebrowser/v2/settings"
	"github.com/filebrowser/filebrowser/v2/users"
)

func init() {
	rootCmd.AddCommand(usersCmd)
}

var usersCmd = &cobra.Command{
	Use:   "users",
	Short: "Users management utility",
	Long:  `Users management utility.`,
	Args:  cobra.NoArgs,
}

func printUsers(usrs []*users.User) {
	w := tabwriter.NewWriter(os.Stdout, 0, 0, 2, ' ', 0)
	fmt.Fprintln(w, "ID\tUsername\tScope\tLocale\tV. Mode\tS.Click\tAdmin\tExecute\tCreate\tRename\tModify\tDelete\tShare\tDownload\tPwd Lock")

	for _, u := range usrs {
		fmt.Fprintf(w, "%d\t%s\t%s\t%s\t%s\t%t\t%t\t%t\t%t\t%t\t%t\t%t\t%t\t%t\t%t\t\n",
			u.ID,
			u.Username,
			u.Scope,
			u.Locale,
			u.ViewMode,
			u.SingleClick,
			u.Perm.Admin,
			u.Perm.Execute,
			u.Perm.Create,
			u.Perm.Rename,
			u.Perm.Modify,
			u.Perm.Delete,
			u.Perm.Share,
			u.Perm.Download,
			u.LockPassword,
		)
	}

	w.Flush()
}

func parseUsernameOrID(arg string) (username string, id uint) {
	id64, err := strconv.ParseUint(arg, 10, 64)
	if err != nil {
		return arg, 0
	}
	return "", uint(id64)
}

func addUserFlags(flags *pflag.FlagSet) {
	flags.Bool("perm.admin", false, "admin perm for users")
	flags.Bool("perm.execute", true, "execute perm for users")
	flags.Bool("perm.create", true, "create perm for users")
	flags.Bool("perm.rename", true, "rename perm for users")
	flags.Bool("perm.modify", true, "modify perm for users")
	flags.Bool("perm.delete", true, "delete perm for users")
	flags.Bool("perm.share", true, "share perm for users")
	flags.Bool("perm.download", true, "download perm for users")
	flags.String("sorting.by", "name", "sorting mode (name, size or modified)")
	flags.Bool("sorting.asc", false, "sorting by ascending order")
	flags.Bool("lockPassword", false, "lock password")
	flags.StringSlice("commands", nil, "a list of the commands a user can execute")
	flags.String("scope", ".", "scope for users")
	flags.String("locale", "en", "locale for users")
	flags.String("viewMode", string(users.ListViewMode), "view mode for users")
	flags.Bool("singleClick", false, "use single clicks only")
	flags.Bool("dateFormat", false, "use date format (true for absolute time, false for relative)")
	flags.Bool("hideDotfiles", false, "hide dotfiles")
	flags.String("aceEditorTheme", "", "ace editor's syntax highlighting theme for users")
	flags.Float64("quota.limit", 0, "quota limit in KB, MB, GB or TB (0 = unlimited)")
	flags.String("quota.unit", "GB", "quota unit (KB, MB, GB or TB)")
	flags.Bool("quota.enforce", false, "enforce quota (hard limit)")
}

func getAndParseViewMode(flags *pflag.FlagSet) (users.ViewMode, error) {
	viewModeStr, err := flags.GetString("viewMode")
	if err != nil {
		return "", err
	}

	viewMode := users.ViewMode(viewModeStr)
	if viewMode != users.ListViewMode && viewMode != users.MosaicViewMode {
		return "", errors.New("view mode must be \"" + string(users.ListViewMode) + "\" or \"" + string(users.MosaicViewMode) + "\"")
	}

	return viewMode, nil
}

func getUserDefaults(flags *pflag.FlagSet, defaults *settings.UserDefaults, all bool) error {
	errs := []error{}

	visit := func(flag *pflag.Flag) {
		var err error
		switch flag.Name {
		case "scope":
			defaults.Scope, err = flags.GetString(flag.Name)
		case "locale":
			defaults.Locale, err = flags.GetString(flag.Name)
		case "viewMode":
			defaults.ViewMode, err = getAndParseViewMode(flags)
		case "singleClick":
			defaults.SingleClick, err = flags.GetBool(flag.Name)
		case "aceEditorTheme":
			defaults.AceEditorTheme, err = flags.GetString(flag.Name)
		case "perm.admin":
			defaults.Perm.Admin, err = flags.GetBool(flag.Name)
		case "perm.execute":
			defaults.Perm.Execute, err = flags.GetBool(flag.Name)
		case "perm.create":
			defaults.Perm.Create, err = flags.GetBool(flag.Name)
		case "perm.rename":
			defaults.Perm.Rename, err = flags.GetBool(flag.Name)
		case "perm.modify":
			defaults.Perm.Modify, err = flags.GetBool(flag.Name)
		case "perm.delete":
			defaults.Perm.Delete, err = flags.GetBool(flag.Name)
		case "perm.share":
			defaults.Perm.Share, err = flags.GetBool(flag.Name)
		case "perm.download":
			defaults.Perm.Download, err = flags.GetBool(flag.Name)
		case "commands":
			defaults.Commands, err = flags.GetStringSlice(flag.Name)
		case "sorting.by":
			defaults.Sorting.By, err = flags.GetString(flag.Name)
		case "sorting.asc":
			defaults.Sorting.Asc, err = flags.GetBool(flag.Name)
		case "hideDotfiles":
			defaults.HideDotfiles, err = flags.GetBool(flag.Name)
		case "quota.limit":
			var quotaLimit float64
			quotaLimit, err = flags.GetFloat64(flag.Name)
			if err == nil {
				var quotaUnit string
				quotaUnit, err = flags.GetString("quota.unit")
				if err == nil {
					defaults.QuotaLimit, err = users.ConvertToBytes(quotaLimit, quotaUnit)
				}
			}
		case "quota.unit":
			defaults.QuotaUnit, err = flags.GetString(flag.Name)
		case "quota.enforce":
			defaults.EnforceQuota, err = flags.GetBool(flag.Name)
		}

		if err != nil {
			errs = append(errs, err)
		}
	}

	if all {
		flags.VisitAll(visit)
	} else {
		flags.Visit(visit)
	}

	return errors.Join(errs...)
}