novel-maker 2.2.0

package/skill/scripts/coordinator/stats_report.py

@@ -0,0 +1,165 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+项目统计报告
+快速扫描 novels/ 和 .novel-maker/ 目录，输出完整的写作进度统计。
+供 `/novel-maker stats` 指令使用。
+"""
+
+import os
+import sys
+import json
+from collections import Counter
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'common'))
+from nm_utils import (
+    list_chapters, read_chapter, count_chinese,
+    extract_characters, extract_locations, estimate_pacing
+)
+
+
+def scan_volume(volume_path):
+    """Scan a single volume directory."""
+    vol_name = os.path.basename(volume_path)
+    chapters_dir = os.path.join(volume_path, 'chapters')
+    if not os.path.isdir(chapters_dir):
+        return None
+
+    chapter_files = list_chapters(chapters_dir)
+    if not chapter_files:
+        return None
+
+    chapters = []
+    all_chars = Counter()
+    all_locs = Counter()
+    pacing_dist = Counter()
+    total_words = 0
+
+    for idx, filepath in enumerate(chapter_files):
+        raw, title, clean, wc = read_chapter(filepath)
+        pacing_label, _ = estimate_pacing(raw)
+        chars = extract_characters(clean)
+        locs = extract_locations(clean)
+
+        chapters.append({
+            'num': idx + 1,
+            'title': title,
+            'words': wc,
+            'pacing': pacing_label,
+            'characters': chars[:4],
+        })
+
+        total_words += wc
+        for c in chars:
+            all_chars[c] += 1
+        for loc in locs:
+            all_locs[loc] += 1
+        pacing_dist[pacing_label] += 1
+
+    return {
+        'name': vol_name,
+        'total_chapters': len(chapters),
+        'total_words': total_words,
+        'avg_words': round(total_words / max(len(chapters), 1), 0),
+        'chapters': chapters,
+        'main_characters': [name for name, _ in all_chars.most_common(8)],
+        'character_appearances': dict(all_chars.most_common(10)),
+        'main_locations': [loc for loc, _ in all_locs.most_common(5)],
+        'pacing_distribution': dict(pacing_dist),
+    }
+
+
+def scan_project(novels_dir):
+    """Scan the entire novels directory."""
+    if not os.path.isdir(novels_dir):
+        return {'error': f'目录不存在: {novels_dir}'}
+
+    # Check for outline
+    outline_path = os.path.join(novels_dir, 'outline.md')
+    has_outline = os.path.exists(outline_path)
+    outline_word_count = 0
+    if has_outline:
+        with open(outline_path, 'r', encoding='utf-8') as f:
+            outline_word_count = count_chinese(f.read())
+
+    # Scan volumes
+    volumes = []
+    for entry in sorted(os.listdir(novels_dir)):
+        vol_path = os.path.join(novels_dir, entry)
+        if os.path.isdir(vol_path) and entry.startswith('volume'):
+            result = scan_volume(vol_path)
+            if result:
+                volumes.append(result)
+
+    # Aggregate stats
+    total_chapters = sum(v['total_chapters'] for v in volumes)
+    total_words = sum(v['total_words'] for v in volumes)
+
+    return {
+        'has_outline': has_outline,
+        'outline_words': outline_word_count,
+        'volumes': len(volumes),
+        'total_chapters': total_chapters,
+        'total_words': total_words,
+        'volume_details': volumes,
+    }
+
+
+def main():
+    import argparse
+    parser = argparse.ArgumentParser(description="Project stats report")
+    parser.add_argument("novels_dir", nargs='?', default='novels', help="Novels directory")
+    parser.add_argument("--json", action="store_true", help="Output as JSON")
+    parser.add_argument("--volume", type=str, help="Show only specific volume")
+    args = parser.parse_args()
+
+    novels_dir = args.novels_dir
+    if not os.path.isdir(novels_dir):
+        # Try to find novels dir
+        base = os.path.dirname(os.path.abspath(novels_dir))
+        alt = os.path.join(base, 'novels')
+        if os.path.isdir(alt):
+            novels_dir = alt
+        else:
+            print(f"错误: 目录不存在 - {novels_dir}", file=sys.stderr)
+            sys.exit(1)
+
+    result = scan_project(novels_dir)
+
+    if args.json:
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+        return
+
+    if 'error' in result:
+        print(result['error'])
+        return
+
+    print(f"\n{'=' * 60}")
+    print(f"📚 NovelMaker 项目统计")
+    print(f"{'=' * 60}")
+    print(f"大纲: {'✅ 已生成' if result['has_outline'] else '❌ 未生成'} ({result['outline_words']}字)")
+    print(f"卷数: {result['volumes']}")
+    print(f"总章节: {result['total_chapters']}")
+    print(f"总字数: {result['total_words']:,}")
+    print()
+
+    for vol in result['volume_details']:
+        if args.volume and vol['name'] != args.volume:
+            continue
+        print(f"  📖 {vol['name']}")
+        print(f"     章节: {vol['total_chapters']} 章 | 字数: {vol['total_words']:,} | 平均: {vol['avg_words']}字/章")
+        print(f"     主要角色: {', '.join(vol['main_characters'][:5])}")
+        print(f"     主要场景: {', '.join(vol['main_locations'][:3])}")
+        pacing = vol['pacing_distribution']
+        pacing_str = ' | '.join(f"{k}:{v}章" for k, v in sorted(pacing.items()))
+        print(f"     节奏分布: {pacing_str}")
+
+        if args.volume == vol['name']:
+            print(f"\n     章节明细:")
+            for ch in vol['chapters']:
+                print(f"       第{ch['num']}章: {ch['title']} ({ch['words']}字, {ch['pacing']})")
+        print()
+
+
+if __name__ == "__main__":
+    main()

package/skill/scripts/coordinator/volume_batch.py

@@ -0,0 +1,121 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+卷级批量章节分析
+批处理卷内所有（或最近N章），输出卷级汇总 JSON。
+"""
+
+import os
+import sys
+import json
+from collections import Counter
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'common'))
+from nm_utils import (
+    list_chapters, read_chapter, extract_characters,
+    extract_locations, detect_hook, detect_hook_type_from_patterns,
+    count_chinese
+)
+
+FORESHADOWING_KEYWORDS = {
+    '血脉': ['血脉', '祖先', '传承', '遗传', '血统'],
+    '身世': ['身世', '出身', '来历', '母亲', '父亲'],
+    '预言': ['预言', '命中注定', '天意', '卦象', '命格'],
+    '阴谋': ['阴谋', '暗算', '布局', '棋局', '幕后'],
+    '身份': ['身份', '真实身份', '秘密', '隐藏'],
+    '仇敌': ['仇人', '仇敌', '恩怨', '血海深仇', '宿敌'],
+}
+
+
+def analyze_volume(chapters_dir, recent_n=None):
+    chapter_files = list_chapters(chapters_dir, recent_n)
+    if not chapter_files:
+        return {'error': f'No chapters found in {chapters_dir}'}
+
+    chapter_summaries = []
+    all_characters = Counter()
+    foreshadowing_counter = Counter()
+    hook_type_counter = Counter()
+    character_matrix = {}
+
+    for idx, filepath in enumerate(chapter_files):
+        raw, title, clean, wc = read_chapter(filepath)
+        chars = extract_characters(clean)
+        hook = detect_hook_type_from_patterns(raw)
+
+        chapter_num = idx + 1
+        chapter_summaries.append({
+            'chapter': chapter_num,
+            'title': title,
+            'word_count': wc,
+            'within_range': 2200 <= wc <= 2800,
+            'characters': chars[:6],
+            'hook_type': hook,
+        })
+
+        for c in chars:
+            all_characters[c] += 1
+        hook_type_counter[hook] += 1
+        character_matrix[f'ch{chapter_num}'] = chars
+
+        for keyword in clean:
+            for ftype, keywords in FORESHADOWING_KEYWORDS.items():
+                if any(kw in keyword for kw in keywords):
+                    foreshadowing_counter[ftype] += 1
+                    break
+
+    main_characters = [name for name, _ in all_characters.most_common(8)]
+
+    return {
+        'total_chapters': len(chapter_files),
+        'total_words': sum(s['word_count'] for s in chapter_summaries),
+        'avg_words': round(sum(s['word_count'] for s in chapter_summaries) / max(len(chapter_summaries), 1), 0),
+        'chapters': chapter_summaries,
+        'main_characters': main_characters,
+        'character_matrix': character_matrix,
+        'foreshadowing_summary': dict(foreshadowing_counter),
+        'hook_distribution': dict(hook_type_counter),
+    }
+
+
+def main():
+    if len(sys.argv) < 2:
+        print('用法: python volume_batch.py <卷章节目录路径> [--recent N] [--json]')
+        return
+    chapters_dir = sys.argv[1]
+    if not os.path.isdir(chapters_dir):
+        print(f'错误: 目录不存在 - {chapters_dir}')
+        return
+    recent_n = None
+    json_only = False
+    if '--recent' in sys.argv:
+        idx = sys.argv.index('--recent')
+        if idx + 1 < len(sys.argv):
+            recent_n = int(sys.argv[idx + 1])
+    json_only = '--json' in sys.argv
+
+    result = analyze_volume(chapters_dir, recent_n)
+
+    if json_only:
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+    else:
+        print(f"=== 卷级批量分析 ===\n")
+        if 'error' in result:
+            print(result['error'])
+            return
+        print(f"总章节: {result['total_chapters']} 章")
+        print(f"总字数: {result['total_words']} 字")
+        print(f"平均每章: {result['avg_words']} 字\n")
+        print(f"主要角色: {', '.join(result['main_characters'][:8])}\n")
+        print(f"钩子类型分布:")
+        for ht, count in sorted(result.get('hook_distribution', {}).items(), key=lambda x: x[1], reverse=True):
+            bar = "█" * count
+            print(f"  {ht}: {bar} ({count})")
+        print(f"\n章节概览:")
+        for ch in result['chapters']:
+            status = "✓" if ch['within_range'] else "⚠"
+            print(f"  {status} 第{ch['chapter']}章: {ch['title']} ({ch['word_count']}字, 角色: {', '.join(ch['characters'][:3])})")
+
+
+if __name__ == '__main__':
+    main()

package/skill/scripts/planner/outline_extractor.py

@@ -0,0 +1,89 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+大纲快速提取
+解析 outline.md 和 vol*/plan.md 的标题层级，输出结构化大纲树。
+供 `/novel-maker memory outline` 指令使用。
+"""
+
+import os
+import sys
+import json
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'common'))
+from nm_utils import parse_outline_headings, count_chinese
+
+
+def extract_outline(filepath):
+    """Extract structured outline from a markdown file."""
+    if not os.path.exists(filepath):
+        return None
+    with open(filepath, 'r', encoding='utf-8') as f:
+        text = f.read()
+    wc = count_chinese(text)
+    tree = parse_outline_headings(text)
+    return {'file': os.path.basename(filepath), 'words': wc, 'headings': tree}
+
+
+def extract_all_outlines(base_dir):
+    """Extract all outline-related files."""
+    results = {}
+
+    # Main outline
+    outline_path = os.path.join(base_dir, 'outline.md')
+    if os.path.exists(outline_path):
+        results['outline'] = extract_outline(outline_path)
+
+    # Volume plans
+    for entry in sorted(os.listdir(base_dir)):
+        vol_path = os.path.join(base_dir, entry)
+        if os.path.isdir(vol_path) and entry.startswith('volume'):
+            plan_path = os.path.join(vol_path, 'plan.md')
+            if os.path.exists(plan_path):
+                results[entry] = extract_outline(plan_path)
+
+            # Act plans
+            act_path = os.path.join(vol_path, 'act-plan.md')
+            if os.path.exists(act_path):
+                results[f"{entry}/act-plan"] = extract_outline(act_path)
+
+    return results
+
+
+def main():
+    import argparse
+    parser = argparse.ArgumentParser(description="Outline extractor")
+    parser.add_argument("base_dir", nargs='?', default='novels', help="Novels base directory")
+    parser.add_argument("--json", action="store_true", help="Output as JSON")
+    parser.add_argument("--file", type=str, help="Specific file to extract")
+    args = parser.parse_args()
+
+    if args.file:
+        if not os.path.isfile(args.file):
+            print(f"Error: {args.file} not found", file=sys.stderr)
+            sys.exit(1)
+        result = {args.file: extract_outline(args.file)}
+    else:
+        if not os.path.isdir(args.base_dir):
+            print(f"Error: {args.base_dir} is not a directory", file=sys.stderr)
+            sys.exit(1)
+        result = extract_all_outlines(args.base_dir)
+
+    if args.json:
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+    else:
+        for key, val in result.items():
+            if val is None:
+                print(f"\n❌ {key}: 文件不存在")
+                continue
+            print(f"\n{'=' * 50}")
+            print(f"📄 {key} ({val['words']}字, {len(val['headings'])}个标题)")
+            print(f"{'=' * 50}")
+            for h in val['headings']:
+                indent = "  " * (h['level'] - 1)
+                marker = {1: '📖', 2: '📑', 3: '📝', 4: '•'}.get(h['level'], '•')
+                print(f"{indent}{marker} {h['title']}")
+
+
+if __name__ == "__main__":
+    main()

package/skill/scripts/planner/planner_context.py

@@ -0,0 +1,220 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+规划师上下文包 — 一键生成规划所需的精简上下文
+
+将规划师需要读取的 10+ 个文件压缩为一个结构化 JSON（~5000 token），
+替代手动读取全部真相文件+大纲（~30,000 token）。
+
+用法:
+    python scripts/planner/planner_context.py --volume 01 --json
+    python scripts/planner/planner_context.py --volume 01 --act 2 --json
+"""
+
+import argparse
+import json
+import os
+import sys
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'common'))
+from nm_utils import (
+    read_truth_section, list_chapters, read_chapter,
+    extract_characters, generate_summary, parse_outline_headings,
+    chapter_sort_key
+)
+
+NOVEL_MAKER_DIR = '.novel-maker'
+TRUTH_DIR = os.path.join(NOVEL_MAKER_DIR, 'truth-files')
+
+
+def _find_project_root(start_path):
+    path = os.path.abspath(start_path)
+    for _ in range(10):
+        if os.path.isdir(os.path.join(path, NOVEL_MAKER_DIR)):
+            return path
+        parent = os.path.dirname(path)
+        if parent == path:
+            break
+        path = parent
+    return None
+
+
+def _read_truth_compact(filepath, max_chars=600):
+    sections = read_truth_section(filepath)
+    if not sections:
+        return None
+    result = {}
+    for title, content in sections.items():
+        if title == 'header':
+            continue
+        lines = [l.strip() for l in content.split('\n') if l.strip() and not l.startswith('#')]
+        compact = '\n'.join(lines[:10])
+        if len(compact) > max_chars:
+            compact = compact[:max_chars] + '...'
+        if compact:
+            result[title] = compact
+    return result
+
+
+def _get_volume_progress(volume_dir):
+    chapters_dir = os.path.join(volume_dir, 'chapters')
+    if not os.path.isdir(chapters_dir):
+        return {'total': 0, 'chapters': []}
+    all_chapters = list_chapters(chapters_dir)
+    chapters = []
+    total_wc = 0
+    for ch_path in all_chapters:
+        raw, title, clean, wc = read_chapter(ch_path)
+        ch_num = chapter_sort_key(ch_path)
+        chars = extract_characters(clean)[:5]
+        summary = generate_summary(raw, first_n=50, last_n=50)
+        chapters.append({
+            'number': ch_num,
+            'title': title,
+            'word_count': wc,
+            'characters': chars,
+            'summary': summary
+        })
+        total_wc += wc
+    return {
+        'total': len(chapters),
+        'total_word_count': total_wc,
+        'chapters': chapters[-5:]  # 只返回最近5章
+    }
+
+
+def _get_outline_context(outline_path, current_chapter=None):
+    if not os.path.exists(outline_path):
+        return None
+    with open(outline_path, encoding='utf-8') as f:
+        text = f.read()
+    headings = parse_outline_headings(text)
+    if current_chapter:
+        result = []
+        found = False
+        for h in headings:
+            if f'第{current_chapter}章' in h['title']:
+                found = True
+            if found:
+                result.append(h)
+                if len(result) >= 10:
+                    break
+        if not result:
+            result = headings[-10:] if len(headings) > 10 else headings
+    else:
+        result = headings[-10:] if len(headings) > 10 else headings
+    return [{'level': h['level'], 'title': h['title']} for h in result]
+
+
+def _get_active_hooks(truth_dir):
+    fpath = os.path.join(truth_dir, 'pending-hooks.md')
+    sections = read_truth_section(fpath)
+    if not sections:
+        return []
+    hooks = []
+    for title, content in sections.items():
+        if 'header' in title or '已回收' in title or '已解决' in title:
+            continue
+        lines = [l.strip() for l in content.split('\n') if l.strip() and not l.startswith('#')]
+        for line in lines:
+            if 'H[' in line or '伏笔' in line or '钩子' in line:
+                hooks.append(line[:100])
+    return hooks[:10]
+
+
+def _get_active_subplots(truth_dir):
+    fpath = os.path.join(truth_dir, 'subplot-board.md')
+    sections = read_truth_section(fpath)
+    if not sections:
+        return []
+    subplots = []
+    for title, content in sections.items():
+        if '活跃' in title:
+            lines = [l.strip() for l in content.split('\n') if l.strip() and not l.startswith('#')]
+            for line in lines:
+                if line.startswith('-') or line.startswith('*'):
+                    subplots.append(line[:100])
+    return subplots[:5]
+
+
+def build_planner_context(volume='01', act=None, root=None, truth_dir=None):
+    if root:
+        pass
+    else:
+        root = _find_project_root('.')
+    if not root:
+        return {'error': '未找到 .novel-maker 目录'}
+
+    truth_path = truth_dir or os.path.join(root, TRUTH_DIR)
+    volume_dir = os.path.join(root, 'novels', f'volume-{volume.zfill(2)}')
+
+    context = {
+        'meta': {'volume': volume, 'act': act},
+        'volume_progress': {},
+        'truth_files': {},
+        'outline': None,
+        'active_hooks': [],
+        'active_subplots': []
+    }
+
+    # 卷进度
+    context['volume_progress'] = _get_volume_progress(volume_dir)
+
+    # 真相文件摘要
+    truth_files = {
+        'characters': 'characters.md',
+        'current_state': 'current-state.md',
+        'world_setting': 'world-setting.md',
+        'power_system': 'power-system.md',
+        'emotional_arcs': 'emotional-arcs.md',
+    }
+    for key, filename in truth_files.items():
+        fpath = os.path.join(truth_path, filename)
+        compact = _read_truth_compact(fpath)
+        if compact:
+            context['truth_files'][key] = compact
+
+    # 大纲
+    outline_path = os.path.join(volume_dir, 'outline.md')
+    current_ch = context['volume_progress']['total'] + 1 if context['volume_progress']['total'] > 0 else None
+    context['outline'] = _get_outline_context(outline_path, current_ch)
+
+    # 活跃伏笔
+    context['active_hooks'] = _get_active_hooks(truth_path)
+
+    # 活跃支线
+    context['active_subplots'] = _get_active_subplots(truth_path)
+
+    return context
+
+
+def main():
+    parser = argparse.ArgumentParser(description='规划师上下文包')
+    parser.add_argument('--volume', '-v', default='01', help='卷号（默认01）')
+    parser.add_argument('--act', '-a', type=int, help='当前幕号')
+    parser.add_argument('--root', help='项目根目录')
+    parser.add_argument('--truth-dir', help='真相文件目录')
+    parser.add_argument('--json', action='store_true', help='JSON输出')
+    args = parser.parse_args()
+
+    ctx = build_planner_context(args.volume, args.act, args.root, args.truth_dir)
+
+    if args.json:
+        print(json.dumps(ctx, ensure_ascii=False, indent=2))
+    else:
+        if 'error' in ctx:
+            print(f"错误: {ctx['error']}")
+            return
+        print(f"=== 规划师上下文: 卷{ctx['meta']['volume']} ===\n")
+        prog = ctx['volume_progress']
+        print(f"进度: {prog['total']}章, {prog.get('total_word_count', 0)}字")
+        if ctx['truth_files']:
+            print(f"真相文件: {len(ctx['truth_files'])}个")
+        if ctx['active_hooks']:
+            print(f"活跃伏笔: {len(ctx['active_hooks'])}个")
+        if ctx['active_subplots']:
+            print(f"活跃支线: {len(ctx['active_subplots'])}条")
+
+
+if __name__ == '__main__':
+    main()