0.0.1

main.go

@@ -12,8 +12,10 @@ import (
 	"x-ui/config"
 	"x-ui/database"
 	"x-ui/logger"
+	"x-ui/v2ui"
 	"x-ui/web"
 	"x-ui/web/global"
+	"x-ui/web/service"
 )
 
 // this function call global.setWebServer
@@ -46,7 +48,8 @@ func runWebServer() {
 	setWebServer(server)
 	err = server.Start()
 	if err != nil {
-		panic(err)
+		log.Println(err)
+		return
 	}
 
 	sigCh := make(chan os.Signal, 1)
@@ -60,7 +63,8 @@ func runWebServer() {
 			setWebServer(server)
 			err = server.Start()
 			if err != nil {
-				panic(err)
+				log.Println(err)
+				return
 			}
 		} else {
 			continue
@@ -68,8 +72,48 @@ func runWebServer() {
 	}
 }
 
-func v2ui(dbPath string) {
-	// migrate from v2-ui
+func resetSetting() {
+	err := database.InitDB(config.GetDBPath())
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+
+	settingService := service.SettingService{}
+	err = settingService.ResetSettings()
+	if err != nil {
+		fmt.Println("reset setting failed:", err)
+	} else {
+		fmt.Println("reset setting success")
+	}
+}
+
+func updateSetting(port int, username string, password string) {
+	err := database.InitDB(config.GetDBPath())
+	if err != nil {
+		fmt.Println(err)
+		return
+	}
+
+	settingService := service.SettingService{}
+
+	if port > 0 {
+		err := settingService.SetPort(port)
+		if err != nil {
+			fmt.Println("set port failed:", err)
+		} else {
+			fmt.Printf("set port %v success", port)
+		}
+	}
+	if username != "" || password != "" {
+		userService := service.UserService{}
+		err := userService.UpdateFirstUser(username, password)
+		if err != nil {
+			fmt.Println("set username and password failed:", err)
+		} else {
+			fmt.Println("set username and password success")
+		}
+	}
 }
 
 func main() {
@@ -78,24 +122,77 @@ func main() {
 		return
 	}
 
+	var showVersion bool
+	flag.BoolVar(&showVersion, "v", false, "show version")
+
 	runCmd := flag.NewFlagSet("run", flag.ExitOnError)
 
 	v2uiCmd := flag.NewFlagSet("v2-ui", flag.ExitOnError)
 	var dbPath string
 	v2uiCmd.StringVar(&dbPath, "db", "/etc/v2-ui/v2-ui.db", "set v2-ui db file path")
 
-	switch flag.Arg(0) {
+	settingCmd := flag.NewFlagSet("setting", flag.ExitOnError)
+	var port int
+	var username string
+	var password string
+	var reset bool
+	settingCmd.BoolVar(&reset, "reset", false, "reset all setting")
+	settingCmd.IntVar(&port, "port", 0, "set panel port")
+	settingCmd.StringVar(&username, "username", "", "set login username")
+	settingCmd.StringVar(&password, "password", "", "set login password")
+
+	oldUsage := flag.Usage
+	flag.Usage = func() {
+		oldUsage()
+		fmt.Println()
+		fmt.Println("Commands:")
+		fmt.Println("    run            run web panel")
+		fmt.Println("    v2-ui          migrate form v2-ui")
+		fmt.Println("    setting        set settings")
+	}
+
+	flag.Parse()
+	if showVersion {
+		fmt.Println(config.GetVersion())
+		return
+	}
+
+	switch os.Args[1] {
 	case "run":
-		runCmd.Parse(os.Args[2:])
+		err := runCmd.Parse(os.Args[2:])
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
 		runWebServer()
 	case "v2-ui":
-		v2uiCmd.Parse(os.Args[2:])
-		v2ui(dbPath)
+		err := v2uiCmd.Parse(os.Args[2:])
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		err = v2ui.MigrateFromV2UI(dbPath)
+		if err != nil {
+			logger.Error("migrate from v2-ui failed:", err)
+		}
+	case "setting":
+		err := settingCmd.Parse(os.Args[2:])
+		if err != nil {
+			fmt.Println(err)
+			return
+		}
+		if reset {
+			resetSetting()
+		} else {
+			updateSetting(port, username, password)
+		}
 	default:
-		fmt.Println("excepted 'run' or 'v2-ui' subcommands")
+		fmt.Println("except 'run' or 'v2-ui' or 'setting' subcommands")
 		fmt.Println()
 		runCmd.Usage()
 		fmt.Println()
 		v2uiCmd.Usage()
+		fmt.Println()
+		settingCmd.Usage()
 	}
 }