1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889 |
- // Copyright (C) 2019-2023 Nicola Murino
- //
- // This program is free software: you can redistribute it and/or modify
- // it under the terms of the GNU Affero General Public License as published
- // by the Free Software Foundation, version 3.
- //
- // This program is distributed in the hope that it will be useful,
- // but WITHOUT ANY WARRANTY; without even the implied warranty of
- // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- // GNU Affero General Public License for more details.
- //
- // You should have received a copy of the GNU Affero General Public License
- // along with this program. If not, see <https://www.gnu.org/licenses/>.
- package cmd
- import (
- "bufio"
- "os"
- "strings"
- "github.com/rs/zerolog"
- "github.com/spf13/cobra"
- "github.com/spf13/viper"
- "github.com/drakkan/sftpgo/v2/internal/config"
- "github.com/drakkan/sftpgo/v2/internal/dataprovider"
- "github.com/drakkan/sftpgo/v2/internal/logger"
- "github.com/drakkan/sftpgo/v2/internal/util"
- )
- var (
- resetProviderForce bool
- resetProviderCmd = &cobra.Command{
- Use: "resetprovider",
- Short: "Reset the configured provider, any data will be lost",
- Long: `This command reads the data provider connection details from the specified
- configuration file and resets the provider by deleting all data and schemas.
- This command is not supported for the memory provider.
- Please take a look at the usage below to customize the options.`,
- Run: func(_ *cobra.Command, _ []string) {
- logger.DisableLogger()
- logger.EnableConsoleLogger(zerolog.DebugLevel)
- configDir = util.CleanDirInput(configDir)
- err := config.LoadConfig(configDir, configFile)
- if err != nil {
- logger.WarnToConsole("Unable to initialize data provider, config load error: %v", err)
- os.Exit(1)
- }
- kmsConfig := config.GetKMSConfig()
- err = kmsConfig.Initialize()
- if err != nil {
- logger.ErrorToConsole("unable to initialize KMS: %v", err)
- os.Exit(1)
- }
- providerConf := config.GetProviderConf()
- if !resetProviderForce {
- logger.WarnToConsole("You are about to delete all the SFTPGo data for provider %q, config file: %q",
- providerConf.Driver, viper.ConfigFileUsed())
- logger.WarnToConsole("Are you sure? (Y/n)")
- reader := bufio.NewReader(os.Stdin)
- answer, err := reader.ReadString('\n')
- if err != nil {
- logger.ErrorToConsole("unable to read your answer: %v", err)
- os.Exit(1)
- }
- if strings.ToUpper(strings.TrimSpace(answer)) != "Y" {
- logger.InfoToConsole("command aborted")
- os.Exit(1)
- }
- }
- logger.InfoToConsole("Resetting provider: %q, config file: %q", providerConf.Driver, viper.ConfigFileUsed())
- err = dataprovider.ResetDatabase(providerConf, configDir)
- if err != nil {
- logger.WarnToConsole("Error resetting provider: %v", err)
- os.Exit(1)
- }
- logger.InfoToConsole("Tha data provider was successfully reset")
- },
- }
- )
- func init() {
- addConfigFlags(resetProviderCmd)
- resetProviderCmd.Flags().BoolVar(&resetProviderForce, "force", false, `reset the provider without asking for confirmation`)
- rootCmd.AddCommand(resetProviderCmd)
- }
|