@xata.io/client 0.20.2 → 0.21.1

package/dist/index.mjs

@@ -1,5 +1,6 @@
-const defaultTrace = async (_name, fn, _options) => {
+const defaultTrace = async (name, fn, _options) => {
   return await fn({
+    name,
     setAttributes: () => {
       return;
     }
@@ -41,6 +42,18 @@ function isStringArray(value) {
 function isNumber(value) {
   return isDefined(value) && typeof value === "number";
 }
+function parseNumber(value) {
+  if (isNumber(value)) {
+    return value;
+  }
+  if (isString(value)) {
+    const parsed = Number(value);
+    if (!Number.isNaN(parsed)) {
+      return parsed;
+    }
+  }
+  return void 0;
+}
 function toBase64(value) {
   try {
     return btoa(value);
@@ -67,10 +80,13 @@ function chunk(array, chunkSize) {
   }
   return result;
 }
+async function timeout(ms) {
+  return new Promise((resolve) => setTimeout(resolve, ms));
+}
 
 function getEnvironment() {
   try {
-    if (isObject(process) && isObject(process.env)) {
+    if (isDefined(process) && isDefined(process.env)) {
       return {
         apiKey: process.env.XATA_API_KEY ?? getGlobalApiKey(),
         databaseURL: process.env.XATA_DATABASE_URL ?? getGlobalDatabaseURL(),
@@ -179,6 +195,29 @@ function getAPIKey() {
   }
 }
 
+var __accessCheck$8 = (obj, member, msg) => {
+  if (!member.has(obj))
+    throw TypeError("Cannot " + msg);
+};
+var __privateGet$8 = (obj, member, getter) => {
+  __accessCheck$8(obj, member, "read from private field");
+  return getter ? getter.call(obj) : member.get(obj);
+};
+var __privateAdd$8 = (obj, member, value) => {
+  if (member.has(obj))
+    throw TypeError("Cannot add the same private member more than once");
+  member instanceof WeakSet ? member.add(obj) : member.set(obj, value);
+};
+var __privateSet$8 = (obj, member, value, setter) => {
+  __accessCheck$8(obj, member, "write to private field");
+  setter ? setter.call(obj, value) : member.set(obj, value);
+  return value;
+};
+var __privateMethod$4 = (obj, member, method) => {
+  __accessCheck$8(obj, member, "access private method");
+  return method;
+};
+var _fetch, _queue, _concurrency, _enqueue, enqueue_fn;
 function getFetchImplementation(userFetch) {
   const globalFetch = typeof fetch !== "undefined" ? fetch : void 0;
   const fetchImpl = userFetch ?? globalFetch;
@@ -189,8 +228,74 @@ function getFetchImplementation(userFetch) {
   }
   return fetchImpl;
 }
+class ApiRequestPool {
+  constructor(concurrency = 10) {
+    __privateAdd$8(this, _enqueue);
+    __privateAdd$8(this, _fetch, void 0);
+    __privateAdd$8(this, _queue, void 0);
+    __privateAdd$8(this, _concurrency, void 0);
+    __privateSet$8(this, _queue, []);
+    __privateSet$8(this, _concurrency, concurrency);
+    this.running = 0;
+    this.started = 0;
+  }
+  setFetch(fetch2) {
+    __privateSet$8(this, _fetch, fetch2);
+  }
+  getFetch() {
+    if (!__privateGet$8(this, _fetch)) {
+      throw new Error("Fetch not set");
+    }
+    return __privateGet$8(this, _fetch);
+  }
+  request(url, options) {
+    const start = new Date();
+    const fetch2 = this.getFetch();
+    const runRequest = async (stalled = false) => {
+      const response = await fetch2(url, options);
+      if (response.status === 429) {
+        const rateLimitReset = parseNumber(response.headers?.get("x-ratelimit-reset")) ?? 1;
+        await timeout(rateLimitReset * 1e3);
+        return await runRequest(true);
+      }
+      if (stalled) {
+        const stalledTime = new Date().getTime() - start.getTime();
+        console.warn(`A request to Xata hit your workspace limits, was retried and stalled for ${stalledTime}ms`);
+      }
+      return response;
+    };
+    return __privateMethod$4(this, _enqueue, enqueue_fn).call(this, async () => {
+      return await runRequest();
+    });
+  }
+}
+_fetch = new WeakMap();
+_queue = new WeakMap();
+_concurrency = new WeakMap();
+_enqueue = new WeakSet();
+enqueue_fn = function(task) {
+  const promise = new Promise((resolve) => __privateGet$8(this, _queue).push(resolve)).finally(() => {
+    this.started--;
+    this.running++;
+  }).then(() => task()).finally(() => {
+    this.running--;
+    const next = __privateGet$8(this, _queue).shift();
+    if (next !== void 0) {
+      this.started++;
+      next();
+    }
+  });
+  if (this.running + this.started < __privateGet$8(this, _concurrency)) {
+    const next = __privateGet$8(this, _queue).shift();
+    if (next !== void 0) {
+      this.started++;
+      next();
+    }
+  }
+  return promise;
+};
 
-const VERSION = "0.20.2";
+const VERSION = "0.21.1";
 
 class ErrorWithCause extends Error {
   constructor(message, options) {
@@ -233,6 +338,7 @@ function getMessage(data) {
   }
 }
 
+const pool = new ApiRequestPool();
 const resolveUrl = (url, queryParams = {}, pathParams = {}) => {
   const cleanQueryParams = Object.entries(queryParams).reduce((acc, [key, value]) => {
     if (value === void 0 || value === null)
@@ -281,11 +387,13 @@ async function fetch$1({
   signal,
   clientID,
   sessionID,
+  clientName,
   fetchOptions = {}
 }) {
-  return trace(
+  pool.setFetch(fetchImpl);
+  return await trace(
     `${method.toUpperCase()} ${path}`,
-    async ({ setAttributes }) => {
+    async ({ name, setAttributes }) => {
       const baseUrl = buildBaseUrl({ endpoint, path, workspacesApiUrl, pathParams, apiUrl });
       const fullUrl = resolveUrl(baseUrl, queryParams, pathParams);
       const url = fullUrl.includes("localhost") ? fullUrl.replace(/^[^.]+\./, "http://") : fullUrl;
@@ -293,24 +401,27 @@ async function fetch$1({
         [TraceAttributes.HTTP_URL]: url,
         [TraceAttributes.HTTP_TARGET]: resolveUrl(path, queryParams, pathParams)
       });
-      const response = await fetchImpl(url, {
+      const xataAgent = compact([
+        ["client", "TS_SDK"],
+        ["version", VERSION],
+        isDefined(clientName) ? ["service", clientName] : void 0
+      ]).map(([key, value]) => `${key}=${value}`).join("; ");
+      const response = await pool.request(url, {
         ...fetchOptions,
         method: method.toUpperCase(),
         body: body ? JSON.stringify(body) : void 0,
         headers: {
+          "Accept-Encoding": "identity",
           "Content-Type": "application/json",
-          "User-Agent": `Xata client-ts/${VERSION}`,
           "X-Xata-Client-ID": clientID ?? "",
           "X-Xata-Session-ID": sessionID ?? "",
+          "X-Xata-Agent": xataAgent,
           ...headers,
           ...hostHeader(fullUrl),
           Authorization: `Bearer ${apiKey}`
         },
         signal
       });
-      if (response.status === 204) {
-        return {};
-      }
       const { host, protocol } = parseUrl(response.url);
       const requestId = response.headers?.get("x-request-id") ?? void 0;
       setAttributes({
@@ -320,6 +431,12 @@ async function fetch$1({
         [TraceAttributes.HTTP_HOST]: host,
         [TraceAttributes.HTTP_SCHEME]: protocol?.replace(":", "")
       });
+      if (response.status === 204) {
+        return {};
+      }
+      if (response.status === 429) {
+        throw new FetcherError(response.status, "Rate limit exceeded", requestId);
+      }
       try {
         const jsonResponse = await response.json();
         if (response.ok) {
@@ -771,7 +888,8 @@ class XataApiClient {
       workspacesApiUrl: getHostUrl(provider, "workspaces"),
       fetchImpl: getFetchImplementation(options.fetch),
       apiKey,
-      trace
+      trace,
+      clientName: options.clientName
     });
   }
   get user() {
@@ -2303,13 +2421,19 @@ class RestRepository extends Query {
         const result = await this.read(a, columns);
         return result;
       }
-      if (isString(a) && isObject(b)) {
-        const columns = isStringArray(c) ? c : void 0;
-        return __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a, b, columns, { ifVersion });
-      }
-      if (isObject(a) && isString(a.id)) {
-        const columns = isStringArray(b) ? b : void 0;
-        return __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a.id, { ...a, id: void 0 }, columns, { ifVersion });
+      try {
+        if (isString(a) && isObject(b)) {
+          const columns = isStringArray(c) ? c : void 0;
+          return await __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a, b, columns, { ifVersion });
+        }
+        if (isObject(a) && isString(a.id)) {
+          const columns = isStringArray(b) ? b : void 0;
+          return await __privateMethod$2(this, _updateRecordWithID, updateRecordWithID_fn).call(this, a.id, { ...a, id: void 0 }, columns, { ifVersion });
+        }
+      } catch (error) {
+        if (error.status === 422)
+          return null;
+        throw error;
       }
       throw new Error("Invalid arguments for update method");
     });
@@ -2440,7 +2564,9 @@ class RestRepository extends Query {
           prefix: options.prefix,
           highlight: options.highlight,
           filter: options.filter,
-          boosters: options.boosters
+          boosters: options.boosters,
+          page: options.page,
+          target: options.target
         },
         ...fetchProps
       });
@@ -2758,10 +2884,10 @@ const initObject = (db, schemaTables, table, object, selectedColumns) => {
     const value = result[column.name];
     switch (column.type) {
       case "datetime": {
-        const date = value !== void 0 ? new Date(value) : void 0;
-        if (date && isNaN(date.getTime())) {
+        const date = value !== void 0 ? new Date(value) : null;
+        if (date !== null && isNaN(date.getTime())) {
           console.error(`Failed to parse date ${value} for field ${column.name}`);
-        } else if (date) {
+        } else {
           result[column.name] = date;
         }
         break;
@@ -3029,10 +3155,10 @@ _schemaTables = new WeakMap();
 _search = new WeakSet();
 search_fn = async function(query, options, getFetchProps) {
   const fetchProps = await getFetchProps();
-  const { tables, fuzziness, highlight, prefix } = options ?? {};
+  const { tables, fuzziness, highlight, prefix, page } = options ?? {};
   const { records } = await searchBranch({
     pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", region: "{region}" },
-    body: { tables, query, fuzziness, prefix, highlight },
+    body: { tables, query, fuzziness, prefix, highlight, page },
     ...fetchProps
   });
   return records;
@@ -3050,6 +3176,22 @@ getSchemaTables_fn = async function(getFetchProps) {
   return schema.tables;
 };
 
+class TransactionPlugin extends XataPlugin {
+  build({ getFetchProps }) {
+    return {
+      run: async (operations) => {
+        const fetchProps = await getFetchProps();
+        const response = await branchTransaction({
+          pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", region: "{region}" },
+          body: { operations },
+          ...fetchProps
+        });
+        return response;
+      }
+    };
+  }
+}
+
 const isBranchStrategyBuilder = (strategy) => {
   return typeof strategy === "function";
 };
@@ -3177,8 +3319,10 @@ const buildClient = (plugins) => {
       };
       const db = new SchemaPlugin(schemaTables).build(pluginOptions);
       const search = new SearchPlugin(db, schemaTables).build(pluginOptions);
+      const transactions = new TransactionPlugin().build(pluginOptions);
       this.db = db;
       this.search = search;
+      this.transactions = transactions;
       for (const [key, namespace] of Object.entries(plugins ?? {})) {
         if (namespace === void 0)
           continue;
@@ -3210,6 +3354,7 @@ const buildClient = (plugins) => {
     const apiKey = options?.apiKey || getAPIKey();
     const cache = options?.cache ?? new SimpleCache({ defaultQueryTTL: 0 });
     const trace = options?.trace ?? defaultTrace;
+    const clientName = options?.clientName;
     const branch = async () => options?.branch !== void 0 ? await __privateMethod(this, _evaluateBranch, evaluateBranch_fn).call(this, options.branch) : await getCurrentBranchName({ apiKey, databaseURL, fetchImpl: options?.fetch });
     if (!apiKey) {
       throw new Error("Option apiKey is required");
@@ -3217,8 +3362,16 @@ const buildClient = (plugins) => {
     if (!databaseURL) {
       throw new Error("Option databaseURL is required");
     }
-    return { fetch, databaseURL, apiKey, branch, cache, trace, clientID: generateUUID(), enableBrowser };
-  }, _getFetchProps = new WeakSet(), getFetchProps_fn = async function({ fetch, apiKey, databaseURL, branch, trace, clientID }) {
+    return { fetch, databaseURL, apiKey, branch, cache, trace, clientID: generateUUID(), enableBrowser, clientName };
+  }, _getFetchProps = new WeakSet(), getFetchProps_fn = async function({
+    fetch,
+    apiKey,
+    databaseURL,
+    branch,
+    trace,
+    clientID,
+    clientName
+  }) {
     const branchValue = await __privateMethod(this, _evaluateBranch, evaluateBranch_fn).call(this, branch);
     if (!branchValue)
       throw new Error("Unable to resolve branch value");
@@ -3232,7 +3385,8 @@ const buildClient = (plugins) => {
         return databaseURL + newPath;
       },
       trace,
-      clientID
+      clientID,
+      clientName
     };
   }, _evaluateBranch = new WeakSet(), evaluateBranch_fn = async function(param) {
     if (__privateGet(this, _branch))