chore: minor update

Author: kermanx

.gitignore

@@ -6,6 +6,7 @@ __pycache__
 target
 service/uni-token-service
 service/uni-token-service.exe
+service/data.db
 frontend/public/release
 .eslintcache
 docs/.vitepress/cache

service/main.go

@@ -95,64 +95,42 @@ func main() {
 	}
 
 	if len(os.Args) < 2 {
-		// Directly run the executable
 		handleSetup()
 		logic.OpenUI(url.Values{}, false)
 		return
 	}
 
 	command := os.Args[1]
-
-	if command == "run" || command == "debug" {
-		err := s.Run()
-		if err != nil {
-			panic(err)
-		}
-		return
-	}
-
-	if command == "version" {
-		fmt.Println(constants.Version)
-		return
-	}
-
-	if command == "url" {
-		handleUrlScheme(os.Args[2])
-		return
-	}
-
-	if command == "setup" {
-		handleSetup()
-		return
-	}
-
-	if command == "install-and-start" {
-		handleInstallAndStart(&s, serviceName)
-		return
-	}
-
-	if command == "uninstall" {
-		handleSudo(false, []string{"uninstall-impl"})
+	commandHandlers := map[string]func(){
+		"run":               func() { mustRun(s.Run()) },
+		"debug":             func() { mustRun(s.Run()) },
+		"version":           func() { fmt.Println(constants.Version) },
+		"url":               func() { handleUrlScheme(os.Args[2]) },
+		"setup":             handleSetup,
+		"install-and-start": func() { handleInstallAndStart(&s, serviceName) },
+		"uninstall":         func() { handleSudo(false, []string{"uninstall-impl"}) },
+		"uninstall-impl":    func() { handleUninstall(s, serviceName) },
+		"sudo":              func() { handleSudoCommand() },
+	}
+
+	if handler, exists := commandHandlers[command]; exists {
+		handler()
 		return
 	}
 
-	if command == "uninstall-impl" {
-		handleUninstall(s, serviceName)
-		return
-	}
+	prg.logger.Infof("Executing command: %s\n", command)
+	mustRun(service.Control(s, command))
+}
 
-	if command == "sudo" {
-		if len(os.Args) < 3 {
-			fmt.Println("Usage: sudo <command>")
-			return
-		}
-		handleSudo(false, os.Args[2:])
+func handleSudoCommand() {
+	if len(os.Args) < 3 {
+		fmt.Println("Usage: sudo <command>")
 		return
 	}
+	handleSudo(false, os.Args[2:])
+}
 
-	prg.logger.Infof("Executing command: %s\n", command)
-
-	err = service.Control(s, command)
+func mustRun(err error) {
 	if err != nil {
 		panic(err)
 	}
@@ -222,10 +200,6 @@ func handleSudo(useInstalled bool, args []string) {
 }
 
 func handleUrlScheme(url string) {
-	if !strings.HasPrefix(url, "uni-token://") {
-		fmt.Println("Invalid URL scheme. Expected 'uni-token://'.")
-		return
-	}
 	url = strings.TrimPrefix(url, "uni-token://")
 
 	switch url {

service/server/server.go

@@ -30,7 +30,16 @@ func SetupAPIServer() (int, error) {
 		MaxAge:              240 * time.Hour,
 	}))
 
-	// Setup routes
+	setupRoutes(router)
+
+	port := findAvailablePort()
+	logic.ServerPort = port
+
+	go router.Run(":" + strconv.Itoa(port))
+	return port, nil
+}
+
+func setupRoutes(router *gin.Engine) {
 	SetupActionAPI(router)
 	SetupGatewayAPI(router)
 	SetupAppAPI(router)
@@ -39,33 +48,21 @@ func SetupAPIServer() (int, error) {
 	SetupUsageAPI(router)
 	SetupSiliconFlowAPI(router)
 	SetupAuthAPI(router)
-
-	// Get a random available port
-	port := getPort()
-	logic.ServerPort = port
-
-	// Start server in a goroutine
-	go func() {
-		router.Run(":" + strconv.Itoa(port))
-	}()
-
-	return port, nil
 }
 
-func checkPortAvailability(port int) bool {
+func isPortAvailable(port int) bool {
 	_, err := net.Dial("tcp", "localhost:"+strconv.Itoa(port))
 	return err != nil
 }
 
-const PORT_RANGE_START = 18760
+const portRangeStart = 18760
 
-func getPort() int {
-	for i := PORT_RANGE_START; i < PORT_RANGE_START+10; i++ {
-		if checkPortAvailability(i) {
-			return i
-		} else {
-			log.Printf("Port %d is not available", i)
+func findAvailablePort() int {
+	for port := portRangeStart; port < portRangeStart+10; port++ {
+		if isPortAvailable(port) {
+			return port
 		}
+		log.Printf("Port %d is not available", port)
 	}
-	panic(fmt.Sprintf("No available port found in the range %d-%d", PORT_RANGE_START, PORT_RANGE_START+10))
+	panic(fmt.Sprintf("No available port found in range %d-%d", portRangeStart, portRangeStart+10))
 }