#!/usr/bin/env python3
"""把大脑决策文件 Rnn-brain.md 里的 `### Tk` 各节原样抽成 Rnn-Tk-task.md。

用法:
    python split_tasks.py <运行目录>/R04-brain.md

规则（与 SKILL.md / templates.md 一致）:
- 只抽取 "## 任务" 与 "## 向用户提问" 之间的 `### Tk` 节，正文一字不改。
- 模型取该节 "- 模型:" 行；写了 fable 则改回 opus（Fable 只属于大脑），并在 stdout 提示。
- 报告路径优先取该节 "- 报告格式:" 行里反引号中的路径（大脑常自己指定文件名）；
  大脑用 `...\\` 省略了目录前缀时，用运行目录补全；没写则用默认 Rnn-Tk-report.md。
- 末尾追加 worker 报告格式模板。
输出: 每个任务一行 "Tk <model> <task-file> <report-path>"，供调度员直接派 worker。
"""
import re
import sys
from pathlib import Path

REPORT_TEMPLATE = '''
---
## 报告格式
写到 `{report}`，结构如下：

```
# {rnd}-{tk} 报告
- 结论: 完成 | 部分完成 | 失败 | 受阻
- 模型: {model}

## 做了什么
<按步骤列出>

## 改动的文件
<路径 + 一句话说明；没有就写"无">

## 证据
<关键命令及输出摘录、测量值、测试结果、日志片段>

## 验收标准逐条
1. <标准> —— 满足/不满足，证据: ...

## 问题与建议
<遇到的意外、任务单没覆盖到需要大脑决定的点>
```
验收标准要求的各张表、设计说明、数值结果放在"证据"一节或其后的独立小节中。
'''


def main(path: str) -> None:
    brain = Path(path).resolve()
    rundir = brain.parent
    rnd = brain.stem.split('-')[0]           # R04 / R02b
    text = brain.read_text(encoding='utf-8')
    if '## 任务' not in text:
        sys.exit('决策文件里没有 "## 任务" 节')
    body = text.split('## 任务', 1)[1]
    for stop in ('## 向用户提问', '## 总结'):
        if stop in body:
            body = body.split(stop, 1)[0]
    parts = re.split(r'(?m)^### (T\w+)\s*$', body)
    if len(parts) < 3:
        sys.exit('没有找到任何 "### Tk" 节')
    for i in range(1, len(parts), 2):
        tk, sec = parts[i], parts[i + 1].strip('\n')
        m = re.search(r'-\s*模型:\s*(\w+)', sec)
        model = m.group(1).lower() if m else 'opus'
        if model == 'fable':
            print(f'[warn] {tk} 指定了 fable，改回 opus（Fable 只属于大脑）', file=sys.stderr)
            model = 'opus'
        rep = None
        m = re.search(r'-\s*报告格式:.*?`([^`]+\.md)`', sec)
        if m:
            rep = m.group(1)
            if rep.startswith('...'):
                rep = str(rundir / Path(rep.replace('...', '').lstrip('\\/')).name)
            elif not Path(rep).is_absolute():
                rep = str(rundir / rep)
        if not rep:
            rep = str(rundir / f'{rnd}-{tk}-report.md')
        out = (f'# {rnd}-{tk} 任务单（模型: {model}）\n\n### {tk}\n{sec}\n'
               + REPORT_TEMPLATE.format(report=rep, rnd=rnd, tk=tk, model=model))
        task_file = rundir / f'{rnd}-{tk}-task.md'
        task_file.write_text(out, encoding='utf-8')
        print(f'{tk} {model} {task_file} {rep}')


if __name__ == '__main__':
    if len(sys.argv) != 2:
        sys.exit(__doc__)
    main(sys.argv[1])