@pidgeon-ai/sdk 1.2.0 → 1.4.0

package/dist/index.js

@@ -43,7 +43,7 @@ function createSseClient({
   ...options
 }) {
   let lastEventId;
-  const sleep = sseSleepFn ?? ((ms) => new Promise((resolve) => setTimeout(resolve, ms)));
+  const sleep2 = sseSleepFn ?? ((ms) => new Promise((resolve) => setTimeout(resolve, ms)));
   const createStream = async function* () {
     let retryDelay = sseDefaultRetryDelay ?? 3e3;
     let attempt = 0;
@@ -147,7 +147,7 @@ function createSseClient({
           break;
         }
         const backoff = Math.min(retryDelay * 2 ** (attempt - 1), sseMaxRetryDelay ?? 3e4);
-        await sleep(backoff);
+        await sleep2(backoff);
       }
     }
   };
@@ -987,7 +987,7 @@ var messagesSend = (options) => (options.client ?? client).post({
     ...options.headers
   }
 });
-var messagesSearch = (options) => (options.client ?? client).get({
+var messagesSearch = (options) => (options?.client ?? client).get({
   security: [{ scheme: "bearer", type: "http" }],
   url: "/messages/search",
   ...options
@@ -1223,6 +1223,11 @@ var appPasswordsRevoke = (options) => (options.client ?? client).delete({
   url: "/app-passwords/{id}",
   ...options
 });
+var meGet = (options) => (options?.client ?? client).get({
+  security: [{ scheme: "bearer", type: "http" }],
+  url: "/me",
+  ...options
+});
 var workspaceGet = (options) => (options?.client ?? client).get({
   security: [{ scheme: "bearer", type: "http" }],
   url: "/workspace",
@@ -1238,6 +1243,11 @@ var eventsList = (options) => (options?.client ?? client).get({
   url: "/events",
   ...options
 });
+var eventsStream = (options) => (options?.client ?? client).sse.get({
+  security: [{ scheme: "bearer", type: "http" }],
+  url: "/events/stream",
+  ...options
+});
 var eventsGet = (options) => (options.client ?? client).get({
   security: [{ scheme: "bearer", type: "http" }],
   url: "/events/{id}",
@@ -1459,7 +1469,16 @@ var attachmentsDownloadUrl = (options) => (options.client ?? client).get({
 var SIGNATURE_HEADER = "x-pidgeon-signature";
 var TIMESTAMP_HEADER = "x-pidgeon-timestamp";
 async function verifyWebhook(input) {
-  const get = (name) => input.headers instanceof Headers ? input.headers.get(name) ?? void 0 : input.headers[name] ?? input.headers[name.toLowerCase()];
+  const get = (name) => {
+    if (input.headers instanceof Headers) return input.headers.get(name) ?? void 0;
+    const direct = input.headers[name];
+    if (direct !== void 0) return direct;
+    const wanted = name.toLowerCase();
+    for (const [key2, value] of Object.entries(input.headers)) {
+      if (key2.toLowerCase() === wanted) return value;
+    }
+    return void 0;
+  };
   const timestamp = get(TIMESTAMP_HEADER);
   const signature = get(SIGNATURE_HEADER);
   if (!timestamp || !signature) throw new Error("This request is not signed.");
@@ -1496,6 +1515,105 @@ function constantTimeEqual(a, b) {
   }
   return difference === 0;
 }
+
+// src/stream.ts
+async function* streamEvents(options) {
+  const base = (options.baseUrl ?? "https://pidgeon.ai/api/v1").replace(/\/$/, "");
+  let lastEventId;
+  let backoffMs = 1e3;
+  while (!options.signal?.aborted) {
+    const url = new URL(`${base}/events/stream`);
+    if (options.identityId) url.searchParams.set("identity_id", options.identityId);
+    if (options.type) url.searchParams.set("type", options.type);
+    if (options.after && !lastEventId) url.searchParams.set("after", options.after);
+    let response;
+    try {
+      response = await fetch(url, {
+        headers: {
+          authorization: `Bearer ${options.apiKey}`,
+          accept: "text/event-stream",
+          ...lastEventId ? { "last-event-id": lastEventId } : {}
+        },
+        signal: options.signal
+      });
+    } catch {
+      if (options.signal?.aborted) return;
+      await sleep(backoffMs, options.signal);
+      backoffMs = Math.min(backoffMs * 2, 3e4);
+      continue;
+    }
+    if (response.status === 401 || response.status === 403 || response.status === 404) {
+      throw new Error(
+        `Pidgeon refused the stream: ${response.status} ${await response.text().catch(() => "")}`.trim()
+      );
+    }
+    if (!response.ok || !response.body) {
+      await sleep(backoffMs, options.signal);
+      backoffMs = Math.min(backoffMs * 2, 3e4);
+      continue;
+    }
+    backoffMs = 1e3;
+    for await (const frame of frames(response.body, options.signal)) {
+      if (frame.id) lastEventId = frame.id;
+      if (!frame.data) continue;
+      if (frame.event === "error") continue;
+      let parsed;
+      try {
+        parsed = JSON.parse(frame.data);
+      } catch {
+        continue;
+      }
+      yield parsed;
+    }
+  }
+}
+async function* frames(body, signal) {
+  const reader = body.getReader();
+  const decoder = new TextDecoder();
+  let buffer = "";
+  try {
+    while (!signal?.aborted) {
+      const { done, value } = await reader.read();
+      if (done) return;
+      buffer += decoder.decode(value, { stream: true });
+      let boundary = buffer.indexOf("\n\n");
+      while (boundary !== -1) {
+        const block = buffer.slice(0, boundary);
+        buffer = buffer.slice(boundary + 2);
+        boundary = buffer.indexOf("\n\n");
+        const frame = {};
+        const data = [];
+        for (const line of block.split("\n")) {
+          if (line.startsWith(":")) continue;
+          const colon = line.indexOf(":");
+          if (colon === -1) continue;
+          const field = line.slice(0, colon);
+          const value2 = line.slice(colon + 1).replace(/^ /, "");
+          if (field === "id") frame.id = value2;
+          else if (field === "event") frame.event = value2;
+          else if (field === "data") data.push(value2);
+        }
+        if (data.length > 0) frame.data = data.join("\n");
+        if (frame.id || frame.data) yield frame;
+      }
+    }
+  } finally {
+    await reader.cancel().catch(() => void 0);
+  }
+}
+function sleep(ms, signal) {
+  return new Promise((resolve) => {
+    const timer = setTimeout(resolve, ms);
+    signal?.addEventListener(
+      "abort",
+      () => {
+        clearTimeout(timer);
+        resolve();
+      },
+      { once: true }
+    );
+  });
+}
 export {
   SIGNATURE_HEADER,
   TIMESTAMP_HEADER,
@@ -1539,6 +1657,7 @@ export {
   draftsUpdate,
   eventsGet,
   eventsList,
+  eventsStream,
   identitiesAccessCreate,
   identitiesAccessDelete,
   identitiesAccessGet,
@@ -1567,6 +1686,7 @@ export {
   mailboxesEmptyTrash,
   mailboxesList,
   mailboxesListFolder,
+  meGet,
   messagesAct,
   messagesForward,
   messagesGet,
@@ -1584,6 +1704,7 @@ export {
   routingDelete,
   routingList,
   routingUpdate,
+  streamEvents,
   threadsGet,
   threadsLabelsCreate,
   threadsLabelsDelete,