#!/usr/bin/env bash
# line-needs-own-worktree.sh 的迴歸測試（inkstone/ISEP#109 → comment 5391）。
#
# **離線、不打網路、不花錢、每次結果一樣**——判準只有兩個，兩個都是 git 自己回答的：
# ①這條命令會不會移動 HEAD ②那個目錄是不是主工作目錄。所以不需要語意判官。
#
# 每個案例都自己開一顆真的 git repo（外加一份真的 linked worktree）。
#
# 三群，兩個方向都要有證據：
#   A 群「不該擋」——總管、唯讀動作、還原檔案、已經在自己的 worktree 裡、逃生門、
#                    worktree add 本身（**這群紅了比漏擋嚴重**：擋掉它等於線動不了工）
#   B 群「該擋」  ——線在共用目錄裡 checkout／switch／-b，含 -C 與複合命令
#   C 群「訊息與副作用」——講得出目錄與指令、不要開在 /tmp、逃生門、
#                    以及**真的把說謊的登記清掉**（驗收條件 7）
#
# 用法：hooks/tests/line-needs-own-worktree.test.sh [hook 路徑]

set -u
HERE="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
HOOK="${1:-$(cd "$HERE/.." && pwd)/line-needs-own-worktree.sh}"

PASS=0; FAIL=0; N=0
ROOT=$(mktemp -d "${TMPDIR:-/tmp}/isep-wt.XXXXXX")
clean() { rm -rf "$ROOT"; }
trap clean EXIT

# 🔴 macOS 的 $TMPDIR 是 /var/folders/… 而 git 回的是解析過的 /private/var/…
#    測試比對前先把路徑解析掉，不然會拿「同一個目錄的兩種寫法」互比而假紅。
ROOT=$(cd "$ROOT" && pwd -P)
REPO="$ROOT/shared-repo"
mkdir -p "$REPO"
git -C "$REPO" init -q -b main
git -C "$REPO" config user.email t@t; git -C "$REPO" config user.name t
echo hi > "$REPO/a.txt"; git -C "$REPO" add -A; git -C "$REPO" commit -qm init
git -C "$REPO" worktree add -q "$ROOT/own-wt" -b mine
OWN="$ROOT/own-wt"
PLAIN="$ROOT/not-a-repo"; mkdir -p "$PLAIN"

# payload <command> [cwd]
payload() {
  python3 - "$1" "${2:-$REPO}" <<'PY'
import json, sys
print(json.dumps({"session_id": "S-WT", "hook_event_name": "PreToolUse", "cwd": sys.argv[2],
                  "tool_name": "Bash", "tool_input": {"command": sys.argv[1]}},
                 ensure_ascii=False))
PY
}

# t <期望 exit> <說明> <command> [cwd] [env 前綴]
t() {
  local want="$1" desc="$2" cmd="$3" cwd="${4:-$REPO}" envs="${5:-CLAUDE_CODE_CHILD_SESSION=1}"
  N=$((N+1))
  local err rc
  err=$(printf '%s' "$(payload "$cmd" "$cwd")" | env $envs bash "$HOOK" 2>&1 >/dev/null); rc=$?
  LAST_ERR="$err"
  if [ "$rc" -eq "$want" ]; then
    printf '  ✅ %s\n' "$desc"; PASS=$((PASS+1))
  else
    printf '  ❌ %s —— 期望 exit=%s，實得 exit=%s\n' "$desc" "$want" "$rc"
    printf '%s\n' "$err" | sed -n '1,8p' | sed 's/^/     /'
    FAIL=$((FAIL+1))
  fi
}
has()   { N=$((N+1)); if printf '%s' "$LAST_ERR" | grep -qF -- "$1"; then printf '  ✅ %s\n' "$2"; PASS=$((PASS+1)); else printf '  ❌ %s —— 找不到「%s」\n' "$2" "$1"; FAIL=$((FAIL+1)); fi; }
hasnt() { N=$((N+1)); if printf '%s' "$LAST_ERR" | grep -qF -- "$1"; then printf '  ❌ %s —— 出現了「%s」\n' "$2" "$1"; FAIL=$((FAIL+1)); else printf '  ✅ %s\n' "$2"; PASS=$((PASS+1)); fi; }

echo
echo "── A 群：不該擋（誤攔比漏擋嚴重——擋錯了線就動不了工）──"

t 0 "① 總管（不是線）在共用目錄切分支 ⇒ 不管（那是他自己的目錄）" \
  "git checkout -b feat/x" "$REPO" "CLAUDE_CODE_CHILD_SESSION="
t 0 "② git status ⇒ 唯讀，不管"                 "git status"
t 0 "③ git log --oneline ⇒ 唯讀，不管"          "git log --oneline -5"
t 0 "④ git branch -a ⇒ 唯讀，不管"              "git branch -a"
t 0 "⑤ git checkout -- a.txt（還原檔案，不動 HEAD）⇒ 放行"     "git checkout -- a.txt"
t 0 "⑥ git checkout HEAD -- a.txt ⇒ 放行"                      "git checkout HEAD -- a.txt"
t 0 "⑦ git restore a.txt ⇒ 放行（根本不是 checkout）"          "git restore a.txt"
t 0 "⑧ git checkout -p ⇒ 互動式挑 hunk，不動 HEAD，放行"       "git checkout -p"
t 0 "⑨ **已經在自己的 worktree 裡**切分支 ⇒ 放行（這是我們要它做的事）" \
  "git checkout -b feat/inside" "$OWN"
t 0 "⑩ git worktree add ⇒ 放行（那是解法本身，不是病）" \
  "git worktree add $ROOT/wt2 -b feat/y"
t 0 "⑪ 逃生門 WORKTREE_OK=1 ⇒ 放行" \
  "git checkout -b feat/z" "$REPO" "CLAUDE_CODE_CHILD_SESSION=1 WORKTREE_OK=1"
t 0 "⑫ 不是 git 命令 ⇒ 不管"                    "npm test"
t 0 "⑬ 目錄根本不是 git repo ⇒ 不管"            "git checkout -b feat/x" "$PLAIN"
t 0 "⑭ 命令裡有 checkout 這個字但不是 git 的（不看措辭，看動作）" \
  "echo '別忘了 git checkout 回 main'"
t 0 "⑮ payload 壞掉 ⇒ 放行（不擋在不確定上）"   ""

N=$((N+1))
brc=$(printf 'not json' | CLAUDE_CODE_CHILD_SESSION=1 bash "$HOOK" >/dev/null 2>&1; echo $?)
if [ "$brc" -eq 0 ]; then printf '  ✅ ⑯ 完全不是 JSON 的 payload ⇒ 放行\n'; PASS=$((PASS+1));
else printf '  ❌ ⑯ payload 壞掉卻擋下來了（exit=%s）\n' "$brc"; FAIL=$((FAIL+1)); fi

echo
echo "── B 群：該擋 ──"

t 2 "⑰ 線在共用目錄 git checkout -b ⇒ 擋"        "git checkout -b feat/new"
t 2 "⑱ 線在共用目錄 git switch -c ⇒ 擋"          "git switch -c feat/new2"
t 2 "⑲ 線在共用目錄 git checkout main ⇒ 擋（切回去也是切）"  "git checkout main"
t 2 "⑳ git -C <共用目錄> checkout（人在別的地方）⇒ 擋"       "git -C $REPO checkout -b feat/c" "$PLAIN"
t 2 "㉑ 複合命令 cd … && git checkout ⇒ 擋（一段一段拆）"    "cd $REPO && git checkout -b feat/d"
t 2 "㉒ 前面掛 env 的 git checkout ⇒ 擋"                     "FOO=1 git checkout -b feat/e"
t 2 "㉓ git switch main ⇒ 擋"                                "git switch main"

echo
echo "── C 群：訊息與副作用 ──"

t 2 "㉔ 擋下來（下面幾條檢查這一次的訊息）" "git checkout -b feat/msg"
has "$REPO"                    "㉕ 說得出是哪個目錄"
has "worktree add"             "㉖ 給得出開自己 worktree 的指令"
has "不要開在 /tmp"            "㉗ 講得出不要開在 /tmp（08-29 實測：那些全 prunable）"
has "WORKTREE_OK=1"            "㉘ 講得出逃生門"
has "worktree prune"           "㉙ 講得出收工要怎麼收"
hasnt '$('                     "㉚ 訊息沒有被 shell 展開"

# 驗收條件 7：說謊的登記真的被清掉
git -C "$REPO" worktree add -q "$ROOT/ghost" -b ghosted
rm -rf "$ROOT/ghost"
N=$((N+1))
if git -C "$REPO" worktree list 2>/dev/null | grep -q prunable; then
  printf '  ✅ ㉛ 造出一筆「指向空氣」的登記（目錄砍掉、登記還在）\n'; PASS=$((PASS+1))
else
  printf '  ❌ ㉛ 造不出 prunable 的登記，下一條測不到東西\n'; FAIL=$((FAIL+1))
fi
t 2 "㉜ 這時候擋下來" "git checkout -b feat/prune"
has "指向空氣"       "㉝ 訊息說得出清掉了幾筆說謊的登記"
N=$((N+1))
if git -C "$REPO" worktree list 2>/dev/null | grep -q prunable; then
  printf '  ❌ ㉞ prune 沒有真的執行——登記簿還在說謊\n'; FAIL=$((FAIL+1))
else
  printf '  ✅ ㉞ 說謊的登記真的被清掉了（prune 是收工動作，不是偶爾整理）\n'; PASS=$((PASS+1))
fi
N=$((N+1))
if [ -d "$OWN" ] && git -C "$REPO" worktree list | grep -q "own-wt"; then
  printf '  ✅ ㉟ 而且還在的那份 worktree 一根寒毛都沒動\n'; PASS=$((PASS+1))
else
  printf '  ❌ ㉟ prune 把還在的 worktree 也弄掉了——那是災難\n'; FAIL=$((FAIL+1))
fi

clean
echo
echo "══ $PASS/$N 通過${FAIL:+，$FAIL 個失敗} ══"
[ "$FAIL" -eq 0 ]