feat: web config add stroage dir item (#428)

cmd/web/flags.go

@@ -9,11 +9,12 @@ import (
 )
 
 type args struct {
-	Address  *string `json:"address"`
-	Port     *int    `json:"port"`
-	Username *string `json:"username"`
-	Password *string `json:"password"`
-	ApiToken *string `json:"apiToken"`
+	Address    *string `json:"address"`
+	Port       *int    `json:"port"`
+	Username   *string `json:"username"`
+	Password   *string `json:"password"`
+	ApiToken   *string `json:"apiToken"`
+	StorageDir *string `json:"storageDir"`
 
 	configPath *string
 }
@@ -25,6 +26,7 @@ func parse() *args {
 	cliArgs.Username = flag.String("u", "gopeed", "HTTP Basic Auth Username")
 	cliArgs.Password = flag.String("p", "", "HTTP Basic Auth Pwd")
 	cliArgs.ApiToken = flag.String("T", "", "API token, that can only be used when basic authentication is enabled.")
+	cliArgs.StorageDir = flag.String("d", "", "Storage directory")
 	cliArgs.configPath = flag.String("c", "./config.json", "Config file path")
 	flag.Parse()
 
@@ -45,6 +47,9 @@ func parse() *args {
 	if cfgArgs.ApiToken == nil {
 		cfgArgs.ApiToken = cliArgs.ApiToken
 	}
+	if cfgArgs.StorageDir == nil {
+		cfgArgs.StorageDir = cliArgs.StorageDir
+	}
 	return cfgArgs
 }
 

cmd/web/main.go

@@ -31,11 +31,16 @@ func main() {
 		}
 	}
 
-	exe, err := os.Executable()
-	if err != nil {
-		panic(err)
+	var dir string
+	if args.StorageDir != nil && *args.StorageDir != "" {
+		dir = *args.StorageDir
+	} else {
+		exe, err := os.Executable()
+		if err != nil {
+			panic(err)
+		}
+		dir = filepath.Dir(exe)
 	}
-	dir := filepath.Dir(exe)
 
 	cfg := &model.StartConfig{
 		Network:        "tcp",