w1196396546
7 months ago
committed by
GitHub
GitHub
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
24 changed files with 2211 additions and 1933 deletions
-
272.github/workflows/auto-tag-release.yml
-
37.github/workflows/release.yml
-
47.gitignore
-
86.goreleaser.yml
-
53CHANGELOG.md
-
21LICENSE
-
305README.md
-
330cmd/cursor-id-modifier/main.go
-
8go.mod
-
16go.sum
-
BINimg/wx_public.jpg
-
BINimg/wx_public_2.png
-
153internal/config/config.go
-
187internal/lang/lang.go
-
216internal/process/manager.go
-
94internal/ui/display.go
-
20internal/ui/logo.go
-
122internal/ui/spinner.go
-
1050main.go
-
116pkg/idgen/generator.go
-
146scripts/build_all.bat
-
108scripts/build_all.sh
-
421scripts/install.ps1
-
336scripts/install.sh
@ -0,0 +1,272 @@ |
|||
# This workflow requires Ubuntu 22.04 or 24.04 |
|||
|
|||
name: Auto Tag & Release |
|||
|
|||
on: |
|||
push: |
|||
branches: |
|||
- master |
|||
- main |
|||
tags: |
|||
- "v*" |
|||
paths-ignore: |
|||
- "**.md" |
|||
- "LICENSE" |
|||
- ".gitignore" |
|||
workflow_call: {} |
|||
|
|||
permissions: |
|||
contents: write |
|||
packages: write |
|||
actions: write |
|||
|
|||
jobs: |
|||
pre_job: |
|||
runs-on: ubuntu-22.04 |
|||
outputs: |
|||
should_skip: ${{ steps.skip_check.outputs.should_skip }} |
|||
steps: |
|||
- id: skip_check |
|||
uses: fkirc/skip-duplicate-actions@v5.3.0 |
|||
with: |
|||
cancel_others: "true" |
|||
concurrent_skipping: "same_content" |
|||
|
|||
auto-tag-release: |
|||
needs: pre_job |
|||
if: | |
|||
needs.pre_job.outputs.should_skip != 'true' || |
|||
startsWith(github.ref, 'refs/tags/v') |
|||
runs-on: ubuntu-22.04 |
|||
timeout-minutes: 15 |
|||
outputs: |
|||
version: ${{ steps.get_latest_tag.outputs.version }} |
|||
concurrency: |
|||
group: ${{ github.workflow }}-${{ github.ref }} |
|||
cancel-in-progress: true |
|||
|
|||
steps: |
|||
- name: Checkout |
|||
uses: actions/checkout@v3 |
|||
with: |
|||
fetch-depth: 0 |
|||
lfs: true |
|||
submodules: recursive |
|||
|
|||
- name: Setup Go |
|||
uses: actions/setup-go@v3 |
|||
with: |
|||
go-version: "1.21" |
|||
check-latest: true |
|||
cache: true |
|||
|
|||
- name: Cache |
|||
uses: actions/cache@v3 |
|||
with: |
|||
path: | |
|||
~/.cache/go-build |
|||
~/go/pkg/mod |
|||
~/.cache/git |
|||
key: ${{ runner.os }}-build-${{ hashFiles('**/go.sum') }} |
|||
restore-keys: | |
|||
${{ runner.os }}-build- |
|||
${{ runner.os }}- |
|||
|
|||
# 只在非tag推送时执行自动打tag |
|||
- name: Get latest tag |
|||
if: ${{ !startsWith(github.ref, 'refs/tags/v') }} |
|||
id: get_latest_tag |
|||
run: | |
|||
set -euo pipefail |
|||
git fetch --tags --force || { |
|||
echo "::error::Failed to fetch tags" |
|||
exit 1 |
|||
} |
|||
latest_tag=$(git tag -l 'v*' --sort=-v:refname | head -n 1) |
|||
if [ -z "$latest_tag" ]; then |
|||
new_version="v0.1.0" |
|||
else |
|||
major=$(echo $latest_tag | cut -d. -f1) |
|||
minor=$(echo $latest_tag | cut -d. -f2) |
|||
patch=$(echo $latest_tag | cut -d. -f3) |
|||
new_patch=$((patch + 1)) |
|||
new_version="$major.$minor.$new_patch" |
|||
fi |
|||
echo "version=$new_version" >> "$GITHUB_OUTPUT" |
|||
echo "Generated version: $new_version" |
|||
|
|||
- name: Validate version |
|||
if: ${{ !startsWith(github.ref, 'refs/tags/v') }} |
|||
run: | |
|||
set -euo pipefail |
|||
new_tag="${{ steps.get_latest_tag.outputs.version }}" |
|||
echo "Validating version: $new_tag" |
|||
if [[ ! $new_tag =~ ^v[0-9]+\.[0-9]+\.[0-9]+$ ]]; then |
|||
echo "::error::Invalid version format: $new_tag" |
|||
exit 1 |
|||
fi |
|||
major=$(echo $new_tag | cut -d. -f1 | tr -d 'v') |
|||
minor=$(echo $new_tag | cut -d. -f2) |
|||
patch=$(echo $new_tag | cut -d. -f3) |
|||
if [[ $major -gt 99 || $minor -gt 99 || $patch -gt 999 ]]; then |
|||
echo "::error::Version numbers out of valid range" |
|||
exit 1 |
|||
fi |
|||
echo "Version validation passed" |
|||
|
|||
- name: Create new tag |
|||
if: ${{ !startsWith(github.ref, 'refs/tags/v') }} |
|||
env: |
|||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
|||
run: | |
|||
new_tag=${{ steps.get_latest_tag.outputs.version }} |
|||
git config --global user.name 'github-actions[bot]' |
|||
git config --global user.email 'github-actions[bot]@users.noreply.github.com' |
|||
git tag -a $new_tag -m "Release $new_tag" |
|||
git push origin $new_tag |
|||
|
|||
# 在 Run GoReleaser 之前添加配置检查步骤 |
|||
- name: Check GoReleaser config |
|||
run: | |
|||
if [ ! -f ".goreleaser.yml" ] && [ ! -f ".goreleaser.yaml" ]; then |
|||
echo "::error::GoReleaser configuration file not found" |
|||
exit 1 |
|||
fi |
|||
|
|||
# 添加依赖检查步骤 |
|||
- name: Check Dependencies |
|||
run: | |
|||
go mod verify |
|||
go mod download |
|||
# 如果使用 vendor 模式，则执行以下命令 |
|||
if [ -d "vendor" ]; then |
|||
go mod vendor |
|||
fi |
|||
|
|||
# 添加构建环境准备步骤 |
|||
- name: Prepare Build Environment |
|||
run: | |
|||
echo "Building version: ${VERSION:-development}" |
|||
echo "GOOS=${GOOS:-$(go env GOOS)}" >> $GITHUB_ENV |
|||
echo "GOARCH=${GOARCH:-$(go env GOARCH)}" >> $GITHUB_ENV |
|||
echo "GO111MODULE=on" >> $GITHUB_ENV |
|||
|
|||
# 添加清理步骤 |
|||
- name: Cleanup workspace |
|||
run: | |
|||
rm -rf /tmp/go/ |
|||
rm -rf .cache/ |
|||
rm -rf dist/ |
|||
git clean -fdx |
|||
git status |
|||
|
|||
# 修改 GoReleaser 步骤 |
|||
- name: Run GoReleaser |
|||
if: ${{ startsWith(github.ref, 'refs/tags/v') || (success() && steps.get_latest_tag.outputs.version != '') }} |
|||
uses: goreleaser/goreleaser-action@v3 |
|||
with: |
|||
distribution: goreleaser |
|||
version: latest |
|||
args: release --clean --timeout 60m |
|||
env: |
|||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
|||
VERSION: ${{ steps.get_latest_tag.outputs.version }} |
|||
CGO_ENABLED: 0 |
|||
GOPATH: /tmp/go |
|||
GOROOT: ${{ env.GOROOT }} |
|||
GOCACHE: /tmp/.cache/go-build |
|||
GOMODCACHE: /tmp/go/pkg/mod |
|||
GORELEASER_DEBUG: 1 |
|||
GORELEASER_CURRENT_TAG: ${{ steps.get_latest_tag.outputs.version }} |
|||
# 添加额外的构建信息 |
|||
BUILD_TIME: ${{ steps.get_latest_tag.outputs.version }} |
|||
BUILD_COMMIT: ${{ github.sha }} |
|||
|
|||
# 优化 vendor 同步步骤 |
|||
- name: Sync vendor directory |
|||
run: | |
|||
echo "Syncing vendor directory..." |
|||
go mod tidy |
|||
go mod vendor |
|||
go mod verify |
|||
# 验证 vendor 目录 |
|||
if [ -d "vendor" ]; then |
|||
echo "Verifying vendor directory..." |
|||
go mod verify |
|||
# 检查是否有未跟踪的文件 |
|||
if [ -n "$(git status --porcelain vendor/)" ]; then |
|||
echo "Warning: Vendor directory has uncommitted changes" |
|||
git status vendor/ |
|||
fi |
|||
fi |
|||
|
|||
# 添加错误检查步骤 |
|||
- name: Check GoReleaser Output |
|||
if: failure() |
|||
run: | |
|||
echo "::group::GoReleaser Debug Info" |
|||
cat dist/artifacts.json || true |
|||
echo "::endgroup::" |
|||
|
|||
echo "::group::GoReleaser Config" |
|||
cat .goreleaser.yml |
|||
echo "::endgroup::" |
|||
|
|||
echo "::group::Environment Info" |
|||
go version |
|||
go env |
|||
echo "::endgroup::" |
|||
|
|||
- name: Set Release Version |
|||
if: startsWith(github.ref, 'refs/tags/v') |
|||
run: | |
|||
echo "VERSION=${GITHUB_REF#refs/tags/}" >> $GITHUB_ENV |
|||
|
|||
# 改进验证步骤 |
|||
- name: Verify Release |
|||
if: ${{ startsWith(github.ref, 'refs/tags/v') || (success() && steps.get_latest_tag.outputs.version != '') }} |
|||
run: | |
|||
echo "Verifying release artifacts..." |
|||
if [ ! -d "dist" ]; then |
|||
echo "::error::Release artifacts not found" |
|||
exit 1 |
|||
fi |
|||
# 验证生成的二进制文件 |
|||
for file in dist/cursor-id-modifier_*; do |
|||
if [ -f "$file" ]; then |
|||
echo "Verifying: $file" |
|||
if [[ "$file" == *.exe ]]; then |
|||
# Windows 二进制文件检查 |
|||
if ! [ -x "$file" ]; then |
|||
echo "::error::$file is not executable" |
|||
exit 1 |
|||
fi |
|||
else |
|||
# Unix 二进制文件检查 |
|||
if ! [ -x "$file" ]; then |
|||
echo "::error::$file is not executable" |
|||
exit 1 |
|||
fi |
|||
fi |
|||
fi |
|||
done |
|||
|
|||
- name: Notify on failure |
|||
if: failure() |
|||
run: | |
|||
echo "::error::Release process failed" |
|||
|
|||
# 修改构建摘要步骤 |
|||
- name: Build Summary |
|||
if: always() |
|||
run: | |
|||
echo "## Build Summary" >> $GITHUB_STEP_SUMMARY |
|||
echo "- Go Version: $(go version)" >> $GITHUB_STEP_SUMMARY |
|||
echo "- Release Version: ${VERSION:-N/A}" >> $GITHUB_STEP_SUMMARY |
|||
echo "- GPG Signing: Disabled" >> $GITHUB_STEP_SUMMARY |
|||
echo "- Build Status: ${{ job.status }}" >> $GITHUB_STEP_SUMMARY |
|||
|
|||
if [ -d "dist" ]; then |
|||
echo "### Generated Artifacts" >> $GITHUB_STEP_SUMMARY |
|||
ls -lh dist/ | awk '{print "- "$9" ("$5")"}' >> $GITHUB_STEP_SUMMARY |
|||
fi |
|||
@ -1,37 +0,0 @@ |
|||
name: Release |
|||
|
|||
on: |
|||
push: |
|||
tags: |
|||
- 'v*' |
|||
|
|||
permissions: |
|||
contents: write |
|||
|
|||
jobs: |
|||
goreleaser: |
|||
runs-on: ubuntu-latest |
|||
steps: |
|||
- name: Checkout |
|||
uses: actions/checkout@v4 |
|||
with: |
|||
fetch-depth: 0 |
|||
|
|||
- name: Set up Go |
|||
uses: actions/setup-go@v5 |
|||
with: |
|||
check-latest: true |
|||
|
|||
- name: Set Repository Variables |
|||
run: | |
|||
echo "GITHUB_REPOSITORY_OWNER=$(echo ${{ github.repository }} | cut -d '/' -f 1)" >> $GITHUB_ENV |
|||
echo "GITHUB_REPOSITORY_NAME=$(echo ${{ github.repository }} | cut -d '/' -f 2)" >> $GITHUB_ENV |
|||
|
|||
- name: Run GoReleaser |
|||
uses: goreleaser/goreleaser-action@v5 |
|||
with: |
|||
distribution: goreleaser |
|||
version: latest |
|||
args: release --clean |
|||
env: |
|||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
|||
@ -1,23 +1,42 @@ |
|||
# Binary files |
|||
*.dll |
|||
*.so |
|||
*.dylib |
|||
|
|||
# Build directories |
|||
releases/ |
|||
cursor-id-modifier |
|||
# Compiled binary |
|||
/cursor-id-modifier |
|||
/cursor-id-modifier.exe |
|||
|
|||
# Build output directories |
|||
bin/ |
|||
dist/ |
|||
|
|||
# Go specific |
|||
go.sum |
|||
go/ |
|||
.cache/ |
|||
|
|||
# Temporary files |
|||
*.tmp |
|||
*~ |
|||
# IDE and editor files |
|||
.vscode/ |
|||
.idea/ |
|||
*.swp |
|||
*.swo |
|||
|
|||
# System files |
|||
# OS specific |
|||
.DS_Store |
|||
Thumbs.db |
|||
|
|||
.vscode |
|||
/.idea |
|||
# Build and release artifacts |
|||
releases/ |
|||
*.syso |
|||
*.exe |
|||
*.exe~ |
|||
*.dll |
|||
*.so |
|||
*.dylib |
|||
|
|||
# Test files |
|||
*.test |
|||
*.out |
|||
coverage.txt |
|||
|
|||
# Temporary files |
|||
*.tmp |
|||
*~ |
|||
*.bak |
|||
*.log |
|||
@ -0,0 +1,53 @@ |
|||
# 📝 Changelog |
|||
|
|||
All notable changes to this project will be documented in this file. |
|||
|
|||
The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), |
|||
and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html). |
|||
|
|||
## [0.1.23] - 2024-12-29 🚀 |
|||
|
|||
### ✨ Features |
|||
- **Initial Release**: First public release of Cursor ID Modifier |
|||
- **Multi-Platform Support**: |
|||
- 🪟 Windows (x64, x86) |
|||
- 🍎 macOS (Intel & Apple Silicon) |
|||
- 🐧 Linux (x64, x86, ARM64) |
|||
- **Installation**: |
|||
- Automated installation scripts for all platforms |
|||
- One-line installation commands |
|||
- Secure download and verification |
|||
- **Core Functionality**: |
|||
- Telemetry ID modification for Cursor IDE |
|||
- Automatic process management |
|||
- Secure configuration handling |
|||
|
|||
### 🐛 Bug Fixes |
|||
- **Installation**: |
|||
- Fixed environment variable preservation in sudo operations |
|||
- Enhanced error handling during installation |
|||
- Improved binary download reliability |
|||
- **Process Management**: |
|||
- Improved Cursor process detection accuracy |
|||
- Enhanced process termination reliability |
|||
- Better handling of edge cases |
|||
- **User Experience**: |
|||
- Enhanced error messages and user feedback |
|||
- Improved progress indicators |
|||
- Better handling of system permissions |
|||
|
|||
### 🔧 Technical Improvements |
|||
- Optimized binary size with proper build flags |
|||
- Enhanced cross-platform compatibility |
|||
- Improved error handling and logging |
|||
- Better system resource management |
|||
|
|||
### 📚 Documentation |
|||
- Added comprehensive installation instructions |
|||
- Included platform-specific guidelines |
|||
- Enhanced error troubleshooting guide |
|||
|
|||
--- |
|||
*For more details about the changes, please refer to the [commit history](https://github.com/yuaotian/go-cursor-help/commits/main).* |
|||
|
|||
[0.1.22]: https://github.com/yuaotian/go-cursor-help/releases/tag/v0.1.23 |
|||
@ -0,0 +1,21 @@ |
|||
MIT License |
|||
|
|||
Copyright (c) 2024 dacrab |
|||
|
|||
Permission is hereby granted, free of charge, to any person obtaining a copy |
|||
of this software and associated documentation files (the "Software"), to deal |
|||
in the Software without restriction, including without limitation the rights |
|||
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
|||
copies of the Software, and to permit persons to whom the Software is |
|||
furnished to do so, subject to the following conditions: |
|||
|
|||
The above copyright notice and this permission notice shall be included in all |
|||
copies or substantial portions of the Software. |
|||
|
|||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|||
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|||
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|||
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
|||
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
|||
SOFTWARE. |
|||
@ -0,0 +1,330 @@ |
|||
package main |
|||
|
|||
import ( |
|||
"bufio" |
|||
"flag" |
|||
"fmt" |
|||
"os" |
|||
"os/exec" |
|||
"os/user" |
|||
"runtime" |
|||
"runtime/debug" |
|||
"strings" |
|||
|
|||
"github.com/yuaotian/go-cursor-help/internal/config" |
|||
"github.com/yuaotian/go-cursor-help/internal/lang" |
|||
"github.com/yuaotian/go-cursor-help/internal/process" |
|||
"github.com/yuaotian/go-cursor-help/internal/ui" |
|||
"github.com/yuaotian/go-cursor-help/pkg/idgen" |
|||
|
|||
"github.com/sirupsen/logrus" |
|||
) |
|||
|
|||
// Global variables
|
|||
var ( |
|||
version = "dev" |
|||
setReadOnly = flag.Bool("r", false, "set storage.json to read-only mode") |
|||
showVersion = flag.Bool("v", false, "show version information") |
|||
log = logrus.New() |
|||
) |
|||
|
|||
func main() { |
|||
setupErrorRecovery() |
|||
handleFlags() |
|||
setupLogger() |
|||
|
|||
username := getCurrentUser() |
|||
log.Debug("Running as user:", username) |
|||
|
|||
// Initialize components
|
|||
display := ui.NewDisplay(nil) |
|||
configManager := initConfigManager(username) |
|||
generator := idgen.NewGenerator() |
|||
processManager := process.NewManager(nil, log) |
|||
|
|||
// Check and handle privileges
|
|||
if err := handlePrivileges(display); err != nil { |
|||
return |
|||
} |
|||
|
|||
// Setup display
|
|||
setupDisplay(display) |
|||
|
|||
text := lang.GetText() |
|||
|
|||
// Handle Cursor processes
|
|||
if err := handleCursorProcesses(display, processManager); err != nil { |
|||
return |
|||
} |
|||
|
|||
// Handle configuration
|
|||
oldConfig := readExistingConfig(display, configManager, text) |
|||
newConfig := generateNewConfig(display, generator, oldConfig, text) |
|||
|
|||
if err := saveConfiguration(display, configManager, newConfig); err != nil { |
|||
return |
|||
} |
|||
|
|||
// Show completion messages
|
|||
showCompletionMessages(display) |
|||
|
|||
if os.Getenv("AUTOMATED_MODE") != "1" { |
|||
waitExit() |
|||
} |
|||
} |
|||
|
|||
func setupErrorRecovery() { |
|||
defer func() { |
|||
if r := recover(); r != nil { |
|||
log.Errorf("Panic recovered: %v\n", r) |
|||
debug.PrintStack() |
|||
waitExit() |
|||
} |
|||
}() |
|||
} |
|||
|
|||
func handleFlags() { |
|||
flag.Parse() |
|||
if *showVersion { |
|||
fmt.Printf("Cursor ID Modifier v%s\n", version) |
|||
os.Exit(0) |
|||
} |
|||
} |
|||
|
|||
func setupLogger() { |
|||
log.SetFormatter(&logrus.TextFormatter{ |
|||
FullTimestamp: true, |
|||
DisableLevelTruncation: true, |
|||
PadLevelText: true, |
|||
}) |
|||
log.SetLevel(logrus.InfoLevel) |
|||
} |
|||
|
|||
func getCurrentUser() string { |
|||
if username := os.Getenv("SUDO_USER"); username != "" { |
|||
return username |
|||
} |
|||
|
|||
user, err := user.Current() |
|||
if err != nil { |
|||
log.Fatal(err) |
|||
} |
|||
return user.Username |
|||
} |
|||
|
|||
func initConfigManager(username string) *config.Manager { |
|||
configManager, err := config.NewManager(username) |
|||
if err != nil { |
|||
log.Fatal(err) |
|||
} |
|||
return configManager |
|||
} |
|||
|
|||
func handlePrivileges(display *ui.Display) error { |
|||
isAdmin, err := checkAdminPrivileges() |
|||
if err != nil { |
|||
log.Error(err) |
|||
waitExit() |
|||
return err |
|||
} |
|||
|
|||
if !isAdmin { |
|||
if runtime.GOOS == "windows" { |
|||
return handleWindowsPrivileges(display) |
|||
} |
|||
display.ShowPrivilegeError( |
|||
lang.GetText().PrivilegeError, |
|||
lang.GetText().RunWithSudo, |
|||
lang.GetText().SudoExample, |
|||
) |
|||
waitExit() |
|||
return fmt.Errorf("insufficient privileges") |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func handleWindowsPrivileges(display *ui.Display) error { |
|||
message := "\nRequesting administrator privileges..." |
|||
if lang.GetCurrentLanguage() == lang.CN { |
|||
message = "\n请求管理员权限..." |
|||
} |
|||
fmt.Println(message) |
|||
|
|||
if err := selfElevate(); err != nil { |
|||
log.Error(err) |
|||
display.ShowPrivilegeError( |
|||
lang.GetText().PrivilegeError, |
|||
lang.GetText().RunAsAdmin, |
|||
lang.GetText().RunWithSudo, |
|||
lang.GetText().SudoExample, |
|||
) |
|||
waitExit() |
|||
return err |
|||
} |
|||
return nil |
|||
} |
|||
|
|||
func setupDisplay(display *ui.Display) { |
|||
if err := display.ClearScreen(); err != nil { |
|||
log.Warn("Failed to clear screen:", err) |
|||
} |
|||
display.ShowLogo() |
|||
fmt.Println() |
|||
} |
|||
|
|||
func handleCursorProcesses(display *ui.Display, processManager *process.Manager) error { |
|||
if os.Getenv("AUTOMATED_MODE") == "1" { |
|||
log.Debug("Running in automated mode, skipping Cursor process closing") |
|||
return nil |
|||
} |
|||
|
|||
display.ShowProgress("Closing Cursor...") |
|||
log.Debug("Attempting to close Cursor processes") |
|||
|
|||
if err := processManager.KillCursorProcesses(); err != nil { |
|||
log.Error("Failed to close Cursor:", err) |
|||
display.StopProgress() |
|||
display.ShowError("Failed to close Cursor. Please close it manually and try again.") |
|||
waitExit() |
|||
return err |
|||
} |
|||
|
|||
if processManager.IsCursorRunning() { |
|||
log.Error("Cursor processes still detected after closing") |
|||
display.StopProgress() |
|||
display.ShowError("Failed to close Cursor completely. Please close it manually and try again.") |
|||
waitExit() |
|||
return fmt.Errorf("cursor still running") |
|||
} |
|||
|
|||
log.Debug("Successfully closed all Cursor processes") |
|||
display.StopProgress() |
|||
fmt.Println() |
|||
return nil |
|||
} |
|||
|
|||
func readExistingConfig(display *ui.Display, configManager *config.Manager, text lang.TextResource) *config.StorageConfig { |
|||
fmt.Println() |
|||
display.ShowProgress(text.ReadingConfig) |
|||
oldConfig, err := configManager.ReadConfig() |
|||
if err != nil { |
|||
log.Warn("Failed to read existing config:", err) |
|||
oldConfig = nil |
|||
} |
|||
display.StopProgress() |
|||
fmt.Println() |
|||
return oldConfig |
|||
} |
|||
|
|||
func generateNewConfig(display *ui.Display, generator *idgen.Generator, oldConfig *config.StorageConfig, text lang.TextResource) *config.StorageConfig { |
|||
display.ShowProgress(text.GeneratingIds) |
|||
newConfig := &config.StorageConfig{} |
|||
|
|||
if machineID, err := generator.GenerateMachineID(); err != nil { |
|||
log.Fatal("Failed to generate machine ID:", err) |
|||
} else { |
|||
newConfig.TelemetryMachineId = machineID |
|||
} |
|||
|
|||
if macMachineID, err := generator.GenerateMacMachineID(); err != nil { |
|||
log.Fatal("Failed to generate MAC machine ID:", err) |
|||
} else { |
|||
newConfig.TelemetryMacMachineId = macMachineID |
|||
} |
|||
|
|||
if deviceID, err := generator.GenerateDeviceID(); err != nil { |
|||
log.Fatal("Failed to generate device ID:", err) |
|||
} else { |
|||
newConfig.TelemetryDevDeviceId = deviceID |
|||
} |
|||
|
|||
if oldConfig != nil && oldConfig.TelemetrySqmId != "" { |
|||
newConfig.TelemetrySqmId = oldConfig.TelemetrySqmId |
|||
} else if sqmID, err := generator.GenerateSQMID(); err != nil { |
|||
log.Fatal("Failed to generate SQM ID:", err) |
|||
} else { |
|||
newConfig.TelemetrySqmId = sqmID |
|||
} |
|||
|
|||
display.StopProgress() |
|||
fmt.Println() |
|||
return newConfig |
|||
} |
|||
|
|||
func saveConfiguration(display *ui.Display, configManager *config.Manager, newConfig *config.StorageConfig) error { |
|||
display.ShowProgress("Saving configuration...") |
|||
if err := configManager.SaveConfig(newConfig, *setReadOnly); err != nil { |
|||
log.Error(err) |
|||
waitExit() |
|||
return err |
|||
} |
|||
display.StopProgress() |
|||
fmt.Println() |
|||
return nil |
|||
} |
|||
|
|||
func showCompletionMessages(display *ui.Display) { |
|||
display.ShowSuccess(lang.GetText().SuccessMessage, lang.GetText().RestartMessage) |
|||
fmt.Println() |
|||
|
|||
message := "Operation completed!" |
|||
if lang.GetCurrentLanguage() == lang.CN { |
|||
message = "操作完成！" |
|||
} |
|||
display.ShowInfo(message) |
|||
} |
|||
|
|||
func waitExit() { |
|||
fmt.Print(lang.GetText().PressEnterToExit) |
|||
os.Stdout.Sync() |
|||
bufio.NewReader(os.Stdin).ReadString('\n') |
|||
} |
|||
|
|||
func checkAdminPrivileges() (bool, error) { |
|||
switch runtime.GOOS { |
|||
case "windows": |
|||
cmd := exec.Command("net", "session") |
|||
return cmd.Run() == nil, nil |
|||
|
|||
case "darwin", "linux": |
|||
currentUser, err := user.Current() |
|||
if err != nil { |
|||
return false, fmt.Errorf("failed to get current user: %w", err) |
|||
} |
|||
return currentUser.Uid == "0", nil |
|||
|
|||
default: |
|||
return false, fmt.Errorf("unsupported operating system: %s", runtime.GOOS) |
|||
} |
|||
} |
|||
|
|||
func selfElevate() error { |
|||
os.Setenv("AUTOMATED_MODE", "1") |
|||
|
|||
switch runtime.GOOS { |
|||
case "windows": |
|||
verb := "runas" |
|||
exe, _ := os.Executable() |
|||
cwd, _ := os.Getwd() |
|||
args := strings.Join(os.Args[1:], " ") |
|||
|
|||
cmd := exec.Command("cmd", "/C", "start", verb, exe, args) |
|||
cmd.Dir = cwd |
|||
return cmd.Run() |
|||
|
|||
case "darwin", "linux": |
|||
exe, err := os.Executable() |
|||
if err != nil { |
|||
return err |
|||
} |
|||
|
|||
cmd := exec.Command("sudo", append([]string{exe}, os.Args[1:]...)...) |
|||
cmd.Stdin = os.Stdin |
|||
cmd.Stdout = os.Stdout |
|||
cmd.Stderr = os.Stderr |
|||
return cmd.Run() |
|||
|
|||
default: |
|||
return fmt.Errorf("unsupported operating system: %s", runtime.GOOS) |
|||
} |
|||
} |
|||
@ -1,11 +1,15 @@ |
|||
module cursor-id-modifier |
|||
module github.com/yuaotian/go-cursor-help |
|||
|
|||
go 1.21 |
|||
|
|||
require github.com/fatih/color v1.15.0 |
|||
require ( |
|||
github.com/fatih/color v1.15.0 |
|||
github.com/sirupsen/logrus v1.9.3 |
|||
) |
|||
|
|||
require ( |
|||
github.com/mattn/go-colorable v0.1.13 // indirect |
|||
github.com/mattn/go-isatty v0.0.17 // indirect |
|||
github.com/stretchr/testify v1.10.0 // indirect |
|||
golang.org/x/sys v0.13.0 // indirect |
|||
) |
|||
|
After Width: 258 | Height: 258 | Size: 27 KiB |
|
After Width: 580 | Height: 206 | Size: 30 KiB |
@ -0,0 +1,153 @@ |
|||
package config |
|||
|
|||
import ( |
|||
"encoding/json" |
|||
"fmt" |
|||
"os" |
|||
"path/filepath" |
|||
"runtime" |
|||
"sync" |
|||
"time" |
|||
) |
|||
|
|||
// StorageConfig represents the storage configuration
|
|||
type StorageConfig struct { |
|||
TelemetryMacMachineId string `json:"telemetry.macMachineId"` |
|||
TelemetryMachineId string `json:"telemetry.machineId"` |
|||
TelemetryDevDeviceId string `json:"telemetry.devDeviceId"` |
|||
TelemetrySqmId string `json:"telemetry.sqmId"` |
|||
LastModified string `json:"lastModified"` |
|||
Version string `json:"version"` |
|||
} |
|||
|
|||
// Manager handles configuration operations
|
|||
type Manager struct { |
|||
configPath string |
|||
mu sync.RWMutex |
|||
} |
|||
|
|||
// NewManager creates a new configuration manager
|
|||
func NewManager(username string) (*Manager, error) { |
|||
configPath, err := getConfigPath(username) |
|||
if err != nil { |
|||
return nil, fmt.Errorf("failed to get config path: %w", err) |
|||
} |
|||
return &Manager{configPath: configPath}, nil |
|||
} |
|||
|
|||
// ReadConfig reads the existing configuration
|
|||
func (m *Manager) ReadConfig() (*StorageConfig, error) { |
|||
m.mu.RLock() |
|||
defer m.mu.RUnlock() |
|||
|
|||
data, err := os.ReadFile(m.configPath) |
|||
if err != nil { |
|||
if os.IsNotExist(err) { |
|||
return nil, nil |
|||
} |
|||
return nil, fmt.Errorf("failed to read config file: %w", err) |
|||
} |
|||
|
|||
var config StorageConfig |
|||
if err := json.Unmarshal(data, &config); err != nil { |
|||
return nil, fmt.Errorf("failed to parse config file: %w", err) |
|||
} |
|||
|
|||
return &config, nil |
|||
} |
|||
|
|||
// SaveConfig saves the configuration
|
|||
func (m *Manager) SaveConfig(config *StorageConfig, readOnly bool) error { |
|||
m.mu.Lock() |
|||
defer m.mu.Unlock() |
|||
|
|||
// Ensure parent directories exist
|
|||
if err := os.MkdirAll(filepath.Dir(m.configPath), 0755); err != nil { |
|||
return fmt.Errorf("failed to create config directory: %w", err) |
|||
} |
|||
|
|||
// Prepare updated configuration
|
|||
updatedConfig := m.prepareUpdatedConfig(config) |
|||
|
|||
// Write configuration
|
|||
if err := m.writeConfigFile(updatedConfig, readOnly); err != nil { |
|||
return err |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
// prepareUpdatedConfig merges existing config with updates
|
|||
func (m *Manager) prepareUpdatedConfig(config *StorageConfig) map[string]interface{} { |
|||
// Read existing config
|
|||
originalFile := make(map[string]interface{}) |
|||
if data, err := os.ReadFile(m.configPath); err == nil { |
|||
json.Unmarshal(data, &originalFile) |
|||
} |
|||
|
|||
// Update fields
|
|||
originalFile["telemetry.sqmId"] = config.TelemetrySqmId |
|||
originalFile["telemetry.macMachineId"] = config.TelemetryMacMachineId |
|||
originalFile["telemetry.machineId"] = config.TelemetryMachineId |
|||
originalFile["telemetry.devDeviceId"] = config.TelemetryDevDeviceId |
|||
originalFile["lastModified"] = time.Now().UTC().Format(time.RFC3339) |
|||
// originalFile["version"] = "1.0.1"
|
|||
|
|||
return originalFile |
|||
} |
|||
|
|||
// writeConfigFile handles the atomic write of the config file
|
|||
func (m *Manager) writeConfigFile(config map[string]interface{}, readOnly bool) error { |
|||
// Marshal with indentation
|
|||
content, err := json.MarshalIndent(config, "", " ") |
|||
if err != nil { |
|||
return fmt.Errorf("failed to marshal config: %w", err) |
|||
} |
|||
|
|||
// Write to temporary file
|
|||
tmpPath := m.configPath + ".tmp" |
|||
if err := os.WriteFile(tmpPath, content, 0666); err != nil { |
|||
return fmt.Errorf("failed to write temporary file: %w", err) |
|||
} |
|||
|
|||
// Set final permissions
|
|||
fileMode := os.FileMode(0666) |
|||
if readOnly { |
|||
fileMode = 0444 |
|||
} |
|||
|
|||
if err := os.Chmod(tmpPath, fileMode); err != nil { |
|||
os.Remove(tmpPath) |
|||
return fmt.Errorf("failed to set temporary file permissions: %w", err) |
|||
} |
|||
|
|||
// Atomic rename
|
|||
if err := os.Rename(tmpPath, m.configPath); err != nil { |
|||
os.Remove(tmpPath) |
|||
return fmt.Errorf("failed to rename file: %w", err) |
|||
} |
|||
|
|||
// Sync directory
|
|||
if dir, err := os.Open(filepath.Dir(m.configPath)); err == nil { |
|||
defer dir.Close() |
|||
dir.Sync() |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
// getConfigPath returns the path to the configuration file
|
|||
func getConfigPath(username string) (string, error) { |
|||
var configDir string |
|||
switch runtime.GOOS { |
|||
case "windows": |
|||
configDir = filepath.Join(os.Getenv("APPDATA"), "Cursor", "User", "globalStorage") |
|||
case "darwin": |
|||
configDir = filepath.Join("/Users", username, "Library", "Application Support", "Cursor", "User", "globalStorage") |
|||
case "linux": |
|||
configDir = filepath.Join("/home", username, ".config", "Cursor", "User", "globalStorage") |
|||
default: |
|||
return "", fmt.Errorf("unsupported operating system: %s", runtime.GOOS) |
|||
} |
|||
return filepath.Join(configDir, "storage.json"), nil |
|||
} |
|||
@ -0,0 +1,187 @@ |
|||
package lang |
|||
|
|||
import ( |
|||
"os" |
|||
"os/exec" |
|||
"strings" |
|||
"sync" |
|||
) |
|||
|
|||
// Language represents a supported language code
|
|||
type Language string |
|||
|
|||
const ( |
|||
// CN represents Chinese language
|
|||
CN Language = "cn" |
|||
// EN represents English language
|
|||
EN Language = "en" |
|||
) |
|||
|
|||
// TextResource contains all translatable text resources
|
|||
type TextResource struct { |
|||
// Success messages
|
|||
SuccessMessage string |
|||
RestartMessage string |
|||
|
|||
// Progress messages
|
|||
ReadingConfig string |
|||
GeneratingIds string |
|||
CheckingProcesses string |
|||
ClosingProcesses string |
|||
ProcessesClosed string |
|||
PleaseWait string |
|||
|
|||
// Error messages
|
|||
ErrorPrefix string |
|||
PrivilegeError string |
|||
|
|||
// Instructions
|
|||
RunAsAdmin string |
|||
RunWithSudo string |
|||
SudoExample string |
|||
PressEnterToExit string |
|||
SetReadOnlyMessage string |
|||
|
|||
// Info messages
|
|||
ConfigLocation string |
|||
} |
|||
|
|||
var ( |
|||
currentLanguage Language |
|||
currentLanguageOnce sync.Once |
|||
languageMutex sync.RWMutex |
|||
) |
|||
|
|||
// GetCurrentLanguage returns the current language, detecting it if not already set
|
|||
func GetCurrentLanguage() Language { |
|||
currentLanguageOnce.Do(func() { |
|||
currentLanguage = detectLanguage() |
|||
}) |
|||
|
|||
languageMutex.RLock() |
|||
defer languageMutex.RUnlock() |
|||
return currentLanguage |
|||
} |
|||
|
|||
// SetLanguage sets the current language
|
|||
func SetLanguage(lang Language) { |
|||
languageMutex.Lock() |
|||
defer languageMutex.Unlock() |
|||
currentLanguage = lang |
|||
} |
|||
|
|||
// GetText returns the TextResource for the current language
|
|||
func GetText() TextResource { |
|||
return texts[GetCurrentLanguage()] |
|||
} |
|||
|
|||
// detectLanguage detects the system language
|
|||
func detectLanguage() Language { |
|||
// Check environment variables first
|
|||
if isChineseEnvVar() { |
|||
return CN |
|||
} |
|||
|
|||
// Then check OS-specific locale
|
|||
if isWindows() { |
|||
if isWindowsChineseLocale() { |
|||
return CN |
|||
} |
|||
} else if isUnixChineseLocale() { |
|||
return CN |
|||
} |
|||
|
|||
return EN |
|||
} |
|||
|
|||
func isChineseEnvVar() bool { |
|||
for _, envVar := range []string{"LANG", "LANGUAGE", "LC_ALL"} { |
|||
if lang := os.Getenv(envVar); lang != "" && strings.Contains(strings.ToLower(lang), "zh") { |
|||
return true |
|||
} |
|||
} |
|||
return false |
|||
} |
|||
|
|||
func isWindows() bool { |
|||
return os.Getenv("OS") == "Windows_NT" |
|||
} |
|||
|
|||
func isWindowsChineseLocale() bool { |
|||
// Check Windows UI culture
|
|||
cmd := exec.Command("powershell", "-Command", |
|||
"[System.Globalization.CultureInfo]::CurrentUICulture.Name") |
|||
output, err := cmd.Output() |
|||
if err == nil && strings.HasPrefix(strings.ToLower(strings.TrimSpace(string(output))), "zh") { |
|||
return true |
|||
} |
|||
|
|||
// Check Windows locale
|
|||
cmd = exec.Command("wmic", "os", "get", "locale") |
|||
output, err = cmd.Output() |
|||
return err == nil && strings.Contains(string(output), "2052") |
|||
} |
|||
|
|||
func isUnixChineseLocale() bool { |
|||
cmd := exec.Command("locale") |
|||
output, err := cmd.Output() |
|||
return err == nil && strings.Contains(strings.ToLower(string(output)), "zh_cn") |
|||
} |
|||
|
|||
// texts contains all translations
|
|||
var texts = map[Language]TextResource{ |
|||
CN: { |
|||
// Success messages
|
|||
SuccessMessage: "[√] 配置文件已成功更新！", |
|||
RestartMessage: "[!] 请手动重启 Cursor 以使更新生效", |
|||
|
|||
// Progress messages
|
|||
ReadingConfig: "正在读取配置文件...", |
|||
GeneratingIds: "正在生成新的标识符...", |
|||
CheckingProcesses: "正在检查运行中的 Cursor 实例...", |
|||
ClosingProcesses: "正在关闭 Cursor 实例...", |
|||
ProcessesClosed: "所有 Cursor 实例已关闭", |
|||
PleaseWait: "请稍候...", |
|||
|
|||
// Error messages
|
|||
ErrorPrefix: "程序发生严重错误: %v", |
|||
PrivilegeError: "\n[!] 错误：需要管理员权限", |
|||
|
|||
// Instructions
|
|||
RunAsAdmin: "请右键点击程序，选择「以管理员身份运行」", |
|||
RunWithSudo: "请使用 sudo 命令运行此程序", |
|||
SudoExample: "示例: sudo %s", |
|||
PressEnterToExit: "\n按回车键退出程序...", |
|||
SetReadOnlyMessage: "设置 storage.json 为只读模式, 这将导致 workspace 记录信息丢失等问题", |
|||
|
|||
// Info messages
|
|||
ConfigLocation: "配置文件位置:", |
|||
}, |
|||
EN: { |
|||
// Success messages
|
|||
SuccessMessage: "[√] Configuration file updated successfully!", |
|||
RestartMessage: "[!] Please restart Cursor manually for changes to take effect", |
|||
|
|||
// Progress messages
|
|||
ReadingConfig: "Reading configuration file...", |
|||
GeneratingIds: "Generating new identifiers...", |
|||
CheckingProcesses: "Checking for running Cursor instances...", |
|||
ClosingProcesses: "Closing Cursor instances...", |
|||
ProcessesClosed: "All Cursor instances have been closed", |
|||
PleaseWait: "Please wait...", |
|||
|
|||
// Error messages
|
|||
ErrorPrefix: "Program encountered a serious error: %v", |
|||
PrivilegeError: "\n[!] Error: Administrator privileges required", |
|||
|
|||
// Instructions
|
|||
RunAsAdmin: "Please right-click and select 'Run as Administrator'", |
|||
RunWithSudo: "Please run this program with sudo", |
|||
SudoExample: "Example: sudo %s", |
|||
PressEnterToExit: "\nPress Enter to exit...", |
|||
SetReadOnlyMessage: "Set storage.json to read-only mode, which will cause issues such as lost workspace records", |
|||
|
|||
// Info messages
|
|||
ConfigLocation: "Config file location:", |
|||
}, |
|||
} |
|||
@ -0,0 +1,216 @@ |
|||
package process |
|||
|
|||
import ( |
|||
"fmt" |
|||
"os/exec" |
|||
"runtime" |
|||
"strings" |
|||
"time" |
|||
|
|||
"github.com/sirupsen/logrus" |
|||
) |
|||
|
|||
// Config holds process manager configuration
|
|||
type Config struct { |
|||
MaxAttempts int // Maximum number of attempts to kill processes
|
|||
RetryDelay time.Duration // Delay between retry attempts
|
|||
ProcessPatterns []string // Process names to look for
|
|||
} |
|||
|
|||
// DefaultConfig returns the default configuration
|
|||
func DefaultConfig() *Config { |
|||
return &Config{ |
|||
MaxAttempts: 3, |
|||
RetryDelay: 2 * time.Second, |
|||
ProcessPatterns: []string{ |
|||
"Cursor.exe", // Windows executable
|
|||
"Cursor ", // Linux/macOS executable with space
|
|||
"cursor ", // Linux/macOS executable lowercase with space
|
|||
"cursor", // Linux/macOS executable lowercase
|
|||
"Cursor", // Linux/macOS executable
|
|||
"*cursor*", // Any process containing cursor
|
|||
"*Cursor*", // Any process containing Cursor
|
|||
}, |
|||
} |
|||
} |
|||
|
|||
// Manager handles process-related operations
|
|||
type Manager struct { |
|||
config *Config |
|||
log *logrus.Logger |
|||
} |
|||
|
|||
// NewManager creates a new process manager with optional config and logger
|
|||
func NewManager(config *Config, log *logrus.Logger) *Manager { |
|||
if config == nil { |
|||
config = DefaultConfig() |
|||
} |
|||
if log == nil { |
|||
log = logrus.New() |
|||
} |
|||
return &Manager{ |
|||
config: config, |
|||
log: log, |
|||
} |
|||
} |
|||
|
|||
// IsCursorRunning checks if any Cursor process is currently running
|
|||
func (m *Manager) IsCursorRunning() bool { |
|||
processes, err := m.getCursorProcesses() |
|||
if err != nil { |
|||
m.log.Warn("Failed to get Cursor processes:", err) |
|||
return false |
|||
} |
|||
return len(processes) > 0 |
|||
} |
|||
|
|||
// KillCursorProcesses attempts to kill all running Cursor processes
|
|||
func (m *Manager) KillCursorProcesses() error { |
|||
for attempt := 1; attempt <= m.config.MaxAttempts; attempt++ { |
|||
processes, err := m.getCursorProcesses() |
|||
if err != nil { |
|||
return fmt.Errorf("failed to get processes: %w", err) |
|||
} |
|||
|
|||
if len(processes) == 0 { |
|||
return nil |
|||
} |
|||
|
|||
// Try graceful shutdown first on Windows
|
|||
if runtime.GOOS == "windows" { |
|||
for _, pid := range processes { |
|||
exec.Command("taskkill", "/PID", pid).Run() |
|||
time.Sleep(500 * time.Millisecond) |
|||
} |
|||
} |
|||
|
|||
// Force kill remaining processes
|
|||
remainingProcesses, _ := m.getCursorProcesses() |
|||
for _, pid := range remainingProcesses { |
|||
m.killProcess(pid) |
|||
} |
|||
|
|||
time.Sleep(m.config.RetryDelay) |
|||
|
|||
if processes, _ := m.getCursorProcesses(); len(processes) == 0 { |
|||
return nil |
|||
} |
|||
} |
|||
|
|||
return nil |
|||
} |
|||
|
|||
// getCursorProcesses returns PIDs of running Cursor processes
|
|||
func (m *Manager) getCursorProcesses() ([]string, error) { |
|||
cmd := m.getProcessListCommand() |
|||
if cmd == nil { |
|||
return nil, fmt.Errorf("unsupported operating system: %s", runtime.GOOS) |
|||
} |
|||
|
|||
output, err := cmd.Output() |
|||
if err != nil { |
|||
return nil, fmt.Errorf("failed to execute command: %w", err) |
|||
} |
|||
|
|||
return m.parseProcessList(string(output)), nil |
|||
} |
|||
|
|||
// getProcessListCommand returns the appropriate command to list processes based on OS
|
|||
func (m *Manager) getProcessListCommand() *exec.Cmd { |
|||
switch runtime.GOOS { |
|||
case "windows": |
|||
return exec.Command("tasklist", "/FO", "CSV", "/NH") |
|||
case "darwin": |
|||
return exec.Command("ps", "-ax") |
|||
case "linux": |
|||
return exec.Command("ps", "-A") |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
|
|||
// parseProcessList extracts Cursor process PIDs from process list output
|
|||
func (m *Manager) parseProcessList(output string) []string { |
|||
var processes []string |
|||
for _, line := range strings.Split(output, "\n") { |
|||
lowerLine := strings.ToLower(line) |
|||
|
|||
if m.isOwnProcess(lowerLine) { |
|||
continue |
|||
} |
|||
|
|||
if pid := m.findCursorProcess(line, lowerLine); pid != "" { |
|||
processes = append(processes, pid) |
|||
} |
|||
} |
|||
return processes |
|||
} |
|||
|
|||
// isOwnProcess checks if the process belongs to this application
|
|||
func (m *Manager) isOwnProcess(line string) bool { |
|||
return strings.Contains(line, "cursor-id-modifier") || |
|||
strings.Contains(line, "cursor-helper") |
|||
} |
|||
|
|||
// findCursorProcess checks if a process line matches Cursor patterns and returns its PID
|
|||
func (m *Manager) findCursorProcess(line, lowerLine string) string { |
|||
for _, pattern := range m.config.ProcessPatterns { |
|||
if m.matchPattern(lowerLine, strings.ToLower(pattern)) { |
|||
return m.extractPID(line) |
|||
} |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
// matchPattern checks if a line matches a pattern, supporting wildcards
|
|||
func (m *Manager) matchPattern(line, pattern string) bool { |
|||
switch { |
|||
case strings.HasPrefix(pattern, "*") && strings.HasSuffix(pattern, "*"): |
|||
search := pattern[1 : len(pattern)-1] |
|||
return strings.Contains(line, search) |
|||
case strings.HasPrefix(pattern, "*"): |
|||
return strings.HasSuffix(line, pattern[1:]) |
|||
case strings.HasSuffix(pattern, "*"): |
|||
return strings.HasPrefix(line, pattern[:len(pattern)-1]) |
|||
default: |
|||
return line == pattern |
|||
} |
|||
} |
|||
|
|||
// extractPID extracts process ID from a process list line based on OS format
|
|||
func (m *Manager) extractPID(line string) string { |
|||
switch runtime.GOOS { |
|||
case "windows": |
|||
parts := strings.Split(line, ",") |
|||
if len(parts) >= 2 { |
|||
return strings.Trim(parts[1], "\"") |
|||
} |
|||
case "darwin", "linux": |
|||
parts := strings.Fields(line) |
|||
if len(parts) >= 1 { |
|||
return parts[0] |
|||
} |
|||
} |
|||
return "" |
|||
} |
|||
|
|||
// killProcess forcefully terminates a process by PID
|
|||
func (m *Manager) killProcess(pid string) error { |
|||
cmd := m.getKillCommand(pid) |
|||
if cmd == nil { |
|||
return fmt.Errorf("unsupported operating system: %s", runtime.GOOS) |
|||
} |
|||
return cmd.Run() |
|||
} |
|||
|
|||
// getKillCommand returns the appropriate command to kill a process based on OS
|
|||
func (m *Manager) getKillCommand(pid string) *exec.Cmd { |
|||
switch runtime.GOOS { |
|||
case "windows": |
|||
return exec.Command("taskkill", "/F", "/PID", pid) |
|||
case "darwin", "linux": |
|||
return exec.Command("kill", "-9", pid) |
|||
default: |
|||
return nil |
|||
} |
|||
} |
|||
@ -0,0 +1,94 @@ |
|||
package ui |
|||
|
|||
import ( |
|||
"fmt" |
|||
"os" |
|||
"os/exec" |
|||
"runtime" |
|||
"strings" |
|||
|
|||
"github.com/fatih/color" |
|||
) |
|||
|
|||
// Display handles UI operations for terminal output
|
|||
type Display struct { |
|||
spinner *Spinner |
|||
} |
|||
|
|||
// NewDisplay creates a new display instance with an optional spinner
|
|||
func NewDisplay(spinner *Spinner) *Display { |
|||
if spinner == nil { |
|||
spinner = NewSpinner(nil) |
|||
} |
|||
return &Display{spinner: spinner} |
|||
} |
|||
|
|||
// Terminal Operations
|
|||
|
|||
// ClearScreen clears the terminal screen based on OS
|
|||
func (d *Display) ClearScreen() error { |
|||
var cmd *exec.Cmd |
|||
switch runtime.GOOS { |
|||
case "windows": |
|||
cmd = exec.Command("cmd", "/c", "cls") |
|||
default: |
|||
cmd = exec.Command("clear") |
|||
} |
|||
cmd.Stdout = os.Stdout |
|||
return cmd.Run() |
|||
} |
|||
|
|||
// Progress Indicator
|
|||
|
|||
// ShowProgress displays a progress message with a spinner
|
|||
func (d *Display) ShowProgress(message string) { |
|||
d.spinner.SetMessage(message) |
|||
d.spinner.Start() |
|||
} |
|||
|
|||
// StopProgress stops the progress spinner
|
|||
func (d *Display) StopProgress() { |
|||
d.spinner.Stop() |
|||
} |
|||
|
|||
// Message Display
|
|||
|
|||
// ShowSuccess displays success messages in green
|
|||
func (d *Display) ShowSuccess(messages ...string) { |
|||
green := color.New(color.FgGreen) |
|||
for _, msg := range messages { |
|||
green.Println(msg) |
|||
} |
|||
} |
|||
|
|||
// ShowInfo displays an info message in cyan
|
|||
func (d *Display) ShowInfo(message string) { |
|||
cyan := color.New(color.FgCyan) |
|||
cyan.Println(message) |
|||
} |
|||
|
|||
// ShowError displays an error message in red
|
|||
func (d *Display) ShowError(message string) { |
|||
red := color.New(color.FgRed) |
|||
red.Println(message) |
|||
} |
|||
|
|||
// ShowPrivilegeError displays privilege error messages with instructions
|
|||
func (d *Display) ShowPrivilegeError(messages ...string) { |
|||
red := color.New(color.FgRed, color.Bold) |
|||
yellow := color.New(color.FgYellow) |
|||
|
|||
// Main error message
|
|||
red.Println(messages[0]) |
|||
fmt.Println() |
|||
|
|||
// Additional instructions
|
|||
for _, msg := range messages[1:] { |
|||
if strings.Contains(msg, "%s") { |
|||
exe, _ := os.Executable() |
|||
yellow.Printf(msg+"\n", exe) |
|||
} else { |
|||
yellow.Println(msg) |
|||
} |
|||
} |
|||
} |
|||
@ -0,0 +1,20 @@ |
|||
package ui |
|||
|
|||
import ( |
|||
"github.com/fatih/color" |
|||
) |
|||
|
|||
const cyberpunkLogo = ` |
|||
██████╗██╗ ██╗██████╗ ███████╗ ██████╗ ██████╗ |
|||
██╔════╝██║ ██║██╔══██╗██╔════╝██╔═══██╗██╔══██╗ |
|||
██║ ██║ ██║██████╔╝███████╗██║ ██║██████╔╝ |
|||
██║ ██║ ██║██╔══██╗╚════██║██║ ██║██╔══██╗ |
|||
╚██████╗╚██████╔╝██║ ██║███████║╚██████╔╝██║ ██║ |
|||
╚═════╝ ╚═════╝ ╚═╝ ╚═╝╚══════╝ ╚═════╝ ╚═╝ ╚═╝ |
|||
` |
|||
|
|||
// ShowLogo displays the application logo
|
|||
func (d *Display) ShowLogo() { |
|||
cyan := color.New(color.FgCyan, color.Bold) |
|||
cyan.Println(cyberpunkLogo) |
|||
} |
|||
@ -0,0 +1,122 @@ |
|||
package ui |
|||
|
|||
import ( |
|||
"fmt" |
|||
"sync" |
|||
"time" |
|||
|
|||
"github.com/fatih/color" |
|||
) |
|||
|
|||
// SpinnerConfig defines spinner configuration
|
|||
type SpinnerConfig struct { |
|||
Frames []string // Animation frames for the spinner
|
|||
Delay time.Duration // Delay between frame updates
|
|||
} |
|||
|
|||
// DefaultSpinnerConfig returns the default spinner configuration
|
|||
func DefaultSpinnerConfig() *SpinnerConfig { |
|||
return &SpinnerConfig{ |
|||
Frames: []string{"⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"}, |
|||
Delay: 100 * time.Millisecond, |
|||
} |
|||
} |
|||
|
|||
// Spinner represents a progress spinner
|
|||
type Spinner struct { |
|||
config *SpinnerConfig |
|||
message string |
|||
current int |
|||
active bool |
|||
stopCh chan struct{} |
|||
mu sync.RWMutex |
|||
} |
|||
|
|||
// NewSpinner creates a new spinner with the given configuration
|
|||
func NewSpinner(config *SpinnerConfig) *Spinner { |
|||
if config == nil { |
|||
config = DefaultSpinnerConfig() |
|||
} |
|||
return &Spinner{ |
|||
config: config, |
|||
stopCh: make(chan struct{}), |
|||
} |
|||
} |
|||
|
|||
// State management
|
|||
|
|||
// SetMessage sets the spinner message
|
|||
func (s *Spinner) SetMessage(message string) { |
|||
s.mu.Lock() |
|||
defer s.mu.Unlock() |
|||
s.message = message |
|||
} |
|||
|
|||
// IsActive returns whether the spinner is currently active
|
|||
func (s *Spinner) IsActive() bool { |
|||
s.mu.RLock() |
|||
defer s.mu.RUnlock() |
|||
return s.active |
|||
} |
|||
|
|||
// Control methods
|
|||
|
|||
// Start begins the spinner animation
|
|||
func (s *Spinner) Start() { |
|||
s.mu.Lock() |
|||
if s.active { |
|||
s.mu.Unlock() |
|||
return |
|||
} |
|||
s.active = true |
|||
s.mu.Unlock() |
|||
|
|||
go s.run() |
|||
} |
|||
|
|||
// Stop halts the spinner animation
|
|||
func (s *Spinner) Stop() { |
|||
s.mu.Lock() |
|||
defer s.mu.Unlock() |
|||
|
|||
if !s.active { |
|||
return |
|||
} |
|||
|
|||
s.active = false |
|||
close(s.stopCh) |
|||
s.stopCh = make(chan struct{}) |
|||
fmt.Print("\r") // Clear the spinner line
|
|||
} |
|||
|
|||
// Internal methods
|
|||
|
|||
func (s *Spinner) run() { |
|||
ticker := time.NewTicker(s.config.Delay) |
|||
defer ticker.Stop() |
|||
|
|||
cyan := color.New(color.FgCyan, color.Bold) |
|||
message := s.message |
|||
|
|||
// Print initial state
|
|||
fmt.Printf("\r %s %s", cyan.Sprint(s.config.Frames[0]), message) |
|||
|
|||
for { |
|||
select { |
|||
case <-s.stopCh: |
|||
return |
|||
case <-ticker.C: |
|||
s.mu.RLock() |
|||
if !s.active { |
|||
s.mu.RUnlock() |
|||
return |
|||
} |
|||
frame := s.config.Frames[s.current%len(s.config.Frames)] |
|||
s.current++ |
|||
s.mu.RUnlock() |
|||
|
|||
fmt.Printf("\r %s", cyan.Sprint(frame)) |
|||
fmt.Printf("\033[%dG%s", 4, message) // Move cursor and print message
|
|||
} |
|||
} |
|||
} |
|||
1050
main.go
File diff suppressed because it is too large
View File
File diff suppressed because it is too large
View File
@ -0,0 +1,116 @@ |
|||
package idgen |
|||
|
|||
import ( |
|||
"crypto/rand" |
|||
"encoding/hex" |
|||
"fmt" |
|||
"sync" |
|||
) |
|||
|
|||
// Generator handles secure ID generation for machines and devices
|
|||
type Generator struct { |
|||
bufferPool sync.Pool |
|||
} |
|||
|
|||
// NewGenerator creates a new ID generator
|
|||
func NewGenerator() *Generator { |
|||
return &Generator{ |
|||
bufferPool: sync.Pool{ |
|||
New: func() interface{} { |
|||
return make([]byte, 64) |
|||
}, |
|||
}, |
|||
} |
|||
} |
|||
|
|||
// Constants for ID generation
|
|||
const ( |
|||
machineIDPrefix = "auth0|user_" |
|||
uuidFormat = "%s-%s-%s-%s-%s" |
|||
) |
|||
|
|||
// generateRandomHex generates a random hex string of specified length
|
|||
func (g *Generator) generateRandomHex(length int) (string, error) { |
|||
buffer := g.bufferPool.Get().([]byte) |
|||
defer g.bufferPool.Put(buffer) |
|||
|
|||
if _, err := rand.Read(buffer[:length]); err != nil { |
|||
return "", fmt.Errorf("failed to generate random bytes: %w", err) |
|||
} |
|||
return hex.EncodeToString(buffer[:length]), nil |
|||
} |
|||
|
|||
// GenerateMachineID generates a new machine ID with auth0|user_ prefix
|
|||
func (g *Generator) GenerateMachineID() (string, error) { |
|||
randomPart, err := g.generateRandomHex(32) // 生成64字符的十六进制
|
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
|
|||
return fmt.Sprintf("%x%s", []byte(machineIDPrefix), randomPart), nil |
|||
} |
|||
|
|||
// GenerateMacMachineID generates a new 64-byte MAC machine ID
|
|||
func (g *Generator) GenerateMacMachineID() (string, error) { |
|||
return g.generateRandomHex(32) // 生成64字符的十六进制
|
|||
} |
|||
|
|||
// GenerateDeviceID generates a new device ID in UUID format
|
|||
func (g *Generator) GenerateDeviceID() (string, error) { |
|||
id, err := g.generateRandomHex(16) |
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
return fmt.Sprintf(uuidFormat, |
|||
id[0:8], id[8:12], id[12:16], id[16:20], id[20:32]), nil |
|||
} |
|||
|
|||
// GenerateSQMID generates a new SQM ID in UUID format (with braces)
|
|||
func (g *Generator) GenerateSQMID() (string, error) { |
|||
id, err := g.GenerateDeviceID() |
|||
if err != nil { |
|||
return "", err |
|||
} |
|||
return fmt.Sprintf("{%s}", id), nil |
|||
} |
|||
|
|||
// ValidateID validates the format of various ID types
|
|||
func (g *Generator) ValidateID(id string, idType string) bool { |
|||
switch idType { |
|||
case "machineID", "macMachineID": |
|||
return len(id) == 64 && isHexString(id) |
|||
case "deviceID": |
|||
return isValidUUID(id) |
|||
case "sqmID": |
|||
if len(id) < 2 || id[0] != '{' || id[len(id)-1] != '}' { |
|||
return false |
|||
} |
|||
return isValidUUID(id[1 : len(id)-1]) |
|||
default: |
|||
return false |
|||
} |
|||
} |
|||
|
|||
// Helper functions
|
|||
func isHexString(s string) bool { |
|||
_, err := hex.DecodeString(s) |
|||
return err == nil |
|||
} |
|||
|
|||
func isValidUUID(uuid string) bool { |
|||
if len(uuid) != 36 { |
|||
return false |
|||
} |
|||
for i, r := range uuid { |
|||
if i == 8 || i == 13 || i == 18 || i == 23 { |
|||
if r != '-' { |
|||
return false |
|||
} |
|||
continue |
|||
} |
|||
if !((r >= '0' && r <= '9') || (r >= 'a' && r <= 'f') || (r >= 'A' && r <= 'F')) { |
|||
return false |
|||
} |
|||
} |
|||
return true |
|||
} |
|||
@ -1,128 +1,74 @@ |
|||
@echo off |
|||
setlocal EnableDelayedExpansion |
|||
|
|||
:: Build optimization flags |
|||
set "OPTIMIZATION_FLAGS=-trimpath -ldflags=\"-s -w\"" |
|||
set "BUILD_JOBS=4" |
|||
|
|||
:: Messages / 消息 |
|||
set "EN_MESSAGES[0]=Starting build process for version" |
|||
set "EN_MESSAGES[1]=Using optimization flags:" |
|||
set "EN_MESSAGES[2]=Cleaning old builds..." |
|||
set "EN_MESSAGES[3]=Cleanup completed" |
|||
set "EN_MESSAGES[4]=bin directory does not exist, no cleanup needed" |
|||
set "EN_MESSAGES[5]=Starting builds for all platforms..." |
|||
set "EN_MESSAGES[6]=Building for" |
|||
set "EN_MESSAGES[7]=Build successful:" |
|||
set "EN_MESSAGES[8]=Build failed for" |
|||
set "EN_MESSAGES[9]=All builds completed! Total time:" |
|||
set "EN_MESSAGES[10]=seconds" |
|||
|
|||
set "CN_MESSAGES[0]=开始构建版本" |
|||
set "CN_MESSAGES[1]=使用优化标志：" |
|||
set "CN_MESSAGES[2]=正在清理旧的构建文件..." |
|||
set "CN_MESSAGES[3]=清理完成" |
|||
set "CN_MESSAGES[4]=bin 目录不存在，无需清理" |
|||
set "CN_MESSAGES[5]=开始编译所有平台..." |
|||
set "CN_MESSAGES[6]=正在构建" |
|||
set "CN_MESSAGES[7]=构建成功：" |
|||
set "CN_MESSAGES[8]=构建失败：" |
|||
set "CN_MESSAGES[9]=所有构建完成！总耗时：" |
|||
set "CN_MESSAGES[10]=秒" |
|||
|
|||
:: 设置版本信息 / Set version |
|||
set VERSION=2.0.0 |
|||
set "EN_MESSAGES[4]=Starting builds for all platforms..." |
|||
set "EN_MESSAGES[5]=Building for" |
|||
set "EN_MESSAGES[6]=Build successful:" |
|||
set "EN_MESSAGES[7]=All builds completed!" |
|||
|
|||
:: 设置颜色代码 / Set color codes |
|||
:: Colors |
|||
set "GREEN=[32m" |
|||
set "RED=[31m" |
|||
set "YELLOW=[33m" |
|||
set "RESET=[0m" |
|||
|
|||
:: 设置编译优化标志 / Set build optimization flags |
|||
set "LDFLAGS=-s -w" |
|||
set "BUILDMODE=pie" |
|||
set "GCFLAGS=-N -l" |
|||
|
|||
:: 设置 CGO / Set CGO |
|||
set CGO_ENABLED=0 |
|||
|
|||
:: 检测系统语言 / Detect system language |
|||
for /f "tokens=2 delims==" %%a in ('wmic os get OSLanguage /value') do set OSLanguage=%%a |
|||
if "%OSLanguage%"=="2052" (set LANG=cn) else (set LANG=en) |
|||
|
|||
:: 显示编译信息 / Display build info |
|||
echo %YELLOW%!%LANG%_MESSAGES[0]! %VERSION%%RESET% |
|||
echo %YELLOW%!%LANG%_MESSAGES[1]! LDFLAGS=%LDFLAGS%, BUILDMODE=%BUILDMODE%%RESET% |
|||
echo %YELLOW%CGO_ENABLED=%CGO_ENABLED%%RESET% |
|||
|
|||
:: 清理旧的构建文件 / Clean old builds |
|||
echo %YELLOW%!%LANG%_MESSAGES[2]!%RESET% |
|||
:: Cleanup function |
|||
:cleanup |
|||
if exist "..\bin" ( |
|||
rd /s /q "..\bin" |
|||
echo %GREEN%!%LANG%_MESSAGES[3]!%RESET% |
|||
) else ( |
|||
echo %YELLOW%!%LANG%_MESSAGES[4]!%RESET% |
|||
echo %GREEN%!EN_MESSAGES[3]!%RESET% |
|||
) |
|||
|
|||
:: 创建输出目录 / Create output directory |
|||
mkdir "..\bin" 2>nul |
|||
|
|||
:: 定义目标平台数组 / Define target platforms array |
|||
set platforms[0].os=windows |
|||
set platforms[0].arch=amd64 |
|||
set platforms[0].ext=.exe |
|||
set platforms[0].suffix= |
|||
:: Build function with optimizations |
|||
:build |
|||
set "os=%~1" |
|||
set "arch=%~2" |
|||
set "ext=" |
|||
if "%os%"=="windows" set "ext=.exe" |
|||
|
|||
set platforms[1].os=darwin |
|||
set platforms[1].arch=amd64 |
|||
set platforms[1].ext= |
|||
set platforms[1].suffix=_intel |
|||
echo %GREEN%!EN_MESSAGES[5]! %os%/%arch%%RESET% |
|||
|
|||
set platforms[2].os=darwin |
|||
set platforms[2].arch=arm64 |
|||
set platforms[2].ext= |
|||
set platforms[2].suffix=_m1 |
|||
set "CGO_ENABLED=0" |
|||
set "GOOS=%os%" |
|||
set "GOARCH=%arch%" |
|||
|
|||
set platforms[3].os=linux |
|||
set platforms[3].arch=amd64 |
|||
set platforms[3].ext= |
|||
set platforms[3].suffix= |
|||
start /b cmd /c "go build -trimpath -ldflags=\"-s -w\" -o ..\bin\%os%\%arch%\cursor-id-modifier%ext% -a -installsuffix cgo -mod=readonly ..\cmd\cursor-id-modifier" |
|||
exit /b 0 |
|||
|
|||
:: 设置开始时间 / Set start time |
|||
set start_time=%time% |
|||
:: Main execution |
|||
echo %GREEN%!EN_MESSAGES[0]!%RESET% |
|||
echo %GREEN%!EN_MESSAGES[1]! %OPTIMIZATION_FLAGS%%RESET% |
|||
|
|||
:: 编译所有目标 / Build all targets |
|||
echo !%LANG%_MESSAGES[5]! |
|||
call :cleanup |
|||
|
|||
for /L %%i in (0,1,3) do ( |
|||
set "os=!platforms[%%i].os!" |
|||
set "arch=!platforms[%%i].arch!" |
|||
set "ext=!platforms[%%i].ext!" |
|||
set "suffix=!platforms[%%i].suffix!" |
|||
|
|||
echo. |
|||
echo !%LANG%_MESSAGES[6]! !os! !arch!... |
|||
|
|||
set GOOS=!os! |
|||
set GOARCH=!arch! |
|||
|
|||
:: 构建输出文件名 / Build output filename |
|||
set "outfile=..\bin\cursor_id_modifier_v%VERSION%_!os!_!arch!!suffix!!ext!" |
|||
|
|||
:: 执行构建 / Execute build |
|||
go build -trimpath -buildmode=%BUILDMODE% -ldflags="%LDFLAGS%" -gcflags="%GCFLAGS%" -o "!outfile!" ..\main.go |
|||
|
|||
if !errorlevel! equ 0 ( |
|||
echo %GREEN%!%LANG%_MESSAGES[7]! !outfile!%RESET% |
|||
) else ( |
|||
echo %RED%!%LANG%_MESSAGES[8]! !os! !arch!%RESET% |
|||
echo %GREEN%!EN_MESSAGES[4]!%RESET% |
|||
|
|||
:: Start builds in parallel |
|||
set "pending=0" |
|||
for %%o in (windows linux darwin) do ( |
|||
for %%a in (amd64 386) do ( |
|||
call :build %%o %%a |
|||
set /a "pending+=1" |
|||
if !pending! geq %BUILD_JOBS% ( |
|||
timeout /t 1 /nobreak >nul |
|||
set "pending=0" |
|||
) |
|||
) |
|||
) |
|||
|
|||
:: 计算总耗时 / Calculate total time |
|||
set end_time=%time% |
|||
set /a duration = %end_time:~0,2% * 3600 + %end_time:~3,2% * 60 + %end_time:~6,2% - (%start_time:~0,2% * 3600 + %start_time:~3,2% * 60 + %start_time:~6,2%) |
|||
|
|||
echo. |
|||
echo %GREEN%!%LANG%_MESSAGES[9]! %duration% !%LANG%_MESSAGES[10]!%RESET% |
|||
if exist "..\bin" dir /b "..\bin" |
|||
:: Wait for all builds to complete |
|||
:wait_builds |
|||
timeout /t 2 /nobreak >nul |
|||
tasklist /fi "IMAGENAME eq go.exe" 2>nul | find "go.exe" >nul |
|||
if not errorlevel 1 goto wait_builds |
|||
|
|||
pause |
|||
endlocal |
|||
echo %GREEN%!EN_MESSAGES[7]!%RESET% |
|||
@ -1,308 +1,193 @@ |
|||
# Auto-elevate to admin rights if not already running as admin |
|||
if (-NOT ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")) { |
|||
Write-Host "Requesting administrator privileges..." |
|||
$arguments = "-NoProfile -ExecutionPolicy Bypass -File `"$PSCommandPath`" -ExecutionFromElevated" |
|||
Start-Process powershell.exe -ArgumentList $arguments -Verb RunAs |
|||
Exit |
|||
} |
|||
|
|||
# Set TLS to 1.2 / 设置 TLS 为 1.2 |
|||
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12 |
|||
|
|||
# Colors for output / 输出颜色 |
|||
$Red = "`e[31m" |
|||
$Green = "`e[32m" |
|||
$Blue = "`e[36m" |
|||
$Yellow = "`e[33m" |
|||
$Reset = "`e[0m" |
|||
|
|||
# Messages / 消息 |
|||
$EN_MESSAGES = @( |
|||
"Starting installation...", |
|||
"Detected architecture:", |
|||
"Only 64-bit Windows is supported", |
|||
"Latest version:", |
|||
"Creating installation directory...", |
|||
"Downloading latest release from:", |
|||
"Failed to download binary:", |
|||
"Downloaded file not found", |
|||
"Installing binary...", |
|||
"Failed to install binary:", |
|||
"Adding to PATH...", |
|||
"Cleaning up...", |
|||
"Installation completed successfully!", |
|||
"You can now use 'cursor-id-modifier' directly", |
|||
"Checking for running Cursor instances...", |
|||
"Found running Cursor processes. Attempting to close them...", |
|||
"Successfully closed all Cursor instances", |
|||
"Failed to close Cursor instances. Please close them manually", |
|||
"Backing up storage.json...", |
|||
"Backup created at:" |
|||
) |
|||
|
|||
$CN_MESSAGES = @( |
|||
"开始安装...", |
|||
"检测到架构：", |
|||
"仅支持64位Windows系统", |
|||
"最新版本：", |
|||
"正在创建安装目录...", |
|||
"正在从以下地址下载最新版本：", |
|||
"下载二进制文件失败：", |
|||
"未找到下载的文件", |
|||
"正在安装程序...", |
|||
"安装二进制文件失败：", |
|||
"正在添加到PATH...", |
|||
"正在清理...", |
|||
"安装成功完成！", |
|||
"现在可以直接使用 'cursor-id-modifier' 了", |
|||
"正在检查运行中的Cursor进程...", |
|||
"发现正在运行的Cursor进程，尝试关闭...", |
|||
"成功关闭所有Cursor实例", |
|||
"无法关闭Cursor实例，请手动关闭", |
|||
"正在备份storage.json...", |
|||
"备份已创建于：" |
|||
) |
|||
|
|||
# Detect system language / 检测系统语言 |
|||
function Get-SystemLanguage { |
|||
if ((Get-Culture).Name -like "zh-CN") { |
|||
return "cn" |
|||
# Check for admin rights and handle elevation |
|||
$isAdmin = ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator") |
|||
if (-NOT $isAdmin) { |
|||
# Detect PowerShell version and path |
|||
$pwshPath = if (Get-Command "pwsh" -ErrorAction SilentlyContinue) { |
|||
(Get-Command "pwsh").Source # PowerShell 7+ |
|||
} elseif (Test-Path "$env:ProgramFiles\PowerShell\7\pwsh.exe") { |
|||
"$env:ProgramFiles\PowerShell\7\pwsh.exe" |
|||
} else { |
|||
"powershell.exe" # Windows PowerShell |
|||
} |
|||
return "en" |
|||
} |
|||
|
|||
# Get message based on language / 根据语言获取消息 |
|||
function Get-Message($Index) { |
|||
$lang = Get-SystemLanguage |
|||
if ($lang -eq "cn") { |
|||
return $CN_MESSAGES[$Index] |
|||
|
|||
try { |
|||
Write-Host "`nRequesting administrator privileges..." -ForegroundColor Cyan |
|||
$scriptPath = $MyInvocation.MyCommand.Path |
|||
$argList = "-NoProfile -ExecutionPolicy Bypass -File `"$scriptPath`"" |
|||
Start-Process -FilePath $pwshPath -Verb RunAs -ArgumentList $argList -Wait |
|||
exit |
|||
} |
|||
catch { |
|||
Write-Host "`nError: Administrator privileges required" -ForegroundColor Red |
|||
Write-Host "Please run this script from an Administrator PowerShell window" -ForegroundColor Yellow |
|||
Write-Host "`nTo do this:" -ForegroundColor Cyan |
|||
Write-Host "1. Press Win + X" -ForegroundColor White |
|||
Write-Host "2. Click 'Windows Terminal (Admin)' or 'PowerShell (Admin)'" -ForegroundColor White |
|||
Write-Host "3. Run the installation command again" -ForegroundColor White |
|||
Write-Host "`nPress enter to exit..." |
|||
$null = $Host.UI.RawUI.ReadKey('NoEcho,IncludeKeyDown') |
|||
exit 1 |
|||
} |
|||
return $EN_MESSAGES[$Index] |
|||
} |
|||
|
|||
# Functions for colored output / 彩色输出函数 |
|||
function Write-Status($Message) { |
|||
Write-Host "${Blue}[*]${Reset} $Message" |
|||
} |
|||
|
|||
function Write-Success($Message) { |
|||
Write-Host "${Green}[✓]${Reset} $Message" |
|||
} |
|||
|
|||
function Write-Warning($Message) { |
|||
Write-Host "${Yellow}[!]${Reset} $Message" |
|||
} |
|||
# Set TLS to 1.2 |
|||
[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12 |
|||
|
|||
function Write-Error($Message) { |
|||
Write-Host "${Red}[✗]${Reset} $Message" |
|||
Exit 1 |
|||
} |
|||
# Create temporary directory |
|||
$TmpDir = Join-Path $env:TEMP ([System.Guid]::NewGuid().ToString()) |
|||
New-Item -ItemType Directory -Path $TmpDir | Out-Null |
|||
|
|||
# Close Cursor instances / 关闭Cursor实例 |
|||
function Close-CursorInstances { |
|||
Write-Status (Get-Message 14) |
|||
$cursorProcesses = Get-Process "Cursor" -ErrorAction SilentlyContinue |
|||
|
|||
if ($cursorProcesses) { |
|||
Write-Status (Get-Message 15) |
|||
try { |
|||
$cursorProcesses | ForEach-Object { $_.CloseMainWindow() | Out-Null } |
|||
Start-Sleep -Seconds 2 |
|||
$cursorProcesses | Where-Object { !$_.HasExited } | Stop-Process -Force |
|||
Write-Success (Get-Message 16) |
|||
} catch { |
|||
Write-Error (Get-Message 17) |
|||
} |
|||
# Cleanup function |
|||
function Cleanup { |
|||
if (Test-Path $TmpDir) { |
|||
Remove-Item -Recurse -Force $TmpDir |
|||
} |
|||
} |
|||
|
|||
# Backup storage.json / 备份storage.json |
|||
function Backup-StorageJson { |
|||
Write-Status (Get-Message 18) |
|||
$storageJsonPath = "$env:APPDATA\Cursor\User\globalStorage\storage.json" |
|||
if (Test-Path $storageJsonPath) { |
|||
$backupPath = "$storageJsonPath.backup" |
|||
Copy-Item -Path $storageJsonPath -Destination $backupPath -Force |
|||
Write-Success "$(Get-Message 19) $backupPath" |
|||
} |
|||
# Error handler |
|||
trap { |
|||
Write-Host "Error: $_" -ForegroundColor Red |
|||
Cleanup |
|||
Write-Host "Press enter to exit..." |
|||
$null = $Host.UI.RawUI.ReadKey('NoEcho,IncludeKeyDown') |
|||
exit 1 |
|||
} |
|||
|
|||
# Get latest release version from GitHub / 从GitHub获取最新版本 |
|||
function Get-LatestVersion { |
|||
$repo = "yuaotian/go-cursor-help" |
|||
$release = Invoke-RestMethod -Uri "https://api.github.com/repos/$repo/releases/latest" |
|||
return $release.tag_name |
|||
# Detect system architecture |
|||
function Get-SystemArch { |
|||
if ([Environment]::Is64BitOperatingSystem) { |
|||
return "x86_64" |
|||
} else { |
|||
return "i386" |
|||
} |
|||
} |
|||
|
|||
# 在文件开头添加日志函数 |
|||
function Write-Log { |
|||
param( |
|||
[string]$Message, |
|||
[string]$Level = "INFO" |
|||
# Download with progress |
|||
function Get-FileWithProgress { |
|||
param ( |
|||
[string]$Url, |
|||
[string]$OutputFile |
|||
) |
|||
$timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss" |
|||
$logMessage = "[$timestamp] [$Level] $Message" |
|||
$logFile = "$env:TEMP\cursor-id-modifier-install.log" |
|||
Add-Content -Path $logFile -Value $logMessage |
|||
|
|||
# 同时输出到控制台 |
|||
switch ($Level) { |
|||
"ERROR" { Write-Error $Message } |
|||
"WARNING" { Write-Warning $Message } |
|||
"SUCCESS" { Write-Success $Message } |
|||
default { Write-Status $Message } |
|||
try { |
|||
$webClient = New-Object System.Net.WebClient |
|||
$webClient.Headers.Add("User-Agent", "PowerShell Script") |
|||
|
|||
$webClient.DownloadFile($Url, $OutputFile) |
|||
return $true |
|||
} |
|||
catch { |
|||
Write-Host "Failed to download: $_" -ForegroundColor Red |
|||
return $false |
|||
} |
|||
} |
|||
|
|||
# 添加安装前检查函数 |
|||
function Test-Prerequisites { |
|||
Write-Log "Checking prerequisites..." "INFO" |
|||
# Main installation function |
|||
function Install-CursorModifier { |
|||
Write-Host "Starting installation..." -ForegroundColor Cyan |
|||
|
|||
# 检查PowerShell版本 |
|||
if ($PSVersionTable.PSVersion.Major -lt 5) { |
|||
Write-Log "PowerShell 5.0 or higher is required" "ERROR" |
|||
return $false |
|||
# Detect architecture |
|||
$arch = Get-SystemArch |
|||
Write-Host "Detected architecture: $arch" -ForegroundColor Green |
|||
|
|||
# Set installation directory |
|||
$InstallDir = "$env:ProgramFiles\CursorModifier" |
|||
if (!(Test-Path $InstallDir)) { |
|||
New-Item -ItemType Directory -Path $InstallDir | Out-Null |
|||
} |
|||
|
|||
# 检查网络连接 |
|||
# Get latest release |
|||
try { |
|||
$testConnection = Test-Connection -ComputerName "github.com" -Count 1 -Quiet |
|||
if (-not $testConnection) { |
|||
Write-Log "No internet connection available" "ERROR" |
|||
return $false |
|||
$latestRelease = Invoke-RestMethod -Uri "https://api.github.com/repos/yuaotian/go-cursor-help/releases/latest" |
|||
Write-Host "Found latest release: $($latestRelease.tag_name)" -ForegroundColor Cyan |
|||
|
|||
# Look for Windows binary with our architecture |
|||
$version = $latestRelease.tag_name.TrimStart('v') |
|||
Write-Host "Version: $version" -ForegroundColor Cyan |
|||
$possibleNames = @( |
|||
"cursor-id-modifier_${version}_windows_x86_64.exe", |
|||
"cursor-id-modifier_${version}_windows_$($arch).exe" |
|||
) |
|||
|
|||
$asset = $null |
|||
foreach ($name in $possibleNames) { |
|||
Write-Host "Checking for asset: $name" -ForegroundColor Cyan |
|||
$asset = $latestRelease.assets | Where-Object { $_.name -eq $name } |
|||
if ($asset) { |
|||
Write-Host "Found matching asset: $($asset.name)" -ForegroundColor Green |
|||
break |
|||
} |
|||
} |
|||
} catch { |
|||
Write-Log "Failed to check internet connection: $_" "ERROR" |
|||
return $false |
|||
|
|||
if (!$asset) { |
|||
Write-Host "`nAvailable assets:" -ForegroundColor Yellow |
|||
$latestRelease.assets | ForEach-Object { Write-Host "- $($_.name)" } |
|||
throw "Could not find appropriate Windows binary for $arch architecture" |
|||
} |
|||
|
|||
$downloadUrl = $asset.browser_download_url |
|||
} |
|||
catch { |
|||
Write-Host "Failed to get latest release: $_" -ForegroundColor Red |
|||
exit 1 |
|||
} |
|||
|
|||
return $true |
|||
} |
|||
|
|||
# 添加文件验证函数 |
|||
function Test-FileHash { |
|||
param( |
|||
[string]$FilePath, |
|||
[string]$ExpectedHash |
|||
) |
|||
# Download binary |
|||
Write-Host "`nDownloading latest release..." -ForegroundColor Cyan |
|||
$binaryPath = Join-Path $TmpDir "cursor-id-modifier.exe" |
|||
|
|||
$actualHash = Get-FileHash -Path $FilePath -Algorithm SHA256 |
|||
return $actualHash.Hash -eq $ExpectedHash |
|||
} |
|||
|
|||
# 修改下载函数，添加进度条 |
|||
function Download-File { |
|||
param( |
|||
[string]$Url, |
|||
[string]$OutFile |
|||
) |
|||
if (!(Get-FileWithProgress -Url $downloadUrl -OutputFile $binaryPath)) { |
|||
exit 1 |
|||
} |
|||
|
|||
# Install binary |
|||
Write-Host "Installing..." -ForegroundColor Cyan |
|||
try { |
|||
$webClient = New-Object System.Net.WebClient |
|||
$webClient.Headers.Add("User-Agent", "PowerShell Script") |
|||
|
|||
$webClient.DownloadFileAsync($Url, $OutFile) |
|||
Copy-Item -Path $binaryPath -Destination "$InstallDir\cursor-id-modifier.exe" -Force |
|||
|
|||
while ($webClient.IsBusy) { |
|||
Write-Progress -Activity "Downloading..." -Status "Progress:" -PercentComplete -1 |
|||
Start-Sleep -Milliseconds 100 |
|||
# Add to PATH if not already present |
|||
$currentPath = [Environment]::GetEnvironmentVariable("Path", "Machine") |
|||
if ($currentPath -notlike "*$InstallDir*") { |
|||
[Environment]::SetEnvironmentVariable("Path", "$currentPath;$InstallDir", "Machine") |
|||
} |
|||
|
|||
Write-Progress -Activity "Downloading..." -Completed |
|||
return $true |
|||
} |
|||
catch { |
|||
Write-Log "Download failed: $_" "ERROR" |
|||
return $false |
|||
Write-Host "Failed to install: $_" -ForegroundColor Red |
|||
exit 1 |
|||
} |
|||
finally { |
|||
if ($webClient) { |
|||
$webClient.Dispose() |
|||
|
|||
Write-Host "Installation completed successfully!" -ForegroundColor Green |
|||
Write-Host "Running cursor-id-modifier..." -ForegroundColor Cyan |
|||
|
|||
# Run the program |
|||
try { |
|||
& "$InstallDir\cursor-id-modifier.exe" |
|||
if ($LASTEXITCODE -ne 0) { |
|||
Write-Host "Failed to run cursor-id-modifier" -ForegroundColor Red |
|||
exit 1 |
|||
} |
|||
} |
|||
} |
|||
|
|||
# Main installation process / 主安装过程 |
|||
Write-Status (Get-Message 0) |
|||
|
|||
# Close any running Cursor instances |
|||
Close-CursorInstances |
|||
|
|||
# Backup storage.json |
|||
Backup-StorageJson |
|||
|
|||
# Get system architecture / 获取系统架构 |
|||
$arch = if ([Environment]::Is64BitOperatingSystem) { "amd64" } else { "386" } |
|||
Write-Status "$(Get-Message 1) $arch" |
|||
|
|||
if ($arch -ne "amd64") { |
|||
Write-Error (Get-Message 2) |
|||
} |
|||
|
|||
# Get latest version / 获取最新版本 |
|||
$version = Get-LatestVersion |
|||
Write-Status "$(Get-Message 3) $version" |
|||
|
|||
# Set up paths / 设置路径 |
|||
$installDir = "$env:ProgramFiles\cursor-id-modifier" |
|||
$versionWithoutV = $version.TrimStart('v') # 移除版本号前面的 'v' 字符 |
|||
$binaryName = "cursor_id_modifier_${versionWithoutV}_windows_amd64.exe" |
|||
$downloadUrl = "https://github.com/yuaotian/go-cursor-help/releases/download/$version/$binaryName" |
|||
$tempFile = "$env:TEMP\$binaryName" |
|||
|
|||
# Create installation directory / 创建安装目录 |
|||
Write-Status (Get-Message 4) |
|||
if (-not (Test-Path $installDir)) { |
|||
New-Item -ItemType Directory -Path $installDir -Force | Out-Null |
|||
} |
|||
|
|||
# Download binary / 下载二进制文件 |
|||
Write-Status "$(Get-Message 5) $downloadUrl" |
|||
try { |
|||
if (-not (Download-File -Url $downloadUrl -OutFile $tempFile)) { |
|||
Write-Error "$(Get-Message 6)" |
|||
catch { |
|||
Write-Host "Failed to run cursor-id-modifier: $_" -ForegroundColor Red |
|||
exit 1 |
|||
} |
|||
} catch { |
|||
Write-Error "$(Get-Message 6) $_" |
|||
} |
|||
|
|||
# Verify download / 验证下载 |
|||
if (-not (Test-Path $tempFile)) { |
|||
Write-Error (Get-Message 7) |
|||
} |
|||
|
|||
# Install binary / 安装二进制文件 |
|||
Write-Status (Get-Message 8) |
|||
# Run installation |
|||
try { |
|||
Move-Item -Force $tempFile "$installDir\cursor-id-modifier.exe" |
|||
} catch { |
|||
Write-Error "$(Get-Message 9) $_" |
|||
} |
|||
|
|||
# Add to PATH if not already present / 如果尚未添加则添加到PATH |
|||
$userPath = [Environment]::GetEnvironmentVariable("Path", "User") |
|||
if ($userPath -notlike "*$installDir*") { |
|||
Write-Status (Get-Message 10) |
|||
[Environment]::SetEnvironmentVariable( |
|||
"Path", |
|||
"$userPath;$installDir", |
|||
"User" |
|||
) |
|||
} |
|||
|
|||
# Cleanup / 清理 |
|||
Write-Status (Get-Message 11) |
|||
if (Test-Path $tempFile) { |
|||
Remove-Item -Force $tempFile |
|||
} |
|||
|
|||
Write-Success (Get-Message 12) |
|||
Write-Success (Get-Message 13) |
|||
Write-Host "" |
|||
|
|||
# 直接运行程序 |
|||
try { |
|||
Start-Process "$installDir\cursor-id-modifier.exe" -NoNewWindow |
|||
} catch { |
|||
Write-Warning "Failed to start cursor-id-modifier: $_" |
|||
Install-CursorModifier |
|||
} |
|||
catch { |
|||
Write-Host "Installation failed: $_" -ForegroundColor Red |
|||
Cleanup |
|||
Write-Host "Press enter to exit..." |
|||
$null = $Host.UI.RawUI.ReadKey('NoEcho,IncludeKeyDown') |
|||
exit 1 |
|||
} |
|||
finally { |
|||
Cleanup |
|||
if ($LASTEXITCODE -ne 0) { |
|||
Write-Host "Press enter to exit..." -ForegroundColor Green |
|||
$null = $Host.UI.RawUI.ReadKey('NoEcho,IncludeKeyDown') |
|||
} |
|||
} |
|||
Write
Preview
Loading…
Cancel
Save
Reference in new issue