|
@ -0,0 +1,172 @@ |
|
|
|
|
|
package main |
|
|
|
|
|
|
|
|
|
|
|
import ( |
|
|
|
|
|
"crypto/rand" |
|
|
|
|
|
"crypto/sha256" |
|
|
|
|
|
"encoding/hex" |
|
|
|
|
|
"encoding/json" |
|
|
|
|
|
"fmt" |
|
|
|
|
|
"os" |
|
|
|
|
|
"path/filepath" |
|
|
|
|
|
"runtime" |
|
|
|
|
|
) |
|
|
|
|
|
|
|
|
|
|
|
// StorageConfig 存储配置结构体
|
|
|
|
|
|
type StorageConfig struct { |
|
|
|
|
|
TelemetryMacMachineId string `json:"telemetry.macMachineId"` |
|
|
|
|
|
TelemetryMachineId string `json:"telemetry.machineId"` |
|
|
|
|
|
TelemetryDevDeviceId string `json:"telemetry.devDeviceId"` |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 生成类似原始machineId的字符串 (64位小写hex)
|
|
|
|
|
|
func generateMachineId() string { |
|
|
|
|
|
// 生成一些随机数据
|
|
|
|
|
|
data := make([]byte, 32) |
|
|
|
|
|
rand.Read(data) |
|
|
|
|
|
|
|
|
|
|
|
// 使用SHA256生成hash
|
|
|
|
|
|
hash := sha256.New() |
|
|
|
|
|
hash.Write(data) |
|
|
|
|
|
|
|
|
|
|
|
// 转换为小写的hex字符串
|
|
|
|
|
|
return hex.EncodeToString(hash.Sum(nil)) |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 生成类似原始macMachineId的字符串 (64位小写hex)
|
|
|
|
|
|
func generateMacMachineId() string { |
|
|
|
|
|
return generateMachineId() // 使用相同的格式
|
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 生成类似原始devDeviceId的字符串 (标准UUID格式)
|
|
|
|
|
|
func generateDevDeviceId() string { |
|
|
|
|
|
// 生成 UUID v4
|
|
|
|
|
|
uuid := make([]byte, 16) |
|
|
|
|
|
rand.Read(uuid) |
|
|
|
|
|
|
|
|
|
|
|
// 设置版本 (4) 和变体位
|
|
|
|
|
|
uuid[6] = (uuid[6] & 0x0f) | 0x40 // 版本 4
|
|
|
|
|
|
uuid[8] = (uuid[8] & 0x3f) | 0x80 // RFC 4122 变体
|
|
|
|
|
|
|
|
|
|
|
|
// 格式化为标准 UUID 字符串
|
|
|
|
|
|
return fmt.Sprintf("%x-%x-%x-%x-%x", |
|
|
|
|
|
uuid[0:4], |
|
|
|
|
|
uuid[4:6], |
|
|
|
|
|
uuid[6:8], |
|
|
|
|
|
uuid[8:10], |
|
|
|
|
|
uuid[10:16]) |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 获取配置文件路径
|
|
|
|
|
|
func getConfigPath() (string, error) { |
|
|
|
|
|
var configDir string |
|
|
|
|
|
switch runtime.GOOS { |
|
|
|
|
|
case "darwin": |
|
|
|
|
|
homeDir, err := os.UserHomeDir() |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
return "", err |
|
|
|
|
|
} |
|
|
|
|
|
configDir = filepath.Join(homeDir, "Library", "Application Support", "Cursor", "User", "globalStorage") |
|
|
|
|
|
case "windows": |
|
|
|
|
|
appData := os.Getenv("APPDATA") |
|
|
|
|
|
configDir = filepath.Join(appData, "Cursor", "User", "globalStorage") |
|
|
|
|
|
case "linux": |
|
|
|
|
|
homeDir, err := os.UserHomeDir() |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
return "", err |
|
|
|
|
|
} |
|
|
|
|
|
configDir = filepath.Join(homeDir, ".config", "Cursor", "User", "globalStorage") |
|
|
|
|
|
default: |
|
|
|
|
|
return "", fmt.Errorf("不支持的操作系统: %s", runtime.GOOS) |
|
|
|
|
|
} |
|
|
|
|
|
return filepath.Join(configDir, "storage.json"), nil |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 修改文件权限
|
|
|
|
|
|
func setFilePermissions(filePath string) error { |
|
|
|
|
|
if runtime.GOOS == "windows" { |
|
|
|
|
|
// Windows 使用 ACL 权限系统,这里仅设置为只读
|
|
|
|
|
|
return os.Chmod(filePath, 0444) |
|
|
|
|
|
} else { |
|
|
|
|
|
// Linux 和 macOS
|
|
|
|
|
|
return os.Chmod(filePath, 0444) |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 获取Cursor可执行文件路径
|
|
|
|
|
|
func getCursorExePath() (string, error) { |
|
|
|
|
|
switch runtime.GOOS { |
|
|
|
|
|
case "windows": |
|
|
|
|
|
// Windows下通常在LocalAppData目录
|
|
|
|
|
|
localAppData := os.Getenv("LOCALAPPDATA") |
|
|
|
|
|
return filepath.Join(localAppData, "Programs", "Cursor", "Cursor.exe"), nil |
|
|
|
|
|
case "darwin": |
|
|
|
|
|
// macOS下通常在Applications目录
|
|
|
|
|
|
return "/Applications/Cursor.app/Contents/MacOS/Cursor", nil |
|
|
|
|
|
case "linux": |
|
|
|
|
|
// Linux下可能在usr/bin目录
|
|
|
|
|
|
return "/usr/bin/cursor", nil |
|
|
|
|
|
default: |
|
|
|
|
|
return "", fmt.Errorf("不支持的操作系统: %s", runtime.GOOS) |
|
|
|
|
|
} |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
func main() { |
|
|
|
|
|
// 获取配置文件路径
|
|
|
|
|
|
configPath, err := getConfigPath() |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
fmt.Printf("获取配置文件路径失败: %v\n", err) |
|
|
|
|
|
return |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 读取原始文件内容
|
|
|
|
|
|
content, err := os.ReadFile(configPath) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
fmt.Printf("读取配置文件失败: %v\n", err) |
|
|
|
|
|
return |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 解析 JSON
|
|
|
|
|
|
var config map[string]interface{} |
|
|
|
|
|
if err := json.Unmarshal(content, &config); err != nil { |
|
|
|
|
|
fmt.Printf("解析 JSON 失败: %v\n", err) |
|
|
|
|
|
return |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 修改指定字段,使用更准确的生成方法
|
|
|
|
|
|
config["telemetry.macMachineId"] = generateMacMachineId() |
|
|
|
|
|
config["telemetry.machineId"] = generateMachineId() |
|
|
|
|
|
config["telemetry.devDeviceId"] = generateDevDeviceId() |
|
|
|
|
|
|
|
|
|
|
|
// 转换回 JSON,保持原有的格式
|
|
|
|
|
|
newContent, err := json.MarshalIndent(config, "", " ") |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
fmt.Printf("生成 JSON 失败: %v\n", err) |
|
|
|
|
|
return |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 先确保文件可写
|
|
|
|
|
|
err = os.Chmod(configPath, 0666) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
fmt.Printf("修改文件权限失败: %v\n", err) |
|
|
|
|
|
return |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 写入文件
|
|
|
|
|
|
err = os.WriteFile(configPath, newContent, 0666) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
fmt.Printf("写入文件失败: %v\n", err) |
|
|
|
|
|
return |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
// 设置文件为只读
|
|
|
|
|
|
err = setFilePermissions(configPath) |
|
|
|
|
|
if err != nil { |
|
|
|
|
|
fmt.Printf("设置文件只读权限失败: %v\n", err) |
|
|
|
|
|
return |
|
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
fmt.Println("配置文件已成功更新,请手动重启Cursor以使更改生效。") |
|
|
|
|
|
} |