import WebSocket from "ws";
import fs from "fs";

const TENANT_NAME = "YOUR_TENANT_NAME";
const ACCESS_TOKEN = "YOUR_ACCESS_TOKEN";
const ENVIRONMENT = "eu"; // or "us"

const WSS_URL = `wss://api.${ENVIRONMENT}.corti.app/audio-bridge/v2/transcribe?tenant-name=${TENANT_NAME}&token=Bearer%20${ACCESS_TOKEN}`;

const ws = new WebSocket(WSS_URL);

ws.on("open", () => {
  console.log("✅ WebSocket connected");

  // Step 1: Send config immediately (must be within 10 seconds)
  const config = {
    type: "config",
    configuration: {
      primaryLanguage: "en",
      automaticPunctuation: true,
      formatting: {
        numbers: "numerals_above_nine",
        measurements: "abbreviated",
      },
    },
  };

  ws.send(JSON.stringify(config));
  console.log("📤 Sent config");
});

ws.on("message", (data) => {
  // Audio binary frames come back as Buffer — skip those
  if (Buffer.isBuffer(data) && !isJson(data)) return;

  const message = JSON.parse(data.toString());
  console.log("📨 Received:", JSON.stringify(message, null, 2));

  switch (message.type) {
    case "CONFIG_ACCEPTED":
      console.log("✅ Config accepted — session:", message.sessionId);
      // Step 2: Start sending audio now that config is accepted
      sendAudio();
      break;

    case "CONFIG_DENIED":
    case "CONFIG_TIMEOUT":
      console.error("❌ Config error:", message);
      ws.close();
      break;

    case "transcript":
      if (message.data.isFinal) {
        console.log(`🗣  [${message.data.start}s → ${message.data.end}s] ${message.data.text}`);
      } else {
        console.log(`💬 Interim: ${message.data.text}`);
      }
      break;

    case "command":
      console.log(`🎙  Command detected [${message.data.id}]:`, message.data.variables);
      break;

    case "flushed":
      console.log("🔄 Buffer flushed");
      break;

    case "usage":
      console.log(`💳 Credits used: ${message.credits}`);
      break;

    case "ended":
      console.log("🏁 Session ended — server closing socket");
      break;

    case "error":
      console.error("❌ Runtime error:", message.error);
      break;
  }
});

ws.on("close", (code, reason) => {
  console.log(`🔌 Connection closed [${code}]: ${reason}`);
});

ws.on("error", (err) => {
  console.error("🚨 WebSocket error:", err.message);
});

// --- Audio sending ---

function sendAudio() {
  const AUDIO_FILE = "./sample.webm"; // swap with your audio file path

  if (!fs.existsSync(AUDIO_FILE)) {
    console.warn("⚠️  No audio file found — sending silence simulation");
    simulateAudioAndEnd();
    return;
  }

  const audioBuffer = fs.readFileSync(AUDIO_FILE);
  const CHUNK_SIZE = 8192; // ~250–500ms

commands: [
    {
      id: "delete_range",
      phrases: ["delete {delete_range}"],
      variables: [
        {
          key: "delete_range",
          type: "enum",
          enum: ["everything", "the last word", "the last sentence", "that"]
        }
      ]
    }
]