feat: Add resetPassword cli tool

Author: sosyz

cmd/command.go

@@ -20,11 +20,13 @@
 package answercmd
 
 import (
+	"context"
 	"fmt"
 	"os"
 	"strings"
 
 	"github.com/apache/answer/internal/base/conf"
+	"github.com/apache/answer/internal/base/path"
 	"github.com/apache/answer/internal/cli"
 	"github.com/apache/answer/internal/install"
 	"github.com/apache/answer/internal/migrations"
@@ -53,6 +55,10 @@ var (
 	i18nSourcePath string
 	// i18nTargetPath i18n to path
 	i18nTargetPath string
+	// resetPasswordEmail user email for password reset
+	resetPasswordEmail string
+	// resetPasswordPassword new password for password reset
+	resetPasswordPassword string
 )
 
 func init() {
@@ -76,7 +82,10 @@ func init() {
 
 	i18nCmd.Flags().StringVarP(&i18nTargetPath, "target", "t", "", "i18n target path, eg: -t ./i18n/target")
 
-	for _, cmd := range []*cobra.Command{initCmd, checkCmd, runCmd, dumpCmd, upgradeCmd, buildCmd, pluginCmd, configCmd, i18nCmd} {
+	resetPasswordCmd.Flags().StringVarP(&resetPasswordEmail, "email", "e", "", "user email address")
+	resetPasswordCmd.Flags().StringVarP(&resetPasswordPassword, "password", "p", "", "new password (not recommended, will be recorded in shell history)")
+
+	for _, cmd := range []*cobra.Command{initCmd, checkCmd, runCmd, dumpCmd, upgradeCmd, buildCmd, pluginCmd, configCmd, i18nCmd, resetPasswordCmd} {
 		rootCmd.AddCommand(cmd)
 	}
 }
@@ -96,8 +105,8 @@ To run answer, use:
 		Short: "Run Answer",
 		Long:  `Start running Answer`,
 		Run: func(_ *cobra.Command, _ []string) {
-			cli.FormatAllPath(dataDirPath)
-			fmt.Println("config file path: ", cli.GetConfigFilePath())
+			path.FormatAllPath(dataDirPath)
+			fmt.Println("config file path: ", path.GetConfigFilePath())
 			fmt.Println("Answer is starting..........................")
 			runApp()
 		},
@@ -111,10 +120,10 @@ To run answer, use:
 			// check config file and database. if config file exists and database is already created, init done
 			cli.InstallAllInitialEnvironment(dataDirPath)
 
-			configFileExist := cli.CheckConfigFile(cli.GetConfigFilePath())
+			configFileExist := cli.CheckConfigFile(path.GetConfigFilePath())
 			if configFileExist {
 				fmt.Println("config file exists, try to read the config...")
-				c, err := conf.ReadConfig(cli.GetConfigFilePath())
+				c, err := conf.ReadConfig(path.GetConfigFilePath())
 				if err != nil {
 					fmt.Println("read config failed: ", err.Error())
 					return
@@ -128,7 +137,7 @@ To run answer, use:
 			}
 
 			// start installation server to install
-			install.Run(cli.GetConfigFilePath())
+			install.Run(path.GetConfigFilePath())
 		},
 	}
 
@@ -138,9 +147,9 @@ To run answer, use:
 		Long:  `Upgrade Answer to the latest version`,
 		Run: func(_ *cobra.Command, _ []string) {
 			log.SetLogger(log.NewStdLogger(os.Stdout))
-			cli.FormatAllPath(dataDirPath)
+			path.FormatAllPath(dataDirPath)
 			cli.InstallI18nBundle(true)
-			c, err := conf.ReadConfig(cli.GetConfigFilePath())
+			c, err := conf.ReadConfig(path.GetConfigFilePath())
 			if err != nil {
 				fmt.Println("read config failed: ", err.Error())
 				return
@@ -159,8 +168,8 @@ To run answer, use:
 		Long:  `Back up database into an SQL file`,
 		Run: func(_ *cobra.Command, _ []string) {
 			fmt.Println("Answer is backing up data")
-			cli.FormatAllPath(dataDirPath)
-			c, err := conf.ReadConfig(cli.GetConfigFilePath())
+			path.FormatAllPath(dataDirPath)
+			c, err := conf.ReadConfig(path.GetConfigFilePath())
 			if err != nil {
 				fmt.Println("read config failed: ", err.Error())
 				return
@@ -179,9 +188,9 @@ To run answer, use:
 		Short: "Check the required environment",
 		Long:  `Check if the current environment meets the startup requirements`,
 		Run: func(_ *cobra.Command, _ []string) {
-			cli.FormatAllPath(dataDirPath)
+			path.FormatAllPath(dataDirPath)
 			fmt.Println("Start checking the required environment...")
-			if cli.CheckConfigFile(cli.GetConfigFilePath()) {
+			if cli.CheckConfigFile(path.GetConfigFilePath()) {
 				fmt.Println("config file exists [✔]")
 			} else {
 				fmt.Println("config file not exists [x]")
@@ -193,7 +202,7 @@ To run answer, use:
 				fmt.Println("upload directory not exists [x]")
 			}
 
-			c, err := conf.ReadConfig(cli.GetConfigFilePath())
+			c, err := conf.ReadConfig(path.GetConfigFilePath())
 			if err != nil {
 				fmt.Println("read config failed: ", err.Error())
 				return
@@ -246,9 +255,9 @@ To run answer, use:
 		Short: "Set some config to default value",
 		Long:  `Set some config to default value`,
 		Run: func(_ *cobra.Command, _ []string) {
-			cli.FormatAllPath(dataDirPath)
+			path.FormatAllPath(dataDirPath)
 
-			c, err := conf.ReadConfig(cli.GetConfigFilePath())
+			c, err := conf.ReadConfig(path.GetConfigFilePath())
 			if err != nil {
 				fmt.Println("read config failed: ", err.Error())
 				return
@@ -297,6 +306,32 @@ To run answer, use:
 			}
 		},
 	}
+
+	resetPasswordCmd = &cobra.Command{
+		Use:     "passwd",
+		Aliases: []string{"password", "reset-password"},
+		Short:   "Reset user password",
+		Long:    "Reset user password by email address.",
+		Example: `  # Interactive mode (recommended, safest)
+  answer passwd -C ./answer-data
+
+  # Specify email only (will prompt for password securely)
+  answer passwd -C ./answer-data --email user@example.com
+  answer passwd -C ./answer-data -e user@example.com
+
+  # Specify email and password (NOT recommended, will be recorded in shell history)
+  answer passwd -C ./answer-data -e user@example.com -p newpassword123`,
+		Run: func(cmd *cobra.Command, args []string) {
+			opts := &cli.ResetPasswordOptions{
+				Email:    resetPasswordEmail,
+				Password: resetPasswordPassword,
+			}
+			if err := cli.ResetPassword(context.Background(), dataDirPath, opts); err != nil {
+				fmt.Fprintf(os.Stderr, "Error: %v\n", err)
+				os.Exit(1)
+			}
+		},
+	}
 )
 
 // Execute adds all child commands to the root command and sets flags appropriately.

cmd/main.go

@@ -28,7 +28,7 @@ import (
 	"github.com/apache/answer/internal/base/conf"
 	"github.com/apache/answer/internal/base/constant"
 	"github.com/apache/answer/internal/base/cron"
-	"github.com/apache/answer/internal/cli"
+	"github.com/apache/answer/internal/base/path"
 	"github.com/apache/answer/internal/schema"
 	"github.com/gin-gonic/gin"
 	"github.com/segmentfault/pacman"
@@ -67,7 +67,7 @@ func Main() {
 }
 
 func runApp() {
-	c, err := conf.ReadConfig(cli.GetConfigFilePath())
+	c, err := conf.ReadConfig(path.GetConfigFilePath())
 	if err != nil {
 		panic(err)
 	}

go.mod

@@ -60,6 +60,7 @@ require (
 	golang.org/x/crypto v0.36.0
 	golang.org/x/image v0.20.0
 	golang.org/x/net v0.38.0
+	golang.org/x/term v0.30.0
 	golang.org/x/text v0.23.0
 	gopkg.in/gomail.v2 v2.0.0-20160411212932-81ebce5c23df
 	gopkg.in/yaml.v3 v3.0.1

go.sum

@@ -785,6 +785,8 @@ golang.org/x/term v0.0.0-20201117132131-f5c789dd3221/go.mod h1:Nr5EML6q2oocZ2LXR
 golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo=
 golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8=
 golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k=
+golang.org/x/term v0.30.0 h1:PQ39fJZ+mfadBm0y5WlL4vlM7Sx1Hgf13sMIY2+QS9Y=
+golang.org/x/term v0.30.0/go.mod h1:NYYFdzHoI5wRh/h5tDMdMqCqPJZEuNqVR5xJLd/n67g=
 golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ=
 golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk=
 golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ=

internal/base/conf/conf.go

@@ -25,9 +25,9 @@ import (
 	"path/filepath"
 
 	"github.com/apache/answer/internal/base/data"
+	"github.com/apache/answer/internal/base/path"
 	"github.com/apache/answer/internal/base/server"
 	"github.com/apache/answer/internal/base/translator"
-	"github.com/apache/answer/internal/cli"
 	"github.com/apache/answer/internal/router"
 	"github.com/apache/answer/internal/service/service_config"
 	"github.com/apache/answer/pkg/writer"
@@ -98,7 +98,7 @@ func (c *AllConfig) SetEnvironmentOverrides() {
 // ReadConfig read config
 func ReadConfig(configFilePath string) (c *AllConfig, err error) {
 	if len(configFilePath) == 0 {
-		configFilePath = filepath.Join(cli.ConfigFileDir, cli.DefaultConfigFileName)
+		configFilePath = filepath.Join(path.ConfigFileDir, path.DefaultConfigFileName)
 	}
 	c = &AllConfig{}
 	config, err := viper.NewWithPath(configFilePath)

internal/base/path/path.go

@@ -0,0 +1,53 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package path
+
+import (
+	"path/filepath"
+	"sync"
+)
+
+const (
+	DefaultConfigFileName                  = "config.yaml"
+	DefaultCacheFileName                   = "cache.db"
+	DefaultReservedUsernamesConfigFileName = "reserved-usernames.json"
+)
+
+var (
+	ConfigFileDir     = "/conf/"
+	UploadFilePath    = "/uploads/"
+	I18nPath          = "/i18n/"
+	CacheDir          = "/cache/"
+	formatAllPathOnce sync.Once
+)
+
+func FormatAllPath(dataDirPath string) {
+	formatAllPathOnce.Do(func() {
+		ConfigFileDir = filepath.Join(dataDirPath, ConfigFileDir)
+		UploadFilePath = filepath.Join(dataDirPath, UploadFilePath)
+		I18nPath = filepath.Join(dataDirPath, I18nPath)
+		CacheDir = filepath.Join(dataDirPath, CacheDir)
+	})
+}
+
+// GetConfigFilePath get config file path
+func GetConfigFilePath() string {
+	return filepath.Join(ConfigFileDir, DefaultConfigFileName)
+}

internal/cli/install.go

@@ -23,53 +23,25 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
-	"sync"
 
 	"github.com/apache/answer/configs"
 	"github.com/apache/answer/i18n"
+	"github.com/apache/answer/internal/base/path"
 	"github.com/apache/answer/pkg/dir"
 	"github.com/apache/answer/pkg/writer"
 )
 
-const (
-	DefaultConfigFileName                  = "config.yaml"
-	DefaultCacheFileName                   = "cache.db"
-	DefaultReservedUsernamesConfigFileName = "reserved-usernames.json"
-)
-
-var (
-	ConfigFileDir     = "/conf/"
-	UploadFilePath    = "/uploads/"
-	I18nPath          = "/i18n/"
-	CacheDir          = "/cache/"
-	formatAllPathONCE sync.Once
-)
-
-// GetConfigFilePath get config file path
-func GetConfigFilePath() string {
-	return filepath.Join(ConfigFileDir, DefaultConfigFileName)
-}
-
-func FormatAllPath(dataDirPath string) {
-	formatAllPathONCE.Do(func() {
-		ConfigFileDir = filepath.Join(dataDirPath, ConfigFileDir)
-		UploadFilePath = filepath.Join(dataDirPath, UploadFilePath)
-		I18nPath = filepath.Join(dataDirPath, I18nPath)
-		CacheDir = filepath.Join(dataDirPath, CacheDir)
-	})
-}
-
 // InstallAllInitialEnvironment install all initial environment
 func InstallAllInitialEnvironment(dataDirPath string) {
-	FormatAllPath(dataDirPath)
+	path.FormatAllPath(dataDirPath)
 	installUploadDir()
 	InstallI18nBundle(false)
 	fmt.Println("install all initial environment done")
 }
 
 func InstallConfigFile(configFilePath string) error {
 	if len(configFilePath) == 0 {
-		configFilePath = filepath.Join(ConfigFileDir, DefaultConfigFileName)
+		configFilePath = filepath.Join(path.ConfigFileDir, path.DefaultConfigFileName)
 	}
 	fmt.Println("[config-file] try to create at ", configFilePath)
 
@@ -79,7 +51,7 @@ func InstallConfigFile(configFilePath string) error {
 		return nil
 	}
 
-	if err := dir.CreateDirIfNotExist(ConfigFileDir); err != nil {
+	if err := dir.CreateDirIfNotExist(path.ConfigFileDir); err != nil {
 		fmt.Printf("[config-file] create directory fail %s\n", err.Error())
 		return fmt.Errorf("create directory fail %s", err.Error())
 	}
@@ -95,10 +67,10 @@ func InstallConfigFile(configFilePath string) error {
 
 func installUploadDir() {
 	fmt.Println("[upload-dir] try to install...")
-	if err := dir.CreateDirIfNotExist(UploadFilePath); err != nil {
+	if err := dir.CreateDirIfNotExist(path.UploadFilePath); err != nil {
 		fmt.Printf("[upload-dir] install fail %s\n", err.Error())
 	} else {
-		fmt.Printf("[upload-dir] install success, upload directory is %s\n", UploadFilePath)
+		fmt.Printf("[upload-dir] install success, upload directory is %s\n", path.UploadFilePath)
 	}
 }
 
@@ -108,7 +80,7 @@ func InstallI18nBundle(replace bool) {
 	if len(os.Getenv("SKIP_REPLACE_I18N")) > 0 {
 		replace = false
 	}
-	if err := dir.CreateDirIfNotExist(I18nPath); err != nil {
+	if err := dir.CreateDirIfNotExist(path.I18nPath); err != nil {
 		fmt.Println(err.Error())
 		return
 	}
@@ -120,7 +92,7 @@ func InstallI18nBundle(replace bool) {
 	}
 	fmt.Printf("[i18n] find i18n bundle %d\n", len(i18nList))
 	for _, item := range i18nList {
-		path := filepath.Join(I18nPath, item.Name())
+		path := filepath.Join(path.I18nPath, item.Name())
 		content, err := i18n.I18n.ReadFile(item.Name())
 		if err != nil {
 			continue

internal/cli/install_check.go

@@ -23,6 +23,7 @@ import (
 	"fmt"
 
 	"github.com/apache/answer/internal/base/data"
+	"github.com/apache/answer/internal/base/path"
 	"github.com/apache/answer/internal/entity"
 	"github.com/apache/answer/pkg/dir"
 )
@@ -32,7 +33,7 @@ func CheckConfigFile(configPath string) bool {
 }
 
 func CheckUploadDir() bool {
-	return dir.CheckDirExist(UploadFilePath)
+	return dir.CheckDirExist(path.UploadFilePath)
 }
 
 // CheckDBConnection check database whether the connection is normal