@xata.io/client 0.0.0-alpha.vf0c7bc2 → 0.0.0-alpha.vf0e0021

package/dist/index.mjs

@@ -2,13 +2,11 @@ const defaultTrace = async (_name, fn, _options) => {
   return await fn({
     setAttributes: () => {
       return;
-    },
-    onError: () => {
-      return;
     }
   });
 };
 const TraceAttributes = {
+  KIND: "xata.trace.kind",
   VERSION: "xata.sdk.version",
   TABLE: "xata.table",
   HTTP_REQUEST_ID: "http.request_id",
@@ -146,13 +144,13 @@ function getFetchImplementation(userFetch) {
   const fetchImpl = userFetch ?? globalFetch;
   if (!fetchImpl) {
     throw new Error(
-      `The \`fetch\` option passed to the Xata client is resolving to a falsy value and may not be correctly imported.`
+      `Couldn't find \`fetch\`. Install a fetch implementation such as \`node-fetch\` and pass it explicitly.`
     );
   }
   return fetchImpl;
 }
 
-const VERSION = "0.0.0-alpha.vf0c7bc2";
+const VERSION = "0.0.0-alpha.vf0e0021";
 
 class ErrorWithCause extends Error {
   constructor(message, options) {
@@ -203,7 +201,10 @@ const resolveUrl = (url, queryParams = {}, pathParams = {}) => {
   }, {});
   const query = new URLSearchParams(cleanQueryParams).toString();
   const queryString = query.length > 0 ? `?${query}` : "";
-  return url.replace(/\{\w*\}/g, (key) => pathParams[key.slice(1, -1)]) + queryString;
+  const cleanPathParams = Object.entries(pathParams).reduce((acc, [key, value]) => {
+    return { ...acc, [key]: encodeURIComponent(String(value ?? "")).replace("%3A", ":") };
+  }, {});
+  return url.replace(/\{\w*\}/g, (key) => cleanPathParams[key.slice(1, -1)]) + queryString;
 };
 function buildBaseUrl({
   path,
@@ -211,10 +212,10 @@ function buildBaseUrl({
   apiUrl,
   pathParams
 }) {
-  if (!pathParams?.workspace)
+  if (pathParams?.workspace === void 0)
     return `${apiUrl}${path}`;
   const url = typeof workspacesApiUrl === "string" ? `${workspacesApiUrl}${path}` : workspacesApiUrl(path, pathParams);
-  return url.replace("{workspaceId}", pathParams.workspace);
+  return url.replace("{workspaceId}", String(pathParams.workspace));
 }
 function hostHeader(url) {
   const pattern = /.*:\/\/(?<host>[^/]+).*/;
@@ -236,7 +237,7 @@ async function fetch$1({
 }) {
   return trace(
     `${method.toUpperCase()} ${path}`,
-    async ({ setAttributes, onError }) => {
+    async ({ setAttributes }) => {
       const baseUrl = buildBaseUrl({ path, workspacesApiUrl, pathParams, apiUrl });
       const fullUrl = resolveUrl(baseUrl, queryParams, pathParams);
       const url = fullUrl.includes("localhost") ? fullUrl.replace(/^[^.]+\./, "http://") : fullUrl;
@@ -261,6 +262,7 @@ async function fetch$1({
       const { host, protocol } = parseUrl(response.url);
       const requestId = response.headers?.get("x-request-id") ?? void 0;
       setAttributes({
+        [TraceAttributes.KIND]: "http",
         [TraceAttributes.HTTP_REQUEST_ID]: requestId,
         [TraceAttributes.HTTP_STATUS_CODE]: response.status,
         [TraceAttributes.HTTP_HOST]: host,
@@ -273,9 +275,7 @@ async function fetch$1({
         }
         throw new FetcherError(response.status, jsonResponse, requestId);
       } catch (error) {
-        const fetcherError = new FetcherError(response.status, error, requestId);
-        onError(fetcherError.message);
-        throw fetcherError;
+        throw new FetcherError(response.status, error, requestId);
       }
     },
     { [TraceAttributes.HTTP_METHOD]: method.toUpperCase(), [TraceAttributes.HTTP_ROUTE]: path }
@@ -386,6 +386,7 @@ const getDatabaseMetadata = (variables) => fetch$1({
   method: "get",
   ...variables
 });
+const updateDatabaseMetadata = (variables) => fetch$1({ url: "/dbs/{dbName}/metadata", method: "patch", ...variables });
 const getGitBranchesMapping = (variables) => fetch$1({ url: "/dbs/{dbName}/gitBranches", method: "get", ...variables });
 const addGitBranchesEntry = (variables) => fetch$1({ url: "/dbs/{dbName}/gitBranches", method: "post", ...variables });
 const removeGitBranchesEntry = (variables) => fetch$1({ url: "/dbs/{dbName}/gitBranches", method: "delete", ...variables });
@@ -394,6 +395,22 @@ const resolveBranch = (variables) => fetch$1({
   method: "get",
   ...variables
 });
+const listMigrationRequests = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/list", method: "post", ...variables });
+const createMigrationRequest = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations", method: "post", ...variables });
+const getMigrationRequest = (variables) => fetch$1({
+  url: "/dbs/{dbName}/migrations/{mrNumber}",
+  method: "get",
+  ...variables
+});
+const updateMigrationRequest = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}", method: "patch", ...variables });
+const listMigrationRequestsCommits = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}/commits", method: "post", ...variables });
+const compareMigrationRequest = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}/compare", method: "post", ...variables });
+const getMigrationRequestIsMerged = (variables) => fetch$1({ url: "/dbs/{dbName}/migrations/{mrNumber}/merge", method: "get", ...variables });
+const mergeMigrationRequest = (variables) => fetch$1({
+  url: "/dbs/{dbName}/migrations/{mrNumber}/merge",
+  method: "post",
+  ...variables
+});
 const getBranchDetails = (variables) => fetch$1({
   url: "/db/{dbBranchName}",
   method: "get",
@@ -418,6 +435,16 @@ const getBranchMetadata = (variables) => fetch$1({
 const getBranchMigrationHistory = (variables) => fetch$1({ url: "/db/{dbBranchName}/migrations", method: "get", ...variables });
 const executeBranchMigrationPlan = (variables) => fetch$1({ url: "/db/{dbBranchName}/migrations/execute", method: "post", ...variables });
 const getBranchMigrationPlan = (variables) => fetch$1({ url: "/db/{dbBranchName}/migrations/plan", method: "post", ...variables });
+const compareBranchWithUserSchema = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/compare", method: "post", ...variables });
+const compareBranchSchemas = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/compare/{branchName}", method: "post", ...variables });
+const updateBranchSchema = (variables) => fetch$1({
+  url: "/db/{dbBranchName}/schema/update",
+  method: "post",
+  ...variables
+});
+const previewBranchSchemaEdit = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/preview", method: "post", ...variables });
+const applyBranchSchemaEdit = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/apply", method: "post", ...variables });
+const getBranchSchemaHistory = (variables) => fetch$1({ url: "/db/{dbBranchName}/schema/history", method: "post", ...variables });
 const getBranchStats = (variables) => fetch$1({
   url: "/db/{dbBranchName}/stats",
   method: "get",
@@ -503,6 +530,26 @@ const searchBranch = (variables) => fetch$1({
   method: "post",
   ...variables
 });
+const summarizeTable = (variables) => fetch$1({
+  url: "/db/{dbBranchName}/tables/{tableName}/summarize",
+  method: "post",
+  ...variables
+});
+const cPgetDatabaseList = (variables) => fetch$1({
+  url: "/workspaces/{workspaceId}/dbs",
+  method: "get",
+  ...variables
+});
+const cPcreateDatabase = (variables) => fetch$1({ url: "/workspaces/{workspaceId}/dbs/{dbName}", method: "put", ...variables });
+const cPdeleteDatabase = (variables) => fetch$1({
+  url: "/workspaces/{workspaceId}/dbs/{dbName}",
+  method: "delete",
+  ...variables
+});
+const cPgetCPDatabaseMetadata = (variables) => fetch$1(
+  { url: "/workspaces/{workspaceId}/dbs/{dbName}/metadata", method: "get", ...variables }
+);
+const cPupdateCPDatabaseMetadata = (variables) => fetch$1({ url: "/workspaces/{workspaceId}/dbs/{dbName}/metadata", method: "patch", ...variables });
 const operationsByTag = {
   users: { getUser, updateUser, deleteUser, getUserAPIKeys, createUserAPIKey, deleteUserAPIKey },
   workspaces: {
@@ -525,6 +572,7 @@ const operationsByTag = {
     createDatabase,
     deleteDatabase,
     getDatabaseMetadata,
+    updateDatabaseMetadata,
     getGitBranchesMapping,
     addGitBranchesEntry,
     removeGitBranchesEntry,
@@ -537,10 +585,28 @@ const operationsByTag = {
     deleteBranch,
     updateBranchMetadata,
     getBranchMetadata,
+    getBranchStats
+  },
+  migrationRequests: {
+    listMigrationRequests,
+    createMigrationRequest,
+    getMigrationRequest,
+    updateMigrationRequest,
+    listMigrationRequestsCommits,
+    compareMigrationRequest,
+    getMigrationRequestIsMerged,
+    mergeMigrationRequest
+  },
+  branchSchema: {
     getBranchMigrationHistory,
     executeBranchMigrationPlan,
     getBranchMigrationPlan,
-    getBranchStats
+    compareBranchWithUserSchema,
+    compareBranchSchemas,
+    updateBranchSchema,
+    previewBranchSchemaEdit,
+    applyBranchSchemaEdit,
+    getBranchSchemaHistory
   },
   table: {
     createTable,
@@ -564,14 +630,22 @@ const operationsByTag = {
     bulkInsertTableRecords,
     queryTable,
     searchTable,
-    searchBranch
+    searchBranch,
+    summarizeTable
+  },
+  databases: {
+    cPgetDatabaseList,
+    cPcreateDatabase,
+    cPdeleteDatabase,
+    cPgetCPDatabaseMetadata,
+    cPupdateCPDatabaseMetadata
   }
 };
 
 function getHostUrl(provider, type) {
-  if (isValidAlias(provider)) {
+  if (isHostProviderAlias(provider)) {
     return providers[provider][type];
-  } else if (isValidBuilder(provider)) {
+  } else if (isHostProviderBuilder(provider)) {
     return provider[type];
   }
   throw new Error("Invalid API provider");
@@ -586,10 +660,10 @@ const providers = {
     workspaces: "https://{workspaceId}.staging.xatabase.co"
   }
 };
-function isValidAlias(alias) {
+function isHostProviderAlias(alias) {
   return isString(alias) && Object.keys(providers).includes(alias);
 }
-function isValidBuilder(builder) {
+function isHostProviderBuilder(builder) {
   return isObject(builder) && isString(builder.main) && isString(builder.workspaces);
 }
 
@@ -660,6 +734,16 @@ class XataApiClient {
       __privateGet$7(this, _namespaces).records = new RecordsApi(__privateGet$7(this, _extraProps));
     return __privateGet$7(this, _namespaces).records;
   }
+  get migrationRequests() {
+    if (!__privateGet$7(this, _namespaces).migrationRequests)
+      __privateGet$7(this, _namespaces).migrationRequests = new MigrationRequestsApi(__privateGet$7(this, _extraProps));
+    return __privateGet$7(this, _namespaces).migrationRequests;
+  }
+  get branchSchema() {
+    if (!__privateGet$7(this, _namespaces).branchSchema)
+      __privateGet$7(this, _namespaces).branchSchema = new BranchSchemaApi(__privateGet$7(this, _extraProps));
+    return __privateGet$7(this, _namespaces).branchSchema;
+  }
 }
 _extraProps = new WeakMap();
 _namespaces = new WeakMap();
@@ -805,6 +889,13 @@ class DatabaseApi {
       ...this.extraProps
     });
   }
+  updateDatabaseMetadata(workspace, dbName, options = {}) {
+    return operationsByTag.database.updateDatabaseMetadata({
+      pathParams: { workspace, dbName },
+      body: options,
+      ...this.extraProps
+    });
+  }
   getGitBranchesMapping(workspace, dbName) {
     return operationsByTag.database.getGitBranchesMapping({
       pathParams: { workspace, dbName },
@@ -876,27 +967,6 @@ class BranchApi {
       ...this.extraProps
     });
   }
-  getBranchMigrationHistory(workspace, database, branch, options = {}) {
-    return operationsByTag.branch.getBranchMigrationHistory({
-      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
-      body: options,
-      ...this.extraProps
-    });
-  }
-  executeBranchMigrationPlan(workspace, database, branch, migrationPlan) {
-    return operationsByTag.branch.executeBranchMigrationPlan({
-      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
-      body: migrationPlan,
-      ...this.extraProps
-    });
-  }
-  getBranchMigrationPlan(workspace, database, branch, schema) {
-    return operationsByTag.branch.getBranchMigrationPlan({
-      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
-      body: schema,
-      ...this.extraProps
-    });
-  }
   getBranchStats(workspace, database, branch) {
     return operationsByTag.branch.getBranchStats({
       pathParams: { workspace, dbBranchName: `${database}:${branch}` },
@@ -1052,6 +1122,138 @@ class RecordsApi {
       ...this.extraProps
     });
   }
+  summarizeTable(workspace, database, branch, tableName, query) {
+    return operationsByTag.records.summarizeTable({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}`, tableName },
+      body: query,
+      ...this.extraProps
+    });
+  }
+}
+class MigrationRequestsApi {
+  constructor(extraProps) {
+    this.extraProps = extraProps;
+  }
+  listMigrationRequests(workspace, database, options = {}) {
+    return operationsByTag.migrationRequests.listMigrationRequests({
+      pathParams: { workspace, dbName: database },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  createMigrationRequest(workspace, database, options) {
+    return operationsByTag.migrationRequests.createMigrationRequest({
+      pathParams: { workspace, dbName: database },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  getMigrationRequest(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.getMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+  updateMigrationRequest(workspace, database, migrationRequest, options) {
+    return operationsByTag.migrationRequests.updateMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  listMigrationRequestsCommits(workspace, database, migrationRequest, options = {}) {
+    return operationsByTag.migrationRequests.listMigrationRequestsCommits({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  compareMigrationRequest(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.compareMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+  getMigrationRequestIsMerged(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.getMigrationRequestIsMerged({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+  mergeMigrationRequest(workspace, database, migrationRequest) {
+    return operationsByTag.migrationRequests.mergeMigrationRequest({
+      pathParams: { workspace, dbName: database, mrNumber: migrationRequest },
+      ...this.extraProps
+    });
+  }
+}
+class BranchSchemaApi {
+  constructor(extraProps) {
+    this.extraProps = extraProps;
+  }
+  getBranchMigrationHistory(workspace, database, branch, options = {}) {
+    return operationsByTag.branchSchema.getBranchMigrationHistory({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: options,
+      ...this.extraProps
+    });
+  }
+  executeBranchMigrationPlan(workspace, database, branch, migrationPlan) {
+    return operationsByTag.branchSchema.executeBranchMigrationPlan({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: migrationPlan,
+      ...this.extraProps
+    });
+  }
+  getBranchMigrationPlan(workspace, database, branch, schema) {
+    return operationsByTag.branchSchema.getBranchMigrationPlan({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: schema,
+      ...this.extraProps
+    });
+  }
+  compareBranchWithUserSchema(workspace, database, branch, schema) {
+    return operationsByTag.branchSchema.compareBranchWithUserSchema({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: { schema },
+      ...this.extraProps
+    });
+  }
+  compareBranchSchemas(workspace, database, branch, branchName, schema) {
+    return operationsByTag.branchSchema.compareBranchSchemas({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}`, branchName },
+      body: { schema },
+      ...this.extraProps
+    });
+  }
+  updateBranchSchema(workspace, database, branch, migration) {
+    return operationsByTag.branchSchema.updateBranchSchema({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: migration,
+      ...this.extraProps
+    });
+  }
+  previewBranchSchemaEdit(workspace, database, branch, migration) {
+    return operationsByTag.branchSchema.previewBranchSchemaEdit({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: migration,
+      ...this.extraProps
+    });
+  }
+  applyBranchSchemaEdit(workspace, database, branch, edits) {
+    return operationsByTag.branchSchema.applyBranchSchemaEdit({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: { edits },
+      ...this.extraProps
+    });
+  }
+  getBranchSchemaHistory(workspace, database, branch, options = {}) {
+    return operationsByTag.branchSchema.getBranchSchemaHistory({
+      pathParams: { workspace, dbBranchName: `${database}:${branch}` },
+      body: options,
+      ...this.extraProps
+    });
+  }
 }
 
 class XataApiPlugin {
@@ -1177,9 +1379,14 @@ var __privateSet$5 = (obj, member, value, setter) => {
   setter ? setter.call(obj, value) : member.set(obj, value);
   return value;
 };
-var _table$1, _repository, _data;
+var __privateMethod$3 = (obj, member, method) => {
+  __accessCheck$5(obj, member, "access private method");
+  return method;
+};
+var _table$1, _repository, _data, _cleanFilterConstraint, cleanFilterConstraint_fn;
 const _Query = class {
   constructor(repository, table, data, rawParent) {
+    __privateAdd$5(this, _cleanFilterConstraint);
     __privateAdd$5(this, _table$1, void 0);
     __privateAdd$5(this, _repository, void 0);
     __privateAdd$5(this, _data, { filter: {} });
@@ -1236,11 +1443,14 @@ const _Query = class {
   }
   filter(a, b) {
     if (arguments.length === 1) {
-      const constraints = Object.entries(a).map(([column, constraint]) => ({ [column]: constraint }));
+      const constraints = Object.entries(a ?? {}).map(([column, constraint]) => ({
+        [column]: __privateMethod$3(this, _cleanFilterConstraint, cleanFilterConstraint_fn).call(this, column, constraint)
+      }));
       const $all = compact([__privateGet$5(this, _data).filter?.$all].flat().concat(constraints));
       return new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), { filter: { $all } }, __privateGet$5(this, _data));
     } else {
-      const $all = compact([__privateGet$5(this, _data).filter?.$all].flat().concat([{ [a]: b }]));
+      const constraints = isDefined(a) && isDefined(b) ? [{ [a]: __privateMethod$3(this, _cleanFilterConstraint, cleanFilterConstraint_fn).call(this, a, b) }] : void 0;
+      const $all = compact([__privateGet$5(this, _data).filter?.$all].flat().concat(constraints));
       return new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), { filter: { $all } }, __privateGet$5(this, _data));
     }
   }
@@ -1278,11 +1488,20 @@ const _Query = class {
     }
   }
   async getMany(options = {}) {
-    const page = await this.getPaginated(options);
+    const { pagination = {}, ...rest } = options;
+    const { size = PAGINATION_DEFAULT_SIZE, offset } = pagination;
+    const batchSize = size <= PAGINATION_MAX_SIZE ? size : PAGINATION_MAX_SIZE;
+    let page = await this.getPaginated({ ...rest, pagination: { size: batchSize, offset } });
+    const results = [...page.records];
+    while (page.hasNextPage() && results.length < size) {
+      page = await page.nextPage();
+      results.push(...page.records);
+    }
     if (page.hasNextPage() && options.pagination?.size === void 0) {
       console.trace("Calling getMany does not return all results. Paginate to get all results or call getAll.");
     }
-    return page.records;
+    const array = new RecordArray(page, results.slice(0, size));
+    return array;
   }
   async getAll(options = {}) {
     const { batchSize = PAGINATION_MAX_SIZE, ...rest } = options;
@@ -1296,6 +1515,12 @@ const _Query = class {
     const records = await this.getMany({ ...options, pagination: { size: 1 } });
     return records[0] ?? null;
   }
+  async getFirstOrThrow(options = {}) {
+    const records = await this.getMany({ ...options, pagination: { size: 1 } });
+    if (records[0] === void 0)
+      throw new Error("No results found.");
+    return records[0];
+  }
   cache(ttl) {
     return new _Query(__privateGet$5(this, _repository), __privateGet$5(this, _table$1), { cache: ttl }, __privateGet$5(this, _data));
   }
@@ -1319,6 +1544,17 @@ let Query = _Query;
 _table$1 = new WeakMap();
 _repository = new WeakMap();
 _data = new WeakMap();
+_cleanFilterConstraint = new WeakSet();
+cleanFilterConstraint_fn = function(column, value) {
+  const columnType = __privateGet$5(this, _table$1).schema?.columns.find(({ name }) => name === column)?.type;
+  if (columnType === "multiple" && (isString(value) || isStringArray(value))) {
+    return { $includes: value };
+  }
+  if (columnType === "link" && isObject(value) && isString(value.id)) {
+    return value.id;
+  }
+  return value;
+};
 function cleanParent(data, parent) {
   if (isCursorPaginationOptions(data.pagination)) {
     return { ...parent, sorting: void 0, filter: void 0 };
@@ -1385,7 +1621,11 @@ class Repository extends Query {
 }
 class RestRepository extends Query {
   constructor(options) {
-    super(null, options.table, {});
+    super(
+      null,
+      { name: options.table, schema: options.schemaTables?.find((table) => table.name === options.table) },
+      {}
+    );
     __privateAdd$4(this, _insertRecordWithoutId);
     __privateAdd$4(this, _insertRecordWithId);
     __privateAdd$4(this, _bulkInsertTableRecords);
@@ -1411,6 +1651,7 @@ class RestRepository extends Query {
       return trace(name, fn, {
         ...options2,
         [TraceAttributes.TABLE]: __privateGet$4(this, _table),
+        [TraceAttributes.KIND]: "sdk-operation",
         [TraceAttributes.VERSION]: VERSION
       });
     });
@@ -1448,16 +1689,16 @@ class RestRepository extends Query {
       if (Array.isArray(a)) {
         if (a.length === 0)
           return [];
-        const ids = a.map((item) => isString(item) ? item : item.id).filter((id2) => isString(id2));
-        const finalObjects = await this.getAll({ filter: { id: { $any: ids } }, columns });
+        const ids = a.map((item) => extractId(item));
+        const finalObjects = await this.getAll({ filter: { id: { $any: compact(ids) } }, columns });
         const dictionary = finalObjects.reduce((acc, object) => {
           acc[object.id] = object;
           return acc;
         }, {});
-        return ids.map((id2) => dictionary[id2] ?? null);
+        return ids.map((id2) => dictionary[id2 ?? ""] ?? null);
       }
-      const id = isString(a) ? a : a.id;
-      if (isString(id)) {
+      const id = extractId(a);
+      if (id) {
         const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
         try {
           const response = await getRecord({
@@ -1471,7 +1712,7 @@ class RestRepository extends Query {
             ...fetchProps
           });
           const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-          return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response);
+          return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
         } catch (e) {
           if (isObject(e) && e.status === 404) {
             return null;
@@ -1482,6 +1723,25 @@ class RestRepository extends Query {
       return null;
     });
   }
+  async readOrThrow(a, b) {
+    return __privateGet$4(this, _trace).call(this, "readOrThrow", async () => {
+      const result = await this.read(a, b);
+      if (Array.isArray(result)) {
+        const missingIds = compact(
+          a.filter((_item, index) => result[index] === null).map((item) => extractId(item))
+        );
+        if (missingIds.length > 0) {
+          throw new Error(`Could not find records with ids: ${missingIds.join(", ")}`);
+        }
+        return result;
+      }
+      if (result === null) {
+        const id = extractId(a) ?? "unknown";
+        throw new Error(`Record with id ${id} not found`);
+      }
+      return result;
+    });
+  }
   async update(a, b, c) {
     return __privateGet$4(this, _trace).call(this, "update", async () => {
       if (Array.isArray(a)) {
@@ -1504,6 +1764,25 @@ class RestRepository extends Query {
       throw new Error("Invalid arguments for update method");
     });
   }
+  async updateOrThrow(a, b, c) {
+    return __privateGet$4(this, _trace).call(this, "updateOrThrow", async () => {
+      const result = await this.update(a, b, c);
+      if (Array.isArray(result)) {
+        const missingIds = compact(
+          a.filter((_item, index) => result[index] === null).map((item) => extractId(item))
+        );
+        if (missingIds.length > 0) {
+          throw new Error(`Could not find records with ids: ${missingIds.join(", ")}`);
+        }
+        return result;
+      }
+      if (result === null) {
+        const id = extractId(a) ?? "unknown";
+        throw new Error(`Record with id ${id} not found`);
+      }
+      return result;
+    });
+  }
   async createOrUpdate(a, b, c) {
     return __privateGet$4(this, _trace).call(this, "createOrUpdate", async () => {
       if (Array.isArray(a)) {
@@ -1526,28 +1805,43 @@ class RestRepository extends Query {
       throw new Error("Invalid arguments for createOrUpdate method");
     });
   }
-  async delete(a) {
+  async delete(a, b) {
     return __privateGet$4(this, _trace).call(this, "delete", async () => {
       if (Array.isArray(a)) {
         if (a.length === 0)
-          return;
+          return [];
         if (a.length > 100) {
           console.warn("Bulk delete operation is not optimized in the Xata API yet, this request might be slow");
         }
-        await Promise.all(a.map((id) => this.delete(id)));
-        return;
+        return Promise.all(a.map((id) => this.delete(id, b)));
       }
       if (isString(a)) {
-        await __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a);
-        return;
+        return __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a, b);
       }
       if (isObject(a) && isString(a.id)) {
-        await __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a.id);
-        return;
+        return __privateMethod$2(this, _deleteRecord, deleteRecord_fn).call(this, a.id, b);
       }
       throw new Error("Invalid arguments for delete method");
     });
   }
+  async deleteOrThrow(a, b) {
+    return __privateGet$4(this, _trace).call(this, "deleteOrThrow", async () => {
+      const result = await this.delete(a, b);
+      if (Array.isArray(result)) {
+        const missingIds = compact(
+          a.filter((_item, index) => result[index] === null).map((item) => extractId(item))
+        );
+        if (missingIds.length > 0) {
+          throw new Error(`Could not find records with ids: ${missingIds.join(", ")}`);
+        }
+        return result;
+      } else if (result === null) {
+        const id = extractId(a) ?? "unknown";
+        throw new Error(`Record with id ${id} not found`);
+      }
+      return result;
+    });
+  }
   async search(query, options = {}) {
     return __privateGet$4(this, _trace).call(this, "search", async () => {
       const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
@@ -1564,7 +1858,7 @@ class RestRepository extends Query {
         ...fetchProps
       });
       const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-      return records.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item));
+      return records.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item, ["*"]));
     });
   }
   async query(query) {
@@ -1574,7 +1868,7 @@ class RestRepository extends Query {
         return new Page(query, cacheQuery.meta, cacheQuery.records);
       const data = query.getQueryOptions();
       const body = {
-        filter: Object.values(data.filter ?? {}).some(Boolean) ? data.filter : void 0,
+        filter: cleanFilter(data.filter),
         sort: data.sort !== void 0 ? buildSortFilter(data.sort) : void 0,
         page: data.pagination,
         columns: data.columns
@@ -1586,7 +1880,9 @@ class RestRepository extends Query {
         ...fetchProps
       });
       const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-      const records = objects.map((record) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), record));
+      const records = objects.map(
+        (record) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), record, data.columns ?? ["*"])
+      );
       await __privateMethod$2(this, _setCacheQuery, setCacheQuery_fn).call(this, query, meta, records);
       return new Page(query, meta, records);
     });
@@ -1613,7 +1909,7 @@ insertRecordWithoutId_fn = async function(object, columns = ["*"]) {
     ...fetchProps
   });
   const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-  return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response);
+  return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
 };
 _insertRecordWithId = new WeakSet();
 insertRecordWithId_fn = async function(recordId, object, columns = ["*"]) {
@@ -1631,7 +1927,7 @@ insertRecordWithId_fn = async function(recordId, object, columns = ["*"]) {
     ...fetchProps
   });
   const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-  return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response);
+  return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
 };
 _bulkInsertTableRecords = new WeakSet();
 bulkInsertTableRecords_fn = async function(objects, columns = ["*"]) {
@@ -1647,20 +1943,27 @@ bulkInsertTableRecords_fn = async function(objects, columns = ["*"]) {
     throw new Error("Request included columns but server didn't include them");
   }
   const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-  return response.records?.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item));
+  return response.records?.map((item) => initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), item, columns));
 };
 _updateRecordWithID = new WeakSet();
 updateRecordWithID_fn = async function(recordId, object, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
   const record = transformObjectLinks(object);
-  const response = await updateRecordWithID({
-    pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
-    queryParams: { columns },
-    body: record,
-    ...fetchProps
-  });
-  const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-  return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response);
+  try {
+    const response = await updateRecordWithID({
+      pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
+      queryParams: { columns },
+      body: record,
+      ...fetchProps
+    });
+    const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
+    return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
+  } catch (e) {
+    if (isObject(e) && e.status === 404) {
+      return null;
+    }
+    throw e;
+  }
 };
 _upsertRecordWithID = new WeakSet();
 upsertRecordWithID_fn = async function(recordId, object, columns = ["*"]) {
@@ -1672,15 +1975,25 @@ upsertRecordWithID_fn = async function(recordId, object, columns = ["*"]) {
     ...fetchProps
   });
   const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
-  return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response);
+  return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
 };
 _deleteRecord = new WeakSet();
-deleteRecord_fn = async function(recordId) {
+deleteRecord_fn = async function(recordId, columns = ["*"]) {
   const fetchProps = await __privateGet$4(this, _getFetchProps).call(this);
-  await deleteRecord({
-    pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
-    ...fetchProps
-  });
+  try {
+    const response = await deleteRecord({
+      pathParams: { workspace: "{workspaceId}", dbBranchName: "{dbBranch}", tableName: __privateGet$4(this, _table), recordId },
+      queryParams: { columns },
+      ...fetchProps
+    });
+    const schemaTables = await __privateMethod$2(this, _getSchemaTables$1, getSchemaTables_fn$1).call(this);
+    return initObject(__privateGet$4(this, _db), schemaTables, __privateGet$4(this, _table), response, columns);
+  } catch (e) {
+    if (isObject(e) && e.status === 404) {
+      return null;
+    }
+    throw e;
+  }
 };
 _setCacheQuery = new WeakSet();
 setCacheQuery_fn = async function(query, meta, records) {
@@ -1717,7 +2030,7 @@ const transformObjectLinks = (object) => {
     return { ...acc, [key]: isIdentifiable(value) ? value.id : value };
   }, {});
 };
-const initObject = (db, schemaTables, table, object) => {
+const initObject = (db, schemaTables, table, object, selectedColumns) => {
   const result = {};
   const { xata, ...rest } = object ?? {};
   Object.assign(result, rest);
@@ -1725,6 +2038,8 @@ const initObject = (db, schemaTables, table, object) => {
   if (!columns)
     console.error(`Table ${table} not found in schema`);
   for (const column of columns ?? []) {
+    if (!isValidColumn(selectedColumns, column))
+      continue;
     const value = result[column.name];
     switch (column.type) {
       case "datetime": {
@@ -1741,10 +2056,28 @@ const initObject = (db, schemaTables, table, object) => {
         if (!linkTable) {
           console.error(`Failed to parse link for field ${column.name}`);
         } else if (isObject(value)) {
-          result[column.name] = initObject(db, schemaTables, linkTable, value);
+          const selectedLinkColumns = selectedColumns.reduce((acc, item) => {
+            if (item === column.name) {
+              return [...acc, "*"];
+            }
+            if (item.startsWith(`${column.name}.`)) {
+              const [, ...path] = item.split(".");
+              return [...acc, path.join(".")];
+            }
+            return acc;
+          }, []);
+          result[column.name] = initObject(db, schemaTables, linkTable, value, selectedLinkColumns);
+        } else {
+          result[column.name] = null;
         }
         break;
       }
+      default:
+        result[column.name] = value ?? null;
+        if (column.notNull === true && value === null) {
+          console.error(`Parse error, column ${column.name} is non nullable and value resolves null`);
+        }
+        break;
     }
   }
   result.read = function(columns2) {
@@ -1768,6 +2101,28 @@ const initObject = (db, schemaTables, table, object) => {
 function isResponseWithRecords(value) {
   return isObject(value) && Array.isArray(value.records);
 }
+function extractId(value) {
+  if (isString(value))
+    return value;
+  if (isObject(value) && isString(value.id))
+    return value.id;
+  return void 0;
+}
+function cleanFilter(filter) {
+  if (!filter)
+    return void 0;
+  const values = Object.values(filter).filter(Boolean).filter((value) => Array.isArray(value) ? value.length > 0 : true);
+  return values.length > 0 ? filter : void 0;
+}
+function isValidColumn(columns, column) {
+  if (columns.includes("*"))
+    return true;
+  if (column.type === "link") {
+    const linkColumns = columns.filter((item) => item.startsWith(column.name));
+    return linkColumns.length > 0;
+  }
+  return columns.includes(column.name);
+}
 
 var __accessCheck$3 = (obj, member, msg) => {
   if (!member.has(obj))
@@ -1818,18 +2173,25 @@ class SimpleCache {
 }
 _map = new WeakMap();
 
-const gt = (value) => ({ $gt: value });
-const ge = (value) => ({ $ge: value });
-const gte = (value) => ({ $ge: value });
-const lt = (value) => ({ $lt: value });
-const lte = (value) => ({ $le: value });
-const le = (value) => ({ $le: value });
+const greaterThan = (value) => ({ $gt: value });
+const gt = greaterThan;
+const greaterThanEquals = (value) => ({ $ge: value });
+const greaterEquals = greaterThanEquals;
+const gte = greaterThanEquals;
+const ge = greaterThanEquals;
+const lessThan = (value) => ({ $lt: value });
+const lt = lessThan;
+const lessThanEquals = (value) => ({ $le: value });
+const lessEquals = lessThanEquals;
+const lte = lessThanEquals;
+const le = lessThanEquals;
 const exists = (column) => ({ $exists: column });
 const notExists = (column) => ({ $notExists: column });
 const startsWith = (value) => ({ $startsWith: value });
 const endsWith = (value) => ({ $endsWith: value });
 const pattern = (value) => ({ $pattern: value });
 const is = (value) => ({ $is: value });
+const equals = is;
 const isNot = (value) => ({ $isNot: value });
 const contains = (value) => ({ $contains: value });
 const includes = (value) => ({ $includes: value });
@@ -1926,7 +2288,7 @@ class SearchPlugin extends XataPlugin {
         const schemaTables = await __privateMethod$1(this, _getSchemaTables, getSchemaTables_fn).call(this, getFetchProps);
         return records.map((record) => {
           const { table = "orphan" } = record.xata;
-          return { table, record: initObject(this.db, schemaTables, table, record) };
+          return { table, record: initObject(this.db, schemaTables, table, record, ["*"]) };
         });
       },
       byTable: async (query, options = {}) => {
@@ -1935,7 +2297,7 @@ class SearchPlugin extends XataPlugin {
         return records.reduce((acc, record) => {
           const { table = "orphan" } = record.xata;
           const items = acc[table] ?? [];
-          const item = initObject(this.db, schemaTables, table, record);
+          const item = initObject(this.db, schemaTables, table, record, ["*"]);
           return { ...acc, [table]: [...items, item] };
         }, {});
       }
@@ -2116,8 +2478,11 @@ const buildClient = (plugins) => {
     const cache = options?.cache ?? new SimpleCache({ defaultQueryTTL: 0 });
     const trace = options?.trace ?? defaultTrace;
     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 (!databaseURL || !apiKey) {
-      throw new Error("Options databaseURL and apiKey are required");
+    if (!apiKey) {
+      throw new Error("Option apiKey is required");
+    }
+    if (!databaseURL) {
+      throw new Error("Option databaseURL is required");
     }
     return { fetch, databaseURL, apiKey, branch, cache, trace };
   }, _getFetchProps = new WeakSet(), getFetchProps_fn = async function({ fetch, apiKey, databaseURL, branch, trace }) {
@@ -2130,7 +2495,7 @@ const buildClient = (plugins) => {
       apiUrl: "",
       workspacesApiUrl: (path, params) => {
         const hasBranch = params.dbBranchName ?? params.branch;
-        const newPath = path.replace(/^\/db\/[^/]+/, hasBranch ? `:${branchValue}` : "");
+        const newPath = path.replace(/^\/db\/[^/]+/, hasBranch !== void 0 ? `:${branchValue}` : "");
         return databaseURL + newPath;
       },
       trace
@@ -2245,5 +2610,5 @@ class XataError extends Error {
   }
 }
 
-export { BaseClient, operationsByTag as Operations, PAGINATION_DEFAULT_OFFSET, PAGINATION_DEFAULT_SIZE, PAGINATION_MAX_OFFSET, PAGINATION_MAX_SIZE, Page, Query, RecordArray, Repository, RestRepository, SchemaPlugin, SearchPlugin, Serializer, SimpleCache, XataApiClient, XataApiPlugin, XataError, XataPlugin, acceptWorkspaceMemberInvite, addGitBranchesEntry, addTableColumn, buildClient, buildWorkerRunner, bulkInsertTableRecords, cancelWorkspaceMemberInvite, contains, createBranch, createDatabase, createTable, createUserAPIKey, createWorkspace, deleteBranch, deleteColumn, deleteDatabase, deleteRecord, deleteTable, deleteUser, deleteUserAPIKey, deleteWorkspace, deserialize, endsWith, executeBranchMigrationPlan, exists, ge, getAPIKey, getBranchDetails, getBranchList, getBranchMetadata, getBranchMigrationHistory, getBranchMigrationPlan, getBranchStats, getColumn, getCurrentBranchDetails, getCurrentBranchName, getDatabaseList, getDatabaseMetadata, getDatabaseURL, getGitBranchesMapping, getRecord, getTableColumns, getTableSchema, getUser, getUserAPIKeys, getWorkspace, getWorkspaceMembersList, getWorkspacesList, gt, gte, includes, includesAll, includesAny, includesNone, insertRecord, insertRecordWithID, inviteWorkspaceMember, is, isCursorPaginationOptions, isIdentifiable, isNot, isXataRecord, le, lt, lte, notExists, operationsByTag, pattern, queryTable, removeGitBranchesEntry, removeWorkspaceMember, resendWorkspaceMemberInvite, resolveBranch, searchBranch, searchTable, serialize, setTableSchema, startsWith, updateBranchMetadata, updateColumn, updateRecordWithID, updateTable, updateUser, updateWorkspace, updateWorkspaceMemberInvite, updateWorkspaceMemberRole, upsertRecordWithID };
+export { BaseClient, operationsByTag as Operations, PAGINATION_DEFAULT_OFFSET, PAGINATION_DEFAULT_SIZE, PAGINATION_MAX_OFFSET, PAGINATION_MAX_SIZE, Page, Query, RecordArray, Repository, RestRepository, SchemaPlugin, SearchPlugin, Serializer, SimpleCache, XataApiClient, XataApiPlugin, XataError, XataPlugin, acceptWorkspaceMemberInvite, addGitBranchesEntry, addTableColumn, applyBranchSchemaEdit, buildClient, buildWorkerRunner, bulkInsertTableRecords, cPcreateDatabase, cPdeleteDatabase, cPgetCPDatabaseMetadata, cPgetDatabaseList, cPupdateCPDatabaseMetadata, cancelWorkspaceMemberInvite, compareBranchSchemas, compareBranchWithUserSchema, compareMigrationRequest, contains, createBranch, createDatabase, createMigrationRequest, createTable, createUserAPIKey, createWorkspace, deleteBranch, deleteColumn, deleteDatabase, deleteRecord, deleteTable, deleteUser, deleteUserAPIKey, deleteWorkspace, deserialize, endsWith, equals, executeBranchMigrationPlan, exists, ge, getAPIKey, getBranchDetails, getBranchList, getBranchMetadata, getBranchMigrationHistory, getBranchMigrationPlan, getBranchSchemaHistory, getBranchStats, getColumn, getCurrentBranchDetails, getCurrentBranchName, getDatabaseList, getDatabaseMetadata, getDatabaseURL, getGitBranchesMapping, getMigrationRequest, getMigrationRequestIsMerged, getRecord, getTableColumns, getTableSchema, getUser, getUserAPIKeys, getWorkspace, getWorkspaceMembersList, getWorkspacesList, greaterEquals, greaterThan, greaterThanEquals, gt, gte, includes, includesAll, includesAny, includesNone, insertRecord, insertRecordWithID, inviteWorkspaceMember, is, isCursorPaginationOptions, isIdentifiable, isNot, isXataRecord, le, lessEquals, lessThan, lessThanEquals, listMigrationRequests, listMigrationRequestsCommits, lt, lte, mergeMigrationRequest, notExists, operationsByTag, pattern, previewBranchSchemaEdit, queryTable, removeGitBranchesEntry, removeWorkspaceMember, resendWorkspaceMemberInvite, resolveBranch, searchBranch, searchTable, serialize, setTableSchema, startsWith, summarizeTable, updateBranchMetadata, updateBranchSchema, updateColumn, updateDatabaseMetadata, updateMigrationRequest, updateRecordWithID, updateTable, updateUser, updateWorkspace, updateWorkspaceMemberInvite, updateWorkspaceMemberRole, upsertRecordWithID };
 //# sourceMappingURL=index.mjs.map