#!/usr/bin/env bash # installer-opensource.sh — Open-source installer for loongsuite-pilot # # Install (first time): # curl -fsSL https://loongcollector-community-edition.oss-cn-shanghai.aliyuncs.com/loongsuite-pilot/installer.sh | bash # curl -fsSL /installer.sh | bash -s -- install \ # --sls-endpoint "https://cn-hangzhou.log.aliyuncs.com" \ # --sls-project "my-project" \ # --sls-logstore "my-logstore" \ # --sls-ak-id "your-ak-id" \ # --sls-ak-secret "your-ak-secret" # # Install a specific version: # curl -fsSL /installer.sh | bash -s -- install --version 1.2.0 # # Upgrade (preserve config, auto-rollback on failure): # curl -fsSL /installer.sh | bash -s -- upgrade # # Uninstall: # curl -fsSL /installer.sh | bash -s -- uninstall # curl -fsSL /installer.sh | bash -s -- uninstall --purge set -euo pipefail # ============================================================ # Constants # ============================================================ PACKAGE_NAME="loongsuite-pilot" PERMANENT_DIR="$HOME/.loongsuite-pilot/package" DEFAULT_DATA_DIR="$HOME/.loongsuite-pilot" # OSS download base URL _OSS_BASE_URL="https://loongcollector-community-edition.oss-cn-shanghai.aliyuncs.com/loongsuite-pilot" # ============================================================ # Parse sub-command # ============================================================ COMMAND="" PACKAGE_URL="${LOONGSUITE_PILOT_PACKAGE_URL:-}" INSTALL_VERSION="" SLS_ENDPOINT="" SLS_PROJECT="" SLS_LOGSTORE="" SLS_AK_ID="" SLS_AK_SECRET="" DATA_DIR="$DEFAULT_DATA_DIR" LOG_LEVEL="" USER_ID="" COLLECT_LOG="" COLLECT_TRACE="" CMS_LICENSE_KEY="" CMS_ENDPOINT="" CMS_WORKSPACE="" SERVICE_NAME_PREFIX="" SELECTED_AGENTS="" MASK_MODE="" MASK_TYPES="" HAS_SUDO=0 PURGE=0 SYSTEM_SERVICE=0 # First arg is sub-command (or option -> default to install) if [[ $# -gt 0 ]]; then case "$1" in install|upgrade|uninstall) COMMAND="$1"; shift ;; -*) COMMAND="install" ;; *) COMMAND="install" ;; esac else COMMAND="install" fi while [[ $# -gt 0 ]]; do case "$1" in --sls-endpoint) SLS_ENDPOINT="$2"; shift 2 ;; --sls-endpoint=*) SLS_ENDPOINT="${1#*=}"; shift ;; --sls-project) SLS_PROJECT="$2"; shift 2 ;; --sls-project=*) SLS_PROJECT="${1#*=}"; shift ;; --sls-logstore) SLS_LOGSTORE="$2"; shift 2 ;; --sls-logstore=*) SLS_LOGSTORE="${1#*=}"; shift ;; --sls-ak-id) SLS_AK_ID="$2"; shift 2 ;; --sls-ak-id=*) SLS_AK_ID="${1#*=}"; shift ;; --sls-ak-secret) SLS_AK_SECRET="$2"; shift 2 ;; --sls-ak-secret=*) SLS_AK_SECRET="${1#*=}"; shift ;; --package-url) PACKAGE_URL="$2"; shift 2 ;; --package-url=*) PACKAGE_URL="${1#--package-url=}"; shift ;; --data-dir) DATA_DIR="$2"; shift 2 ;; --data-dir=*) DATA_DIR="${1#*=}"; shift ;; --log-level) LOG_LEVEL="$2"; shift 2 ;; --log-level=*) LOG_LEVEL="${1#*=}"; shift ;; --userId|--user.id) USER_ID="$2"; shift 2 ;; --userId=*|--user.id=*) USER_ID="${1#*=}"; shift ;; --lang) export LOONGSUITE_PILOT_LANG="$2"; shift 2 ;; --lang=*) export LOONGSUITE_PILOT_LANG="${1#--lang=}"; shift ;; --version) INSTALL_VERSION="$2"; shift 2 ;; --version=*) INSTALL_VERSION="${1#*=}"; shift ;; --collect-log) COLLECT_LOG="$2"; shift 2 ;; --collect-log=*) COLLECT_LOG="${1#*=}"; shift ;; --collect-trace) COLLECT_TRACE="$2"; shift 2 ;; --collect-trace=*) COLLECT_TRACE="${1#*=}"; shift ;; --cms-license-key) CMS_LICENSE_KEY="$2"; shift 2 ;; --cms-license-key=*) CMS_LICENSE_KEY="${1#*=}"; shift ;; --cms-endpoint) CMS_ENDPOINT="$2"; shift 2 ;; --cms-endpoint=*) CMS_ENDPOINT="${1#*=}"; shift ;; --cms-workspace) CMS_WORKSPACE="$2"; shift 2 ;; --cms-workspace=*) CMS_WORKSPACE="${1#*=}"; shift ;; --service-name-prefix) SERVICE_NAME_PREFIX="$2"; shift 2 ;; --service-name-prefix=*) SERVICE_NAME_PREFIX="${1#*=}"; shift ;; --agents) SELECTED_AGENTS="$2"; shift 2 ;; --agents=*) SELECTED_AGENTS="${1#*=}"; shift ;; --mask-mode) MASK_MODE="$2"; shift 2 ;; --mask-mode=*) MASK_MODE="${1#*=}"; shift ;; --mask-types) MASK_TYPES="$2"; shift 2 ;; --mask-types=*) MASK_TYPES="${1#*=}"; shift ;; --purge) PURGE=1; shift ;; --system-service) SYSTEM_SERVICE=1; shift ;; *) echo "Unknown option: $1" >&2 exit 1 ;; esac done if [ -n "$MASK_MODE" ]; then case "$MASK_MODE" in all|none|custom) ;; *) echo "❌ Unknown mask mode: $MASK_MODE (use 'all', 'custom', or 'none')" >&2 exit 1 ;; esac fi if [ "$MASK_MODE" = "custom" ] && [ -z "$MASK_TYPES" ]; then echo "❌ --mask-types is required when --mask-mode custom" >&2 exit 1 fi if [ -n "$MASK_TYPES" ] && [ "$MASK_MODE" != "custom" ]; then echo "❌ --mask-types can only be used with --mask-mode custom" >&2 exit 1 fi # Validate current user and sudo access on Linux validate_install_user() { case "$(uname -s)" in Linux) local current_user current_user=$(whoami) if [ "$(id -u)" -eq 0 ]; then HAS_SUDO=1 SYSTEM_SERVICE=1 msg " ✅ 以 root 身份安装（自动使用系统级服务）" \ " ✅ Installing as root (auto system-level service)" elif [ "$SYSTEM_SERVICE" -eq 1 ]; then if sudo -n true 2>/dev/null; then HAS_SUDO=1 msg " ✅ sudo 权限校验通过 (user: $current_user)" \ " ✅ sudo access verified (user: $current_user)" elif sudo -v 2>/dev/null; then HAS_SUDO=1 msg " ✅ sudo 权限校验通过 (user: $current_user)" \ " ✅ sudo access verified (user: $current_user)" else HAS_SUDO=0 SYSTEM_SERVICE=0 msg "⚠️ 无 sudo 权限 — 无法注册系统级服务。将使用用户态 systemd 服务。" \ "⚠️ No sudo access — cannot register system-level service. Using user-level systemd." fi else msg " Install user: $current_user（服务类型将在启动时检测）" \ " Install user: $current_user (service type determined at start)" fi ;; esac } # Resolve PACKAGE_URL from OSS if not explicitly set if [ -z "$PACKAGE_URL" ]; then if [ -n "$INSTALL_VERSION" ]; then PACKAGE_URL="${_OSS_BASE_URL}/${INSTALL_VERSION}/${PACKAGE_NAME}.tar.gz" else PACKAGE_URL="${_OSS_BASE_URL}/latest/${PACKAGE_NAME}.tar.gz" fi fi # ============================================================ # Language detection # ============================================================ detect_lang() { if [ -n "${LOONGSUITE_PILOT_LANG:-}" ]; then echo "$LOONGSUITE_PILOT_LANG"; return; fi for v in "${LANGUAGE:-}" "${LC_ALL:-}" "${LC_MESSAGES:-}" "${LANG:-}"; do if echo "$v" | grep -qi "zh"; then echo "zh"; return; fi done if [ "$(uname)" = "Darwin" ]; then local al al=$(defaults read -g AppleLanguages 2>/dev/null | grep -i "zh" | head -1 || true) if [ -n "$al" ]; then echo "zh"; return; fi fi echo "en" } LANG_MODE=$(detect_lang) msg() { [ "$LANG_MODE" = "zh" ] && echo "$1" || echo "$2"; } # ============================================================ # Common: check dependencies # ============================================================ _resolve_realpath() { realpath "$1" 2>/dev/null || readlink -f "$1" 2>/dev/null || echo "$1" } _node_is_app_bundle() { local resolved resolved=$(_resolve_realpath "$1") case "$resolved" in /Applications/*.app/Contents/*|/System/Applications/*.app/Contents/*|"$HOME"/Applications/*.app/Contents/*) return 0 ;; esac return 1 } _node_is_suitable() { local bin="$1" [ -x "$bin" ] || return 1 _node_is_app_bundle "$bin" && return 1 local ver ver="$("$bin" --version 2>/dev/null)" || return 1 local major="${ver#v}" major="${major%%.*}" [[ "$major" =~ ^[0-9]+$ ]] && (( major >= 18 )) || return 1 return 0 } resolve_node() { local _candidates=() local _nvm_candidates=("$HOME/.nvm/versions/node"/*/bin/node) local i for (( i=${#_nvm_candidates[@]}-1; i>=0; i-- )); do _candidates+=("${_nvm_candidates[i]}") done _candidates+=( "$HOME/.volta/bin/node" "$HOME/.fnm/aliases/default/bin/node" /opt/homebrew/bin/node /usr/local/bin/node "$HOME/.local/bin/node" ) if command -v node >/dev/null 2>&1; then _candidates+=("$(command -v node)") fi for candidate in "${_candidates[@]}"; do if _node_is_suitable "$candidate"; then _resolve_realpath "$candidate" return 0 fi done return 1 } check_deps() { msg "==> 检查依赖..." "==> Checking dependencies..." NODE_BIN=$(resolve_node) || { msg "❌ 缺少依赖: node，请先安装后重试" \ "❌ Missing dependency: node — please install it first" exit 1 } NODE_MAJOR=$("$NODE_BIN" -e "process.stdout.write(String(process.versions.node.split('.')[0]))") if [ "$NODE_MAJOR" -lt 18 ]; then msg "❌ 需要 Node.js >= 18，当前版本: $("$NODE_BIN" --version)" \ "❌ Requires Node.js >= 18, current: $("$NODE_BIN" --version)" exit 1 fi # Pin the node binary path mkdir -p "$DATA_DIR" 2>/dev/null || true echo "$NODE_BIN" > "$DATA_DIR/node-bin" # Derive npm from the same installation NPM_BIN="$(dirname "$NODE_BIN")/npm" if [ ! -x "$NPM_BIN" ]; then if command -v npm &>/dev/null; then NPM_BIN=$(command -v npm) else msg "❌ 缺少依赖: npm，请先安装后重试" \ "❌ Missing dependency: npm — please install it first" exit 1 fi fi if [ "$(uname)" = "Darwin" ]; then local sys_arch; sys_arch=$(uname -m) local node_arch; node_arch=$("$NODE_BIN" -e "process.stdout.write(process.arch)") if [ "$sys_arch" = "arm64" ] && [ "$node_arch" = "x64" ]; then msg "⚠️ 架构不匹配: 系统为 arm64 (Apple Silicon)，但 Node.js 为 x64 (Intel)" \ "⚠️ Architecture mismatch: system is arm64 but Node.js is x64 (Intel)" msg " 原生模块可能无法正常加载，建议安装 arm64 版本的 Node.js" \ " Native modules may fail to load. Please install arm64 Node.js" fi fi if ! command -v curl &>/dev/null && ! command -v wget &>/dev/null; then msg "❌ 需要 curl 或 wget，请先安装" \ "❌ curl or wget is required — please install one first" exit 1 fi msg " ✅ node $("$NODE_BIN" --version) npm $("$NPM_BIN" --version)" \ " ✅ node $("$NODE_BIN" --version) npm $("$NPM_BIN" --version)" msg " node pinned: $NODE_BIN" " node pinned: $NODE_BIN" echo "" } # ============================================================ # Common: download and extract package -> sets INSTALL_SRC # ============================================================ download_and_extract() { TMP_DIR="$(mktemp -d)" # TMP_DIR cleanup is handled by the caller's trap msg "==> 下载安装包: $PACKAGE_URL" \ "==> Downloading: $PACKAGE_URL" if command -v curl &>/dev/null; then curl -fsSL "$PACKAGE_URL" -o "$TMP_DIR/package.tar.gz" else wget -q "$PACKAGE_URL" -O "$TMP_DIR/package.tar.gz" fi msg " ✅ 下载完成" " ✅ Downloaded" echo "" msg "==> 解压安装包..." "==> Extracting..." if tar --warning=no-unknown-keyword -xzf "$TMP_DIR/package.tar.gz" -C "$TMP_DIR" 2>/dev/null; then : else tar -xzf "$TMP_DIR/package.tar.gz" -C "$TMP_DIR" fi if [ -d "$TMP_DIR/$PACKAGE_NAME" ]; then INSTALL_SRC="$TMP_DIR/$PACKAGE_NAME" elif [ -f "$TMP_DIR/package.json" ]; then INSTALL_SRC="$TMP_DIR" else INSTALL_SRC=$(find "$TMP_DIR" -name "package.json" -maxdepth 2 -exec dirname {} \; | head -1 || true) if [ -z "$INSTALL_SRC" ]; then msg "❌ 解压后未找到 package.json，安装包结构异常" \ "❌ package.json not found — unexpected package structure" exit 1 fi fi msg " ✅ 解压完成" " ✅ Extracted" echo "" } # ============================================================ # Agent probe: detect available agents via Node.js CLI probe # ============================================================ PROBE_RESULT="[]" probe_agents() { msg "==> 探测 AI Agent..." "==> Probing AI Agents..." PROBE_RESULT=$("$NODE_BIN" "$INSTALL_SRC/dist/cli-probe.cjs" 2>/dev/null) || { msg " ⚠️ Agent 探测失败，将跳过选择" " ⚠️ Agent probe failed, skipping selection" PROBE_RESULT="[]" return 0 } local count count=$("$NODE_BIN" -e "const r=JSON.parse(process.argv[1]);process.stdout.write(String(r.length))" "$PROBE_RESULT" 2>/dev/null || echo "0") msg " ✅ 探测到 ${count} 个 Agent 定义" " ✅ Found ${count} agent definitions" echo "" } # ============================================================ # Agent selection: interactive menu or --agents flag # ============================================================ select_agents() { if [ -n "$SELECTED_AGENTS" ]; then msg " 使用指定的 Agent: $SELECTED_AGENTS" " Using specified agents: $SELECTED_AGENTS" echo "" return 0 fi local agent_count agent_count=$("$NODE_BIN" -e "const r=JSON.parse(process.argv[1]);process.stdout.write(String(r.length))" "$PROBE_RESULT" 2>/dev/null || echo "0") if [ "$agent_count" = "0" ]; then return 0 fi # Non-interactive: auto-select all detected agents if [ ! -t 0 ]; then SELECTED_AGENTS=$("$NODE_BIN" -e " const r = JSON.parse(process.argv[1]); const detected = r.filter(a => a.detected).map(a => a.id); process.stdout.write(detected.join(',')); " "$PROBE_RESULT" 2>/dev/null || true) msg " (非交互模式) 自动选择已检测到的 Agent: $SELECTED_AGENTS" \ " (non-interactive) Auto-selected detected agents: $SELECTED_AGENTS" echo "" return 0 fi # Interactive menu "$NODE_BIN" -e " const r = JSON.parse(process.argv[1]); const lang = process.argv[2]; const defaults = []; for (let i = 0; i < r.length; i++) { const a = r[i]; const status = lang === 'zh' ? (a.detected ? '已检测到: ' + a.reason : '未检测到') : (a.detected ? 'detected: ' + a.reason : 'not detected'); console.log(' [' + (i+1) + '] ' + a.displayName.padEnd(16) + '(' + status + ')'); if (a.detected) defaults.push(i+1); } console.log(''); if (lang === 'zh') { console.log(' 默认选择已检测到的 Agent: ' + defaults.join(',')); console.log(' 输入要启用的编号 (逗号分隔)，直接回车使用默认:'); } else { console.log(' Default selection (detected): ' + defaults.join(',')); console.log(' Enter numbers to enable (comma-separated), press Enter for default:'); } " "$PROBE_RESULT" "$LANG_MODE" # Read user input local select_input read -r select_input # Compute final selection: empty input = detected agents, otherwise use exact input SELECTED_AGENTS=$("$NODE_BIN" -e " const r = JSON.parse(process.argv[1]); const input = process.argv[2] || ''; let indices; if (!input.trim()) { indices = r.map((a, i) => a.detected ? i : -1).filter(i => i >= 0); } else { indices = [...new Set(input.trim().split(/[\s,]+/).map(Number).filter(n => n >= 1 && n <= r.length))].map(n => n - 1); } const ids = indices.sort((a,b) => a-b).map(i => r[i].id); process.stdout.write(ids.join(',')); " "$PROBE_RESULT" "$select_input" 2>/dev/null || true) if [ -n "$SELECTED_AGENTS" ]; then msg " 已选择: $SELECTED_AGENTS" " Selected: $SELECTED_AGENTS" else msg " 未选择任何 Agent" " No agents selected" fi echo "" } # ============================================================ # Interactive: prompt for userId (skipped when --userId given or non-interactive) # ============================================================ prompt_user_id() { if [ -n "$USER_ID" ]; then return 0; fi if [ ! -t 0 ]; then return 0; fi local existing_uid="" local config_file="$DATA_DIR/config.json" if [ -f "$config_file" ]; then existing_uid=$("$NODE_BIN" -e " try { const c=JSON.parse(require('fs').readFileSync(process.argv[1],'utf-8')); process.stdout.write(c.userId||''); } catch {} " -- "$config_file" 2>/dev/null || true) fi echo "" if [ -n "$existing_uid" ]; then msg " 当前 userId: $existing_uid" \ " Current userId: $existing_uid" msg " 直接回车保留，或输入新值:" \ " Press Enter to keep, or type a new value:" else msg " 请输入你的 userId（用于数据归属，可直接回车跳过）:" \ " Enter your userId (for data attribution, press Enter to skip):" fi printf " > " local input read -r input input=$(echo "$input" | tr -d '[:space:]') if [ -n "$input" ]; then USER_ID="$input" elif [ -n "$existing_uid" ]; then USER_ID="$existing_uid" fi } # ============================================================ # Interactive: confirm config overwrite when key fields differ # ============================================================ confirm_config_overwrite() { local config_file="$DATA_DIR/config.json" if [ ! -f "$config_file" ]; then return 0; fi local diffs diffs=$("$NODE_BIN" -e " const fs = require('fs'); let old = {}; try { old = JSON.parse(fs.readFileSync(process.argv[1], 'utf-8')); } catch { process.exit(0); } const newVals = JSON.parse(process.argv[2]); const normalizeCsv = value => String(value || '').split(',').map(v => v.trim()).filter(Boolean).join(','); const checks = [ { label: 'sls.endpoint', oldVal: (old.sls||{}).endpoint||'', newVal: newVals.slsEndpoint }, { label: 'sls.project', oldVal: (old.sls||{}).project||'', newVal: newVals.slsProject }, { label: 'sls.logstore', oldVal: (old.sls||{}).logstore||'', newVal: newVals.slsLogstore }, { label: 'cms.licenseKey', oldVal: (old.cms||{}).licenseKey||'', newVal: newVals.cmsLicenseKey }, { label: 'cms.endpoint', oldVal: (old.cms||{}).endpoint||'', newVal: newVals.cmsEndpoint }, { label: 'cms.workspace', oldVal: (old.cms||{}).workspace||'', newVal: newVals.cmsWorkspace }, { label: 'serviceNamePrefix', oldVal: old.serviceNamePrefix||'', newVal: newVals.serviceNamePrefix }, { label: 'mask.mode', oldVal: (old.mask||{}).mode||'', newVal: newVals.maskMode }, { label: 'mask.types', oldVal: Array.isArray((old.mask||{}).types) ? normalizeCsv(old.mask.types.join(',')) : '', newVal: normalizeCsv(newVals.maskTypes) }, ]; const changed = checks.filter(c => c.newVal && c.oldVal && c.newVal !== c.oldVal); if (!changed.length) process.exit(0); for (const c of changed) { console.log(c.label + ': ' + c.oldVal + ' -> ' + c.newVal); } " -- "$config_file" "$(printf '{"slsEndpoint":"%s","slsProject":"%s","slsLogstore":"%s","cmsLicenseKey":"%s","cmsEndpoint":"%s","cmsWorkspace":"%s","serviceNamePrefix":"%s","maskMode":"%s","maskTypes":"%s"}' \ "$SLS_ENDPOINT" "$SLS_PROJECT" "$SLS_LOGSTORE" "$CMS_LICENSE_KEY" "$CMS_ENDPOINT" "$CMS_WORKSPACE" "$SERVICE_NAME_PREFIX" "$MASK_MODE" "$MASK_TYPES")" 2>/dev/null || true) if [ -z "$diffs" ]; then return 0; fi echo "" msg "⚠️ 以下配置将被覆盖:" "⚠️ The following config will be overwritten:" echo "$diffs" | while IFS= read -r line; do echo " $line" done if [ -t 0 ]; then echo "" msg " 确认覆盖? (y/N):" " Confirm overwrite? (y/N):" printf " > " local answer read -r answer case "$answer" in y|Y|yes|YES) ;; *) msg "已取消安装" "Installation cancelled" exit 0 ;; esac else msg " (非交互模式) 继续覆盖" \ " (non-interactive) Proceeding with overwrite" fi } # ============================================================ # Common: deploy bootstrap scripts from the current version # ============================================================ deploy_bootstrap_scripts() { local src_dir="$PERMANENT_DIR/scripts" local boot_dir="$HOME/.loongsuite-pilot/bin" mkdir -p "$boot_dir" cp -f "$src_dir/collector-daemon.js" "$boot_dir/" [ -f "$src_dir/updater-daemon.js" ] && cp -f "$src_dir/updater-daemon.js" "$boot_dir/" || true } # ============================================================ # Common: deploy package to versions/ directory # ============================================================ deploy_package() { local src="$1" local cache_dir="$HOME/.loongsuite-pilot" local versions_dir="$cache_dir/versions" local current_file="$cache_dir/current" local previous_file="$cache_dir/previous" local ver="" commit="" if [ -f "$src/VERSION" ]; then ver=$(grep '^version=' "$src/VERSION" | cut -d= -f2) commit=$(grep '^git_commit=' "$src/VERSION" | cut -d= -f2) fi if [ -n "$ver" ] && [ -n "$commit" ]; then local dir_name="${ver}_${commit}" local target="$versions_dir/$dir_name" if [ -f "$current_file" ]; then local old_dir old_dir=$(cat "$current_file" 2>/dev/null | tr -d '[:space:]') if [ -n "$old_dir" ] && [ "$old_dir" != "$dir_name" ]; then echo "$old_dir" > "$previous_file" fi fi msg "==> 部署到 $target ..." "==> Deploying to $target ..." mkdir -p "$versions_dir" rm -rf "$target" cp -r "$src" "$target" echo "$dir_name" > "$current_file.tmp" mv -f "$current_file.tmp" "$current_file" PERMANENT_DIR="$target" else msg "==> 部署到 $PERMANENT_DIR ..." \ "==> Deploying to $PERMANENT_DIR ..." mkdir -p "$(dirname "$PERMANENT_DIR")" rm -rf "$PERMANENT_DIR" cp -r "$src" "$PERMANENT_DIR" fi msg " ✅ 部署完成" " ✅ Deployed" echo "" deploy_bootstrap_scripts msg "==> 安装依赖..." "==> Installing dependencies..." (cd "$PERMANENT_DIR" && "$NPM_BIN" install --production --no-optional 2>&1 | tail -1) msg " ✅ 依赖安装完成" " ✅ Dependencies installed" echo "" msg "==> 部署 hook 脚本..." "==> Deploying hook scripts..." if [ -f scripts/postinstall.js ]; then "$NODE_BIN" scripts/postinstall.js fi msg " ✅ Hook 脚本已部署" " ✅ Hook scripts deployed" msg " 如使用 Codex 桌面版，首次启动需在桌面端手动信任 hooks" \ " If using Codex desktop app, please manually trust hooks on first launch" echo "" } # ============================================================ # Migrate legacy single-directory layout to versions/ layout # ============================================================ migrate_legacy_layout() { local cache_dir="$HOME/.loongsuite-pilot" local current_file="$cache_dir/current" local legacy_dir="$cache_dir/package" local versions_dir="$cache_dir/versions" if [ -f "$current_file" ]; then return 0 fi if [ ! -d "$legacy_dir" ] || [ ! -f "$legacy_dir/dist/index.js" ]; then return 0 fi msg "==> 迁移旧版本目录结构..." "==> Migrating legacy directory layout..." local ver="" commit="" if [ -f "$legacy_dir/VERSION" ]; then ver=$(grep '^version=' "$legacy_dir/VERSION" | cut -d= -f2) commit=$(grep '^git_commit=' "$legacy_dir/VERSION" | cut -d= -f2) fi ver="${ver:-0.0.0}" commit="${commit:-legacy}" local dir_name="${ver}_${commit}" local target="$versions_dir/$dir_name" mkdir -p "$versions_dir" cp -r "$legacy_dir" "$target" echo "$dir_name" > "$current_file" PERMANENT_DIR="$target" msg " ✅ 已迁移到 $target" " ✅ Migrated to $target" echo "" } # ============================================================ # Common: write / merge config.json # ============================================================ write_config() { local config_file="$DATA_DIR/config.json" msg "==> 写入配置文件 $config_file ..." \ "==> Writing config to $config_file ..." mkdir -p "$DATA_DIR" "$NODE_BIN" -e " const fs = require('fs'); const path = '$config_file'; let existing = {}; try { existing = JSON.parse(fs.readFileSync(path, 'utf-8')); } catch {} const config = { ...existing, enabled: true, dataDir: '$DATA_DIR', }; delete config.internal; if (config.userId === undefined && config['user.id'] !== undefined) { config.userId = config['user.id']; } delete config['user.id']; const slsEndpoint = '${SLS_ENDPOINT}'; const slsProject = '${SLS_PROJECT}'; const slsLogstore = '${SLS_LOGSTORE}'; const slsAkId = '${SLS_AK_ID}'; const slsAkSecret = '${SLS_AK_SECRET}'; const logLevel = '${LOG_LEVEL}'; const userId = '${USER_ID}'; if (slsEndpoint || slsProject || slsLogstore) { config.sls = config.sls || {}; delete config.sls.destinationOverride; if (slsEndpoint) { config.sls.endpoint = slsEndpoint; } if (slsAkId && slsAkSecret) { config.sls.mode = 'ak'; config.sls.accessKeyId = slsAkId; config.sls.accessKeySecret = slsAkSecret; } if (slsProject && slsLogstore) { config.sls.project = slsProject; config.sls.logstore = slsLogstore; delete config.sls.endpoints; } } if (logLevel) { config.logLevel = logLevel; } if (userId) { config.userId = userId; delete config.identity; } const collectLog = '${COLLECT_LOG}'; const collectTrace = '${COLLECT_TRACE}'; const cmsLicenseKey = '${CMS_LICENSE_KEY}'; const cmsEndpoint = '${CMS_ENDPOINT}'; const cmsWorkspace = '${CMS_WORKSPACE}'; const serviceNamePrefix = '${SERVICE_NAME_PREFIX}'; const selectedAgents = '${SELECTED_AGENTS}'; const maskMode = '${MASK_MODE}'; const maskTypes = '${MASK_TYPES}'; if (collectLog) config.collectLog = collectLog === 'true'; if (collectTrace) config.collectTrace = collectTrace === 'true'; if (cmsLicenseKey || cmsEndpoint || cmsWorkspace) { config.cms = config.cms || {}; if (cmsLicenseKey) config.cms.licenseKey = cmsLicenseKey; if (cmsEndpoint) config.cms.endpoint = cmsEndpoint; if (cmsWorkspace) config.cms.workspace = cmsWorkspace; } if (serviceNamePrefix) config.serviceNamePrefix = serviceNamePrefix; if (maskMode) { config.mask = config.mask || {}; config.mask.mode = maskMode; if (maskMode === 'custom') { config.mask.types = maskTypes .split(',') .map(type => type.trim()) .filter(Boolean); } else { delete config.mask.types; } } if (selectedAgents) { config.agents = config.agents || {}; const selected = selectedAgents.split(',').map(s => s.trim()).filter(Boolean); const allAgents = JSON.parse(process.argv[1] || '[]'); for (const agent of allAgents) { config.agents[agent.id] = config.agents[agent.id] || {}; config.agents[agent.id].enabled = selected.includes(agent.id); } } fs.writeFileSync(path, JSON.stringify(config, null, 2) + '\n'); " -- "$PROBE_RESULT" msg " ✅ 配置已写入" " ✅ Config written" echo "" } # ============================================================ # Common: install/update the loongsuite-pilot service management script # ============================================================ install_loongsuite_pilot_command() { msg "==> 安装服务管理脚本..." "==> Installing service management script..." local global_bin_dir="$HOME/.local/bin" mkdir -p "$global_bin_dir" local loongsuite_pilot_cmd="$global_bin_dir/loongsuite-pilot" cp -f "$PERMANENT_DIR/scripts/loongsuite-pilot.sh" "$loongsuite_pilot_cmd" chmod +x "$loongsuite_pilot_cmd" msg " ✅ 已安装: $loongsuite_pilot_cmd" " ✅ Installed: $loongsuite_pilot_cmd" # If /usr/local/bin is writable (root), create a symlink for immediate PATH access if [ -d /usr/local/bin ] && [ -w /usr/local/bin ]; then ln -sf "$loongsuite_pilot_cmd" /usr/local/bin/loongsuite-pilot msg " ✅ 已链接到 /usr/local/bin/loongsuite-pilot" " ✅ Linked to /usr/local/bin/loongsuite-pilot" else ensure_path_block() { local file="$1" if [ ! -f "$file" ]; then touch "$file" 2>/dev/null || return 0 fi if [ ! -w "$file" ]; then msg " ⚠️ $file 不可写，跳过" " ⚠️ $file is not writable, skipping" return 0 fi if grep -q '\.local/bin' "$file" 2>/dev/null; then return 0; fi # Ensure file ends with a newline before appending [ -s "$file" ] && [ "$(tail -c1 "$file" | wc -l)" -eq 0 ] && echo "" >> "$file" cat >> "$file" << 'PATHBLOCK' # loongsuite-pilot: add ~/.local/bin to PATH export PATH="$HOME/.local/bin:$PATH" PATHBLOCK msg " 已将 ~/.local/bin 添加到 PATH ($file)" \ " Added ~/.local/bin to PATH ($file)" } case "${SHELL:-/bin/bash}" in */zsh) ensure_path_block "$HOME/.zshrc" || true ;; */bash) ensure_path_block "$HOME/.bashrc" || true ensure_path_block "$HOME/.bash_profile" || true ;; *) ensure_path_block "$HOME/.bashrc" || true ;; esac fi echo "" # Ensure loongsuite-pilot is on PATH for the rest of this script export PATH="$global_bin_dir:$PATH" } # ============================================================ # Common: read VERSION file fields # ============================================================ get_installed_version() { local cache_dir="$HOME/.loongsuite-pilot" local current_file="$cache_dir/current" local versions_dir="$cache_dir/versions" if [ -f "$current_file" ]; then local dir dir=$(cat "$current_file" 2>/dev/null | tr -d '[:space:]') if [ -n "$dir" ] && [ -f "$versions_dir/$dir/VERSION" ]; then grep '^version=' "$versions_dir/$dir/VERSION" | cut -d= -f2 return 0 fi fi local vf="$PERMANENT_DIR/VERSION" if [ -f "$vf" ]; then grep '^version=' "$vf" | cut -d= -f2 else echo "" fi } get_version_from_dir() { local vf="$1/VERSION" if [ -f "$vf" ]; then grep '^version=' "$vf" | cut -d= -f2 else echo "" fi } get_commit_from_dir() { local vf="$1/VERSION" if [ -f "$vf" ]; then grep '^git_commit=' "$vf" | cut -d= -f2 else echo "" fi } show_version_info() { local dir="$1" local vf="$dir/VERSION" if [ -f "$vf" ]; then local v; v=$(grep '^version=' "$vf" | cut -d= -f2) local c; c=$(grep '^git_commit=' "$vf" | cut -d= -f2) local t; t=$(grep '^build_time=' "$vf" | cut -d= -f2) echo "v${v} (${c}, ${t})" else echo "unknown" fi } # ============================================================ # Common: print summary # ============================================================ # ============================================================ # Remove OTel Claude plugin # ============================================================ remove_otel_plugin() { local OTEL_CLAUDE_DIR="$HOME/.cache/opentelemetry.instrumentation.claude" local OTEL_CODEX_DIR="$HOME/.cache/opentelemetry.instrumentation.codex" # Prevent NODE_OPTIONS --require intercept.js from breaking node commands # after the Claude plugin directory (and intercept.js) is deleted unset NODE_OPTIONS 2>/dev/null || true if [ -f "$OTEL_CLAUDE_DIR/package/scripts/uninstall.sh" ]; then bash "$OTEL_CLAUDE_DIR/package/scripts/uninstall.sh" 2>/dev/null || true msg " ✅ Claude Code 插件 hooks 和 alias 已清理" \ " ✅ Claude Code plugin hooks and alias cleaned" else for rc in "$HOME/.bashrc" "$HOME/.zshrc" "$HOME/.bash_profile"; do [ -f "$rc" ] || continue if grep -q "# BEGIN otel-claude-hook" "$rc" 2>/dev/null; then sed -i.bak '/# BEGIN otel-claude-hook/,/# END otel-claude-hook/d' "$rc" rm -f "${rc}.bak" fi if grep -q "# BEGIN otel-claude-hook-env" "$rc" 2>/dev/null; then sed -i.bak '/# BEGIN otel-claude-hook-env/,/# END otel-claude-hook-env/d' "$rc" rm -f "${rc}.bak" fi done msg " ✅ claude alias 已清理" " ✅ claude alias cleaned" # Clean settings.json hooks (fallback when uninstall.sh is unavailable) local claude_settings="$HOME/.claude/settings.json" if [ -f "$claude_settings" ] && grep -qE "otel-claude-hook|hook-entry\.sh" "$claude_settings" 2>/dev/null && command -v node &>/dev/null; then node -e " const fs = require('fs'); const f = process.argv[1]; const isOurs = c => c.includes('otel-claude-hook') || c.includes('hook-entry.sh'); try { const d = JSON.parse(fs.readFileSync(f, 'utf-8')); if (d && d.hooks) { for (const ev of Object.keys(d.hooks)) { if (!Array.isArray(d.hooks[ev])) continue; d.hooks[ev] = d.hooks[ev].map(m => { if (!Array.isArray(m.hooks)) return m; m.hooks = m.hooks.filter(h => !(h.command && isOurs(h.command))); return m.hooks.length > 0 ? m : null; }).filter(Boolean); if (d.hooks[ev].length === 0) delete d.hooks[ev]; } if (Object.keys(d.hooks).length === 0) delete d.hooks; fs.writeFileSync(f, JSON.stringify(d, null, 2) + '\n'); } } catch {} " "$claude_settings" 2>/dev/null || true msg " ✅ settings.json hooks 已清理" " ✅ settings.json hooks cleaned" fi fi local otel_config="$HOME/.claude/otel-config.json" if [ -f "$otel_config" ] && command -v node &>/dev/null; then node -e " const fs = require('fs'); try { const cfg = JSON.parse(fs.readFileSync(process.argv[1], 'utf-8')); delete cfg.log_enabled; delete cfg.log_dir; delete cfg.log_filename_format; fs.writeFileSync(process.argv[1], JSON.stringify(cfg, null, 2) + '\n'); } catch {} " "$otel_config" 2>/dev/null || true fi if [ -d "$OTEL_CLAUDE_DIR" ]; then if [ "$PURGE" -eq 1 ]; then rm -rf "$OTEL_CLAUDE_DIR" msg " ✅ 插件目录已完全删除 (--purge): $OTEL_CLAUDE_DIR" \ " ✅ Plugin directory fully removed (--purge): $OTEL_CLAUDE_DIR" else find "$OTEL_CLAUDE_DIR" -maxdepth 1 \ ! -name sessions \ ! -name "$(basename "$OTEL_CLAUDE_DIR")" \ -exec rm -rf {} + 2>/dev/null || true msg " ✅ 插件文件已删除（sessions/ 已保留）" \ " ✅ Plugin files removed (sessions/ preserved)" fi fi # --- Codex OTel plugin cleanup --- if [ -f "$OTEL_CODEX_DIR/package/scripts/uninstall.sh" ]; then bash "$OTEL_CODEX_DIR/package/scripts/uninstall.sh" 2>/dev/null || true msg " ✅ Codex 插件 hooks 已清理" \ " ✅ Codex plugin hooks cleaned" else # Clean hooks.json (new format) local codex_hooks_json="$HOME/.codex/hooks.json" if [ -f "$codex_hooks_json" ] && grep -qE "otel-codex-hook|hook-entry\.sh" "$codex_hooks_json" 2>/dev/null && command -v node &>/dev/null; then node -e " const fs = require('fs'); const f = process.argv[1]; const isOurs = c => c.includes('otel-codex-hook') || c.includes('hook-entry.sh'); try { const d = JSON.parse(fs.readFileSync(f, 'utf-8')); if (d && d.hooks) { for (const ev of Object.keys(d.hooks)) { d.hooks[ev] = d.hooks[ev].filter(g => { if (!g.hooks) return true; g.hooks = g.hooks.filter(h => !(h.command && isOurs(h.command))); return g.hooks.length > 0; }); if (d.hooks[ev].length === 0) delete d.hooks[ev]; } if (Object.keys(d.hooks).length === 0) { fs.unlinkSync(f); } else { fs.writeFileSync(f, JSON.stringify(d, null, 2) + '\n'); } } } catch {} " "$codex_hooks_json" 2>/dev/null || true fi # Clean config.toml (legacy hooks + trust block) local codex_config="$HOME/.codex/config.toml" if [ -f "$codex_config" ] && grep -q "otel-codex-hook" "$codex_config" 2>/dev/null; then # Remove legacy hook block (# OpenTelemetry instrumentation hooks ... stop) local marker="# OpenTelemetry instrumentation hooks" local end_str='command = "otel-codex-hook stop"' if grep -q "$marker" "$codex_config" 2>/dev/null && grep -qF "$end_str" "$codex_config" 2>/dev/null; then local tmp; tmp=$(mktemp) awk -v m="$marker" -v e="$end_str" ' BEGIN { skip=0 } skip==0 && index($0, m) { skip=1; next } skip==1 { if (index($0, e)) { skip=2 }; next } skip==2 && /^[[:space:]]*$/ { next } { skip=0; print } ' "$codex_config" > "$tmp" mv "$tmp" "$codex_config" fi # Remove trust entries (逐条精确删除,不用 BEGIN/END 范围删以免误伤用户数据) # Step a: 删 BEGIN/END marker 注释行(仅注释行本身) if grep -qE "# (BEGIN|END) otel-codex-hook trust" "$codex_config" 2>/dev/null; then local tmp; tmp=$(mktemp) grep -v "# BEGIN otel-codex-hook trust\|# END otel-codex-hook trust" "$codex_config" > "$tmp" || true mv "$tmp" "$codex_config" fi # Step b: 删 bypass_hook_trust 行 if grep -q "bypass_hook_trust" "$codex_config" 2>/dev/null; then local tmp; tmp=$(mktemp) grep -v '^\s*bypass_hook_trust\s*=' "$codex_config" > "$tmp" || true mv "$tmp" "$codex_config" fi # Step c: 逐条删 [hooks.state.":::0"] section # 匹配 key 中包含 hooks.json 路径的条目(pilot 写的),不动其他 path 的条目 local codex_hooks_json_path codex_hooks_json_path="$(cd "$HOME/.codex" 2>/dev/null && pwd)/hooks.json" if grep -q "$codex_hooks_json_path" "$codex_config" 2>/dev/null; then local tmp; tmp=$(mktemp) awk -v owned_path="$codex_hooks_json_path" ' /^\[hooks\.state\."/ { if (index($0, owned_path) > 0) { skip=1; next } } /^\[/ && !/^\[hooks\.state\."/ { skip=0 } skip { next } { print } ' "$codex_config" > "$tmp" mv "$tmp" "$codex_config" fi # Step d: 删 otel-codex-hook 相关的剩余行(legacy catch-all,不删 hooks.state section) if grep -q "otel-codex-hook" "$codex_config" 2>/dev/null; then local tmp; tmp=$(mktemp) grep -v "otel-codex-hook" "$codex_config" > "$tmp" || true mv "$tmp" "$codex_config" fi # Clean up codex_hooks = true if grep -q "codex_hooks" "$codex_config" 2>/dev/null; then local tmp; tmp=$(mktemp) grep -v '^\s*codex_hooks\s*=' "$codex_config" > "$tmp" || true mv "$tmp" "$codex_config" fi # Clean up multiple blank lines if [ -f "$codex_config" ]; then local tmp; tmp=$(mktemp) awk 'NF{blank=0} !NF{blank++} blank<=1' "$codex_config" > "$tmp" mv "$tmp" "$codex_config" fi msg " ✅ Codex hooks 已从 config.toml 清理" \ " ✅ Codex hooks cleaned from config.toml" fi fi local codex_otel_config="$HOME/.codex/otel-config.json" if [ -f "$codex_otel_config" ] && command -v node &>/dev/null; then node -e " const fs = require('fs'); try { const cfg = JSON.parse(fs.readFileSync(process.argv[1], 'utf-8')); delete cfg.log_enabled; delete cfg.log_dir; delete cfg.log_filename_format; fs.writeFileSync(process.argv[1], JSON.stringify(cfg, null, 2) + '\n'); } catch {} " "$codex_otel_config" 2>/dev/null || true fi if [ -d "$OTEL_CODEX_DIR" ]; then if [ "$PURGE" -eq 1 ]; then rm -rf "$OTEL_CODEX_DIR" msg " ✅ Codex 插件目录已完全删除 (--purge): $OTEL_CODEX_DIR" \ " ✅ Codex plugin directory fully removed (--purge): $OTEL_CODEX_DIR" else find "$OTEL_CODEX_DIR" -maxdepth 1 \ ! -name sessions \ ! -name "$(basename "$OTEL_CODEX_DIR")" \ -exec rm -rf {} + 2>/dev/null || true msg " ✅ Codex 插件文件已删除（sessions/ 已保留）" \ " ✅ Codex plugin files removed (sessions/ preserved)" fi fi } print_summary() { local action="$1" # install / upgrade local config_file="$DATA_DIR/config.json" echo "============================================================" local ver; ver=$(show_version_info "$PERMANENT_DIR") case "$action" in install) msg "✅ 安装完成！版本: $ver" "✅ Installation complete! Version: $ver" ;; upgrade) msg "✅ 升级完成！版本: $ver" "✅ Upgrade complete! Version: $ver" ;; esac echo "" msg "配置文件: $config_file" "Config file: $config_file" msg "数据目录: $DATA_DIR" "Data directory: $DATA_DIR" msg "Hook 目录: $DATA_DIR/hooks" "Hooks directory: $DATA_DIR/hooks" echo "" if [ -n "$SLS_ENDPOINT" ]; then msg "SLS 后端: $SLS_ENDPOINT" "SLS backend: $SLS_ENDPOINT" [ -n "$SLS_PROJECT" ] && msg " 项目: $SLS_PROJECT" " Project: $SLS_PROJECT" [ -n "$SLS_LOGSTORE" ] && msg " 日志库: $SLS_LOGSTORE" " Logstore: $SLS_LOGSTORE" echo "" fi msg "命令:" "Commands:" echo " loongsuite-pilot # 查看状态 / Status" echo " loongsuite-pilot info # 版本与配置 / Version & config" echo "============================================================" } # ============================================================ # CMD: install # ============================================================ cmd_install() { msg "==> 开始安装 $PACKAGE_NAME ..." \ "==> Installing $PACKAGE_NAME ..." echo "" validate_install_user check_deps # Migrate legacy layout if needed migrate_legacy_layout # Check if already installed local cur_ver; cur_ver=$(get_installed_version) if [ -n "$cur_ver" ]; then msg "⚠️ 检测到已安装版本 v${cur_ver}，将执行重新安装" \ "⚠️ Existing installation v${cur_ver} detected, re-installing" echo "" fi # Stop running service before re-install local pid_file="$DATA_DIR/loongsuite-pilot.pid" if [ -f "$pid_file" ]; then local old_pid old_pid=$(cat "$pid_file") if kill -0 "$old_pid" 2>/dev/null; then msg "==> 停止运行中的服务 (PID $old_pid)..." \ "==> Stopping running service (PID $old_pid)..." kill "$old_pid" 2>/dev/null || true local count=0 while kill -0 "$old_pid" 2>/dev/null && [ $count -lt 10 ]; do sleep 1 count=$((count + 1)) done if kill -0 "$old_pid" 2>/dev/null; then kill -9 "$old_pid" 2>/dev/null || true fi rm -f "$pid_file" msg " ✅ 已停止" " ✅ Stopped" echo "" else rm -f "$pid_file" fi fi trap 'rm -rf "${TMP_DIR:-}"' EXIT download_and_extract probe_agents select_agents prompt_user_id confirm_config_overwrite deploy_package "$INSTALL_SRC" write_config install_loongsuite_pilot_command msg "==> 启动服务..." "==> Starting service..." local _start_args="" if [ "$SYSTEM_SERVICE" -eq 1 ]; then _start_args="--system-service" fi if loongsuite-pilot start $_start_args; then sleep 2 local _status_out _status_out="$(loongsuite-pilot status 2>/dev/null || true)" if echo "$_status_out" | grep -q "is running"; then msg " ✅ 服务已启动" " ✅ Service started" else msg " ⚠️ 服务可能尚未就绪，请检查: loongsuite-pilot status" \ " ⚠️ Service may not be ready. Check: loongsuite-pilot status" fi else msg " ⚠️ 服务启动失败，请手动运行: loongsuite-pilot start" \ " ⚠️ Service failed to start, run manually: loongsuite-pilot start" fi echo "" print_summary "install" } # ============================================================ # CMD: upgrade # ============================================================ cmd_upgrade() { msg "==> 开始升级 $PACKAGE_NAME ..." \ "==> Upgrading $PACKAGE_NAME ..." echo "" validate_install_user # Migrate legacy layout if needed migrate_legacy_layout # Must have an existing installation local old_ver; old_ver=$(get_installed_version) if [ -z "$old_ver" ]; then msg "❌ 未检测到已安装的 loongsuite-pilot，请先执行 install" \ "❌ No existing installation found. Please run install first." exit 1 fi msg " 当前版本: ${old_ver:-unknown}" " Current version: ${old_ver:-unknown}" echo "" check_deps trap 'rm -rf "${TMP_DIR:-}"' EXIT download_and_extract local new_ver; new_ver=$(get_version_from_dir "$INSTALL_SRC") local new_commit; new_commit=$(get_commit_from_dir "$INSTALL_SRC") local old_commit; old_commit=$(get_commit_from_dir "$PERMANENT_DIR") if [ -n "$new_ver" ] && [ "$new_ver" = "$old_ver" ] && [ "$new_commit" = "$old_commit" ]; then msg "✅ 已是最新版本 v${new_ver} (${new_commit})，无需升级" \ "✅ Already at latest version v${new_ver} (${new_commit}), nothing to do" exit 0 fi msg " 新版本: ${new_ver:-unknown} (${new_commit:-unknown})" \ " New version: ${new_ver:-unknown} (${new_commit:-unknown})" echo "" # Stop the running service msg "==> 停止服务..." "==> Stopping service..." if command -v loongsuite-pilot &>/dev/null; then loongsuite-pilot stop 2>/dev/null || true elif [ -f "$HOME/.local/bin/loongsuite-pilot" ]; then "$HOME/.local/bin/loongsuite-pilot" stop 2>/dev/null || true fi echo "" # Deploy new version to versions/_/ # Old version stays untouched; deploy_package writes current/previous pointers deploy_package "$INSTALL_SRC" install_loongsuite_pilot_command # Start the new version msg "==> 启动新版本..." "==> Starting new version..." if loongsuite-pilot start; then sleep 2 local _status_out _status_out="$(loongsuite-pilot status 2>/dev/null || true)" if echo "$_status_out" | grep -q "is running"; then msg " ✅ 新版本启动成功" " ✅ New version started successfully" echo "" # GC: remove old versions beyond current + previous gc_old_versions print_summary "upgrade" return 0 fi fi # --- Rollback via version pointer --- echo "" msg "⚠️ 新版本启动失败，正在回滚..." \ "⚠️ New version failed to start, rolling back..." loongsuite-pilot stop 2>/dev/null || true if command -v loongsuite-pilot &>/dev/null; then loongsuite-pilot rollback 2>/dev/null || true else "$HOME/.local/bin/loongsuite-pilot" rollback 2>/dev/null || true fi msg "❌ 升级失败，已回滚到 v${old_ver:-unknown}" \ "❌ Upgrade failed, rolled back to v${old_ver:-unknown}" msg " 请检查日志: loongsuite-pilot log" " Check logs: loongsuite-pilot log" exit 1 } # ============================================================ # GC: remove old version directories beyond current + previous # ============================================================ gc_old_versions() { local cache_dir="$HOME/.loongsuite-pilot" local versions_dir="$cache_dir/versions" local current_file="$cache_dir/current" local previous_file="$cache_dir/previous" [ -d "$versions_dir" ] || return 0 local keep_current="" keep_previous="" if [ -f "$current_file" ]; then keep_current=$(cat "$current_file" 2>/dev/null | tr -d '[:space:]') fi if [ -f "$previous_file" ]; then keep_previous=$(cat "$previous_file" 2>/dev/null | tr -d '[:space:]') fi for d in "$versions_dir"/*/; do [ -d "$d" ] || continue local name name=$(basename "$d") if [ "$name" = "$keep_current" ] || [ "$name" = "$keep_previous" ]; then continue fi rm -rf "$d" done } # ============================================================ # Remove hook entries injected into tool config files # ============================================================ remove_hook_configs() { local HOOK_MARKER=".loongsuite-pilot" local configs=( "$HOME/.cursor/hooks.json" "$HOME/.qoder/settings.json" "$HOME/.qoderwork/settings.json" "$HOME/.claude/settings.json" "$HOME/.codex/hooks.json" ) for cfg in "${configs[@]}"; do [ -f "$cfg" ] || continue local short="${cfg/#$HOME/\~}" local ok=0 if command -v node &>/dev/null; then node -e " const fs = require('fs'); const cfg = process.argv[1]; const marker = process.argv[2]; try { const data = JSON.parse(fs.readFileSync(cfg, 'utf-8')); const hooks = data.hooks; if (!hooks || typeof hooks !== 'object') process.exit(0); let changed = false; for (const [event, entries] of Object.entries(hooks)) { if (!Array.isArray(entries)) continue; const filtered = entries.filter(e => { const cmd = e.command || ''; const nested = Array.isArray(e.hooks) ? e.hooks : []; const hasMarker = cmd.includes(marker) || nested.some(h => (h.command || '').includes(marker)); if (hasMarker) changed = true; return !hasMarker; }); if (filtered.length === 0) { delete hooks[event]; changed = true; } else hooks[event] = filtered; } if (changed) { fs.writeFileSync(cfg, JSON.stringify(data, null, 2) + '\n', 'utf-8'); process.stdout.write('cleaned'); } else { process.stdout.write('skip'); } } catch(e) { process.stderr.write(e.message); process.exit(1); } " "$cfg" "$HOOK_MARKER" && ok=1 fi if [ "$ok" -eq 1 ]; then msg " ✅ 已清理: $short" " ✅ Cleaned: $short" else msg " ⚠️ 跳过: $short (需手动清理)" " ⚠️ Skipped: $short (manual cleanup needed)" fi done } # ============================================================ # CMD: uninstall # ============================================================ cmd_uninstall() { msg "🗑️ 开始卸载 $PACKAGE_NAME ..." \ "🗑️ Uninstalling $PACKAGE_NAME ..." echo "" # Stop service (also removes autostart) msg "==> 停止服务..." "==> Stopping service..." if command -v loongsuite-pilot &>/dev/null; then loongsuite-pilot stop 2>/dev/null || true elif [ -f "$HOME/.local/bin/loongsuite-pilot" ]; then "$HOME/.local/bin/loongsuite-pilot" stop 2>/dev/null || true else local pid_file="$DATA_DIR/loongsuite-pilot.pid" if [ -f "$pid_file" ]; then local pid; pid=$(cat "$pid_file") kill "$pid" 2>/dev/null || true sleep 2 kill -9 "$pid" 2>/dev/null || true rm -f "$pid_file" fi # Manual autostart cleanup when loongsuite-pilot is unavailable case "$(uname -s)" in Darwin) local _plist="$HOME/Library/LaunchAgents/com.loongsuite-pilot.plist" local _uplist="$HOME/Library/LaunchAgents/com.loongsuite-pilot.updater.plist" for f in "$_uplist" "$_plist"; do if [ -f "$f" ]; then launchctl unload -w "$f" 2>/dev/null || true rm -f "$f" fi done ;; Linux) local _run_user _run_user="$(whoami)" # Clean up user-level systemd units local _user_unit_dir="$HOME/.config/systemd/user" if [ -f "$_user_unit_dir/loongsuite-pilot.service" ]; then systemctl --user disable --now loongsuite-pilot.service &>/dev/null || true systemctl --user disable --now loongsuite-pilot-updater.service &>/dev/null || true rm -f "$_user_unit_dir/loongsuite-pilot.service" rm -f "$_user_unit_dir/loongsuite-pilot-updater.service" systemctl --user daemon-reload &>/dev/null || true fi # Clean up system-level systemd units local _sys_unit="/etc/systemd/system/loongsuite-pilot-${_run_user}.service" local _sys_uunit="/etc/systemd/system/loongsuite-pilot-updater-${_run_user}.service" for f in "$_sys_uunit" "$_sys_unit"; do if [ -f "$f" ]; then sudo systemctl disable --now "$(basename "$f")" &>/dev/null || true sudo rm -f "$f" fi done sudo systemctl daemon-reload &>/dev/null || true # Clean up init.d scripts local _initd="/etc/init.d/loongsuite-pilot-${_run_user}" local _initd_u="/etc/init.d/loongsuite-pilot-updater-${_run_user}" for f in "$_initd_u" "$_initd"; do if [ -f "$f" ]; then sudo "$f" stop &>/dev/null || true local _name; _name=$(basename "$f") if command -v chkconfig &>/dev/null; then sudo chkconfig --del "$_name" &>/dev/null || true elif command -v update-rc.d &>/dev/null; then sudo update-rc.d "$_name" remove &>/dev/null || true; fi sudo rm -f "$f" fi done ;; esac fi msg " ✅ 服务已停止" " ✅ Service stopped" echo "" # Remove package directory msg "==> 删除安装目录..." "==> Removing installation..." rm -rf "$HOME/.loongsuite-pilot" msg " ✅ 已删除 $HOME/.loongsuite-pilot" \ " ✅ Removed $HOME/.loongsuite-pilot" # Remove loongsuite-pilot command msg "==> 删除 loongsuite-pilot 命令..." "==> Removing loongsuite-pilot command..." rm -f "$HOME/.local/bin/loongsuite-pilot" rm -f /usr/local/bin/loongsuite-pilot 2>/dev/null || true msg " ✅ loongsuite-pilot 命令已删除" " ✅ loongsuite-pilot command removed" echo "" # Remove hook entries from tool configs msg "==> 清理 hook 配置..." "==> Cleaning up hook configs..." remove_hook_configs echo "" # Remove OTel Claude plugin msg "==> 清理 Claude/Codex 插件..." "==> Cleaning up Claude/Codex plugins..." remove_otel_plugin echo "" # Data directory if [ "$PURGE" -eq 1 ]; then msg "==> 删除数据目录 (--purge)..." "==> Removing data directory (--purge)..." rm -rf "$DATA_DIR" msg " ✅ 已删除 $DATA_DIR" " ✅ Removed $DATA_DIR" else msg "📁 数据目录已保留: $DATA_DIR" \ "📁 Data directory preserved: $DATA_DIR" msg " (包含配置和日志，如需彻底删除请加 --purge)" \ " (contains config and logs, add --purge to remove)" fi echo "" echo "============================================================" msg "✅ 卸载完成！" "✅ Uninstallation complete!" echo "============================================================" } # ============================================================ # Main dispatcher # ============================================================ case "$COMMAND" in install) cmd_install ;; upgrade) cmd_upgrade ;; uninstall) cmd_uninstall ;; *) echo "Usage: $0 {install|upgrade|uninstall} [options]" exit 1 ;; esac