filebrowser/cmd/config_set.go

package cmd

import (
	"github.com/spf13/cobra"
	"github.com/spf13/pflag"
)

func init() {
	configCmd.AddCommand(configSetCmd)
	addConfigFlags(configSetCmd.Flags())
}

var configSetCmd = &cobra.Command{
	Use:   "set",
	Short: "Updates the configuration",
	Long: `Updates the configuration. Set the flags for the options
you want to change. Other options will remain unchanged.`,
	Args: cobra.NoArgs,
	RunE: python(func(cmd *cobra.Command, _ []string, d *pythonData) error {
		flags := cmd.Flags()

		set, err := d.store.Settings.Get()
		if err != nil {
			return err
		}

		ser, err := d.store.Settings.GetServer()
		if err != nil {
			return err
		}

		hasAuth := false

		flags.Visit(func(flag *pflag.Flag) {
			if err != nil {
				return
			}

			switch flag.Name {
			// Server flags from [addServerFlags]
			case "address":
				ser.Address, err = flags.GetString(flag.Name)
			case "log":
				ser.Log, err = flags.GetString(flag.Name)
			case "port":
				ser.Port, err = flags.GetString(flag.Name)
			case "cert":
				ser.TLSCert, err = flags.GetString(flag.Name)
			case "key":
				ser.TLSKey, err = flags.GetString(flag.Name)
			case "root":
				ser.Root, err = flags.GetString(flag.Name)
			case "socket":
				ser.Socket, err = flags.GetString(flag.Name)
			case "baseURL":
				ser.BaseURL, err = flags.GetString(flag.Name)
			case "tokenExpirationTime":
				ser.TokenExpirationTime, err = flags.GetString(flag.Name)
			case "disableThumbnails":
				ser.EnableThumbnails, err = flags.GetBool(flag.Name)
				ser.EnableThumbnails = !ser.EnableThumbnails
			case "disablePreviewResize":
				ser.ResizePreview, err = flags.GetBool(flag.Name)
				ser.ResizePreview = !ser.ResizePreview
			case "disableExec":
				ser.EnableExec, err = flags.GetBool(flag.Name)
				ser.EnableExec = !ser.EnableExec
			case "disableTypeDetectionByHeader":
				ser.TypeDetectionByHeader, err = flags.GetBool(flag.Name)
				ser.TypeDetectionByHeader = !ser.TypeDetectionByHeader

				// Settings flags from [addConfigFlags]
			case "signup":
				set.Signup, err = flags.GetBool(flag.Name)
			case "hideLoginButton":
				set.HideLoginButton, err = flags.GetBool(flag.Name)
			case "createUserDir":
				set.CreateUserDir, err = flags.GetBool(flag.Name)
			case "minimumPasswordLength":
				set.MinimumPasswordLength, err = flags.GetUint(flag.Name)
			case "shell":
				var shell string
				shell, err = flags.GetString(flag.Name)
				if err != nil {
					return
				}
				set.Shell = convertCmdStrToCmdArray(shell)
			case "auth.method":
				hasAuth = true
			case "branding.name":
				set.Branding.Name, err = flags.GetString(flag.Name)
			case "branding.theme":
				set.Branding.Theme, err = flags.GetString(flag.Name)
			case "branding.color":
				set.Branding.Color, err = flags.GetString(flag.Name)
			case "branding.files":
				set.Branding.Files, err = flags.GetString(flag.Name)
			case "branding.disableExternal":
				set.Branding.DisableExternal, err = flags.GetBool(flag.Name)
			case "branding.disableUsedPercentage":
				set.Branding.DisableUsedPercentage, err = flags.GetBool(flag.Name)
			case "fileMode":
				set.FileMode, err = getAndParseFileMode(flags, flag.Name)
			case "dirMode":
				set.DirMode, err = getAndParseFileMode(flags, flag.Name)
			case "tus.chunkSize":
				set.Tus.ChunkSize, err = flags.GetUint64(flag.Name)
			case "tus.retryCount":
				set.Tus.RetryCount, err = flags.GetUint16(flag.Name)
			}

		})
		if err != nil {
			return err
		}

		err = getUserDefaults(flags, &set.Defaults, false)
		if err != nil {
			return err
		}

		// read the defaults
		auther, err := d.store.Auth.Get(set.AuthMethod)
		if err != nil {
			return err
		}

		// check if there are new flags for existing auth method
		set.AuthMethod, auther, err = getAuthentication(flags, hasAuth, set, auther)
		if err != nil {
			return err
		}

		err = d.store.Auth.Save(auther)
		if err != nil {
			return err
		}
		err = d.store.Settings.Save(set)
		if err != nil {
			return err
		}
		err = d.store.Settings.SaveServer(ser)
		if err != nil {
			return err
		}

		return printSettings(ser, set, auther)
	}, pythonConfig{}),
}