novel-maker 2.2.0

package/skill/scripts/test_scripts.py

@@ -0,0 +1,366 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+单元测试脚本
+测试所有辅助脚本的基本功能。
+
+用法:
+    python scripts/test_scripts.py
+"""
+
+import os
+import sys
+import json
+import tempfile
+import shutil
+from pathlib import Path
+
+# Add common to path
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), 'common'))
+
+# Test results
+test_results = {
+    'passed': 0,
+    'failed': 0,
+    'errors': []
+}
+
+
+def test_passed(test_name):
+    test_results['passed'] += 1
+    print(f"  ✓ {test_name}")
+
+
+def test_failed(test_name, error):
+    test_results['failed'] += 1
+    test_results['errors'].append({'test': test_name, 'error': str(error)})
+    print(f"  ✗ {test_name}: {error}")
+
+
+def create_test_chapter(filepath, title="测试章节", content=None):
+    """Create a test chapter file."""
+    if content is None:
+        content = f"""# {title}
+
+林风站在山巅，望着远方的城市。
+
+"苏婉，你来了。"林风转身说道。
+
+苏婉微微一笑："我当然会来。"
+
+两人相视一笑，心中都明白彼此的心意。
+
+突然，远处传来一声巨响，天空中出现了一道裂缝。
+
+"不好！"林风脸色大变，"危机降临了！"
+
+苏婉握紧手中的宝剑："我们一起面对！"
+
+林风从怀中掏出一本古老的秘籍，这是师父留给他的传承。
+
+"据说这本秘籍具有强大的力量，但需要足够的修为才能使用。"林风说道。
+
+苏婉点头："我相信你一定能突破瓶颈。"
+
+林风闭上眼睛，开始运功修炼。他的修为从筑基期开始提升...
+
+一刻钟后，林风睁开眼睛，眼中闪过一丝精光。
+
+"我突破了！现在是金丹期！"林风兴奋地说道。
+
+苏婉为他高兴："太好了！"
+
+但就在这时，一个神秘人出现在他们面前。
+
+"你们以为这样就结束了吗？"神秘人冷笑道，"真正的危险才刚刚开始。"
+
+林风和苏婉对视一眼，都看到了对方眼中的决心。
+
+无论前方有什么危险，他们都会一起面对。
+
+---
+
+**下章预告**：神秘人的真实身份究竟是什么？林风能否保护苏婉？"""
+
+    with open(filepath, 'w', encoding='utf-8') as f:
+        f.write(content)
+
+
+def create_test_truth_file(filepath, content):
+    """Create a test truth file."""
+    os.makedirs(os.path.dirname(filepath), exist_ok=True)
+    with open(filepath, 'w', encoding='utf-8') as f:
+        f.write(content)
+
+
+def test_scene_builder():
+    """Test scene_builder.py"""
+    print("\n=== 测试 scene_builder.py ===")
+    try:
+        from writer.scene_builder import build_scene, list_scene_types
+
+        # Test list_scene_types
+        result = list_scene_types()
+        assert 'available_types' in result
+        assert len(result['available_types']) > 0
+        test_passed('list_scene_types')
+
+        # Test build_scene
+        result = build_scene('冲突', ['林风', '苏婉'], '紧张')
+        assert 'scene_type' in result
+        assert result['scene_type'] == '冲突'
+        assert 'structure' in result
+        assert len(result['structure']) > 0
+        test_passed('build_scene with type, chars, emotion')
+
+        # Test invalid type
+        result = build_scene('invalid_type')
+        assert 'error' in result
+        test_passed('build_scene with invalid type')
+
+    except Exception as e:
+        test_failed('scene_builder', e)
+
+
+def test_dialogue_checker():
+    """Test dialogue_checker.py"""
+    print("\n=== 测试 dialogue_checker.py ===")
+    try:
+        from auditor.dialogue_checker import check_dialogue_quality, extract_dialogues
+
+        # Test extract_dialogues (using ASCII quotes which the regex also matches)
+        text = '林风说道："你好。"苏婉回答："我很好。"'
+        dialogues = extract_dialogues(text)
+        assert len(dialogues) == 2
+        test_passed('extract_dialogues')
+
+        # Test check_dialogue_quality
+        with tempfile.NamedTemporaryFile(mode='w', suffix='.md', delete=False, encoding='utf-8') as f:
+            f.write('# 测试章节\n\n林风说道："你好。"苏婉回答："我很好。"')
+            temp_file = f.name
+
+        try:
+            result = check_dialogue_quality(temp_file)
+            assert 'dialogue_count' in result
+            assert 'checks' in result
+            test_passed('check_dialogue_quality')
+        finally:
+            os.unlink(temp_file)
+
+    except Exception as e:
+        test_failed('dialogue_checker', e)
+
+
+def test_character_arc_tracker():
+    """Test character_arc_tracker.py"""
+    print("\n=== 测试 character_arc_tracker.py ===")
+    try:
+        from reviewer.character_arc_tracker import build_character_arc
+
+        # Create temp directory with test chapters
+        temp_dir = tempfile.mkdtemp()
+        try:
+            for i in range(3):
+                create_test_chapter(os.path.join(temp_dir, f'第{i+1:02d}章.md'), f'第{i+1}章')
+
+            result = build_character_arc(temp_dir, ['林风'])
+            assert 'total_chapters_analyzed' in result
+            assert result['total_chapters_analyzed'] == 3
+            assert 'characters_tracked' in result
+            test_passed('build_character_arc')
+        finally:
+            shutil.rmtree(temp_dir)
+
+    except Exception as e:
+        test_failed('character_arc_tracker', e)
+
+
+def test_subplot_tracker():
+    """Test subplot_tracker.py"""
+    print("\n=== 测试 subplot_tracker.py ===")
+    try:
+        from reviewer.subplot_tracker import track_subplot_across_chapters
+
+        # Create temp directory with test chapters
+        temp_dir = tempfile.mkdtemp()
+        try:
+            for i in range(3):
+                create_test_chapter(os.path.join(temp_dir, f'第{i+1:02d}章.md'), f'第{i+1}章')
+
+            result = track_subplot_across_chapters(temp_dir)
+            assert 'total_chapters_analyzed' in result
+            assert 'subplots_detected' in result
+            test_passed('track_subplot_across_chapters')
+        finally:
+            shutil.rmtree(temp_dir)
+
+    except Exception as e:
+        test_failed('subplot_tracker', e)
+
+
+def test_chapter_transition():
+    """Test chapter_transition.py"""
+    print("\n=== 测试 chapter_transition.py ===")
+    try:
+        from auditor.chapter_transition import check_transition, list_chapters
+
+        # Create temp directory with test chapters
+        temp_dir = tempfile.mkdtemp()
+        try:
+            for i in range(3):
+                create_test_chapter(os.path.join(temp_dir, f'第{i+1:02d}章.md'), f'第{i+1}章')
+
+            chapter_files = list_chapters(temp_dir)
+            assert len(chapter_files) == 3
+
+            transitions = check_transition(chapter_files)
+            assert len(transitions) == 2  # 3 chapters = 2 transitions
+            test_passed('check_transition')
+        finally:
+            shutil.rmtree(temp_dir)
+
+    except Exception as e:
+        test_failed('chapter_transition', e)
+
+
+def test_emotion_curve():
+    """Test emotion_curve.py"""
+    print("\n=== 测试 emotion_curve.py ===")
+    try:
+        from reviewer.emotion_curve import analyze_emotion_in_text, build_emotion_curve
+
+        # Test analyze_emotion_in_text
+        text = '林风很开心，苏婉很悲伤，两人心情复杂。'
+        result = analyze_emotion_in_text(text)
+        assert 'dominant' in result
+        assert 'scores' in result
+        test_passed('analyze_emotion_in_text')
+
+        # Test build_emotion_curve
+        temp_dir = tempfile.mkdtemp()
+        try:
+            for i in range(3):
+                create_test_chapter(os.path.join(temp_dir, f'第{i+1:02d}章.md'), f'第{i+1}章')
+
+            result = build_emotion_curve(temp_dir)
+            assert 'total_chapters' in result
+            assert 'curve' in result
+            test_passed('build_emotion_curve')
+        finally:
+            shutil.rmtree(temp_dir)
+
+    except Exception as e:
+        test_failed('emotion_curve', e)
+
+
+def test_foreshadowing_tracker():
+    """Test foreshadowing_tracker.py"""
+    print("\n=== 测试 foreshadowing_tracker.py ===")
+    try:
+        from reviewer.foreshadowing_tracker import track_foreshadowing_across_chapters
+
+        # Create temp directory with test chapters
+        temp_dir = tempfile.mkdtemp()
+        try:
+            for i in range(3):
+                create_test_chapter(os.path.join(temp_dir, f'第{i+1:02d}章.md'), f'第{i+1}章')
+
+            result = track_foreshadowing_across_chapters(temp_dir)
+            assert 'total_chapters' in result
+            assert 'total_setups' in result
+            assert 'total_resolutions' in result
+            test_passed('track_foreshadowing_across_chapters')
+        finally:
+            shutil.rmtree(temp_dir)
+
+    except Exception as e:
+        test_failed('foreshadowing_tracker', e)
+
+
+def test_worldbuilding_checker():
+    """Test worldbuilding_checker.py"""
+    print("\n=== 测试 worldbuilding_checker.py ===")
+    try:
+        from auditor.worldbuilding_checker import check_worldbuilding_consistency
+
+        # Create temp directory with test chapters
+        temp_dir = tempfile.mkdtemp()
+        truth_dir = tempfile.mkdtemp()
+        try:
+            for i in range(3):
+                create_test_chapter(os.path.join(temp_dir, f'第{i+1:02d}章.md'), f'第{i+1}章')
+
+            # Create truth file
+            create_test_truth_file(
+                os.path.join(truth_dir, 'power_system.md'),
+                '# 力量体系\n\n- 筑基期\n- 金丹期\n- 元婴期'
+            )
+
+            result = check_worldbuilding_consistency(temp_dir, truth_dir)
+            assert 'total_chapters' in result
+            assert 'dimensions_checked' in result
+            assert 'issues' in result
+            test_passed('check_worldbuilding_consistency')
+        finally:
+            shutil.rmtree(temp_dir)
+            shutil.rmtree(truth_dir)
+
+    except Exception as e:
+        test_failed('worldbuilding_checker', e)
+
+
+def test_pacing_optimizer():
+    """Test pacing_optimizer.py"""
+    print("\n=== 测试 pacing_optimizer.py ===")
+    try:
+        from auditor.pacing_optimizer import analyze_pacing_with_suggestions
+
+        # Create temp directory with test chapters
+        temp_dir = tempfile.mkdtemp()
+        try:
+            for i in range(5):
+                create_test_chapter(os.path.join(temp_dir, f'第{i+1:02d}章.md'), f'第{i+1}章')
+
+            result = analyze_pacing_with_suggestions(temp_dir)
+            assert 'total_chapters' in result
+            assert 'pacing_distribution' in result
+            assert 'suggestions' in result
+            test_passed('analyze_pacing_with_suggestions')
+        finally:
+            shutil.rmtree(temp_dir)
+
+    except Exception as e:
+        test_failed('pacing_optimizer', e)
+
+
+def main():
+    print("=" * 60)
+    print("NovelMaker 辅助脚本单元测试")
+    print("=" * 60)
+
+    # Run all tests
+    test_scene_builder()
+    test_dialogue_checker()
+    test_character_arc_tracker()
+    test_subplot_tracker()
+    test_chapter_transition()
+    test_emotion_curve()
+    test_foreshadowing_tracker()
+    test_worldbuilding_checker()
+    test_pacing_optimizer()
+
+    # Print summary
+    print("\n" + "=" * 60)
+    print(f"测试完成: {test_results['passed']}通过, {test_results['failed']}失败")
+    print("=" * 60)
+
+    if test_results['errors']:
+        print("\n失败详情:")
+        for error in test_results['errors']:
+            print(f"  - {error['test']}: {error['error']}")
+
+    return 0 if test_results['failed'] == 0 else 1
+
+
+if __name__ == '__main__':
+    sys.exit(main())

package/skill/scripts/writer/build_write_context.py

@@ -0,0 +1,255 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+写手上下文构建器 — 一键生成写作所需的精简上下文
+
+将写手需要读取的 15+ 个文件压缩为一个结构化 JSON（~3000 token），
+每章节省约 40,000-60,000 token 的文件读取消耗。
+
+用法:
+    python scripts/writer/build_write_context.py novels/volume-01/chapters/ch15.md
+    python scripts/writer/build_write_context.py --chapter 15 --volume 01
+    python scripts/writer/build_write_context.py --chapter 15 --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, detect_hook,
+    parse_outline_headings
+)
+
+NOVEL_MAKER_DIR = '.novel-maker'
+TRUTH_DIR = os.path.join(NOVEL_MAKER_DIR, 'truth-files')
+
+
+def _find_project_root(start_path):
+    """向上查找包含 .novel-maker 的项目根目录"""
+    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=800):
+    """读取真相文件并压缩为精简摘要"""
+    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[:15])  # 每个section最多15行
+        if len(compact) > max_chars:
+            compact = compact[:max_chars] + '...'
+        if compact:
+            result[title] = compact
+    return result
+
+
+def _get_prev_chapters_info(chapters_dir, current_chapter_num, count=2):
+    """获取前N章的结构化摘要"""
+    all_chapters = list_chapters(chapters_dir)
+    prev = []
+    for ch_path in all_chapters:
+        raw, title, clean, wc = read_chapter(ch_path)
+        # 判断是否在当前章之前
+        from nm_utils import chapter_sort_key
+        ch_num = chapter_sort_key(ch_path)
+        if ch_num >= current_chapter_num:
+            break
+        prev.append((ch_num, ch_path, raw, title, clean, wc))
+
+    # 只取最近N章
+    prev = prev[-count:]
+    results = []
+    for ch_num, ch_path, raw, title, clean, wc in prev:
+        chars = extract_characters(clean)[:8]
+        summary = generate_summary(raw, first_n=100, last_n=100)
+        hook = detect_hook(raw)
+        results.append({
+            'chapter': os.path.basename(ch_path),
+            'number': ch_num,
+            'title': title,
+            'word_count': wc,
+            'characters': chars,
+            'hook': hook['type'],
+            'summary': summary
+        })
+    return results
+
+
+def _get_outline_context(outline_path, chapter_num):
+    """从大纲中提取当前章节的目标"""
+    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)
+    # 找到当前章节附近的大纲节点
+    result = []
+    found_current = False
+    for h in headings:
+        if f'第{chapter_num}章' in h['title'] or f'第{_to_cn(chapter_num)}章' in h['title']:
+            found_current = True
+        if found_current:
+            result.append(h)
+            if len(result) >= 5:
+                break
+    # 如果没找到精确匹配，返回最后几个节点
+    if not result:
+        result = headings[-5:] if len(headings) > 5 else headings
+    return [{'level': h['level'], 'title': h['title']} for h in result]
+
+
+def _to_cn(n):
+    """数字转中文"""
+    cn = '零一二三四五六七八九十'
+    if n <= 10:
+        return cn[n]
+    if n < 20:
+        return '十' + (cn[n - 10] if n > 10 else '')
+    return str(n)
+
+
+def _get_constitution_summary(constitution_path):
+    """提取创作宪法要点"""
+    if not os.path.exists(constitution_path):
+        return None
+    sections = read_truth_section(constitution_path)
+    if not sections:
+        return None
+    key_sections = ['字数要求', '文风配置', '核心创作原则', '禁忌清单']
+    result = {}
+    for title, content in sections.items():
+        if any(k in title for k in key_sections):
+            lines = [l.strip() for l in content.split('\n') if l.strip() and not l.startswith('#')]
+            result[title] = '\n'.join(lines[:8])
+    return result
+
+
+def build_context(chapter_path=None, chapter_num=None, volume=None,
+                  novel_root=None, truth_dir=None, recent_count=2):
+    """构建写手上下文"""
+
+    # 确定项目根目录
+    if novel_root:
+        root = novel_root
+    elif chapter_path:
+        root = _find_project_root(os.path.dirname(chapter_path))
+    else:
+        root = _find_project_root('.')
+
+    if not root:
+        return {'error': '未找到 .novel-maker 目录，请先运行 /novel-maker init'}
+
+    truth_path = truth_dir or os.path.join(root, TRUTH_DIR)
+    context = {'meta': {}, 'truth_files': {}, 'prev_chapters': [], 'outline': None, 'constitution': None}
+
+    # 确定章节号和卷号
+    if chapter_path and os.path.exists(chapter_path):
+        from nm_utils import chapter_sort_key
+        chapter_num = chapter_num or chapter_sort_key(chapter_path)
+        chapters_dir = os.path.dirname(chapter_path)
+        volume = volume or os.path.basename(os.path.dirname(chapters_dir))
+    elif chapter_num and volume:
+        chapters_dir = os.path.join(root, 'novels', f'volume-{volume.zfill(2)}', 'chapters')
+        chapter_path = None
+    else:
+        return {'error': '请指定章节路径或章节号+卷号'}
+
+    context['meta'] = {
+        'chapter_num': chapter_num,
+        'volume': volume,
+        'chapters_dir': chapters_dir
+    }
+
+    # 1. 读取精简版真相文件
+    truth_files = {
+        'characters': 'characters.md',
+        'current_state': 'current-state.md',
+        'world_setting': 'world-setting.md',
+        'power_system': 'power-system.md',
+        'pending_hooks': 'pending-hooks.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
+
+    # 2. 前N章摘要
+    if os.path.isdir(chapters_dir):
+        context['prev_chapters'] = _get_prev_chapters_info(chapters_dir, chapter_num, recent_count)
+
+    # 3. 大纲目标
+    outline_path = os.path.join(root, 'novels', f'volume-{volume.zfill(2) if volume else "01"}', 'outline.md')
+    if not os.path.exists(outline_path):
+        outline_path = os.path.join(root, 'novels', 'outline.md')
+    context['outline'] = _get_outline_context(outline_path, chapter_num)
+
+    # 4. 创作宪法要点
+    constitution_path = os.path.join(root, NOVEL_MAKER_DIR, 'constitution.md')
+    if not os.path.exists(constitution_path):
+        constitution_path = os.path.join(root, 'constitution.md')
+    context['constitution'] = _get_constitution_summary(constitution_path)
+
+    return context
+
+
+def main():
+    parser = argparse.ArgumentParser(description='写手上下文构建器')
+    parser.add_argument('chapter', nargs='?', help='章节文件路径')
+    parser.add_argument('--chapter-num', '-n', type=int, help='章节号')
+    parser.add_argument('--volume', '-v', help='卷号')
+    parser.add_argument('--root', help='项目根目录')
+    parser.add_argument('--truth-dir', help='真相文件目录')
+    parser.add_argument('--recent', type=int, default=2, help='前N章摘要（默认2）')
+    parser.add_argument('--json', action='store_true', help='JSON输出')
+    args = parser.parse_args()
+
+    ctx = build_context(
+        chapter_path=args.chapter,
+        chapter_num=args.chapter_num,
+        volume=args.volume,
+        novel_root=args.root,
+        truth_dir=args.truth_dir,
+        recent_count=args.recent
+    )
+
+    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']['chapter_num']}章 ===\n")
+        if ctx['truth_files']:
+            print("【真相文件摘要】")
+            for name, data in ctx['truth_files'].items():
+                print(f"  {name}: {len(data)}个section")
+        if ctx['prev_chapters']:
+            print(f"\n【前{len(ctx['prev_chapters'])}章摘要】")
+            for ch in ctx['prev_chapters']:
+                print(f"  {ch['chapter']}: {ch['title']} ({ch['word_count']}字) 钩子:{ch['hook']}")
+        if ctx['outline']:
+            print(f"\n【大纲目标】")
+            for h in ctx['outline']:
+                print(f"  {'  ' * (h['level'] - 1)}{h['title']}")
+
+
+if __name__ == '__main__':
+    main()

package/skill/scripts/writer/chapter_info.py

@@ -0,0 +1,67 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+章节结构化信息提取
+将章节全文压缩为结构化数据，AI 可直接读取此输出代替全文，大幅减少 token 消耗。
+"""
+
+import sys
+import json
+import os
+
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', 'common'))
+from nm_utils import (
+    clean_markdown, count_chinese, extract_title, read_chapter,
+    extract_characters, extract_locations, detect_structure,
+    detect_hook, generate_summary
+)
+
+
+def analyze_chapter(filepath):
+    raw, title, clean, wc = read_chapter(filepath)
+    return {
+        'file': os.path.basename(filepath),
+        'title': title,
+        'word_count': wc,
+        'characters': extract_characters(clean),
+        'locations': extract_locations(clean),
+        'structure': detect_structure(raw),
+        'hook': detect_hook(raw),
+        'summary': generate_summary(raw)
+    }
+
+
+def main():
+    if len(sys.argv) < 2:
+        print('用法: python chapter_info.py <章节文件路径> [--json]')
+        print('      python chapter_info.py <文件> --json  # 纯JSON输出给AI用')
+        return
+    filepath = sys.argv[1]
+    json_only = '--json' in sys.argv
+
+    if not os.path.exists(filepath):
+        result = {'error': f'文件不存在: {filepath}'}
+    else:
+        result = analyze_chapter(filepath)
+
+    if json_only:
+        print(json.dumps(result, ensure_ascii=False, indent=2))
+    else:
+        info = result
+        if 'error' in info:
+            print(info['error'])
+            return
+        print(f"章节: {info['title'] or info['file']}")
+        print(f"字数: {info['word_count']}")
+        print(f"角色: {', '.join(info['characters'][:8]) if info['characters'] else '未检测到'}")
+        print(f"地点: {', '.join(info['locations'][:5]) if info['locations'] else '未检测到'}")
+        s = info['structure']
+        print(f"结构: 对话{s['dialogue_pct']}% / 动作{s['action_pct']}% / 描写{s['description_pct']}% ({s['total_paragraphs']}段)")
+        h = info['hook']
+        print(f"钩子: {h['type']} (结尾预览: {h['tail_preview'][:40]}...)")
+        print(f"摘要: 开头={info['summary']['开头'][:50]}...")
+        print(f"      结尾={info['summary']['结尾'][-50:]}...")
+
+
+if __name__ == '__main__':
+    main()