#!/usr/bin/env bash
# 最终验证（在 47 上跑）：清污染 → 重置锚点 → 重验两条路径
#  ① 既有用户 guest：留 <host-a> + 工作区有历史数据 + 实例页正常
#  ② 新用户：落 <host-b> + **文件真的写到 106 的盘** + 实例页正常
set -uo pipefail
. "$(cd "$(dirname "$0")/.." && pwd)/config/load.sh"
export PGPASSWORD=dshs_cluster_2026
Q() { /usr/bin/psql -h 127.0.0.1 -p 15432 -U dshs -d dshs -tAc "$1"; }
M=http://127.0.0.1:3080
DOMAIN="${DSHS_BASE_DOMAIN:?config/platform.env 缺 DSHS_BASE_DOMAIN}"
T47="$DSHS_CLUSTER_AGENT_TOKEN"
T106="$DSH_PEER_AGENT_TOKEN"
SSH106="ssh -n -i /root/.ssh/dshworker_ed25519 -o BatchMode=yes -o StrictHostKeyChecking=accept-new root@"$DSH_PEER_PUBLIC_IP""

mksess() {
  local u="$1" T H
  T=$(openssl rand -hex 32); H=$(printf '%s' "$T" | sha256sum | cut -d' ' -f1)
  Q "insert into sessions (token_hash,user_id,created_at,expires_at,ip,user_agent)
     select '$H', id, (extract(epoch from now())*1000)::bigint, (extract(epoch from now())*1000)::bigint+1800000, '127.0.0.1','switch-verify' from users where username='$u'" >/dev/null
  printf '%s' "$T"
}
owner() { Q "select coalesce(i.host_id,'NULL')||' epoch='||coalesce(i.epoch,0) from users u left join dsh_instances i on i.user_id=u.id where u.username='$1'"; }
agent() { curl -s -m 6 -H "x-dsh-agent-token: $2" "http://127.0.0.1:$1/healthz" | grep -o '"instances":[0-9]*'; }
isapp() { grep -q '<base href=' <<<"$1" && echo "✓实例页" || echo "✗非实例页"; }

echo "########## 0) 清污染：停所有实例 + 删测试用户 ##########"
systemctl restart dshs-worker; sleep 4
$SSH106 'systemctl restart dshs-worker' >/dev/null 2>&1; sleep 4
echo "  重启后 "$DSHS_CLUSTER_HOST_ID"=$(agent 19100 $T47)  "$DSH_PEER_HOST_ID"=$(agent 19000 $T106)"
AT=$(mksess admin); AC="sid=$AT"
for u in $(Q "select id from users where username like 'switchtest%' or username like 'swtest%'"); do
  echo "  删测试用户 $u → $(curl -s -m 20 -X DELETE -b "$AC" "$M/api/admin/users/$u" -o /dev/null -w '%{http_code}')"
done
echo "  剩余用户: $(Q "select string_agg(username,', ') from users")"

echo "########## 1) 重置 guest 锚点（host_id="$DSHS_CLUSTER_HOST_ID"） ##########"
Q "update dsh_instances set host_id='w-1', epoch=0, lease_until=0, status='stopped'
   where user_id=(select id from users where username='guest')" >/dev/null
echo "  $(owner guest)"

echo
echo "########## 2) 路径①：既有用户 guest ##########"
GT=$(mksess guest); GC="sid=$GT"
E=$(curl -s -m 60 -X POST -b "$GC" -H 'content-type: application/json' -d '{}' "$M/api/dsh/enter")
sleep 3
echo "  归属: $(owner guest)    ← 期望 "$DSHS_CLUSTER_HOST_ID""
echo "  "$DSHS_CLUSTER_HOST_ID"=$(agent 19100 $T47)  "$DSH_PEER_HOST_ID"=$(agent 19000 $T106)"
echo "  工作区条目: $(curl -s -m 10 -b "$GC" "$M/api/desktop/tree" | grep -o '"name":"[^"]*"' | head -4 | tr '\n' ' ')"
PAGE=$(curl -s -m 25 -L -b "$GC" -H "Host: guest.$DOMAIN" "$M/" | head -c 300)
echo "  实例页: $(isapp "$PAGE")"

echo
echo "########## 3) 路径②：新用户 ##########"
NU="swtest2$(date +%H%M%S)"
echo "  register=$(curl -s -m 15 -X POST -H 'content-type: application/json' -d "{\"username\":\"$NU\",\"password\":\"SwitchTest123\"}" "$M/api/auth/register" -o /dev/null -w '%{http_code}')"
NID=$(Q "select id from users where username='$NU'")
echo "  approve=$(curl -s -m 15 -X POST -b "$AC" "$M/api/admin/users/$NID/approve" -o /dev/null -w '%{http_code}')"
NC=$(curl -s -m 15 -X POST -H 'content-type: application/json' -d "{\"username\":\"$NU\",\"password\":\"SwitchTest123\"}" -D - "$M/api/auth/login" -o /dev/null | grep -i '^set-cookie' | head -1 | grep -oP 'sid=[^;]+')
echo "  mkdir=$(curl -s -m 15 -X POST -b "$NC" -H 'content-type: application/json' -d '{"path":"proj"}' "$M/api/fs/mkdir" -o /dev/null -w '%{http_code}')  ← 首次触达应把归属钉住"
echo "  钉住后归属: $(owner "$NU")   ← 期望 "$DSH_PEER_HOST_ID"（与下面的 launch 必须同台）"
echo "  upload=$(curl -s -m 20 -X POST -b "$NC" -H 'content-type: application/json' -d "{\"path\":\"proj\",\"name\":\"hello.txt\",\"data\":\"$(printf 'hi-from-switch' | base64 -w0)\"}" "$M/api/fs/upload" -o /dev/null -w '%{http_code}')"
echo "  launch=$(curl -s -m 60 -X POST -b "$NC" -H 'content-type: application/json' -d '{"folder":"proj"}' "$M/api/dsh/launch" -o /dev/null -w '%{http_code}')"
sleep 4
echo "  归属: $(owner "$NU")    ← 期望 "$DSH_PEER_HOST_ID"（粘性保持）"
echo "  "$DSH_PEER_HOST_ID"=$(agent 19000 $T106)  "$DSHS_CLUSTER_HOST_ID"=$(agent 19100 $T47)"
echo "  --- 落盘取证 ---"
L47=$($SSH106 "ls "$DSHS_DATA_ROOT"/users/$NID/ws/proj/hello.txt 2>/dev/null" 2>/dev/null || true)
echo "    106 盘: ${L47:-不存在}"
echo "    47  盘: $(ls "$DSHS_DATA_ROOT"/users/$NID/ws/proj/hello.txt 2>/dev/null || echo 不存在（应不存在 ✓）)"
NP=$(curl -s -m 25 -L -b "$NC" -H "Host: $NU.$DOMAIN" "$M/" | head -c 300)
echo "  实例页(Host: $NU.$DOMAIN): $(isapp "$NP")"

echo
echo "########## 收尾 ##########"
curl -s -m 40 -X POST -b "$GC" "$M/api/dsh/stop" -o /dev/null -w "  guest stop=%{http_code}\n"
curl -s -m 40 -X POST -b "$NC" "$M/api/dsh/stop" -o /dev/null -w "  newuser stop=%{http_code}\n"
echo "  stop 后 guest 归属（**不应再被清空**）: $(owner guest)"
Q "delete from sessions where user_agent='switch-verify'" >/dev/null
echo "  残留临时 session: $(Q "select count(*) from sessions where user_agent='switch-verify'")（应 0）"
echo "  新用户待清: $NU / $NID"