@arbidocs/sdk 0.3.46 → 0.3.48

package/dist/browser.cjs

@@ -3637,6 +3637,23 @@ async function generateEncryptedWorkspaceKey(arbi, wrappedKey, serverSessionKey,
   const workspaceKey = client.sealedBoxDecrypt(wrappedKey, encryptionKeyPair.secretKey);
   return client.sealKeyForSession(workspaceKey, serverSessionKey);
 }
+async function getRawWorkspaceKey(arbi, workspaceId, signingPrivateKeyBase64) {
+  const { data: workspaces, error } = await arbi.fetch.GET("/v1/user/workspaces");
+  if (error || !workspaces) {
+    throw new ArbiError("Failed to fetch workspaces");
+  }
+  const ws = workspaces.find((w2) => w2.external_id === workspaceId);
+  if (!ws || !ws.wrapped_key) {
+    throw new ArbiError(`Workspace ${workspaceId} not found or has no encryption key`);
+  }
+  const signingPrivateKey = client.base64ToBytes(signingPrivateKeyBase64);
+  const ed25519PublicKey = signingPrivateKey.slice(32, 64);
+  const encryptionKeyPair = client.deriveEncryptionKeypairFromSigning({
+    publicKey: ed25519PublicKey,
+    secretKey: signingPrivateKey
+  });
+  return client.sealedBoxDecrypt(ws.wrapped_key, encryptionKeyPair.secretKey);
+}
 async function selectWorkspaceById(arbi, workspaceId, serverSessionKey, signingPrivateKeyBase64) {
   const { data: workspaces, error } = await arbi.fetch.GET("/v1/user/workspaces");
   if (error || !workspaces) {
@@ -4106,6 +4123,7 @@ __export(documents_exports, {
   listAll: () => listAll,
   listDocuments: () => listDocuments,
   listPaginated: () => listPaginated,
+  mimeFromName: () => mimeFromName,
   sanitizeFolderPath: () => sanitizeFolderPath,
   updateDocuments: () => updateDocuments,
   uploadDocumentsDirect: () => uploadDocumentsDirect,
@@ -4160,10 +4178,17 @@ async function listDocuments(arbi) {
   return requireData(await arbi.fetch.GET("/v1/document/list"), "Failed to fetch documents");
 }
 async function* listPaginated(arbi, options = {}) {
-  const { pageSize = 5e3, order = "id_asc", fields, signal, lookahead = 3 } = options;
-  const fetchPage = async (pageOffset) => {
+  const {
+    pageSize = 5e3,
+    firstPageSize,
+    order = "id_asc",
+    fields,
+    signal,
+    lookahead = 1
+  } = options;
+  const fetchPage = async (pageOffset, limit) => {
     const query = {
-      limit: pageSize,
+      limit,
       offset: pageOffset,
       order
     };
@@ -4186,19 +4211,21 @@ async function* listPaginated(arbi, options = {}) {
   const queue = [];
   const tryEnqueue = () => {
     while (!done && queue.length < depth && issued < MAX_PAGES) {
-      const p2 = fetchPage(nextOffsetToIssue);
-      p2.catch(() => {
+      const limit = issued === 0 && firstPageSize !== void 0 ? firstPageSize : pageSize;
+      const promise = fetchPage(nextOffsetToIssue, limit);
+      promise.catch(() => {
       });
-      queue.push(p2);
-      nextOffsetToIssue += pageSize;
+      queue.push({ limit, promise });
+      nextOffsetToIssue += limit;
       issued++;
     }
   };
   tryEnqueue();
   while (queue.length > 0 && !signal?.aborted) {
-    const page = await queue.shift();
+    const { limit, promise } = queue.shift();
+    const page = await promise;
     if (signal?.aborted) return;
-    const isShort = page.length < pageSize;
+    const isShort = page.length < limit;
     if (isShort) done = true;
     tryEnqueue();
     if (page.length > 0) {
@@ -4246,11 +4273,11 @@ async function updateDocuments(arbi, documents) {
     "Failed to update documents"
   );
 }
-async function uploadUrl(arbi, urls, workspaceId, shared = false) {
+async function uploadUrl(arbi, urls, shared = false) {
   return requireData(
     await arbi.fetch.POST("/v1/document/upload-url", {
       params: {
-        query: { urls, workspace_ext_id: workspaceId, shared }
+        query: { urls, shared }
       }
     }),
     "Failed to upload from URLs"
@@ -4263,29 +4290,31 @@ async function getParsedContent(auth, docId, stage) {
   });
   return res.json();
 }
-async function uploadFile(auth, workspaceId, fileData, fileName, options) {
+async function uploadFile(auth, fileData, fileName, options) {
   const formData = new FormData();
   formData.append("files", fileData, fileName);
-  const params = new URLSearchParams({ workspace_ext_id: workspaceId });
+  const params = new URLSearchParams();
   if (options?.folder) params.set("folder", sanitizeFolderPath(options.folder));
   if (options?.configExtId) params.set("config_ext_id", options.configExtId);
+  const qs = params.toString();
   const res = await authenticatedFetch({
     ...auth,
-    path: `/v1/document/upload?${params.toString()}`,
+    path: qs ? `/v1/document/upload?${qs}` : `/v1/document/upload`,
     method: "POST",
     body: formData
   });
   return res.json();
 }
-async function uploadFiles(auth, workspaceId, files, options) {
+async function uploadFiles(auth, files, options) {
   const formData = new FormData();
   for (const f2 of files) formData.append("files", f2.data, f2.name);
-  const params = new URLSearchParams({ workspace_ext_id: workspaceId });
+  const params = new URLSearchParams();
   if (options?.folder) params.set("folder", sanitizeFolderPath(options.folder));
   if (options?.configExtId) params.set("config_ext_id", options.configExtId);
+  const qs = params.toString();
   const res = await authenticatedFetch({
     ...auth,
-    path: `/v1/document/upload?${params.toString()}`,
+    path: qs ? `/v1/document/upload?${qs}` : `/v1/document/upload`,
     method: "POST",
     body: formData
   });
@@ -4297,6 +4326,68 @@ async function downloadDocument(auth, docId) {
     path: `/v1/document/${docId}/download`
   });
 }
+function mimeFromName(name) {
+  const dot = name.lastIndexOf(".");
+  const ext = dot === -1 ? "" : name.slice(dot).toLowerCase();
+  switch (ext) {
+    case ".pdf":
+      return "application/pdf";
+    case ".txt":
+      return "text/plain";
+    case ".md":
+      return "text/markdown";
+    case ".html":
+    case ".htm":
+      return "text/html";
+    case ".doc":
+      return "application/msword";
+    case ".docx":
+      return "application/vnd.openxmlformats-officedocument.wordprocessingml.document";
+    case ".xls":
+      return "application/vnd.ms-excel";
+    case ".xlsx":
+      return "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet";
+    case ".ppt":
+      return "application/vnd.ms-powerpoint";
+    case ".pptx":
+      return "application/vnd.openxmlformats-officedocument.presentationml.presentation";
+    case ".eml":
+      return "message/rfc822";
+    default:
+      return void 0;
+  }
+}
+async function presignedPut(url, body, onProgress) {
+  const XhrGlobal = globalThis.XMLHttpRequest;
+  if (XhrGlobal) {
+    await new Promise((resolve, reject) => {
+      const xhr = new XhrGlobal();
+      xhr.open("PUT", url);
+      if (onProgress) {
+        xhr.upload.addEventListener("progress", (e) => {
+          if (e.lengthComputable) onProgress(e.loaded, e.total);
+        });
+      }
+      xhr.addEventListener("load", () => {
+        if (xhr.status >= 200 && xhr.status < 300) {
+          resolve();
+        } else {
+          const snippet = (xhr.responseText || "").slice(0, 200);
+          reject(new Error(`Presigned PUT failed: ${xhr.status} ${snippet}`));
+        }
+      });
+      xhr.addEventListener("error", () => reject(new Error("Presigned PUT network error")));
+      xhr.addEventListener("abort", () => reject(new Error("Presigned PUT aborted")));
+      xhr.send(body);
+    });
+    return;
+  }
+  const putRes = await fetch(url, { method: "PUT", body });
+  if (!putRes.ok) {
+    const snippet = await putRes.text().catch(() => "");
+    throw new Error(`Presigned PUT failed: ${putRes.status} ${snippet.slice(0, 200)}`);
+  }
+}
 async function uploadDocumentsDirect(arbi, workspaceKey, files, options) {
   if (files.length === 0) return { doc_ext_ids: [], skipped: [] };
   const SECRETBOX_ENVELOPE_OVERHEAD = 40;
@@ -4329,6 +4420,11 @@ async function uploadDocumentsDirect(arbi, workspaceKey, files, options) {
   }
   const skipped = [];
   const toCommit = [];
+  const indexByName = /* @__PURE__ */ new Map();
+  files.forEach((f2, i2) => {
+    if (!indexByName.has(f2.name)) indexByName.set(f2.name, []);
+    indexByName.get(f2.name).push(i2);
+  });
   for (const item of items) {
     if (item.status !== "uploading" || !item.upload_url || !item.doc_ext_id) {
       skipped.push({
@@ -4338,21 +4434,34 @@ async function uploadDocumentsDirect(arbi, workspaceKey, files, options) {
       continue;
     }
     const queue = bytesByName.get(item.file_name);
-    if (!queue || queue.length === 0) {
+    const idxQueue = indexByName.get(item.file_name);
+    if (!queue || queue.length === 0 || !idxQueue || idxQueue.length === 0) {
       throw new Error(`upload-init returned slot for unknown file: ${item.file_name}`);
     }
     const raw = queue.shift();
+    const fileIndex = idxQueue.shift();
     const ciphertext = client.encryptFile(raw, workspaceKey);
-    const putRes = await fetch(item.upload_url, {
-      method: "PUT",
-      body: ciphertext
-    });
-    if (!putRes.ok) {
-      const body = await putRes.text().catch(() => "");
+    const total = ciphertext.length;
+    try {
+      await presignedPut(item.upload_url, ciphertext, (loaded, putTotal) => {
+        options?.onBytesProgress?.({
+          fileIndex,
+          fileName: item.file_name,
+          loaded,
+          total: putTotal
+        });
+      });
+    } catch (err) {
       throw new Error(
-        `Presigned PUT for ${item.doc_ext_id} failed: ${putRes.status} ${body.slice(0, 200)}`
+        `Presigned PUT for ${item.doc_ext_id} failed: ${err instanceof Error ? err.message : String(err)}`
       );
     }
+    options?.onBytesProgress?.({
+      fileIndex,
+      fileName: item.file_name,
+      loaded: total,
+      total
+    });
     toCommit.push(item.doc_ext_id);
   }
   if (toCommit.length === 0) {
@@ -5193,19 +5302,19 @@ var Arbi = class {
     get: (externalIds) => getDocuments(this.requireClient(), externalIds),
     delete: (externalIds) => deleteDocuments(this.requireClient(), externalIds),
     update: (documents) => updateDocuments(this.requireClient(), documents),
-    uploadUrl: (urls, shared, workspaceId) => uploadUrl(
-      this.requireClient(),
-      urls,
-      workspaceId ?? this.requireWorkspace(),
-      shared
-    ),
-    uploadFile: (fileData, fileName, options) => uploadFile(
-      this.getAuthHeaders(),
-      options?.workspaceId ?? this.requireWorkspace(),
-      fileData,
-      fileName,
-      options?.folder ? { folder: options.folder } : void 0
-    ),
+    uploadUrl: (urls, shared) => {
+      this.requireWorkspace();
+      return uploadUrl(this.requireClient(), urls, shared);
+    },
+    uploadFile: (fileData, fileName, options) => {
+      this.requireWorkspace();
+      return uploadFile(
+        this.getAuthHeaders(),
+        fileData,
+        fileName,
+        options?.folder ? { folder: options.folder } : void 0
+      );
+    },
     download: (docId) => downloadDocument(this.getAuthHeaders(), docId),
     getParsedContent: (docId, stage) => getParsedContent(this.getAuthHeaders(), docId, stage)
   };
@@ -5443,6 +5552,7 @@ exports.formatFileSize = formatFileSize;
 exports.formatUserName = formatUserName;
 exports.formatWorkspaceChoices = formatWorkspaceChoices;
 exports.getErrorMessage = getErrorMessage;
+exports.getRawWorkspaceKey = getRawWorkspaceKey;
 exports.health = health_exports;
 exports.parseSSEEvents = parseSSEEvents;
 exports.performPasswordLogin = performPasswordLogin;