#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""stop-dialog-guard.py —— 「禁止用征询句收尾」的 Stop 钩子（WorkBuddy / CodeBuddy）

为什么需要它
────────────
2026-09-15 实测：本工作区日志里 `tool=AskUserQuestion` 调用数 = 09-12: 43 / 09-13: 3 / **09-14: 0 / 09-15: 0**
⇒ 既有「提问闸门」（PreToolUse + matcher ^AskUserQuestion$）**拦的是几乎不走的工具面**，
而真实的上抛（"要我接着做吗 / 请确认 / 说一声即可"）发生在**正文里** —— 没有任何机制覆盖。

本钩子 = 覆盖那条面：**每次回复结束时**读 transcript 的**最后一条 assistant 文本**，
只扫**收尾段**（最后两行有效内容）里的征询句式；命中 → 返回 `{"continue": false, "reason": …}`
让 Agent **继续一轮并自我纠正**（把该自己做的事做掉，或改写成「需要你拍板」一节）。

安全设计（都不许省）
────────────────────
1. **自作用域**：只在 `transcript_path` 落在本工作区（`aliyun-dsh-server`）时生效，其他项目一律放行。
2. **防死循环**：输入里的 `stop_hook_active == true` 时**不再阻拦**（官方语义：本次停止已由 stop hook 触发过）。
3. **绝不添乱**：任何异常 → 静默放行（exit 0）。判定只在**收尾段**做，避免正文引用规则时误伤。
4. **性能**：只读转录**末尾 256 KB**（实测整库最大转录 31.9 MB、全文读 14 MB ≈ 832 ms ⇒ 不可接受），只看 stdin + 该文件。
5. **防跑飞**：同一会话 600 秒内最多拦**一次**。
6. **急停双闸**（无需卸载/重启）：env `DSH_STOP_GUARD_OFF=1`，或新建 `<工作区>/.workbuddy/stop-guard.disabled`。
7. **低频自证日志**：命中才写一行（`<工作区>/.workbuddy/stop-dialog-guard.log`），用来回答"到底有没有触发"。

退出码：始终 0；决策通过 stdout 的 JSON 表达。
安装（settings.json 的 hooks 段 · 见档案 73 / 99）：
  "Stop": [{ "hooks": [{ "type": "command",
      "command": "\"<python>\" \"<此脚本>\"", "timeout": 10 }] }]
⚠️ hooks 是**应用启动时快照** ⇒ 装完必须**完全重启 WorkBuddy**；桌面版无 /hooks 面板，等效。
"""
import io
import json
import os
import re
import sys
import time

SCOPE = 'aliyun-dsh-server'          # 只对本工作区生效
LOG_REL = os.path.join('.workbuddy', 'stop-dialog-guard.log')

# 只扫「收尾段」：出现这些就是"把该自己做的事甩回给用户"
PATTERNS = [
    r'要我(再|接着|继续|现在)?[^。！？\n]{0,12}吗',
    r'要不要我[^。！？\n]{0,12}',
    r'是否要我[^。！？\n]{0,12}',
    r'需要我[^。！？\n]{0,12}吗',
    r'请确认[^。！？\n]{0,16}',
    r'要不要(继续|现在做|我来)[^。！？\n]{0,12}',
    r'是否(继续|需要我)[^。！？\n]{0,12}',
    r'说一声即可',
    r'你看(怎么办|怎么弄|要不要)',
    r'你(决定|拍板)一下',
]
RE_BAN = re.compile('|'.join(PATTERNS))

REASON = (
    '⛔ 收尾句是**征询句**，但按本平台规则（`CODEBUDDY.md §1`「回话前自检」+ `dsh-feature-first §5.3` 铁律 3）'
    '先重判三问：① 命中**真门禁**吗（不可逆破坏性操作 / 边界外六类）？没命中 → **删掉这句，自己做完，改成陈述句**（"我接着做 X"）；'
    '② 是不是在把已经定下来的事再问一遍？是 → 删；③ 这件事用户有客观可判的优劣吗？没有 → 才允许问，且**一轮只问这一句**，'
    '并写进 `dsh-feature-first §5.1` 结论骨架的「**需要你拍板**」一节（一句话 + 可感知差别），不要在结尾甩问题。'
)


TAIL_BYTES = 262144
MAX_BYTES = 4194304      # 扩窗上限 4 MB（防"巨行"时无限读）          # 只读末尾 256 KB（实测：整库最大转录 31.9 MB；全文读 14 MB = 832 ms/轮，不可接受）


def transcribe_last_assistant(path):
    """返回最后一条 assistant 文本（**从尾部向后分块读**；读不到返回 ''）。

    ⚠️ 为什么不是"一次读末尾 256 KB"：一条 assistant 记录可能本身就 > 256 KB
    （长回复 / 被回显的工具输出），此时尾窗会切在 JSON 行中间 ⇒ `json.loads` 失败 ⇒ **静默漏判**。
    做法：从尾部按 TAIL_BYTES 递增扩窗（上限 MAX_BYTES），**直到至少解析出一条 assistant 记录**。
    常见情形（小消息）只花一次 256 KB 读，成本可忽略。
    """
    try:
        size = os.path.getsize(path)
    except OSError:
        return ''
    with io.open(path, 'rb') as f:
        window = TAIL_BYTES
        while True:
            start = max(0, size - window)
            f.seek(start)
            raw = f.read().decode('utf-8', 'replace')
            lines = raw.split('\n')
            if start > 0:
                lines = lines[1:]                    # 丢弃被截断的首行
            for line in reversed(lines):
                line = line.strip()
                if not line:
                    continue
                try:
                    rec = json.loads(line)
                except ValueError:
                    continue
                if rec.get('type') != 'message' or rec.get('role') != 'assistant':
                    continue
                chunks = [c['text'] for c in (rec.get('content') or [])
                          if isinstance(c, dict) and isinstance(c.get('text'), str)]
                chunks += [c for c in (rec.get('content') or []) if isinstance(c, str)]
                if chunks:
                    return '\n'.join(chunks)
            if start == 0 or window >= MAX_BYTES:
                # 放行，但**留痕**（A18：静默失败是负债）——可能是一条 >MAX_BYTES 的巨型记录
                try:
                    os.environ.setdefault('_DSH_SG_MISS', '1')
                    r0 = os.environ.get('CODEBUDDY_PROJECT_DIR') or os.environ.get('DSH_WORKSPACE') or ''
                    if r0:
                        log(r0, '未能解析（窗口 %d 字节仍无 assistant 记录）' % window)
                except Exception:
                    pass
                return ''
            window = min(window * 4, MAX_BYTES)


def tail_lines(text, n=2):
    out = [l.strip() for l in text.strip().split('\n') if l.strip()]
    return '\n'.join(out[-n:])


# 转述/引用豁免：收尾行里带引号或"引用/规则/写着/禁"等词 ⇒ 是在复述规则，不是在问用户
RE_QUOTE = re.compile(r'[「」“”"\']|引用|规则|写着|禁')


RATE_WINDOW = 600            # 秒；同一会话两次「阻止停止」的最小间隔


def _rate_limited(root, sid, peek=False):
    """同一会话 RATE_WINDOW 秒内已拦过 ⇒ 本次直接放行（防连续多轮被拦）。"""
    if not root or not sid:
        return False
    p = os.path.join(root, '.workbuddy', 'cache', 'stop-guard-fires.json')
    try:
        d = json.loads(io.open(p, encoding='utf-8').read()) if os.path.exists(p) else {}
    except Exception:
        d = {}
    now = time.time()
    if now - float(d.get(sid, 0) or 0) < RATE_WINDOW:
        return True
    d = {k: v for k, v in d.items() if now - float(v or 0) < 86400}   # 只留 1 天
    d[sid] = now
    try:
        os.makedirs(os.path.dirname(p), exist_ok=True)
        io.open(p, 'w', encoding='utf-8', newline='\n').write(json.dumps(d))
    except Exception:
        pass
    return False


def log(root, detail):
    try:
        p = os.path.join(root, LOG_REL)
        os.makedirs(os.path.dirname(p), exist_ok=True)
        with io.open(p, 'a', encoding='utf-8') as f:
            f.write('%s\t%s\n' % (time.strftime('%Y-%m-%d %H:%M:%S'), detail))
    except Exception:
        pass


def main():
    raw = sys.stdin.read()
    if not raw.strip():
        return
    try:
        payload = json.loads(raw)
    except ValueError:
        return
    tp = str(payload.get('transcript_path') or '')
    if SCOPE not in tp:                       # 作用域外 → 放行
        return
    if os.environ.get('DSH_STOP_GUARD_OFF'):  # 急停（环境变量）→ 放行
        return
    root0 = os.environ.get('CODEBUDDY_PROJECT_DIR') or os.environ.get('DSH_WORKSPACE') or ''
    if root0 and os.path.exists(os.path.join(root0, '.workbuddy', 'stop-guard.disabled')):
        return                                # 急停（闸刀文件）→ 放行
    if payload.get('stop_hook_active'):       # 防死循环 → 放行
        return
    text = transcribe_last_assistant(tp)
    if not text:
        return
    sid = str(payload.get('session_id') or '')
    if os.environ.get('DSH_SG_DEBUG'):        # 调试：每次调用都留痕（用于验证宿主是否真的调用本钩子）
        log(root0 or '.', 'invoked｜scope=%s｜tail_active=%s' % (SCOPE in tp, bool(payload.get('stop_hook_active'))))
    if _rate_limited(root0, sid, peek=True):   # 只查不记账
        return
    tail = tail_lines(text, 1)          # 只看**最后一行**：命中面越窄，误报越少
    if RE_QUOTE.search(tail):            # 复述/引用规则 → 不是收尾提问
        return
    m = RE_BAN.search(tail)
    if not m:
        return
    root = (os.environ.get('CODEBUDDY_PROJECT_DIR') or os.environ.get('DSH_WORKSPACE') or '')
    _rate_limited(root0, sid)                  # 命中才记账（同一会话 10 分钟最多拦 1 次）
    log(root if os.path.isdir(root) else '.', 'stop-dialog-guard 命中：%s ｜ 收尾：%s'
        % (m.group(0), tail.replace('\n', ' ')[:80]))
    sys.stdout.write(json.dumps({'continue': False, 'reason': REASON}, ensure_ascii=False))


if __name__ == '__main__':
    try:
        main()
    except Exception:
        pass
    sys.exit(0)