Function main

tutorials/08_hook_system.py:315–432

()

# ============================================================

def main():
    print("=" * 60)
    print("Tutorial 08: Hook System 演示")
    print("=" * 60)

    # --- 1. 允许型 Hook（退出码 0）---
    print("\n--- 场景 1: 允许型 Hook ---")
    print("  Hook 命令: printf 'hook says hello'")
    print("  退出码: 0 → 允许执行")

    runner1 = HookRunner(HookConfig(
        pre_tool_use=["printf 'pre-hook: checked tool'"],
        post_tool_use=["printf 'post-hook: tool finished'"],
    ))

    pre_result = runner1.run_pre_tool_use("bash", '{"command": "ls"}')
    print(f"  PreToolUse 结果: denied={pre_result.denied}, messages={pre_result.messages}")

    post_result = runner1.run_post_tool_use("bash", '{"command": "ls"}', "file1.py", False)
    print(f"  PostToolUse 结果: denied={post_result.denied}, messages={post_result.messages}")

    # --- 2. 拒绝型 Hook（退出码 2）---
    print("\n--- 场景 2: 拒绝型 Hook ---")
    print("  Hook 命令: printf 'dangerous command blocked'; exit 2")
    print("  退出码: 2 → 拒绝执行")

    runner2 = HookRunner(HookConfig(
        pre_tool_use=["printf 'dangerous command blocked'; exit 2"],
    ))

    pre_result = runner2.run_pre_tool_use("bash", '{"command": "rm -rf /"}')
    print(f"  PreToolUse 结果: denied={pre_result.denied}, messages={pre_result.messages}")

    # --- 3. 多个 Hook 串联 ---
    print("\n--- 场景 3: 多个 Hook 串联（第二个拒绝）---")
    print("  Hook 1: printf 'first check ok'     → exit 0")
    print("  Hook 2: printf 'second check failed' → exit 2")
    print("  Hook 3: printf 'never reached'       → exit 0")

    runner3 = HookRunner(HookConfig(
        pre_tool_use=[
            "printf 'first check ok'",
            "printf 'second check failed'; exit 2",
            "printf 'never reached'",  # 这个不会执行！
        ],
    ))

    pre_result = runner3.run_pre_tool_use("bash", '{"command": "echo hello"}')
    print(f"  结果: denied={pre_result.denied}")
    print(f"  消息: {pre_result.messages}")
    print("  注意：第三个 hook 没有执行（因为第二个已经拒绝了）")

    # --- 4. Hook 读取环境变量 ---
    print("\n--- 场景 4: Hook 读取环境变量 ---")
    runner4 = HookRunner(HookConfig(
        pre_tool_use=["printf \"tool=%s input=%s\" \"$HOOK_TOOL_NAME\" \"$HOOK_TOOL_INPUT\""],
    ))