248 lines
10 KiB
Python
248 lines
10 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""orphan-lock.py — 判定「孤儿锁」并解除（🔴 用户 2026-10-09 授权：检查会话可自解除）。
|
||
|
||
【这条授权改了什么】
|
||
R9 原口径（用户 2026-09-12）：「AI 一律不得删锁、不得以持有者为由单方面接管」。
|
||
它留下一个死局：持有者会话被强杀 ⇒ 锁永远留在盘上 ⇒ 后续任何会话抢都失败
|
||
⇒ 目标卡死，只能等用户本人到场。
|
||
2026-10-09 实测坐实：`contentm_agent` 一把孤儿锁把第 4 步挡住，机制空转 11 棒（约 5 小时）。
|
||
同日用户逐字授权：
|
||
「如果 工作区 项目机制 对应的会话异常 导致未解锁的情况下， 可以授权 检查会话自解除孤儿锁」
|
||
⇒ R9 的**唯一例外**＝本脚本覆盖的这一个情形：**锁的持有者会话已退出、不可能再来释放**。
|
||
|
||
【判据（⛔ 不是"看它像不像死的"，而是查持有者的真身状态）】
|
||
锁目录 `OWNER` 第 3 行记着 `会话：<sid>`（`handoff-guard.sh` 抢锁时写入）。
|
||
拿 sid 去宿主库 `sessions` 表查 `status`：
|
||
· `working` ⇒ **持有者还活着 ⇒ 不动**（R9 原样）
|
||
· `terminated` / `error` / `archived` ⇒ 异常退出／已归档 ⇒ **可解除**
|
||
· `completed` ⇒ 已收工；再要一道时间闸（`--idle-min`，默认 30 分钟）
|
||
内无活动 ⇒ 可解除
|
||
⚠️ 这道时间闸只为不掐掉「刚收工、用户随时会接着聊」的会话；但它**已经不在执行**了
|
||
⇒ 它**也不会再来释放** ⇒ 锁同样已成孤儿。
|
||
· 查不到该 sid／库读不到／`OWNER` 里没记 sid ⇒ **一律不动**（fail-closed）
|
||
🔴 判据只此一份，⛔ 不靠人目测 `OWNER` 文件猜死活。
|
||
|
||
【用法】
|
||
python orphan-lock.py # 判定并解除（默认执行）
|
||
python orphan-lock.py --dry-run # 只看判定，一把锁都不动
|
||
python orphan-lock.py --idle-min 60 # 改 `completed` 档的时间闸（分钟）
|
||
python orphan-lock.py --locks-root <目录> # 改锁根（自测用）
|
||
退出码：0 = 没有孤儿锁（或已全部解除）｜1 = 有锁未动（持有者活着／判不出）｜2 = 参数错
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import argparse
|
||
import os
|
||
import re
|
||
import shutil
|
||
import sqlite3
|
||
import sys
|
||
import time
|
||
from pathlib import Path
|
||
|
||
|
||
# ── 根目录定位（与包内其它脚本同款：宿主 env > `roots.env` 兜底）────────────
|
||
def _sm_load_roots() -> None:
|
||
"""把本包 `env/roots.env` 以 setdefault 注入 env（宿主 env 优先，⛔ 不覆盖）。"""
|
||
here = os.path.dirname(os.path.abspath(__file__))
|
||
cands = []
|
||
for up in range(4):
|
||
base = os.path.normpath(os.path.join(here, *([".."] * up)))
|
||
for rel in (("env", "roots.env"), ("roots.env",)):
|
||
cands.append(os.path.join(base, *rel))
|
||
conf = os.environ.get("CODEBUDDY_CONFIG_DIR") or os.path.expanduser("~/.workbuddy")
|
||
cands.append(os.path.join(conf, "skills", "session-mechanism", "env", "roots.env"))
|
||
for p in cands:
|
||
if os.path.isfile(p):
|
||
try:
|
||
with open(p, encoding="utf-8") as f:
|
||
for ln in f:
|
||
ln = ln.strip()
|
||
if ln and not ln.startswith("#") and "=" in ln:
|
||
k, v = ln.split("=", 1)
|
||
os.environ.setdefault(k.strip(), v.strip())
|
||
except Exception:
|
||
pass
|
||
return
|
||
|
||
|
||
_sm_load_roots()
|
||
|
||
# 授权放行的三档（持有者不可能再来释放）
|
||
DEAD_STATUS = ("terminated", "error", "archived")
|
||
# 需要再加一道时间闸的一档（已收工，但可能刚收工）
|
||
IDLE_STATUS = ("completed",)
|
||
ALIVE_STATUS = ("working",)
|
||
|
||
SID_RE = re.compile(r"^会话[：:]\s*([0-9A-Za-z][0-9A-Za-z\-]{7,})\s*$", re.M)
|
||
|
||
|
||
def _host_db() -> str:
|
||
"""宿主库路径：`DSH_HOST_DB` > `CODEBUDDY_CONFIG_DIR/workbuddy.db` > `~/.workbuddy/workbuddy.db`。"""
|
||
p = (os.environ.get("DSH_HOST_DB") or "").strip()
|
||
if p and os.path.isfile(p):
|
||
return p
|
||
cands = []
|
||
cfg = (os.environ.get("CODEBUDDY_CONFIG_DIR") or "").strip()
|
||
if cfg:
|
||
cands.append(os.path.join(cfg, "workbuddy.db"))
|
||
cands.append(os.path.join(os.path.expanduser("~"), ".workbuddy", "workbuddy.db"))
|
||
for c in cands:
|
||
if os.path.isfile(c):
|
||
return c
|
||
return ""
|
||
|
||
|
||
def _sess_row(db: str, sid: str):
|
||
"""返回 `(status, last_activity_at_ms)`；⛔ 任何异常 ⇒ `None`（调用方按 fail-closed 处理）。"""
|
||
if not db:
|
||
return None
|
||
try:
|
||
con = sqlite3.connect("file:%s?mode=ro" % db.replace("\\", "/"), uri=True, timeout=5)
|
||
try:
|
||
cur = con.cursor()
|
||
cur.execute("select status, last_activity_at from sessions where id like ?",
|
||
(sid.rstrip() + "%",))
|
||
r = cur.fetchone()
|
||
finally:
|
||
con.close()
|
||
if not r:
|
||
return None
|
||
return (str(r[0] or "").strip().lower(), r[1])
|
||
except Exception:
|
||
return None
|
||
|
||
|
||
def _judge(status: str, last_act, idle_min: float):
|
||
"""→ `(可解除?, 说明)`。判据唯一实现，⛔ 别在别处另写一份。"""
|
||
if not status:
|
||
return False, "判不出（宿主库无此会话）⇒ 按 R9 不动"
|
||
if status in ALIVE_STATUS:
|
||
return False, "持有者仍 `working`（活着）⇒ 按 R9 不动"
|
||
if status in DEAD_STATUS:
|
||
return True, "持有者 `%s`（已退出，不会再来释放）" % status
|
||
if status in IDLE_STATUS:
|
||
try:
|
||
ms = int(last_act or 0)
|
||
except Exception:
|
||
ms = 0
|
||
if not ms:
|
||
return False, "`completed` 但读不到最后活动时刻 ⇒ 不动（fail-closed）"
|
||
idle = (time.time() * 1000 - ms) / 60000.0
|
||
if idle >= idle_min:
|
||
return True, "持有者 `completed` 且已静默 %.0f 分钟（≥%d）" % (idle, idle_min)
|
||
return False, "持有者 `completed`，但仅静默 %.0f 分钟（<%d）⇒ 可能马上续聊，不动" % (idle, idle_min)
|
||
return False, "持有者状态 `%s` 不在已知档 ⇒ 不动（fail-closed）" % status
|
||
|
||
|
||
def _scan(root: Path, exec_lock: Path, dry: bool, idle_min: float, db: str, loglines: list):
|
||
left = 0
|
||
items = []
|
||
if root.is_dir():
|
||
for L in sorted(root.iterdir()):
|
||
if not L.is_dir() or L.name in (".gate", ".migrations"):
|
||
continue
|
||
items.append(L)
|
||
if exec_lock.is_dir():
|
||
items.append(exec_lock)
|
||
|
||
if not items:
|
||
print("· 锁目录里一把锁都没有（%s）" % root)
|
||
return 0
|
||
|
||
for L in items:
|
||
try:
|
||
txt = (L / "OWNER").read_text(encoding="utf-8", errors="replace")
|
||
except Exception as e:
|
||
print("· %-42s ｜ 读不到 OWNER（%s）⇒ 不动" % (L.name, e))
|
||
left += 1
|
||
continue
|
||
owner = (txt.splitlines() or ["?"])[0].strip()
|
||
m = SID_RE.search(txt)
|
||
if not m:
|
||
print("· %-42s ｜ 持有者 %-24s ｜ ⛔ OWNER 里没记会话 id ⇒ 不动（fail-closed）"
|
||
% (L.name[:42], owner[:24]))
|
||
left += 1
|
||
continue
|
||
sid = m.group(1)
|
||
row = _sess_row(db, sid)
|
||
status, last_act = (row if row else ("", None))
|
||
ok, why = _judge(status, last_act, idle_min)
|
||
tag = "⇒ 解除" if ok else "⇒ 不动"
|
||
if ok and not dry:
|
||
try:
|
||
shutil.rmtree(L)
|
||
except Exception as e:
|
||
print("· %-42s ｜ 持有者 %-24s ｜ 解除失败：%s" % (L.name[:42], owner[:24], e))
|
||
left += 1
|
||
continue
|
||
loglines.append("%s\t释放\t%s\t%s\t%s\t%s"
|
||
% (time.strftime("%Y-%m-%dT%H:%M:%S"), L.name, owner, sid, status))
|
||
elif ok and dry:
|
||
tag = "⇒ 解除（dry-run，未动）"
|
||
if not ok:
|
||
left += 1
|
||
print("· %-42s ｜ 持有者 %-24s ｜ sid %s ｜ %-11s ｜ %s"
|
||
% (L.name[:42], owner[:24], sid[:8], status or "?", tag + "：" + why))
|
||
return left
|
||
|
||
|
||
def main() -> int:
|
||
try:
|
||
sys.stdout.reconfigure(encoding="utf-8", errors="replace")
|
||
except Exception:
|
||
pass
|
||
|
||
ap = argparse.ArgumentParser(add_help=True)
|
||
ap.add_argument("--dry-run", action="store_true")
|
||
ap.add_argument("--idle-min", type=float, default=30.0)
|
||
ap.add_argument("--locks-root", default="")
|
||
a = ap.parse_args()
|
||
|
||
docs = (os.environ.get("DSH_DOCS_ROOT") or "").strip()
|
||
if a.locks_root:
|
||
root = Path(a.locks_root)
|
||
exec_lock = root / ".exec-lock"
|
||
elif docs and os.path.isdir(docs):
|
||
base = Path(docs) / "交接单"
|
||
root = base / ".locks"
|
||
exec_lock = base / ".exec-lock"
|
||
else:
|
||
print("✗ 定位不到文档库根（`DSH_DOCS_ROOT`）—— ⛔ 一把锁都没动。"
|
||
"请在有 `roots.env` 的环境里跑，或用 `--locks-root` 指定锁根。", file=sys.stderr)
|
||
return 1
|
||
|
||
db = _host_db()
|
||
print("锁根：%s" % root)
|
||
print("宿主库：%s%s" % (db or "（未找到）", "" if db else " ⇒ 查不到会话状态，将一律不动"))
|
||
print("授权来源：用户 2026-10-09「可以授权 检查会话自解除孤儿锁」（R9 的唯一例外）")
|
||
print("执行者：%s / %s" % (os.environ.get("DSH_SESSION_NAME") or "?",
|
||
(os.environ.get("CODEBUDDY_SESSION_ID") or "?")[:8]))
|
||
print("-" * 96)
|
||
|
||
logs: list = []
|
||
left = _scan(root, exec_lock, a.dry_run, a.idle_min, db, logs)
|
||
|
||
if logs and not a.dry_run:
|
||
try:
|
||
lp = root / ".orphan-release.log"
|
||
with open(lp, "a", encoding="utf-8", newline="\n") as f:
|
||
f.write("".join(x + "\n" for x in logs))
|
||
print("-" * 96)
|
||
print("已记流水：%s" % lp)
|
||
except Exception:
|
||
pass
|
||
|
||
print("-" * 96)
|
||
if left:
|
||
print("结论：另有 %d 把锁**未动**（持有者活着／判不出）—— 那是 R9 的地盘，⛔ 不许绕。"
|
||
"若确实卡住，写 `tmp/supervise-inbox/NEED-USER.md` 并明确喊「需用户介入」。" % left)
|
||
return 1
|
||
print("结论：没有孤儿锁（或在本次判定范围内已全部解除）。")
|
||
return 0
|
||
|
||
|
||
if __name__ == "__main__":
|
||
sys.exit(main())
|