mirror of
https://github.com/yuaotian/go-cursor-help.git
synced 2026-03-07 22:33:39 +00:00
7fd954532e
- 新增 cursor_hook.js 核心Hook模块，拦截child_process、crypto、os等关键模块 - 实现统一ID配置管理，支持环境变量和JSON配置文件双重加载机制 - 开发Unix/macOS注入脚本(inject_hook_unix.sh)，自动化Hook代码注入流程 - 开发Windows注入脚本(inject_hook_win.ps1)，适配PowerShell环境 - 升级Linux修改器脚本，集成新的Hook方案并优化备份机制 - 添加完整的调试日志系统和防重复注入保护机制 - 支持动态import模块Hook，增强对ESM环境的兼容性
267 lines
8.3 KiB
PowerShell
267 lines
8.3 KiB
PowerShell
# ========================================
|
||
# Cursor Hook 注入脚本 (Windows)
|
||
# ========================================
|
||
#
|
||
# 🎯 功能：将 cursor_hook.js 注入到 Cursor 的 main.js 文件顶部
|
||
#
|
||
# 📦 使用方式：
|
||
# 1. 以管理员权限运行 PowerShell
|
||
# 2. 执行: .\inject_hook_win.ps1
|
||
#
|
||
# ⚠️ 注意事项：
|
||
# - 会自动备份原始 main.js 文件
|
||
# - 支持回滚到原始版本
|
||
# - Cursor 更新后需要重新注入
|
||
#
|
||
# ========================================
|
||
|
||
param(
|
||
[switch]$Rollback, # 回滚到原始版本
|
||
[switch]$Force, # 强制重新注入
|
||
[switch]$Debug # 启用调试模式
|
||
)
|
||
|
||
# 颜色定义
|
||
$RED = "`e[31m"
|
||
$GREEN = "`e[32m"
|
||
$YELLOW = "`e[33m"
|
||
$BLUE = "`e[34m"
|
||
$NC = "`e[0m"
|
||
|
||
# 日志函数
|
||
function Write-Log {
|
||
param([string]$Message, [string]$Level = "INFO")
|
||
$timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
|
||
switch ($Level) {
|
||
"INFO" { Write-Host "$GREEN[INFO]$NC $Message" }
|
||
"WARN" { Write-Host "$YELLOW[WARN]$NC $Message" }
|
||
"ERROR" { Write-Host "$RED[ERROR]$NC $Message" }
|
||
"DEBUG" { if ($Debug) { Write-Host "$BLUE[DEBUG]$NC $Message" } }
|
||
}
|
||
}
|
||
|
||
# 获取 Cursor 安装路径
|
||
function Get-CursorPath {
|
||
$possiblePaths = @(
|
||
"$env:LOCALAPPDATA\Programs\cursor\resources\app\out\main.js",
|
||
"$env:LOCALAPPDATA\Programs\Cursor\resources\app\out\main.js",
|
||
"C:\Program Files\Cursor\resources\app\out\main.js",
|
||
"C:\Program Files (x86)\Cursor\resources\app\out\main.js"
|
||
)
|
||
|
||
foreach ($path in $possiblePaths) {
|
||
if (Test-Path $path) {
|
||
return $path
|
||
}
|
||
}
|
||
|
||
return $null
|
||
}
|
||
|
||
# 获取 Hook 脚本路径
|
||
function Get-HookScriptPath {
|
||
$scriptDir = Split-Path -Parent $MyInvocation.MyCommand.Path
|
||
$hookPath = Join-Path $scriptDir "cursor_hook.js"
|
||
|
||
if (Test-Path $hookPath) {
|
||
return $hookPath
|
||
}
|
||
|
||
# 尝试从当前目录查找
|
||
$currentDir = Get-Location
|
||
$hookPath = Join-Path $currentDir "cursor_hook.js"
|
||
|
||
if (Test-Path $hookPath) {
|
||
return $hookPath
|
||
}
|
||
|
||
return $null
|
||
}
|
||
|
||
# 检查是否已注入
|
||
function Test-AlreadyInjected {
|
||
param([string]$MainJsPath)
|
||
|
||
$content = Get-Content $MainJsPath -Raw -Encoding UTF8
|
||
return $content -match "__cursor_patched__"
|
||
}
|
||
|
||
# 备份原始文件
|
||
function Backup-MainJs {
|
||
param([string]$MainJsPath)
|
||
|
||
$backupDir = Join-Path (Split-Path -Parent $MainJsPath) "backups"
|
||
if (-not (Test-Path $backupDir)) {
|
||
New-Item -ItemType Directory -Path $backupDir -Force | Out-Null
|
||
}
|
||
|
||
$timestamp = Get-Date -Format "yyyyMMdd_HHmmss"
|
||
$backupPath = Join-Path $backupDir "main.js.backup_$timestamp"
|
||
|
||
# 检查是否有原始备份
|
||
$originalBackup = Join-Path $backupDir "main.js.original"
|
||
if (-not (Test-Path $originalBackup)) {
|
||
Copy-Item $MainJsPath $originalBackup -Force
|
||
Write-Log "已创建原始备份: $originalBackup"
|
||
}
|
||
|
||
Copy-Item $MainJsPath $backupPath -Force
|
||
Write-Log "已创建时间戳备份: $backupPath"
|
||
|
||
return $originalBackup
|
||
}
|
||
|
||
# 回滚到原始版本
|
||
function Restore-MainJs {
|
||
param([string]$MainJsPath)
|
||
|
||
$backupDir = Join-Path (Split-Path -Parent $MainJsPath) "backups"
|
||
$originalBackup = Join-Path $backupDir "main.js.original"
|
||
|
||
if (Test-Path $originalBackup) {
|
||
Copy-Item $originalBackup $MainJsPath -Force
|
||
Write-Log "已回滚到原始版本" "INFO"
|
||
return $true
|
||
} else {
|
||
Write-Log "未找到原始备份文件" "ERROR"
|
||
return $false
|
||
}
|
||
}
|
||
|
||
# 注入 Hook 代码
|
||
function Inject-Hook {
|
||
param(
|
||
[string]$MainJsPath,
|
||
[string]$HookScriptPath
|
||
)
|
||
|
||
# 读取 Hook 脚本内容
|
||
$hookContent = Get-Content $HookScriptPath -Raw -Encoding UTF8
|
||
|
||
# 读取 main.js 内容
|
||
$mainContent = Get-Content $MainJsPath -Raw -Encoding UTF8
|
||
|
||
# 查找注入点：在 Sentry 初始化代码之后
|
||
# Sentry 初始化代码特征: _sentryDebugIds
|
||
$sentryPattern = '(?<=\}\(\);)\s*(?=var\s+\w+\s*=\s*function)'
|
||
|
||
if ($mainContent -match $sentryPattern) {
|
||
# 在 Sentry 初始化之后注入
|
||
$injectionPoint = $mainContent.IndexOf('}();') + 4
|
||
$newContent = $mainContent.Substring(0, $injectionPoint) + "`n`n// ========== Cursor Hook 注入开始 ==========`n" + $hookContent + "`n// ========== Cursor Hook 注入结束 ==========`n`n" + $mainContent.Substring($injectionPoint)
|
||
} else {
|
||
# 如果找不到 Sentry，直接在文件开头注入（在版权声明之后）
|
||
$copyrightEnd = $mainContent.IndexOf('*/') + 2
|
||
if ($copyrightEnd -gt 2) {
|
||
$newContent = $mainContent.Substring(0, $copyrightEnd) + "`n`n// ========== Cursor Hook 注入开始 ==========`n" + $hookContent + "`n// ========== Cursor Hook 注入结束 ==========`n`n" + $mainContent.Substring($copyrightEnd)
|
||
} else {
|
||
$newContent = "// ========== Cursor Hook 注入开始 ==========`n" + $hookContent + "`n// ========== Cursor Hook 注入结束 ==========`n`n" + $mainContent
|
||
}
|
||
}
|
||
|
||
# 写入修改后的内容
|
||
Set-Content -Path $MainJsPath -Value $newContent -Encoding UTF8 -NoNewline
|
||
|
||
return $true
|
||
}
|
||
|
||
# 关闭 Cursor 进程
|
||
function Stop-CursorProcess {
|
||
$cursorProcesses = Get-Process -Name "Cursor*" -ErrorAction SilentlyContinue
|
||
|
||
if ($cursorProcesses) {
|
||
Write-Log "发现 Cursor 进程正在运行，正在关闭..."
|
||
$cursorProcesses | Stop-Process -Force
|
||
Start-Sleep -Seconds 2
|
||
Write-Log "Cursor 进程已关闭"
|
||
}
|
||
}
|
||
|
||
# 主函数
|
||
function Main {
|
||
Write-Host ""
|
||
Write-Host "$BLUE========================================$NC"
|
||
Write-Host "$BLUE Cursor Hook 注入工具 (Windows) $NC"
|
||
Write-Host "$BLUE========================================$NC"
|
||
Write-Host ""
|
||
|
||
# 获取 Cursor main.js 路径
|
||
$mainJsPath = Get-CursorPath
|
||
if (-not $mainJsPath) {
|
||
Write-Log "未找到 Cursor 安装路径" "ERROR"
|
||
Write-Log "请确保 Cursor 已正确安装" "ERROR"
|
||
exit 1
|
||
}
|
||
Write-Log "找到 Cursor main.js: $mainJsPath"
|
||
|
||
# 回滚模式
|
||
if ($Rollback) {
|
||
Write-Log "执行回滚操作..."
|
||
Stop-CursorProcess
|
||
if (Restore-MainJs -MainJsPath $mainJsPath) {
|
||
Write-Log "回滚成功！" "INFO"
|
||
} else {
|
||
Write-Log "回滚失败！" "ERROR"
|
||
exit 1
|
||
}
|
||
exit 0
|
||
}
|
||
|
||
# 检查是否已注入
|
||
if ((Test-AlreadyInjected -MainJsPath $mainJsPath) -and -not $Force) {
|
||
Write-Log "Hook 已经注入，无需重复操作" "WARN"
|
||
Write-Log "如需强制重新注入，请使用 -Force 参数" "INFO"
|
||
exit 0
|
||
}
|
||
|
||
# 获取 Hook 脚本路径
|
||
$hookScriptPath = Get-HookScriptPath
|
||
if (-not $hookScriptPath) {
|
||
Write-Log "未找到 cursor_hook.js 文件" "ERROR"
|
||
Write-Log "请确保 cursor_hook.js 与此脚本在同一目录" "ERROR"
|
||
exit 1
|
||
}
|
||
Write-Log "找到 Hook 脚本: $hookScriptPath"
|
||
|
||
# 关闭 Cursor 进程
|
||
Stop-CursorProcess
|
||
|
||
# 备份原始文件
|
||
Write-Log "正在备份原始文件..."
|
||
$backupPath = Backup-MainJs -MainJsPath $mainJsPath
|
||
|
||
# 注入 Hook 代码
|
||
Write-Log "正在注入 Hook 代码..."
|
||
try {
|
||
if (Inject-Hook -MainJsPath $mainJsPath -HookScriptPath $hookScriptPath) {
|
||
Write-Log "Hook 注入成功！" "INFO"
|
||
} else {
|
||
Write-Log "Hook 注入失败！" "ERROR"
|
||
exit 1
|
||
}
|
||
} catch {
|
||
Write-Log "注入过程中发生错误: $_" "ERROR"
|
||
Write-Log "正在回滚..." "WARN"
|
||
Restore-MainJs -MainJsPath $mainJsPath
|
||
exit 1
|
||
}
|
||
|
||
Write-Host ""
|
||
Write-Host "$GREEN========================================$NC"
|
||
Write-Host "$GREEN ✅ Hook 注入完成！ $NC"
|
||
Write-Host "$GREEN========================================$NC"
|
||
Write-Host ""
|
||
Write-Log "现在可以启动 Cursor 了"
|
||
Write-Log "ID 配置文件位置: $env:USERPROFILE\.cursor_ids.json"
|
||
Write-Host ""
|
||
Write-Host "$YELLOW提示:$NC"
|
||
Write-Host " - 如需回滚，请运行: .\inject_hook_win.ps1 -Rollback"
|
||
Write-Host " - 如需强制重新注入，请运行: .\inject_hook_win.ps1 -Force"
|
||
Write-Host " - 如需启用调试日志，请运行: .\inject_hook_win.ps1 -Debug"
|
||
Write-Host ""
|
||
}
|
||
|
||
# 执行主函数
|
||
Main
|
||
|