Merge pull request #48 from doraemonkeys/master

feat: add option to set storage.json as read-only.

main.go

@@ -9,6 +9,7 @@ import (
 	"encoding/hex"
 	"encoding/json"
 	"errors"
+	"flag"
 	"fmt"
 	"log"
 	"os"
@@ -60,6 +61,7 @@ type (
 		ClosingProcesses   string
 		ProcessesClosed    string
 		PleaseWait         string
+		SetReadOnlyMessage string
 	}
 
 	// StorageConfig optimized storage configuration / 优化的存储配置
@@ -132,6 +134,7 @@ var (
 			ClosingProcesses:   "正在关闭 Cursor 实例...",
 			ProcessesClosed:    "所有 Cursor 实例已关闭",
 			PleaseWait:         "请稍候...",
+			SetReadOnlyMessage: "设置 storage.json 为只读模式, 这将导致 workspace 记录信息丢失等问题",
 		},
 		EN: {
 			SuccessMessage:     "[√] Configuration file updated successfully!",
@@ -149,10 +152,13 @@ var (
 			ClosingProcesses:   "Closing Cursor instances...",
 			ProcessesClosed:    "All Cursor instances have been closed",
 			PleaseWait:         "Please wait...",
+			SetReadOnlyMessage: "Set storage.json to read-only mode, which will cause issues such as lost workspace records",
 		},
 	}
 )
 
+var setReadOnly *bool = flag.Bool("r", false, "set storage.json to read-only mode")
+
 // Error Implementation / 错误实现
 func (e *AppError) Error() string {
 	if e.Context != nil {
@@ -225,16 +231,6 @@ func saveConfig(config *StorageConfig, username string) error { // Modified to t
 		return err
 	}
 
-	content, err := json.MarshalIndent(config, "", "    ")
-	if err != nil {
-		return &AppError{
-			Type: ErrSystem,
-			Op:   "generate JSON",
-			Path: "",
-			Err:  err,
-		}
-	}
-
 	// Create parent directories with proper permissions
 	dir := filepath.Dir(configPath)
 	if err := os.MkdirAll(dir, 0755); err != nil {
@@ -256,9 +252,53 @@ func saveConfig(config *StorageConfig, username string) error { // Modified to t
 		}
 	}
 
+	originalFileStat, err := os.Stat(configPath)
+	if err != nil {
+		return &AppError{
+			Type: ErrSystem,
+			Op:   "get file mode",
+			Path: configPath,
+			Err:  err,
+		}
+	}
+	originalFileMode := originalFileStat.Mode()
+
+	originalFileContent, err := os.ReadFile(configPath)
+	if err != nil {
+		return &AppError{
+			Type: ErrSystem,
+			Op:   "read original file",
+			Path: configPath,
+			Err:  err,
+		}
+	}
+
+	var originalFile map[string]any
+	if err := json.Unmarshal(originalFileContent, &originalFile); err != nil {
+		return &AppError{
+			Type: ErrSystem,
+			Op:   "unmarshal original file",
+			Path: configPath,
+			Err:  err,
+		}
+	}
+	originalFile["telemetry.sqmId"] = config.TelemetrySqmId
+	originalFile["telemetry.macMachineId"] = config.TelemetryMacMachineId
+	originalFile["telemetry.machineId"] = config.TelemetryMachineId
+	originalFile["telemetry.devDeviceId"] = config.TelemetryDevDeviceId
+	newFileContent, err := json.MarshalIndent(originalFile, "", "    ")
+	if err != nil {
+		return &AppError{
+			Type: ErrSystem,
+			Op:   "marshal new file",
+			Path: configPath,
+			Err:  err,
+		}
+	}
+
 	// Write to temporary file first
 	tmpPath := configPath + ".tmp"
-	if err := os.WriteFile(tmpPath, content, 0666); err != nil {
+	if err := os.WriteFile(tmpPath, newFileContent, 0666); err != nil {
 		return &AppError{
 			Type: ErrSystem,
 			Op:   "write temporary file",
@@ -267,8 +307,12 @@ func saveConfig(config *StorageConfig, username string) error { // Modified to t
 		}
 	}
 
+	if *setReadOnly {
+		originalFileMode = 0444
+	}
+
 	// Ensure proper permissions on temporary file
-	if err := os.Chmod(tmpPath, 0444); err != nil {
+	if err := os.Chmod(tmpPath, originalFileMode); err != nil {
 		os.Remove(tmpPath)
 		return &AppError{
 			Type: ErrSystem,
@@ -495,6 +539,15 @@ func showSuccess() {
 	}
 }
 
+func showReadOnlyMessage() {
+	if *setReadOnly {
+		warningColor := color.New(color.FgYellow, color.Bold)
+		warningColor.Printf("%s\n", texts[currentLanguage].SetReadOnlyMessage)
+		fmt.Println("Press Enter to continue...")
+		bufio.NewReader(os.Stdin).ReadString('\n')
+	}
+}
+
 func showPrivilegeError() {
 	text := texts[currentLanguage]
 	red := color.New(color.FgRed, color.Bold)
@@ -691,6 +744,9 @@ func main() {
 		}
 	}()
 
+	flag.Parse()
+	showReadOnlyMessage()
+
 	var username string
 	if username = os.Getenv("SUDO_USER"); username == "" {
 		user, err := user.Current()