#!/usr/bin/env bun
import { WebClient } from "@slack/web-api";

const client = new WebClient(process.env.SLACK_BOT_TOKEN);
const CHANNEL = "YOUR_CHANNEL";
const USER = "YOUR_USER";

async function main() {
  // Get team ID
  const auth = await client.auth.test();
  const team = auth.team_id;

  // Create thread
  const userMsg = await client.chat.postMessage({
    channel: CHANNEL,
    text: "Testing streaming",
  });
  const thread = userMsg.ts;

  // Set initial status
  await client.assistant.threads.setStatus({
    channel_id: CHANNEL,
    thread_ts: thread,
    status: "Processing...",
    loading_messages: ["Analyzing request"],
  });

  await sleep(2000);

  // Start streaming
  const streamer = client.chatStream({
    channel: CHANNEL,
    thread_ts: thread,
    recipient_team_id: team,
    recipient_user_id: USER,
    markdown_text: "**Starting process...**",
  });

  await sleep(2000);

  // Append chunks with status updates
  await client.assistant.threads.setStatus({
    channel_id: CHANNEL,
    thread_ts: thread,
    status: "Step 1 of 3",
  });
  await streamer.append({ markdown_text: "\n\n**Step 1**: First chunk" });

  await sleep(3000);

  await client.assistant.threads.setStatus({
    channel_id: CHANNEL,
    thread_ts: thread,
    status: "Step 2 of 3",
  });
  await streamer.append({ markdown_text: "\n\n**Step 2**: Second chunk" });

  await sleep(3000);

  await client.assistant.threads.setStatus({
    channel_id: CHANNEL,
    thread_ts: thread,
    status: "Step 3 of 3",
  });
  await streamer.append({ markdown_text: "\n\n**Step 3**: Final chunk" });

  await sleep(2000);

  // Stop stream
  await streamer.stop();
  await client.assistant.threads.setStatus({
    channel_id: CHANNEL,
    thread_ts: thread,
    status: "",
  });
}

function sleep(ms) {
  return new Promise((resolve) => setTimeout(resolve, ms));
}

main().catch(console.error);