use a3s_code_core::{Agent, AgentEvent, SessionOptions};

let session = agent.session("/my-project", Some(
    SessionOptions::new().with_permissive_policy()
))?;

let (mut rx, handle) = session.stream("Refactor the auth module", None).await?;

while let Some(event) = rx.recv().await {
    match event {
        AgentEvent::TextDelta { text } => print!("{text}"),
        AgentEvent::ToolStart { name, .. } => println!("\n🔧 {name}..."),
        AgentEvent::ToolEnd { name, exit_code, .. } => {
            println!(" ✓ (exit={})", exit_code);
        }
        AgentEvent::TurnEnd { turn, usage } => {
            println!("\n└─ Turn {turn} done ({} tokens)", usage.total_tokens);
        }
        AgentEvent::End { text, usage } => {
            println!("\n■ Done — {} tokens total", usage.total_tokens);
            break;
        }
        _ => {}
    }
}

handle.await??;

session = agent.session("/my-project", permissive=True)

async for event in session.stream("Refactor the auth module"):
    t = event.get("type")
    if t == "text_delta":
        print(event["text"], end="", flush=True)
    elif t == "tool_start":
        print(f"\n🔧 {event['name']}...", end="")
    elif t == "tool_end":
        print(f" ✓ (exit={event['exit_code']})")
    elif t == "turn_end":
        print(f"\n└─ Turn {event['turn']} done ({event['usage']['total_tokens']} tokens)")
    elif t == "end":
        print(f"\n■ Done — {event['usage']['total_tokens']} tokens total")
        break

const session = agent.session('/my-project', { permissive: true });

const stream = await session.stream('Refactor the auth module');

for await (const event of stream) {
  switch (event.type) {
    case 'text_delta':
      process.stdout.write(event.text);
      break;
    case 'tool_start':
      process.stdout.write(`\n🔧 ${event.name}...`);
      break;
    case 'tool_end':
      console.log(` ✓ (exit=${event.exitCode})`);
      break;
    case 'turn_end':
      console.log(`\n└─ Turn ${event.turn} done (${event.usage.totalTokens} tokens)`);
      break;
    case 'end':
      console.log(`\n■ Done — ${event.usage.totalTokens} tokens total`);
      break;
  }
}