승인 및 사용자 입력 처리

async def handle_tool_request(tool_name, input_data, context):
    # 사용자에게 프롬프트하고 허용 또는 거부 반환
    ...


options = ClaudeAgentOptions(can_use_tool=handle_tool_request)

import asyncio

from claude_agent_sdk import ClaudeAgentOptions, ResultMessage, query
from claude_agent_sdk.types import (
    HookMatcher,
    PermissionResultAllow,
    PermissionResultDeny,
    ToolPermissionContext,
)


async def can_use_tool(
    tool_name: str, input_data: dict, context: ToolPermissionContext
) -> PermissionResultAllow | PermissionResultDeny:
    # 도구 요청 표시
    print(f"\nTool: {tool_name}")
    if tool_name == "Bash":
        print(f"Command: {input_data.get('command')}")
        if input_data.get("description"):
            print(f"Description: {input_data.get('description')}")
    else:
        print(f"Input: {input_data}")

    # 사용자 승인 받기
    response = input("Allow this action? (y/n): ")

    # 사용자의 응답에 따라 허용 또는 거부 반환
    if response.lower() == "y":
        # 허용: 도구가 원본(또는 수정된) 입력으로 실행됨
        return PermissionResultAllow(updated_input=input_data)
    else:
        # 거부: 도구가 실행되지 않음, Claude가 메시지를 봄
        return PermissionResultDeny(message="User denied this action")


# 필수 해결 방법: 더미 훅이 can_use_tool을 위해 스트림을 열어 둠
async def dummy_hook(input_data, tool_use_id, context):
    return {"continue_": True}


async def prompt_stream():
    yield {
        "type": "user",
        "message": {
            "role": "user",
            "content": "Create a test file in /tmp and then delete it",
        },
    }


async def main():
    async for message in query(
        prompt=prompt_stream(),
        options=ClaudeAgentOptions(
            can_use_tool=can_use_tool,
            hooks={"PreToolUse": [HookMatcher(matcher=None, hooks=[dummy_hook])]},
        ),
    ):
        if isinstance(message, ResultMessage) and message.subtype == "success":
            print(message.result)


asyncio.run(main())

from claude_agent_sdk.types import PermissionResultAllow, PermissionResultDeny

# 도구가 실행되도록 허용
return PermissionResultAllow(updated_input=input_data)

# 도구 차단
return PermissionResultDeny(message="User rejected this action")

import asyncio

from claude_agent_sdk import ClaudeAgentOptions, ResultMessage, query
from claude_agent_sdk.types import HookMatcher, PermissionResultAllow


def parse_response(response: str, options: list) -> str:
    """사용자 입력을 옵션 번호 또는 자유 텍스트로 구문 분석합니다."""
    try:
        indices = [int(s.strip()) - 1 for s in response.split(",")]
        labels = [options[i]["label"] for i in indices if 0 <= i < len(options)]
        return ", ".join(labels) if labels else response
    except ValueError:
        return response


async def handle_ask_user_question(input_data: dict) -> PermissionResultAllow:
    """Claude의 질문을 표시하고 사용자 답변을 수집합니다."""
    answers = {}

    for q in input_data.get("questions", []):
        print(f"\n{q['header']}: {q['question']}")

        options = q["options"]
        for i, opt in enumerate(options):
            print(f"  {i + 1}. {opt['label']} - {opt['description']}")
        if q.get("multiSelect"):
            print("  (Enter numbers separated by commas, or type your own answer)")
        else:
            print("  (Enter a number, or type your own answer)")

        response = input("Your choice: ").strip()
        answers[q["question"]] = parse_response(response, options)

    return PermissionResultAllow(
        updated_input={
            "questions": input_data.get("questions", []),
            "answers": answers,
        }
    )


async def can_use_tool(
    tool_name: str, input_data: dict, context
) -> PermissionResultAllow:
    # AskUserQuestion을 질문 핸들러로 라우팅
    if tool_name == "AskUserQuestion":
        return await handle_ask_user_question(input_data)
    # 이 예제에서는 다른 도구를 자동 승인
    return PermissionResultAllow(updated_input=input_data)


async def prompt_stream():
    yield {
        "type": "user",
        "message": {
            "role": "user",
            "content": "Help me decide on the tech stack for a new mobile app",
        },
    }


# 필수 해결 방법: 더미 훅이 can_use_tool을 위해 스트림을 열어 둠
async def dummy_hook(input_data, tool_use_id, context):
    return {"continue_": True}


async def main():
    async for message in query(
        prompt=prompt_stream(),
        options=ClaudeAgentOptions(
            can_use_tool=can_use_tool,
            hooks={"PreToolUse": [HookMatcher(matcher=None, hooks=[dummy_hook])]},
        ),
    ):
        if isinstance(message, ResultMessage) and message.subtype == "success":
            print(message.result)


asyncio.run(main())