feat(scripts): 添加 PEP 723 脚本元数据并修复 HTTP/2 伪头过滤问题
- 为 build_studio_urls.py 和 lookup_groups.py 添加 PEP 723 依赖声明 - 修复 HTTP/2 伪头导致 requests 抛 InvalidHeader 的问题 - 添加 CDP 端口连通性预检查及启动指引
This commit is contained in:
@@ -1,15 +1,19 @@
|
||||
# =====================================================================
|
||||
# Trae CN 技能安装器（Windows）
|
||||
# 作用：把本项目 skills\ 下全部技能安装到用户级技能目录
|
||||
# %USERPROFILE%\.trae-cn\skills\
|
||||
# 默认安装全部技能；也可用 -Skills 只装一部分，或用 -DryRun 预览。
|
||||
# 技能安装器（Windows，多 agent 目标）
|
||||
# 作用：把本项目 skills\ 下全部技能安装到指定的用户级技能目录：
|
||||
# trae-cn -> %USERPROFILE%\.trae-cn\skills\ （Trae CN / Trae SOLO）
|
||||
# zcode -> %USERPROFILE%\.zcode\skills\ （ZCode CLI）
|
||||
# 默认安装全部技能到全部已知目录；也可用 -Agent、-Skills 只装一部分，
|
||||
# 或用 -DryRun 预览。
|
||||
# 用法：双击同目录下的 install-skills.bat（推荐），或
|
||||
# powershell -NoProfile -ExecutionPolicy Bypass -File install-skills.ps1
|
||||
# powershell ... -Skills yt-studio-url-builder,youtube-studio-csv-download
|
||||
# powershell ... -DryRun
|
||||
# powershell ... -Agent zcode -Skills yt-studio-url-builder,youtube-studio-csv-download
|
||||
# powershell ... -Agent all -DryRun
|
||||
# =====================================================================
|
||||
|
||||
param(
|
||||
# 目标 agent：trae-cn（默认）/ zcode / all。all = 装到下面全部已知目标。
|
||||
[string]$Agent = "trae-cn",
|
||||
# 只安装这些技能（逗号/分号/空格分隔，匹配技能文件夹名，不区分大小写）。缺省=全部。
|
||||
[string[]]$Skills,
|
||||
# 只预览将要安装/备份/跳过的技能，不实际复制。
|
||||
@@ -18,17 +22,30 @@ param(
|
||||
|
||||
$ErrorActionPreference = "Stop"
|
||||
|
||||
# --- 1. 定位源目录与目标目录 ---
|
||||
# --- 已知目标：agent 名 -> 用户级技能目录 -----------------------------------
|
||||
# 新增 agent 支持时在此登记即可，其余逻辑不变。
|
||||
$AgentTargets = @{
|
||||
"trae-cn" = ".trae-cn\skills"
|
||||
"zcode" = ".zcode\skills"
|
||||
}
|
||||
|
||||
# --- 1. 解析目标目录（可能多个） ---
|
||||
if ($AgentTargets.ContainsKey($Agent)) {
|
||||
$destRelList = @($AgentTargets[$Agent])
|
||||
} elseif ($Agent -eq "all") {
|
||||
$destRelList = @($AgentTargets.Values)
|
||||
} else {
|
||||
Write-Host "[错误] 未知 agent: $Agent（可选: $($AgentTargets.Keys -join ', ') | all）" -ForegroundColor Red
|
||||
exit 1
|
||||
}
|
||||
|
||||
$ScriptDir = $PSScriptRoot
|
||||
$ProjectRoot = Split-Path -Parent $ScriptDir
|
||||
$SkillsSource = Join-Path $ProjectRoot "skills"
|
||||
$TraeSkills = Join-Path $env:USERPROFILE ".trae-cn\skills"
|
||||
# 备份放在技能目录外，避免被 Trae 当成技能重复扫描
|
||||
$BackupRoot = Join-Path $env:USERPROFILE ".trae-cn\skills-backup"
|
||||
|
||||
Write-Host "=== Trae CN 技能安装器 ===" -ForegroundColor Cyan
|
||||
Write-Host "=== 技能安装器 ===" -ForegroundColor Cyan
|
||||
Write-Host "技能源目录: $SkillsSource"
|
||||
Write-Host "安装目标: $TraeSkills"
|
||||
Write-Host ("安装目标: " + (($destRelList | ForEach-Object { "$env:USERPROFILE\$_" }) -join ", "))
|
||||
|
||||
# 把 -Skills 参数拆成规范化名字列表（兼容逗号/分号/空格分隔）
|
||||
$skillFilter = @()
|
||||
@@ -83,7 +100,7 @@ if ($targets.Count -eq 0) {
|
||||
}
|
||||
|
||||
# --- 3. 读取每个技能的 frontmatter name，并校验其与文件夹名一致 ---
|
||||
# 预测性关键：Trae 以 frontmatter 的 name 作为技能名，若与文件夹名不一致会引发歧义/旧名残留。
|
||||
# 预测性关键：各 agent 以 frontmatter 的 name 作为技能名，若与文件夹名不一致会引发歧义/旧名残留。
|
||||
$warnCount = 0
|
||||
foreach ($skill in $targets) {
|
||||
$skillFile = Join-Path $skill.FullName "SKILL.md"
|
||||
@@ -111,81 +128,84 @@ foreach ($skill in $targets) {
|
||||
Write-Host ("发现 {0} 个技能: {1}" -f $targets.Count, ($targets.Name -join ", "))
|
||||
Write-Host ""
|
||||
|
||||
# --- 4. 计算操作（备份/安装）并可选预览 ---
|
||||
function Test-SkillInstalled([string]$skillName) {
|
||||
return Test-Path (Join-Path $TraeSkills $skillName)
|
||||
# --- 4. 对每个目标目录执行安装（备份旧版 -> 复制） ---
|
||||
$timestamp = Get-Date -Format "yyyyMMdd-HHmmss"
|
||||
$totalBackup = 0
|
||||
$totalFresh = 0
|
||||
$failedTargets = @()
|
||||
|
||||
foreach ($destRel in $destRelList) {
|
||||
$DestSkills = Join-Path $env:USERPROFILE $destRel
|
||||
$BackupRoot = (Join-Path $env:USERPROFILE ($destRel -replace '\\skills$', '')) + "\skills-backup"
|
||||
|
||||
if ($DryRun) {
|
||||
Write-Host "== 预览 [$env:USERPROFILE\$destRel]（未执行）==" -ForegroundColor Cyan
|
||||
$freshNames = @($targets | Where-Object { -not (Test-Path (Join-Path $DestSkills $_.Name)) })
|
||||
$backupNames = @($targets | Where-Object { Test-Path (Join-Path $DestSkills $_.Name) })
|
||||
if ($freshNames.Count -gt 0) {
|
||||
Write-Host (" 新安装 {0}: {1}" -f $freshNames.Count, ($freshNames.Name -join ", "))
|
||||
}
|
||||
if ($backupNames.Count -gt 0) {
|
||||
Write-Host (" 覆盖安装（旧版将备份）{0}: {1}" -f $backupNames.Count, ($backupNames.Name -join ", "))
|
||||
}
|
||||
continue
|
||||
}
|
||||
|
||||
Write-Host "--- 安装到 $DestSkills ---" -ForegroundColor Cyan
|
||||
New-Item -ItemType Directory -Force -Path $DestSkills | Out-Null
|
||||
|
||||
foreach ($skill in $targets) {
|
||||
$dest = Join-Path $DestSkills $skill.Name
|
||||
|
||||
try {
|
||||
if (Test-Path $dest) {
|
||||
New-Item -ItemType Directory -Force -Path $BackupRoot | Out-Null
|
||||
$backupPath = Join-Path $BackupRoot ("{0}-{1}" -f $skill.Name, $timestamp)
|
||||
Move-Item -Path $dest -Destination $backupPath
|
||||
$totalBackup++
|
||||
Write-Host "[备份] $($skill.Name) 旧版 -> $backupPath" -ForegroundColor Yellow
|
||||
}
|
||||
|
||||
Copy-Item -Path $skill.FullName -Destination $dest -Recurse -Force
|
||||
$totalFresh++
|
||||
Write-Host "[安装] $($skill.Name)" -ForegroundColor Green
|
||||
|
||||
# 校验该技能落盘完整
|
||||
if (-not (Test-Path (Join-Path $DestSkills "$($skill.Name)\SKILL.md"))) {
|
||||
throw "SKILL.md 未落盘"
|
||||
}
|
||||
} catch {
|
||||
Write-Host "[FAIL] $($skill.Name) -> $DestSkills : $_" -ForegroundColor Red
|
||||
$failedTargets += "$destRel/$($skill.Name)"
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
$toBackup = @($targets | Where-Object { Test-SkillInstalled $_.Name })
|
||||
$toFresh = @($targets | Where-Object { -not (Test-SkillInstalled $_.Name) })
|
||||
|
||||
# --- 5. 预览模式收尾 ---
|
||||
if ($DryRun) {
|
||||
Write-Host "== 预览（未执行）==" -ForegroundColor Cyan
|
||||
if ($toFresh.Count -gt 0) {
|
||||
Write-Host (" 新安装 {0}: {1}" -f $toFresh.Count, ($toFresh.Name -join ", "))
|
||||
}
|
||||
if ($toBackup.Count -gt 0) {
|
||||
Write-Host (" 覆盖安装（旧版将备份）{0}: {1}" -f $toBackup.Count, ($toBackup.Name -join ", "))
|
||||
}
|
||||
Write-Host " 总技能数: $($targets.Count)"
|
||||
Write-Host "同步目录: $TraeSkills"
|
||||
Write-Host "备份目录: $BackupRoot"
|
||||
Write-Host ""
|
||||
Write-Host "[DryRun] 完成，未做任何修改。" -ForegroundColor Green
|
||||
exit 0
|
||||
}
|
||||
|
||||
# --- 5. 执行安装 ---
|
||||
New-Item -ItemType Directory -Force -Path $TraeSkills | Out-Null
|
||||
|
||||
$timestamp = Get-Date -Format "yyyyMMdd-HHmmss"
|
||||
$backupCount = 0
|
||||
|
||||
foreach ($skill in $targets) {
|
||||
$dest = Join-Path $TraeSkills $skill.Name
|
||||
|
||||
if (Test-Path $dest) {
|
||||
New-Item -ItemType Directory -Force -Path $BackupRoot | Out-Null
|
||||
$backupPath = Join-Path $BackupRoot ("{0}-{1}" -f $skill.Name, $timestamp)
|
||||
Move-Item -Path $dest -Destination $backupPath
|
||||
$backupCount++
|
||||
Write-Host "[备份] $($skill.Name) 旧版 -> $backupPath" -ForegroundColor Yellow
|
||||
}
|
||||
|
||||
Copy-Item -Path $skill.FullName -Destination $dest -Recurse -Force
|
||||
Write-Host "[安装] $($skill.Name)" -ForegroundColor Green
|
||||
}
|
||||
|
||||
# --- 6. 校验安装结果 ---
|
||||
# --- 6. 汇总与后续提示 ---
|
||||
Write-Host ""
|
||||
Write-Host "=== 校验结果 ==="
|
||||
$failed = @()
|
||||
foreach ($skill in $targets) {
|
||||
if (Test-Path (Join-Path $TraeSkills "$($skill.Name)\SKILL.md")) {
|
||||
Write-Host " [OK] $($skill.Name)" -ForegroundColor Green
|
||||
} else {
|
||||
Write-Host " [FAIL] $($skill.Name)" -ForegroundColor Red
|
||||
$failed += $skill.Name
|
||||
}
|
||||
}
|
||||
|
||||
# --- 7. 汇总与后续提示 ---
|
||||
Write-Host ""
|
||||
if ($failed.Count -gt 0) {
|
||||
Write-Host ("安装失败: {0}" -f ($failed -join ", ")) -ForegroundColor Red
|
||||
if ($failedTargets.Count -gt 0) {
|
||||
Write-Host ("安装失败: {0}" -f ($failedTargets -join ", ")) -ForegroundColor Red
|
||||
exit 2
|
||||
}
|
||||
|
||||
Write-Host "全部安装成功。" -ForegroundColor Green
|
||||
Write-Host ("本次：新建 {0} 个，覆盖 {1} 个（备份到技能目录外的 backups）。" -f $toFresh.Count, $toBackup.Count) -ForegroundColor Green
|
||||
Write-Host ("本次共写入 {0} 个技能实例（新装含覆盖），备份 {1} 个旧版。" -f ($totalFresh), $totalBackup) -ForegroundColor Green
|
||||
if ($warnCount -gt 0) {
|
||||
Write-Host ("注意：{0} 处 name/文件夹不一致或缺 name 的可疑技能（见上方警告）。" -f $warnCount) -ForegroundColor Yellow
|
||||
}
|
||||
if ($backupCount -gt 0) {
|
||||
Write-Host "旧版本备份于: $BackupRoot（确认新版可用后可手动删除）" -ForegroundColor Yellow
|
||||
if ($totalBackup -gt 0) {
|
||||
Write-Host "旧版本备份于各目标目录旁的 skills-backup（确认新版可用后可手动删除）" -ForegroundColor Yellow
|
||||
}
|
||||
|
||||
Write-Host ""
|
||||
Write-Host "下一步: 重启 Trae CN 或新建会话后，在对话中提及技能名或相关意图即可触发:"
|
||||
Write-Host "下一步: 重启对应 agent 或新建会话后，在对话中提及技能名或相关意图即可触发:"
|
||||
foreach ($s in $targets) {
|
||||
Write-Host " - $($s.Name)"
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user