aico-cli 0.3.17 → 0.3.20

package/templates/hooks/requirement/hook-utils.ps1

@@ -0,0 +1,365 @@
+# AICO Hook 通用工具库 - PowerShell 版本
+# 为所有 PowerShell Hook 脚本提供通用功能
+
+# 版本信息
+$AICO_HOOK_UTILS_VERSION = "1.0.0"
+$AICO_HOOK_UTILS_BUILD = "2025-01-16"
+
+# 路径配置
+$GLOBAL:AICO_HOOKS_DIR = Split-Path -Parent (Split-Path -Parent $MyInvocation.MyCommand.Path)
+$GLOBAL:AICO_TEMPLATES_DIR = Split-Path -Parent $AICO_HOOKS_DIR
+$GLOBAL:AICO_TEMP_DIR = Join-Path $env:TEMP "aico-hooks"
+$GLOBAL:AICO_LOG_DIR = Join-Path $env:USERPROFILE ".claude\hooks"
+
+# 确保必要目录存在
+function Initialize-HookEnvironment {
+    @($AICO_TEMP_DIR, $AICO_LOG_DIR) | ForEach-Object {
+        if (!(Test-Path $_)) {
+            New-Item -ItemType Directory -Path $_ -Force | Out-Null
+        }
+    }
+}
+
+Initialize-HookEnvironment
+
+# 日志函数 - 增强版本
+function Write-HookLog {
+    param(
+        [Parameter(Mandatory=$true)]
+        [string]$Message,
+
+        [Parameter(Mandatory=$false)]
+        [ValidateSet("INFO", "WARN", "ERROR", "DEBUG", "SUCCESS")]
+        [string]$Level = "INFO",
+
+        [Parameter(Mandatory=$false)]
+        [string]$LogFile = "$AICO_LOG_DIR\hook-utils.log",
+
+        [Parameter(Mandatory=$false)]
+        [switch]$ConsoleOnly,
+
+        [Parameter(Mandatory=$false)]
+        [string]$Source = "hook-utils"
+    )
+
+    $timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
+
+    # 日志格式化
+    $logEntry = switch ($Level) {
+        "INFO" { "[$Level] $timestamp [$Source] $Message" }
+        "WARN" { "[$Level] $timestamp [$Source] ⚠️ $Message" }
+        "ERROR" { "[$Level] $timestamp [$Source] ❌ $Message" }
+        "DEBUG" { "[$Level] $timestamp [$Source] 🔍 $Message" }
+        "SUCCESS" { "[$LEVEL] $timestamp [$Source] ✅ $Message" }
+        default { "[$Level] $timestamp [$Source] $Message" }
+    }
+
+    # 控制台输出
+    if (!$ConsoleOnly) {
+        switch ($Level) {
+            "ERROR" { Write-Host $logEntry -ForegroundColor Red }
+            "WARN" { Write-Host $logEntry -ForegroundColor Yellow }
+            "SUCCESS" { Write-Host $logEntry -ForegroundColor Green }
+            "DEBUG" { Write-Host $logEntry -ForegroundColor Gray }
+            default { Write-Host $logEntry -ForegroundColor White }
+        }
+    }
+
+    # 文件日志
+    try {
+        Add-Content -Path $LogFile -Value $logEntry -Force -ErrorAction Stop
+    }
+    catch {
+        Write-Debug "文件日志写入失败: $($_.Exception.Message)"
+    }
+}
+
+# Hook 配置文件操作
+function Get-HookConfig {
+    param(
+        [string]$ConfigFile = "$AICO_HOOKS_DIR\hooks-config.json",
+        [string]$HookName = ""
+    )
+
+    try {
+        if (Test-Path $ConfigFile) {
+            $config = Get-Content $ConfigFile -Raw | ConvertFrom-Json -ErrorAction Stop
+
+            if ($HookName) {
+                return $config.hooks.$HookName
+            }
+            return $config
+        }
+    }
+    catch {
+        Write-HookLog -Level "ERROR" -Message "配置文件解析失败: $($_.Exception.Message)" -Source "Get-HookConfig"
+    }
+
+    return $null
+}
+
+# Hook 状态管理
+function Set-HookStatus {
+    param(
+        [Parameter(Mandatory=$true)]
+        [string]$HookName,
+
+        [Parameter(Mandatory=$true)]
+        [ValidateSet("success", "failed", "running", "pending")]
+        [string]$Status,
+
+        [string]$Details = "",
+        [string]$StatusDir = $AICO_TEMP_DIR
+    )
+
+    try {
+        $statusFile = Join-Path $StatusDir "$HookName.status"
+        $statusData = @{
+            Status = $Status
+            Timestamp = Get-Date -Format "yyyy-MM-dd HH:mm:ss"
+            Details = $Details
+            HookName = $HookName
+        }
+
+        $statusData | ConvertTo-Json -Depth 2 | Set-Content -Path $statusFile -Force
+        Write-HookLog -Level "DEBUG" -Message "Hook 状态更新: $HookName = $Status" -Source "Set-HookStatus"
+        return $true
+    }
+    catch {
+        Write-HookLog -Level "ERROR" -Message "状态更新失败: $($_.Exception.Message)" -Source "Set-HookStatus"
+        return $false
+    }
+}
+
+function Get-HookStatus {
+    param(
+        [Parameter(Mandatory=$true)]
+        [string]$HookName,
+        [string]$StatusDir = $AICO_TEMP_DIR
+    )
+
+    try {
+        $statusFile = Join-Path $StatusDir "$HookName.status"
+        if (Test-Path $statusFile) {
+            $statusData = Get-Content $statusFile -Raw | ConvertFrom-Json
+            return $statusData.Status
+        }
+    }
+    catch {
+        Write-HookLog -Level "DEBUG" -Message "状态文件读取失败: $($_.Exception.Message)" -Source "Get-HookStatus"
+    }
+
+    return "unknown"
+}
+
+# 依赖检查
+function Test-HookDependencies {
+    param(
+        [Parameter(Mandatory=$true)]
+        [string]$HookName,
+        [string]$ConfigFile = "$AICO_HOOKS_DIR\hooks-config.json"
+    )
+
+    try {
+        $config = Get-HookConfig -ConfigFile $ConfigFile -HookName $HookName
+        if ($config -and $config.dependencies) {
+            foreach ($dependency in $config.dependencies) {
+                $depStatus = Get-HookStatus -HookName $dependency
+                if ($depStatus -ne "success") {
+                    Write-HookLog -Level "ERROR" -Message "依赖检查失败: $HookName 依赖于 $dependency (状态: $depStatus)" -Source "Test-HookDependencies"
+                    return $false
+                }
+            }
+        }
+        return $true
+    }
+    catch {
+        Write-HookLog -Level "ERROR" -Message "依赖检查异常: $($_.Exception.Message)" -Source "Test-HookDependencies"
+        return $false
+    }
+}
+
+# 安全文件操作
+function Invoke-SafeFileOperation {
+    param(
+        [Parameter(Mandatory=$true)]
+        [ValidateSet("read", "write", "append", "lock", "unlock", "exists")]
+        [string]$Operation,
+
+        [Parameter(Mandatory=$true)]
+        [string]$FilePath,
+
+        [string]$Content = "",
+
+        [int]$LockTimeout = 30,  # 秒
+        [string]$LockSuffix = ".lock"
+    )
+
+    try {
+        $directory = Split-Path -Parent $FilePath
+        if ($Operation -ne "exists" -and ![string]::IsNullOrEmpty($directory)) {
+            if (!(Test-Path $directory)) {
+                New-Item -ItemType Directory -Path $directory -Force | Out-Null
+            }
+        }
+
+        switch ($Operation) {
+            "exists" {
+                return Test-Path $FilePath
+            }
+
+            "read" {
+                if (Test-Path $FilePath) {
+                    return Get-Content $FilePath -Raw -ErrorAction Stop
+                }
+                return ""
+            }
+
+            "write" {
+                $Content | Set-Content -Path $FilePath -Force -ErrorAction Stop
+                Write-HookLog -Level "DEBUG" -Message "文件写入成功: $FilePath" -Source "SafeFileOp"
+                return $true
+            }
+
+            "append" {
+                $Content | Add-Content -Path $FilePath -Force -ErrorAction Stop
+                Write-HookLog -Level "DEBUG" -Message "文件追加成功: $FilePath" -Source "SafeFileOp"
+                return $true
+            }
+
+            "lock" {
+                $lockFile = "$FilePath$LockSuffix"
+                $startTime = Get-Date
+
+                while (Test-Path $lockFile) {
+                    if (((Get-Date) - $startTime).TotalSeconds -gt $LockTimeout) {
+                        Write-HookLog -Level "ERROR" -Message "文件锁定超时: $FilePath" -Source "SafeFileOp"
+                        return $false
+                    }
+                    Start-Sleep -Seconds 1
+                }
+
+                New-Item -ItemType File -Path $lockFile -Force | Out-Null
+                Write-HookLog -Level "DEBUG" -Message "文件锁定成功: $FilePath" -Source "SafeFileOp"
+                return $true
+            }
+
+            "unlock" {
+                $lockFile = "$FilePath$LockSuffix"
+                if (Test-Path $lockFile) {
+                    Remove-Item $lockFile -Force -ErrorAction Stop
+                    Write-HookLog -Level "DEBUG" -Message "文件解锁成功: $FilePath" -Source "SafeFileOp"
+                }
+                return $true
+            }
+        }
+    }
+    catch {
+        Write-HookLog -Level "ERROR" -Message "文件操作失败 (${Operation}): $($_.Exception.Message)" -Source "SafeFileOp"
+        return $false
+    }
+}
+
+# 执行外部 Hook 脚本（带重试机制）
+function Invoke-ExternalHook {
+    param(
+        [Parameter(Mandatory=$true)]
+        [string]$ScriptPath,
+
+        [string[]]$Arguments = @(),
+        [int]$MaxRetries = 3,
+        [int]$RetryDelay = 2  # 秒
+    )
+
+    if (!(Test-Path $ScriptPath)) {
+        Write-HookLog -Level "ERROR" -Message "Hook 脚本不存在: $ScriptPath" -Source "Invoke-ExternalHook"
+        return $false
+    }
+
+    for ($retry = 0; $retry -le $MaxRetries; $retry++) {
+        try {
+            Write-HookLog -Level "DEBUG" -Message "执行外部 Hook: $ScriptPath (尝试 $($retry + 1))" -Source "Invoke-ExternalHook"
+
+            $result = & $ScriptPath @Arguments
+            if ($LASTEXITCODE -eq 0) {
+                Write-HookLog -Level "SUCCESS" -Message "外部 Hook 执行成功: $ScriptPath" -Source "Invoke-ExternalHook"
+                return $true
+            } else {
+                throw "Hook 返回非零退出码: $LASTEXITCODE"
+            }
+        }
+        catch {
+            Write-HookLog -Level "ERROR" -Message "外部 Hook 执行失败 (尝试 $($retry + 1)): $($_.Exception.Message)" -Source "Invoke-ExternalHook"
+
+            if ($retry -lt $MaxRetries) {
+                Write-HookLog -Level "INFO" -Message "等待 ${RetryDelay}秒后重试..." -Source "Invoke-ExternalHook"
+                Start-Sleep -Seconds $RetryDelay
+            }
+        }
+    }
+
+    Write-HookLog -Level "ERROR" -Message "外部 Hook 执行最终失败: $ScriptPath" -Source "Invoke-ExternalHook"
+    return $false
+}
+
+# 性能监控函数
+function Measure-HookPerformance {
+    param(
+        [ScriptBlock]$ScriptBlock,
+        [string]$HookName = "unknown"
+    )
+
+    $stopwatch = [System.Diagnostics.Stopwatch]::StartNew()
+
+    try {
+        $result = & $ScriptBlock
+        $stopwatch.Stop()
+
+        Write-HookLog -Level "DEBUG" -Message "Hook 性能 [$HookName]: $($stopwatch.ElapsedMilliseconds)ms" -Source "Measure-HookPerformance"
+        return $result
+    }
+    catch {
+        $stopwatch.Stop()
+        Write-HookLog -Level "ERROR" -Message "Hook 执行异常 [$HookName] (耗时: $($stopwatch.ElapsedMilliseconds)ms): $($_.Exception.Message)" -Source "Measure-HookPerformance"
+        throw
+    }
+}
+
+# 清理函数
+function Clear-HookTempData {
+    param([string]$HookName = "")
+
+    try {
+        if ($HookName) {
+            # 清理特定 Hook 的临时数据
+            $statusFile = Join-Path $AICO_TEMP_DIR "$HookName.status"
+            if (Test-Path $statusFile) {
+                Remove-Item $statusFile -Force
+                Write-HookLog -Level "DEBUG" -Message "清理临时数据: $HookName" -Source "Clear-HookTempData"
+            }
+        } else {
+            # 清理所有临时数据
+            if (Test-Path $AICO_TEMP_DIR) {
+                Get-ChildItem $AICO_TEMP_DIR -File | Remove-Item -Force
+                Write-HookLog -Level "INFO" -Message "清理所有临时数据" -Source "Clear-HookTempData"
+            }
+        }
+    }
+    catch {
+        Write-HookLog -Level "WARN" -Message "临时数据清理异常: $($_.Exception.Message)" -Source "Clear-HookTempData"
+    }
+}
+
+# 导出函数
+Export-ModuleMember -Function @(
+    'Write-HookLog',
+    'Get-HookConfig',
+    'Set-HookStatus',
+    'Get-HookStatus',
+    'Test-HookDependencies',
+    'Invoke-SafeFileOperation',
+    'Invoke-ExternalHook',
+    'Measure-HookPerformance',
+    'Clear-HookTempData',
+    'Initialize-HookEnvironment'
+)