feat(role): 新增閒置小睡整理

2026-07-28 14:39:26 +08:00
parent bb48a77dd4
commit 945c310551
4 changed files with 180 additions and 24 deletions
@@ -4,9 +4,10 @@
 #       先檢查是否有 AI 正在運行，沒有才進入睡眠並整理記憶：
 #       NREM 鞏固（分類／去噪／去重／合併／優先度）→ REM 整合（跨記憶
 #       連結／抽象化／提取線索）→ 壓縮歸檔 → 日常與其他依使用頻率與優先度遺忘。
-#       另提供 --catchup（cron 未執行時的補跑）、--force（手動立即整理）、
-#       --install-cron／--remove-cron（排程安裝與移除）、--status（狀態）。
-# 更新時間：2026/07/28 12:59:41
+#       另提供 --nap（CLI 閒置時的小睡整理）、--catchup（cron 未執行時的補跑）、
+#       --force（手動立即整理）、--install-cron／--remove-cron（排程安裝與移除）、
+#       --status（狀態）。
+# 更新時間：2026/07/28 14:36:00
 # 相依：bash、node、任一 headless CLI、crontab（僅排程安裝需要）、
 #       同目錄的 role_lib.sh 與 memory.js。
 # 退出碼：0 成功或無事可做；1 參數錯誤或整理失敗（cron 觸發時不影響使用者）。
@@ -18,6 +19,7 @@ SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
 . "${SCRIPT_DIR}/role_lib.sh"

 CRON_MARKER="# jsc-role-sleep"
+NAP_CRON_MARKER="# jsc-role-nap"
 SLEEP_TIMEOUT="${ROLE_SLEEP_TIMEOUT:-180}"
 SLEEP_OUTPUT_LIMIT="${ROLE_SLEEP_OUTPUT_LIMIT:-8000}"

@@ -27,6 +29,7 @@ usage() {
 用法：role_sleep.sh <模式>

  --run           cron 觸發：在睡眠時段內且無 AI 運行時整理記憶
+  --nap           小睡觸發：CLI 閒置一段時間且 inbox 達門檻時整理記憶
  --catchup       補跑：cron 未執行時，由 SessionStart hook 於背景呼叫
  --force         立即整理一次（忽略時段與 AI 運行檢查）
  --install-cron  安裝／更新睡眠排程（每小時檢查一次）
@@ -42,6 +45,50 @@ require_role() {
  [ -f "$(role_file "$ROLE")" ] || { role_log "WRN" "找不到角色定義檔：$(role_file "$ROLE")"; exit 1; }
 }

+positive_int_or_default() {
+  # 讀取正整數環境變數；未設定或不合法時使用預設值
+  local value="$1" fallback="$2" min="${3:-1}" max="${4:-}"
+  case "$value" in
+    ""|*[!0-9]*) printf '%s' "$fallback"; return 0 ;;
+  esac
+  [ "$value" -lt "$min" ] && { printf '%s' "$fallback"; return 0; }
+  if [ -n "$max" ] && [ "$value" -gt "$max" ]; then
+    printf '%s' "$fallback"
+    return 0
+  fi
+  printf '%s' "$value"
+}
+
+nap_enabled() {
+  # 小睡預設啟用；設 ROLE_NAP_ENABLED=0/false/no 可關閉
+  case "${ROLE_NAP_ENABLED:-1}" in
+    0|false|no) return 1 ;;
+  esac
+  return 0
+}
+
+nap_idle_minutes() { positive_int_or_default "${ROLE_NAP_IDLE_MINUTES:-}" 45 1; }
+nap_min_inbox() { positive_int_or_default "${ROLE_NAP_MIN_INBOX:-}" 3 1; }
+nap_interval_minutes() { positive_int_or_default "${ROLE_NAP_INTERVAL_MINUTES:-}" 10 1 59; }
+
+role_sleep_child_running() {
+  # 小睡只避開整理用的 headless 子 CLI；互動式 CLI 閒置時仍可小睡
+  local pid cmd self="$$"
+  for pid in $(pgrep -f '(^|/)(claude|codex|agy|opencode|copilot)([[:space:]]|$)' 2>/dev/null); do
+    if [ "$pid" = "$self" ] || [ "$pid" = "$PPID" ]; then
+      continue
+    fi
+    cmd="$(ps -o args= -p "$pid" 2>/dev/null)"
+    case "$cmd" in
+      *role_sleep.sh*|*role_capture.sh*|*role_load.sh*|*pgrep*) continue ;;
+    esac
+    if printf '%s' "$cmd" | grep -Eq '(^|/)(codex[[:space:]]+exec|claude([[:space:]].*)?[[:space:]]+-p|agy([[:space:]].*)?[[:space:]]+-p|opencode[[:space:]]+run|copilot([[:space:]].*)?[[:space:]]+-p)'; then
+      return 0
+    fi
+  done
+  return 1
+}
+
 # ------------------------------------------------------------------------------
 # 整理主流程
 # ------------------------------------------------------------------------------
@@ -170,20 +217,32 @@ cron_path() {
  printf '%s' "$value"
 }

-cron_line() {
-  # 組出 crontab 條目：睡眠時段內每小時檢查一次
+cron_env_prefix() {
+  # 組出 cron 需要的精簡環境變數
  local env_prefix="PATH=$(cron_quote "$(cron_path)")"
  local var
-  for var in ROLE_ENABLED ROLE_NAME ROLE_HOME ROLE_MEMORY_HOME ROLE_CLI ROLE_MODEL ROLE_SLEEP_START ROLE_SLEEP_END ROLE_SCOPE; do
+  for var in ROLE_ENABLED ROLE_NAME ROLE_HOME ROLE_MEMORY_HOME ROLE_CLI ROLE_MODEL ROLE_SLEEP_START ROLE_SLEEP_END ROLE_SCOPE ROLE_NAP_ENABLED ROLE_NAP_IDLE_MINUTES ROLE_NAP_MIN_INBOX ROLE_NAP_INTERVAL_MINUTES; do
    if [ -n "${!var:-}" ]; then
      env_prefix="${env_prefix} ${var}=$(cron_quote "${!var}")"
    fi
  done
+  printf '%s' "$env_prefix"
+}
+
+cron_line() {
+  # 組出 crontab 條目：睡眠時段內每小時檢查一次
  printf '0 %s * * * %s %s --run >> %s 2>&1 %s\n' \
-    "$(cron_hours)" "$env_prefix" "$(cron_quote "${SCRIPT_DIR}/role_sleep.sh")" \
+    "$(cron_hours)" "$(cron_env_prefix)" "$(cron_quote "${SCRIPT_DIR}/role_sleep.sh")" \
    "$(cron_quote "$(sleep_log_path)")" "$CRON_MARKER"
 }

+nap_cron_line() {
+  # 組出小睡 crontab 條目：全天依間隔檢查閒置狀態
+  printf '*/%s * * * * %s %s --nap >> %s 2>&1 %s\n' \
+    "$(nap_interval_minutes)" "$(cron_env_prefix)" "$(cron_quote "${SCRIPT_DIR}/role_sleep.sh")" \
+    "$(cron_quote "$(sleep_log_path)")" "$NAP_CRON_MARKER"
+}
+
 cron_hours() {
  # 依睡眠時段換算 cron 小時欄位（每小時檢查一次，讓 AI 運行中的情況能在下個小時重試）
  local start end hour hours=""
@@ -209,10 +268,19 @@ install_cron() {
  command -v crontab >/dev/null 2>&1 || { role_log "ERR" "找不到 crontab，無法安裝排程"; return 1; }
  mkdir -p "$(role_home)" 2>/dev/null
  local current new
-  current="$(crontab -l 2>/dev/null | grep -v -F "$CRON_MARKER")"
-  new="$(printf '%s\n%s' "$current" "$(cron_line)" | sed '/^$/d')"
+  current="$(crontab -l 2>/dev/null | grep -v -F "$CRON_MARKER" | grep -v -F "$NAP_CRON_MARKER")"
+  if nap_enabled; then
+    new="$(printf '%s\n%s\n%s' "$current" "$(cron_line)" "$(nap_cron_line)" | sed '/^$/d')"
+  else
+    new="$(printf '%s\n%s' "$current" "$(cron_line)" | sed '/^$/d')"
+  fi
  printf '%s\n' "$new" | crontab - || { role_log "ERR" "寫入 crontab 失敗"; return 1; }
  role_log "INF" "已安裝睡眠排程：每日 $(cron_hours) 時整點檢查（角色 ${ROLE}，時段 $(role_sleep_start)–$(role_sleep_end)）"
+  if nap_enabled; then
+    role_log "INF" "已安裝小睡排程：每 $(nap_interval_minutes) 分鐘檢查，閒置滿 $(nap_idle_minutes) 分鐘且 inbox ≥ $(nap_min_inbox) 則時整理"
+  else
+    role_log "INF" "小睡排程已停用（ROLE_NAP_ENABLED=${ROLE_NAP_ENABLED:-1}）"
+  fi
  role_log "INF" "排程輸出：$(sleep_log_path)"
  if ! pgrep -x cron >/dev/null 2>&1 && ! pgrep -x crond >/dev/null 2>&1; then
    role_log "WRN" "系統 cron 服務未執行（WSL 常見），排程不會觸發；SessionStart 的背景補跑仍會運作"
@@ -223,15 +291,16 @@ install_cron() {
 remove_cron() {
  # 移除本 skill 安裝的排程條目
  command -v crontab >/dev/null 2>&1 || { role_log "ERR" "找不到 crontab"; return 1; }
-  crontab -l 2>/dev/null | grep -v -F "$CRON_MARKER" | crontab -
+  crontab -l 2>/dev/null | grep -v -F "$CRON_MARKER" | grep -v -F "$NAP_CRON_MARKER" | crontab -
  role_log "INF" "已移除睡眠排程"
  return 0
 }

 show_status() {
  # 以表格輸出目前角色與記憶狀態（供 skill 的 --status 使用）
-  local cron_state="未安裝" cron_service="未執行" window="否"
+  local cron_state="未安裝" nap_state="未安裝" cron_service="未執行" window="否"
  crontab -l 2>/dev/null | grep -qF "$CRON_MARKER" && cron_state="已安裝"
+  crontab -l 2>/dev/null | grep -qF "$NAP_CRON_MARKER" && nap_state="已安裝"
  { pgrep -x cron >/dev/null 2>&1 || pgrep -x crond >/dev/null 2>&1; } && cron_service="執行中"
  role_in_sleep_window && window="是"
  printf '| 項目 | 值 |\n| --- | --- |\n'
@@ -240,6 +309,9 @@ show_status() {
  printf '| 睡眠時段 | %s–%s |\n' "$(role_sleep_start)" "$(role_sleep_end)"
  printf '| 目前是否睡眠中 | %s |\n' "$window"
  printf '| cron 排程 | %s |\n' "$cron_state"
+  printf '| 小睡排程 | %s |\n' "$nap_state"
+  printf '| 小睡啟用 | %s |\n' "$(nap_enabled && printf '是' || printf '否')"
+  printf '| 小睡條件 | 閒置 ≥ %s 分鐘，待整理 ≥ %s 則，每 %s 分鐘檢查 |\n' "$(nap_idle_minutes)" "$(nap_min_inbox)" "$(nap_interval_minutes)"
  printf '| cron 服務 | %s |\n' "$cron_service"
  printf '| 摘要 CLI | %s |\n' "$(role_select_cli 2>/dev/null || printf '找不到可用 CLI')"
  printf '\n'
@@ -266,6 +338,23 @@ case "$MODE" in
    fi
    sleep_cycle "cron"
    ;;
+  --nap)
+    role_enabled || exit 0
+    require_role
+    if ! nap_enabled; then
+      role_log "DBG" "小睡已停用（ROLE_NAP_ENABLED=${ROLE_NAP_ENABLED:-1}），略過"
+      exit 0
+    fi
+    if role_sleep_child_running; then
+      role_log "INF" "偵測到整理用 headless CLI 正在運行，本次小睡略過"
+      exit 0
+    fi
+    if [ "$(node "${SCRIPT_DIR}/memory.js" need-nap --role "$ROLE" --idle-minutes "$(nap_idle_minutes)" --min-inbox "$(nap_min_inbox)" 2>/dev/null)" != "yes" ]; then
+      role_log "DBG" "尚未達小睡條件（閒置滿 $(nap_idle_minutes) 分鐘且 inbox ≥ $(nap_min_inbox) 則），略過"
+      exit 0
+    fi
+    sleep_cycle "小睡"
+    ;;
  --catchup)
    role_enabled || exit 0
    require_role