filebrowser/cmd/config_init.go

package cmd

import (
	"fmt"

	"github.com/spf13/cobra"

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

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

var configInitCmd = &cobra.Command{
	Use:   "init",
	Short: "Initialize a new database",
	Long: `Initialize a new database to use with File Browser. All of
this options can be changed in the future with the command
'filebrowser config set'. The user related flags apply
to the defaults when creating new users and you don't
override the options.`,
	Args: cobra.NoArgs,
	RunE: python(func(cmd *cobra.Command, _ []string, d *pythonData) error {
		defaults := settings.UserDefaults{}
		flags := cmd.Flags()
		err := getUserDefaults(flags, &defaults, true)
		if err != nil {
			return err
		}
		authMethod, auther, err := getAuthentication(flags)
		if err != nil {
			return err
		}

		key := generateKey()

		signup, err := getBool(flags, "signup")
		if err != nil {
			return err
		}

		hideLoginButton, err := getBool(flags, "hide-login-button")
		if err != nil {
			return err
		}

		createUserDir, err := getBool(flags, "create-user-dir")
		if err != nil {
			return err
		}

		minLength, err := getUint(flags, "minimum-password-length")
		if err != nil {
			return err
		}

		shell, err := getString(flags, "shell")
		if err != nil {
			return err
		}

		brandingName, err := getString(flags, "branding.name")
		if err != nil {
			return err
		}

		brandingDisableExternal, err := getBool(flags, "branding.disableExternal")
		if err != nil {
			return err
		}

		brandingDisableUsedPercentage, err := getBool(flags, "branding.disableUsedPercentage")
		if err != nil {
			return err
		}

		brandingTheme, err := getString(flags, "branding.theme")
		if err != nil {
			return err
		}

		brandingFiles, err := getString(flags, "branding.files")
		if err != nil {
			return err
		}

		tusChunkSize, err := flags.GetUint64("tus.chunkSize")
		if err != nil {
			return err
		}

		tusRetryCount, err := flags.GetUint16("tus.retryCount")
		if err != nil {
			return err
		}

		s := &settings.Settings{
			Key:                   key,
			Signup:                signup,
			HideLoginButton:       hideLoginButton,
			CreateUserDir:         createUserDir,
			MinimumPasswordLength: minLength,
			Shell:                 convertCmdStrToCmdArray(shell),
			AuthMethod:            authMethod,
			Defaults:              defaults,
			Branding: settings.Branding{
				Name:                  brandingName,
				DisableExternal:       brandingDisableExternal,
				DisableUsedPercentage: brandingDisableUsedPercentage,
				Theme:                 brandingTheme,
				Files:                 brandingFiles,
			},
			Tus: settings.Tus{
				ChunkSize:  tusChunkSize,
				RetryCount: tusRetryCount,
			},
		}

		s.FileMode, err = getMode(flags, "file-mode")
		if err != nil {
			return err
		}

		s.DirMode, err = getMode(flags, "dir-mode")
		if err != nil {
			return err
		}

		address, err := getString(flags, "address")
		if err != nil {
			return err
		}

		socket, err := getString(flags, "socket")
		if err != nil {
			return err
		}

		root, err := getString(flags, "root")
		if err != nil {
			return err
		}

		baseURL, err := getString(flags, "baseurl")
		if err != nil {
			return err
		}

		tlsKey, err := getString(flags, "key")
		if err != nil {
			return err
		}

		cert, err := getString(flags, "cert")
		if err != nil {
			return err
		}

		port, err := getString(flags, "port")
		if err != nil {
			return err
		}

		log, err := getString(flags, "log")
		if err != nil {
			return err
		}

		tokenExpirationTime := getStringParam(flags, "token-expiration-time")

		ser := &settings.Server{
			Address:             address,
			Socket:              socket,
			Root:                root,
			BaseURL:             baseURL,
			TLSKey:              tlsKey,
			TLSCert:             cert,
			Port:                port,
			Log:                 log,
			TokenExpirationTime: tokenExpirationTime,
		}

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

		fmt.Printf(`
Congratulations! You've set up your database to use with File Browser.
Now add your first user via 'filebrowser users add' and then you just
need to call the main command to boot up the server.
`)
		return printSettings(ser, s, auther)
	}, pythonConfig{noDB: true}),
}