agent-cli.sh

#!/usr/bin/env bash
set -uo pipefail

DEFAULT_PACKAGES=(@qoder-ai/qodercli @qodercn-ai/qoderclicn @openai/codex @anthropic-ai/claude-code @mimo-ai/cli)

CONFIG_DIR="$HOME/.config/agent-cli"
PACKAGES_FILE="$CONFIG_DIR/packages.conf"

packages=()
selected=()

CHECK_ONLY=0
NONINTERACTIVE=0
ADD_PKGS=()
DEL_PKGS=()
UNINSTALL_PKGS=()

show_help() {
  cat <<EOF
用法: agent-cli [选项]

选项:
  -c, --check              仅检查最新版本，不安装（非交互）
  -y, --yes               非交互，直接检查并更新所有包（不进入菜单）
  --add PKG               非交互，追加一个包并持久化（可多次使用）
  -a PKG                  同上
  --delete PKG            非交互，从列表中移除一个包（不卸载，可多次使用）
  -d PKG                  同上
  --uninstall PKG         非交互，从列表移除并 npm uninstall（可多次使用）
  -u PKG                  同上
  -h, --help              显示此帮助

无参数: 进入交互式菜单（空格勾选，↑/↓ 移动选择，←/→ 调整顺序，
          a 追加，d 删除(移出列表)，u 卸载(移出列表并 npm uninstall)，
          c 仅检查，Enter 更新）
EOF
}

while [[ $# -gt 0 ]]; do
  case "$1" in
    -c|--check) CHECK_ONLY=1 ;;
    -y|--yes) NONINTERACTIVE=1 ;;
    --add) shift; ADD_PKGS+=("${1:-}") ;;
    -a) shift; ADD_PKGS+=("${1:-}") ;;
    --delete) shift; DEL_PKGS+=("${1:-}") ;;
    -d) shift; DEL_PKGS+=("${1:-}") ;;
    --uninstall) shift; UNINSTALL_PKGS+=("${1:-}") ;;
    -u) shift; UNINSTALL_PKGS+=("${1:-}") ;;
    -h|--help) show_help; exit 0 ;;
    *) echo "未知参数: $1"; echo "使用 -h 查看帮助"; exit 1 ;;
  esac
  shift
done

load_packages() {
  packages=()
  selected=()
  mkdir -p "$CONFIG_DIR"
  if [ ! -f "$PACKAGES_FILE" ]; then
    for p in "${DEFAULT_PACKAGES[@]}"; do
      echo "$p" >> "$PACKAGES_FILE"
    done
  fi
  if [ -f "$PACKAGES_FILE" ]; then
    while IFS= read -r line; do
      [ -z "$line" ] && continue
      packages+=("$line"); selected+=(1)
    done < "$PACKAGES_FILE"
  fi
}

save_packages() {
  mkdir -p "$CONFIG_DIR"
  : > "$PACKAGES_FILE"
  for p in "${packages[@]:-}"; do
    [ -z "$p" ] && continue
    echo "$p" >> "$PACKAGES_FILE"
  done
}

add_pkg() {
  local p="$1" q dup=0
  [ -z "$p" ] && return
  for q in "${packages[@]}"; do
    [ "$q" = "$p" ] && dup=1 && break
  done
  if [ "$dup" = "0" ]; then
    packages+=("$p"); selected+=(1)
    save_packages
  fi
}

delete_pkg_name() {
  local p="$1" i newpkgs=() newsel=() found=0
  for i in "${!packages[@]}"; do
    if [ "${packages[i]}" = "$p" ]; then found=1; continue; fi
    newpkgs+=("${packages[i]}"); newsel+=("${selected[i]}")
  done
  if [ "$found" = "0" ]; then
    echo "未找到包: $p"
    return
  fi
  packages=("${newpkgs[@]}"); selected=("${newsel[@]}")
  save_packages
}

uninstall_pkg() {
  local p="$1"
  for q in "${packages[@]}"; do
    [ "$q" = "$p" ] && break
    q=""
  done
  if [ -z "$q" ]; then
    echo "未找到包: $p"
    return
  fi
  delete_pkg_name "$p"
  echo "[卸载] $p: 正在执行 npm uninstall -g ..."
  npm uninstall -g "$p" </dev/null || echo "[失败] $p 卸载失败"
}

get_installed_version() {
  npm list -g "$1" --depth=0 2>/dev/null | grep "$1@" | head -1 | sed 's/.*@//' || echo ""
}

get_latest_version() {
  npm view "$1" version 2>/dev/null || echo ""
}

check_only() {
  stty -echo 2>/dev/null || true
  trap 'stty echo 2>/dev/null || true' RETURN EXIT INT TERM
  echo "检查结果："
  echo
  for i in "${!packages[@]}"; do
    [ "${selected[i]:-0}" = "1" ] || continue
    pkg="${packages[i]}"
    installed=$(get_installed_version "$pkg")
    latest=$(get_latest_version "$pkg")
    if [ -z "$latest" ]; then
      echo "[ERROR] $pkg: 无法获取最新版本"
    elif [ -z "$installed" ]; then
      echo "[未安装] $pkg: 最新 $latest"
    elif [ "$installed" != "$latest" ]; then
      echo "[可更新] $pkg: $installed -> $latest"
    else
      echo "[最新]   $pkg: $installed"
     fi
  done
  while IFS= read -t 0 -rsn1 -d '' _ 2>/dev/null; do :; done
}

run() {
  stty -echo 2>/dev/null || true
  trap 'stty echo 2>/dev/null || true' RETURN EXIT INT TERM
  echo "开始检查/安装所选包..."
  echo
  for i in "${!packages[@]}"; do
    [ "${selected[i]:-0}" = "1" ] || continue
    pkg="${packages[i]}"
    installed=$(get_installed_version "$pkg")
    latest=$(get_latest_version "$pkg")

    if [ -z "$latest" ]; then
      echo "[ERROR] $pkg: 无法获取最新版本"
      continue
    fi

    if [ -z "$installed" ]; then
      echo "[安装] $pkg: 未安装, 正在安装 $latest ..."
      npm install -g "$pkg@latest" </dev/null || echo "[失败] $pkg 安装失败"
    elif [ "$installed" != "$latest" ]; then
      echo "[更新] $pkg: $installed -> $latest"
      npm install -g "$pkg@latest" </dev/null || echo "[失败] $pkg 更新失败"
    else
      echo "[最新] $pkg: $installed"
    fi
  done
  while IFS= read -t 0 -rsn1 -d '' _ 2>/dev/null; do :; done
}

read_line() {
  local prompt="$1" input="" ch
  echo -n "$prompt"
  while IFS= read -rsn1 ch; do
    if [ -z "$ch" ]; then
      echo
      READ_LINE_RESULT="$input"
      return 0
    fi
    if [ "$ch" = $'\e' ]; then
      echo
      echo "（已取消）"
      READ_LINE_RESULT=""
      return 1
    fi
    if [ "$ch" = $'\177' ] || [ "$ch" = $'\b' ]; then
      if [ -n "$input" ]; then
        input="${input%?}"
        printf '\b \b'
      fi
      continue
    fi
    input+="$ch"
    printf '%s' "$ch"
  done
  READ_LINE_RESULT="$input"
  return 0
}

checklist() {
  local cursor=0 key rest p newpkgs
  while true; do
    clear
    echo "选择要检查/安装的包（空格勾选，↑/↓ 移动，←/→ 调整顺序，a 追加，d 删除(移出列表)，u 卸载(移出列表并 npm uninstall)，c 仅检查，Enter 更新）:"
    echo
    for i in "${!packages[@]}"; do
       if [ "${selected[i]:-0}" = "1" ]; then mark="x"; else mark=" "; fi
       if [ "$i" = "$cursor" ]; then
         echo " > [$mark] ${packages[i]}"
       else
         echo "   [$mark] ${packages[i]}"
       fi
    done
    echo
    IFS= read -rsn1 key
    case "$key" in
      $'\e')
        rest=
        IFS= read -rsn2 -t 1 rest || true
        key+=$rest
        case "$key" in
          $'\e[A'|$'\eOA')
            if (( cursor > 0 )); then cursor=$((cursor - 1)); fi ;;
          $'\e[B'|$'\eOB')
            if (( cursor < ${#packages[@]} - 1 )); then cursor=$((cursor + 1)); fi ;;
          $'\e[D'|$'\eOD')
            if (( cursor > 0 )); then
              j=$((cursor - 1))
              tmp="${packages[cursor]}"; packages[cursor]="${packages[j]}"; packages[j]="$tmp"
              tmp="${selected[cursor]}"; selected[cursor]="${selected[j]}"; selected[j]="$tmp"
              cursor=$((cursor - 1))
              save_packages
            fi ;;
          $'\e[C'|$'\eOC')
            if (( cursor < ${#packages[@]} - 1 )); then
              j=$((cursor + 1))
              tmp="${packages[cursor]}"; packages[cursor]="${packages[j]}"; packages[j]="$tmp"
              tmp="${selected[cursor]}"; selected[cursor]="${selected[j]}"; selected[j]="$tmp"
              cursor=$((cursor + 1))
              save_packages
            fi ;;
        esac ;;
      ' ')
        if [ "${selected[cursor]:-0}" = "1" ]; then selected[cursor]=0; else selected[cursor]=1; fi ;;
      a|A)
        echo
        if read_line "输入要追加的包名（多个用空格分隔，Esc 取消）: "; then
          read -ra newpkgs <<< "$READ_LINE_RESULT"
          for p in "${newpkgs[@]}"; do
            [ -z "$p" ] && continue
            add_pkg "$p"
          done
        fi
        continue ;;
      d|D)
        p="${packages[cursor]}"
        delete_pkg_name "$p"
        if (( cursor >= ${#packages[@]} )); then cursor=$(( ${#packages[@]} - 1 )); fi
        continue ;;
      u|U)
        p="${packages[cursor]}"
        uninstall_pkg "$p"
        if (( cursor >= ${#packages[@]} )); then cursor=$(( ${#packages[@]} - 1 )); fi
        continue ;;
      c|C)
        check_only
        exit 0 ;;
      '')
        break ;;
    esac
  done
}

load_packages

if [ "${#ADD_PKGS[@]}" -gt 0 ] || [ "${#DEL_PKGS[@]}" -gt 0 ] || [ "${#UNINSTALL_PKGS[@]}" -gt 0 ]; then
  for p in "${ADD_PKGS[@]:-}"; do add_pkg "$p"; done
  for p in "${DEL_PKGS[@]:-}"; do delete_pkg_name "$p"; done
  for p in "${UNINSTALL_PKGS[@]:-}"; do uninstall_pkg "$p"; done
  echo "已更新包列表（持久化至 $PACKAGES_FILE）："
  echo "  当前包列表: ${packages[*]:-无}"
  [ "$NONINTERACTIVE" = "1" ] || exit 0
fi

if [ "$CHECK_ONLY" = "1" ]; then
  check_only
elif [ "$NONINTERACTIVE" = "1" ]; then
  run
else
  checklist
  run
fi