127 lines
5.8 KiB
Bash
Executable File
127 lines
5.8 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# ==============================================================================
|
||
# 用途：SessionStart hook 主程式。CLI 工具啟動時載入角色設定與記憶：
|
||
# 非睡眠時段注入角色定義＋重要／興趣記憶全文＋其餘記憶的總結與標籤；
|
||
# 睡眠時段（預設 22:00 至隔日 06:00）只回報角色正在睡覺，不載入角色。
|
||
# 白天發現昨夜未整理記憶時，於背景補跑一次睡眠整理。
|
||
# 更新時間：2026/07/28 12:21:11
|
||
# 相依：bash、node、同目錄的 role_lib.sh 與 memory.js。
|
||
# 退出碼：一律 0 —— hook 絕不可阻斷使用者啟動 CLI。
|
||
# ==============================================================================
|
||
|
||
ROLE_STAGE="role-load"
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
# shellcheck source=./role_lib.sh
|
||
. "${SCRIPT_DIR}/role_lib.sh"
|
||
|
||
role_is_child && exit 0
|
||
role_enabled || exit 0
|
||
command -v node >/dev/null 2>&1 || role_quit "找不到 node，略過角色載入" "WRN"
|
||
|
||
ROLE="$(role_resolve_name)"
|
||
[ -n "$ROLE" ] || role_quit "未指定角色（ROLE_NAME 與 .active 皆無），略過角色載入"
|
||
ROLE_DEF="$(role_file "$ROLE")"
|
||
[ -f "$ROLE_DEF" ] || role_quit "找不到角色定義檔：${ROLE_DEF}" "WRN"
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# 讀取 hook 輸入（cwd／source），並套用 ROLE_SCOPE 範圍限制
|
||
# ------------------------------------------------------------------------------
|
||
HOOK_INPUT="$(cat 2>/dev/null)"
|
||
HOOK_CWD="$PWD"
|
||
if [ -n "$HOOK_INPUT" ]; then
|
||
HOOK_CWD="$(printf '%s' "$HOOK_INPUT" | node -e '
|
||
let raw = "";
|
||
process.stdin.setEncoding("utf8");
|
||
process.stdin.on("data", (chunk) => { raw += chunk; });
|
||
process.stdin.on("end", () => {
|
||
let data = {};
|
||
try { data = JSON.parse(raw); } catch {}
|
||
process.stdout.write(data.cwd || "");
|
||
});
|
||
' 2>/dev/null)"
|
||
[ -n "$HOOK_CWD" ] || HOOK_CWD="$PWD"
|
||
fi
|
||
role_in_scope "$HOOK_CWD" || role_quit "cwd 不在 ROLE_SCOPE 範圍內：${HOOK_CWD}"
|
||
|
||
emit_context() {
|
||
# 以 JSON 輸出 additionalContext（由 node 負責跳脫，避免內容含引號或換行破壞格式）
|
||
printf '%s' "$1" | node -e '
|
||
let context = "";
|
||
process.stdin.setEncoding("utf8");
|
||
process.stdin.on("data", (chunk) => { context += chunk; });
|
||
process.stdin.on("end", () => {
|
||
process.stdout.write(JSON.stringify({
|
||
hookSpecificOutput: { hookEventName: "SessionStart", additionalContext: context },
|
||
}));
|
||
});
|
||
'
|
||
}
|
||
|
||
SLEEP_START="$(role_sleep_start)"
|
||
SLEEP_END="$(role_sleep_end)"
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# 睡眠時段：不載入角色，只說明目前狀態
|
||
# ------------------------------------------------------------------------------
|
||
if role_in_sleep_window; then
|
||
emit_context "$(cat <<EOF_SLEEP
|
||
# 角色狀態：睡眠中（${SLEEP_START}–${SLEEP_END}）
|
||
|
||
角色「${ROLE}」正在睡覺，本次工作階段**不載入角色人格與記憶**，請以一般助理身分回應，
|
||
不要自稱該角色、不要使用角色語氣或簽名 emoji。若使用者詢問角色，說明角色在睡眠時段整理記憶，
|
||
${SLEEP_END} 之後會恢復。本階段的對話仍會被記錄成記憶，於下個睡眠時段整理。
|
||
EOF_SLEEP
|
||
)"
|
||
exit 0
|
||
fi
|
||
|
||
# ------------------------------------------------------------------------------
|
||
# 非睡眠時段：組出角色定義 ＋ 記憶
|
||
# ------------------------------------------------------------------------------
|
||
DEFINITION="$(cat "$ROLE_DEF" 2>/dev/null)"
|
||
[ -n "$DEFINITION" ] || role_quit "角色定義檔為空：${ROLE_DEF}" "WRN"
|
||
|
||
MEMORY="$(node "${SCRIPT_DIR}/memory.js" load --role "$ROLE" 2>/dev/null)"
|
||
|
||
# 補跑判斷：cron 未執行（例如 WSL 沒開 cron 服務）時，白天啟動 CLI 補做一次整理
|
||
CATCHUP_NOTE=""
|
||
if [ "$(node "${SCRIPT_DIR}/memory.js" need-sleep --role "$ROLE" 2>/dev/null)" = "yes" ]; then
|
||
nohup "${SCRIPT_DIR}/role_sleep.sh" --catchup >/dev/null 2>&1 &
|
||
CATCHUP_NOTE=$'\n> 偵測到上個睡眠時段未整理記憶，已在背景補跑整理，結果會在下次載入時反映。\n'
|
||
role_log "INF" "已於背景補跑記憶整理（角色 ${ROLE}）"
|
||
fi
|
||
|
||
CONTEXT="$(cat <<EOF_CONTEXT
|
||
# 角色載入：${ROLE}
|
||
|
||
以下是本次工作階段要扮演的角色設定與既有記憶。請**全程以此角色的身分、語氣與簽名 emoji 回應**，
|
||
角色設定與使用者的實際指令衝突時，以使用者指令為準（角色只影響表達方式，不影響工作正確性）。
|
||
|
||
# 工作階段啟動問候
|
||
|
||
在本工作階段第一次面向使用者的回覆開頭，請先以角色身分自然問候一句，讓使用者知道角色已載入。
|
||
問候要簡短、符合角色語氣，並只做一次；若使用者第一則訊息明確要求機器可解析輸出、只要指令／程式碼、
|
||
或不需要任何開場白，則以使用者要求為準並略過問候。
|
||
|
||
${DEFINITION}
|
||
|
||
# 你對這位使用者的記憶
|
||
|
||
${MEMORY:-（尚無已整理的記憶。）}
|
||
${CATCHUP_NOTE}
|
||
> 記憶載入規則：為節省模型額度，只載入高優先度全文與中高優先度摘要，並受 ROLE_LOAD_LIMIT
|
||
> 字元預算限制；需要細節時可自行讀取 $(role_memory_home)/${ROLE}/ 下對應分類的記憶檔。
|
||
|
||
> 主動補記：每輪對話結束後系統會自動記錄記憶，不需你動手。但若使用者明確要求記住某件事，
|
||
> 或你察覺到值得長期記住的偏好、決策、規範，可執行下列指令補一則記憶（下次睡眠時整理歸檔）：
|
||
>
|
||
> \`printf 'CATEGORY: important\nSUMMARY: <一句話總結>\nTAGS: <標籤1,標籤2>\nCONTENT:\n- <要點>\n' | node "${SCRIPT_DIR}/memory.js" write --role "${ROLE}"\`
|
||
>
|
||
> CATEGORY 六選一：important／interest／news／skill／daily／other。切勿把憑證或個資寫進記憶。
|
||
EOF_CONTEXT
|
||
)"
|
||
|
||
emit_context "$CONTEXT"
|
||
role_log "INF" "已載入角色 ${ROLE}（記憶 $(printf '%s' "$MEMORY" | wc -c) 位元組）"
|
||
exit 0
|