import asyncio

from copilot import CopilotClient
from copilot.generated.rpc import PermissionDecisionApproveOnce

input_log: list[str] = []


async def auto_approve_permission(request, invocation):
    return PermissionDecisionApproveOnce()


async def auto_approve_tool(input_data, invocation):
    return {"permissionDecision": "allow"}


async def handle_user_input(request, invocation):
    input_log.append(f"question: {request['question']}")
    return {"answer": "Paris", "wasFreeform": True}


async def main():
    client = CopilotClient()

    try:
        session = await client.create_session(
            model="claude-haiku-4.5",
            on_permission_request=auto_approve_permission,
            on_user_input_request=handle_user_input,
            hooks={"on_pre_tool_use": auto_approve_tool},
        )

        response = await session.send_and_wait(
            "I want to learn about a city. Use the ask_user tool to ask me "
            "which city I'm interested in. Then tell me about that city."
        )

        if response:
            print(response.data.content)

        await session.disconnect()

        print("\n--- User input log ---")
        for entry in input_log:
            print(f"  {entry}")
        print(f"\nTotal user input requests: {len(input_log)}")
    finally:
        await client.stop()


asyncio.run(main())