@xata.io/client 0.0.0-beta.4e4e7fc → 0.0.0-beta.53c906e

package/dist/schema/repository.js

@@ -26,17 +26,24 @@ var __asyncValues = (this && this.__asyncValues) || function (o) {
     function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
     function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
 };
-var _RestRepository_instances, _RestRepository_client, _RestRepository_fetch, _RestRepository_table, _RestRepository_getFetchProps, _BaseClient_links, _BaseClient_branch;
+var _RestRepository_instances, _RestRepository_client, _RestRepository_fetch, _RestRepository_table, _RestRepository_getFetchProps, _RestRepository_insertRecordWithoutId, _RestRepository_insertRecordWithId, _RestRepository_bulkInsertTableRecords, _RestRepository_updateRecordWithID, _RestRepository_upsertRecordWithID, _RestRepository_deleteRecord, _BaseClient_links, _BaseClient_branch;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BaseClient = exports.RestRespositoryFactory = exports.RestRepository = exports.Repository = void 0;
 const api_1 = require("../api");
-const filters_1 = require("./filters");
+const fetch_1 = require("../util/fetch");
+const lang_1 = require("../util/lang");
+const config_1 = require("../util/config");
 const pagination_1 = require("./pagination");
 const query_1 = require("./query");
+const record_1 = require("./record");
+const sorting_1 = require("./sorting");
+/**
+ * Common interface for performing operations on a table.
+ */
 class Repository extends query_1.Query {
 }
 exports.Repository = Repository;
-class RestRepository extends Repository {
+class RestRepository extends query_1.Query {
     constructor(client, table) {
         super(null, table, {});
         _RestRepository_instances.add(this);
@@ -45,107 +52,151 @@ class RestRepository extends Repository {
         _RestRepository_table.set(this, void 0);
         __classPrivateFieldSet(this, _RestRepository_client, client, "f");
         __classPrivateFieldSet(this, _RestRepository_table, table, "f");
-        // TODO: Remove when integrating with API client
-        const fetchImpl = typeof fetch !== 'undefined' ? fetch : __classPrivateFieldGet(this, _RestRepository_client, "f").options.fetch;
-        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.`);
-        }
-        __classPrivateFieldSet(this, _RestRepository_fetch, fetchImpl, "f");
+        __classPrivateFieldSet(this, _RestRepository_fetch, (0, fetch_1.getFetchImplementation)(__classPrivateFieldGet(this, _RestRepository_client, "f").options.fetch), "f");
     }
-    create(object) {
+    create(a, b) {
         return __awaiter(this, void 0, void 0, function* () {
-            const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
-            const record = transformObjectLinks(object);
-            const response = object.id
-                ? yield (0, api_1.insertRecordWithID)(Object.assign({ pathParams: {
-                        workspace: '{workspaceId}',
-                        dbBranchName: '{dbBranch}',
-                        tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"),
-                        recordId: object.id
-                    }, body: record }, fetchProps))
-                : yield (0, api_1.insertRecord)(Object.assign({ pathParams: {
-                        workspace: '{workspaceId}',
-                        dbBranchName: '{dbBranch}',
-                        tableName: __classPrivateFieldGet(this, _RestRepository_table, "f")
-                    }, body: record }, fetchProps));
-            const finalObject = yield this.read(response.id);
-            if (!finalObject) {
-                throw new Error('The server failed to save the record');
+            // Create many records
+            if (Array.isArray(a)) {
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_bulkInsertTableRecords).call(this, a);
+            }
+            // Create one record with id as param
+            if ((0, lang_1.isString)(a) && (0, lang_1.isObject)(b)) {
+                if (a === '')
+                    throw new Error("The id can't be empty");
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_insertRecordWithId).call(this, a, b);
+            }
+            // Create one record with id as property
+            if ((0, lang_1.isObject)(a) && (0, lang_1.isString)(a.id)) {
+                if (a.id === '')
+                    throw new Error("The id can't be empty");
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_insertRecordWithId).call(this, a.id, Object.assign(Object.assign({}, a), { id: undefined }));
             }
-            return finalObject;
+            // Create one record without id
+            if ((0, lang_1.isObject)(a)) {
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_insertRecordWithoutId).call(this, a);
+            }
+            throw new Error('Invalid arguments for create method');
         });
     }
-    createMany(objects) {
+    // TODO: Add column support: https://github.com/xataio/openapi/issues/139
+    read(recordId) {
         return __awaiter(this, void 0, void 0, function* () {
             const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
-            const records = objects.map((object) => transformObjectLinks(object));
-            const response = yield (0, api_1.bulkInsertTableRecords)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f") }, body: { records } }, fetchProps));
-            // TODO: Use filer.$any() to get all the records
-            const finalObjects = yield Promise.all(response.recordIDs.map((id) => this.read(id)));
-            if (finalObjects.some((object) => !object)) {
-                throw new Error('The server failed to save the record');
+            try {
+                const response = yield (0, api_1.getRecord)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId } }, fetchProps));
+                return __classPrivateFieldGet(this, _RestRepository_client, "f").initObject(__classPrivateFieldGet(this, _RestRepository_table, "f"), response);
+            }
+            catch (e) {
+                if ((0, lang_1.isObject)(e) && e.status === 404) {
+                    return null;
+                }
+                throw e;
             }
-            return finalObjects;
         });
     }
-    read(recordId) {
+    update(a, b) {
         return __awaiter(this, void 0, void 0, function* () {
-            const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
-            const response = yield (0, api_1.getRecord)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId } }, fetchProps));
-            return __classPrivateFieldGet(this, _RestRepository_client, "f").initObject(__classPrivateFieldGet(this, _RestRepository_table, "f"), response);
+            // Update many records
+            if (Array.isArray(a)) {
+                if (a.length > 100) {
+                    // TODO: Implement bulk update when API has support for it
+                    console.warn('Bulk update operation is not optimized in the Xata API yet, this request might be slow');
+                }
+                return Promise.all(a.map((object) => this.update(object)));
+            }
+            // Update one record with id as param
+            if ((0, lang_1.isString)(a) && (0, lang_1.isObject)(b)) {
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_updateRecordWithID).call(this, a, b);
+            }
+            // Update one record with id as property
+            if ((0, lang_1.isObject)(a) && (0, lang_1.isString)(a.id)) {
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_updateRecordWithID).call(this, a.id, Object.assign(Object.assign({}, a), { id: undefined }));
+            }
+            throw new Error('Invalid arguments for update method');
         });
     }
-    update(recordId, object) {
+    createOrUpdate(a, b) {
         return __awaiter(this, void 0, void 0, function* () {
-            const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
-            const response = yield (0, api_1.updateRecordWithID)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId }, body: object }, fetchProps));
-            const item = yield this.read(response.id);
-            if (!item)
-                throw new Error('The server failed to save the record');
-            return item;
+            // Create or update many records
+            if (Array.isArray(a)) {
+                if (a.length > 100) {
+                    // TODO: Implement bulk update when API has support for it
+                    console.warn('Bulk update operation is not optimized in the Xata API yet, this request might be slow');
+                }
+                return Promise.all(a.map((object) => this.createOrUpdate(object)));
+            }
+            // Create or update one record with id as param
+            if ((0, lang_1.isString)(a) && (0, lang_1.isObject)(b)) {
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_upsertRecordWithID).call(this, a, b);
+            }
+            // Create or update one record with id as property
+            if ((0, lang_1.isObject)(a) && (0, lang_1.isString)(a.id)) {
+                return __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_upsertRecordWithID).call(this, a.id, Object.assign(Object.assign({}, a), { id: undefined }));
+            }
+            throw new Error('Invalid arguments for createOrUpdate method');
         });
     }
-    upsert(recordId, object) {
+    delete(recordId) {
         return __awaiter(this, void 0, void 0, function* () {
-            const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
-            const response = yield (0, api_1.upsertRecordWithID)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId }, body: object }, fetchProps));
-            const item = yield this.read(response.id);
-            if (!item)
-                throw new Error('The server failed to save the record');
-            return item;
+            // Delete many records
+            if (Array.isArray(recordId)) {
+                if (recordId.length > 100) {
+                    // TODO: Implement bulk delete when API has support for it
+                    console.warn('Bulk delete operation is not optimized in the Xata API yet, this request might be slow');
+                }
+                yield Promise.all(recordId.map((id) => this.delete(id)));
+                return;
+            }
+            // Delete one record with id as param
+            if ((0, lang_1.isString)(recordId)) {
+                yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_deleteRecord).call(this, recordId);
+                return;
+            }
+            // Delete one record with id as property
+            if ((0, lang_1.isObject)(recordId) && (0, lang_1.isString)(recordId.id)) {
+                yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_deleteRecord).call(this, recordId.id);
+                return;
+            }
+            throw new Error('Invalid arguments for delete method');
         });
     }
-    delete(recordId) {
+    search(query, options = {}) {
         return __awaiter(this, void 0, void 0, function* () {
             const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
-            yield (0, api_1.deleteRecord)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId } }, fetchProps));
+            const { records } = yield (0, api_1.searchBranch)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}' }, body: { tables: [__classPrivateFieldGet(this, _RestRepository_table, "f")], query, fuzziness: options.fuzziness } }, fetchProps));
+            return records.map((item) => __classPrivateFieldGet(this, _RestRepository_client, "f").initObject(__classPrivateFieldGet(this, _RestRepository_table, "f"), item));
         });
     }
-    query(query, options) {
-        var _a, _b, _c;
+    query(query) {
+        var _a;
         return __awaiter(this, void 0, void 0, function* () {
             const data = query.getQueryOptions();
             const body = {
-                filter: Object.values(data.filter).some(Boolean) ? data.filter : undefined,
-                sort: (_a = (0, filters_1.buildSortFilter)(options === null || options === void 0 ? void 0 : options.sort)) !== null && _a !== void 0 ? _a : data.sort,
-                page: (_b = options === null || options === void 0 ? void 0 : options.page) !== null && _b !== void 0 ? _b : data.page,
-                columns: (_c = options === null || options === void 0 ? void 0 : options.columns) !== null && _c !== void 0 ? _c : data.columns
+                filter: Object.values((_a = data.filter) !== null && _a !== void 0 ? _a : {}).some(Boolean) ? data.filter : undefined,
+                sort: data.sort ? (0, sorting_1.buildSortFilter)(data.sort) : undefined,
+                page: data.page,
+                columns: data.columns
             };
             const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
             const { meta, records: objects } = yield (0, api_1.queryTable)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f") }, body }, fetchProps));
             const records = objects.map((record) => __classPrivateFieldGet(this, _RestRepository_client, "f").initObject(__classPrivateFieldGet(this, _RestRepository_table, "f"), record));
-            // TODO: We should properly type this any
             return new pagination_1.Page(query, meta, records);
         });
     }
 }
 exports.RestRepository = RestRepository;
 _RestRepository_client = new WeakMap(), _RestRepository_fetch = new WeakMap(), _RestRepository_table = new WeakMap(), _RestRepository_instances = new WeakSet(), _RestRepository_getFetchProps = function _RestRepository_getFetchProps() {
+    var _a;
     return __awaiter(this, void 0, void 0, function* () {
         const branch = yield __classPrivateFieldGet(this, _RestRepository_client, "f").getBranch();
+        const apiKey = (_a = __classPrivateFieldGet(this, _RestRepository_client, "f").options.apiKey) !== null && _a !== void 0 ? _a : (0, config_1.getAPIKey)();
+        if (!apiKey) {
+            throw new Error('Could not resolve a valid apiKey');
+        }
         return {
             fetchImpl: __classPrivateFieldGet(this, _RestRepository_fetch, "f"),
-            apiKey: __classPrivateFieldGet(this, _RestRepository_client, "f").options.apiKey,
+            apiKey,
             apiUrl: '',
             // Instead of using workspace and dbBranch, we inject a probably CNAME'd URL
             workspacesApiUrl: (path, params) => {
@@ -157,6 +208,72 @@ _RestRepository_client = new WeakMap(), _RestRepository_fetch = new WeakMap(), _
             }
         };
     });
+}, _RestRepository_insertRecordWithoutId = function _RestRepository_insertRecordWithoutId(object) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
+        const record = transformObjectLinks(object);
+        const response = yield (0, api_1.insertRecord)(Object.assign({ pathParams: {
+                workspace: '{workspaceId}',
+                dbBranchName: '{dbBranch}',
+                tableName: __classPrivateFieldGet(this, _RestRepository_table, "f")
+            }, body: record }, fetchProps));
+        const finalObject = yield this.read(response.id);
+        if (!finalObject) {
+            throw new Error('The server failed to save the record');
+        }
+        return finalObject;
+    });
+}, _RestRepository_insertRecordWithId = function _RestRepository_insertRecordWithId(recordId, object) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
+        const record = transformObjectLinks(object);
+        const response = yield (0, api_1.insertRecordWithID)(Object.assign({ pathParams: {
+                workspace: '{workspaceId}',
+                dbBranchName: '{dbBranch}',
+                tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"),
+                recordId
+            }, body: record, queryParams: { createOnly: true } }, fetchProps));
+        const finalObject = yield this.read(response.id);
+        if (!finalObject) {
+            throw new Error('The server failed to save the record');
+        }
+        return finalObject;
+    });
+}, _RestRepository_bulkInsertTableRecords = function _RestRepository_bulkInsertTableRecords(objects) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
+        const records = objects.map((object) => transformObjectLinks(object));
+        const response = yield (0, api_1.bulkInsertTableRecords)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f") }, body: { records } }, fetchProps));
+        const finalObjects = yield this.any(...response.recordIDs.map((id) => this.filter('id', id))).getAll();
+        if (finalObjects.length !== objects.length) {
+            throw new Error('The server failed to save some records');
+        }
+        return finalObjects;
+    });
+}, _RestRepository_updateRecordWithID = function _RestRepository_updateRecordWithID(recordId, object) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
+        const record = transformObjectLinks(object);
+        const response = yield (0, api_1.updateRecordWithID)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId }, body: record }, fetchProps));
+        const item = yield this.read(response.id);
+        if (!item)
+            throw new Error('The server failed to save the record');
+        return item;
+    });
+}, _RestRepository_upsertRecordWithID = function _RestRepository_upsertRecordWithID(recordId, object) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
+        const response = yield (0, api_1.upsertRecordWithID)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId }, body: object }, fetchProps));
+        const item = yield this.read(response.id);
+        if (!item)
+            throw new Error('The server failed to save the record');
+        return item;
+    });
+}, _RestRepository_deleteRecord = function _RestRepository_deleteRecord(recordId) {
+    return __awaiter(this, void 0, void 0, function* () {
+        const fetchProps = yield __classPrivateFieldGet(this, _RestRepository_instances, "m", _RestRepository_getFetchProps).call(this);
+        yield (0, api_1.deleteRecord)(Object.assign({ pathParams: { workspace: '{workspaceId}', dbBranchName: '{dbBranch}', tableName: __classPrivateFieldGet(this, _RestRepository_table, "f"), recordId } }, fetchProps));
+    });
 };
 class RestRespositoryFactory {
     createRepository(client, table) {
@@ -164,54 +281,69 @@ class RestRespositoryFactory {
     }
 }
 exports.RestRespositoryFactory = RestRespositoryFactory;
+function resolveXataClientOptions(options) {
+    const databaseURL = (options === null || options === void 0 ? void 0 : options.databaseURL) || (0, config_1.getDatabaseURL)() || '';
+    const apiKey = (options === null || options === void 0 ? void 0 : options.apiKey) || (0, config_1.getAPIKey)() || '';
+    const branch = (options === null || options === void 0 ? void 0 : options.branch) || (() => (0, config_1.getCurrentBranchName)({ apiKey, databaseURL, fetchImpl: options === null || options === void 0 ? void 0 : options.fetch }));
+    if (!databaseURL || !apiKey) {
+        throw new Error('Options databaseURL and apiKey are required');
+    }
+    return Object.assign(Object.assign({}, options), { databaseURL,
+        apiKey,
+        branch });
+}
 class BaseClient {
-    constructor(options, links) {
+    constructor(options = {}, links = {}) {
         _BaseClient_links.set(this, void 0);
         _BaseClient_branch.set(this, void 0);
-        if (!options.databaseURL || !options.apiKey || !options.branch) {
-            throw new Error('Options databaseURL, apiKey and branch are required');
-        }
-        this.options = options;
+        this.options = resolveXataClientOptions(options);
+        // Make this property not enumerable so it doesn't show up in console.dir, etc.
+        Object.defineProperty(this.options, 'apiKey', { enumerable: false });
         __classPrivateFieldSet(this, _BaseClient_links, links, "f");
+        const factory = this.options.repositoryFactory || new RestRespositoryFactory();
+        this.db = new Proxy({}, {
+            get: (_target, prop) => {
+                if (!(0, lang_1.isString)(prop))
+                    throw new Error('Invalid table name');
+                return factory.createRepository(this, prop);
+            }
+        });
+    }
+    search(query, options) {
+        var _a;
+        return __awaiter(this, void 0, void 0, function* () {
+            const tables = (_a = options === null || options === void 0 ? void 0 : options.tables) !== null && _a !== void 0 ? _a : Object.keys(this.db);
+            // TODO: Implement global search with a single call, REST repository abstraction needed
+            const results = yield Promise.all(tables.map((table) => this.db[table].search(query, options).then((results) => [table, results])));
+            return Object.fromEntries(results);
+        });
     }
     initObject(table, object) {
-        const o = {};
-        Object.assign(o, object);
+        const result = {};
+        Object.assign(result, object);
         const tableLinks = __classPrivateFieldGet(this, _BaseClient_links, "f")[table] || [];
         for (const link of tableLinks) {
             const [field, linkTable] = link;
-            const value = o[field];
-            if (value && typeof value === 'object') {
-                const { id } = value;
-                if (Object.keys(value).find((col) => col === 'id')) {
-                    o[field] = this.initObject(linkTable, value);
-                }
-                else if (id) {
-                    o[field] = {
-                        id,
-                        get: () => {
-                            this.db[linkTable].read(id);
-                        }
-                    };
-                }
+            const value = result[field];
+            if (value && (0, lang_1.isObject)(value)) {
+                result[field] = this.initObject(linkTable, value);
             }
         }
         const db = this.db;
-        o.read = function () {
-            return db[table].read(o['id']);
+        result.read = function () {
+            return db[table].read(result['id']);
         };
-        o.update = function (data) {
-            return db[table].update(o['id'], data);
+        result.update = function (data) {
+            return db[table].update(result['id'], data);
         };
-        o.delete = function () {
-            return db[table].delete(o['id']);
+        result.delete = function () {
+            return db[table].delete(result['id']);
         };
         for (const prop of ['read', 'update', 'delete']) {
-            Object.defineProperty(o, prop, { enumerable: false });
+            Object.defineProperty(result, prop, { enumerable: false });
         }
-        // TODO: links and rev links
-        Object.freeze(o);
-        return o;
+        Object.freeze(result);
+        return result;
     }
     getBranch() {
         var e_1, _a;
@@ -249,12 +381,12 @@ _BaseClient_links = new WeakMap(), _BaseClient_branch = new WeakMap();
 const isBranchStrategyBuilder = (strategy) => {
     return typeof strategy === 'function';
 };
-// TODO: We can find a better implementation for links
 const transformObjectLinks = (object) => {
     return Object.entries(object).reduce((acc, [key, value]) => {
-        if (value && typeof value === 'object' && typeof value.id === 'string') {
-            return Object.assign(Object.assign({}, acc), { [key]: value.id });
-        }
-        return Object.assign(Object.assign({}, acc), { [key]: value });
+        // Ignore internal properties
+        if (key === 'xata')
+            return acc;
+        // Transform links to identifier
+        return Object.assign(Object.assign({}, acc), { [key]: (0, record_1.isIdentifiable)(value) ? value.id : value });
     }, {});
 };

package/dist/schema/selection.d.ts

@@ -1,14 +1,25 @@
-import { XataRecord } from '..';
-import { PartialBy } from '../util/lang';
-import { StringKeys, UnionToIntersection, Values } from '../util/types';
-import { Query } from './query';
-declare type Queries<T> = {
-    [key in keyof T as T[key] extends Query<any> ? key : never]: T[key];
-};
-declare type InternalProperties = keyof Omit<XataRecord, 'id'>;
-export declare type Selectable<T extends XataRecord> = Omit<PartialBy<T, 'id'>, InternalProperties>;
-export declare type SelectableColumn<O> = '*' | (O extends Array<unknown> ? never : O extends Record<string, any> ? '*' | Values<{
-    [K in StringKeys<O>]: O[K] extends Record<string, any> ? `${K}.${SelectableColumn<O[K]>}` : K;
-}> : '');
-export declare type Select<T, K extends SelectableColumn<T>> = UnionToIntersection<K extends keyof T ? Pick<T, K> : T> & Queries<T> & XataRecord;
+import { If, IsArray, IsObject, StringKeys, UnionToIntersection, Values } from '../util/types';
+import { Link, XataRecord } from './record';
+export declare type SelectableColumn<O, RecursivePath extends any[] = []> = '*' | 'id' | DataProps<O> | NestedColumns<O, RecursivePath>;
+export declare type SelectedPick<O extends XataRecord, Key extends SelectableColumn<O>[]> = XataRecord & UnionToIntersection<Values<{
+    [K in Key[number]]: NestedValueAtColumn<O, K> & XataRecord;
+}>>;
+export declare type ValueAtColumn<O, P extends SelectableColumn<O>> = P extends '*' ? Values<O> : P extends 'id' ? string : P extends keyof O ? O[P] : P extends `${infer K}.${infer V}` ? K extends keyof O ? Values<O[K] extends XataRecord ? (V extends SelectableColumn<O[K]> ? {
+    V: ValueAtColumn<O[K], V>;
+} : never) : O[K]> : never : never;
+declare type MAX_RECURSION = 5;
+declare type NestedColumns<O, RecursivePath extends any[]> = RecursivePath['length'] extends MAX_RECURSION ? never : If<IsObject<O>, Values<{
+    [K in DataProps<O>]: If<IsArray<NonNullable<O[K]>>, K, // If the property is an array, we stop recursion. We don't support object arrays yet
+    If<IsObject<NonNullable<O[K]>>, NonNullable<O[K]> extends XataRecord ? SelectableColumn<NonNullable<O[K]>, [...RecursivePath, O[K]]> extends string ? K | `${K}.${SelectableColumn<NonNullable<O[K]>, [...RecursivePath, O[K]]>}` : never : `${K}.${StringKeys<NonNullable<O[K]>> | '*'}`, // This allows usage of objects that are not links
+    K>>;
+}>, never>;
+declare type DataProps<O> = Exclude<StringKeys<O>, StringKeys<XataRecord>>;
+declare type NestedValueAtColumn<O, Key extends SelectableColumn<O>> = Key extends `${infer N}.${infer M}` ? N extends DataProps<O> ? {
+    [K in N]: M extends SelectableColumn<NonNullable<O[K]>> ? NonNullable<O[K]> extends XataRecord ? ForwardNullable<O[K], NestedValueAtColumn<NonNullable<O[K]>, M> & XataRecord> : ForwardNullable<O[K], NestedValueAtColumn<NonNullable<O[K]>, M>> : unknown;
+} : unknown : Key extends DataProps<O> ? {
+    [K in Key]: NonNullable<O[K]> extends XataRecord ? ForwardNullable<O[K], SelectedPick<NonNullable<O[K]>, ['*']>> : O[K];
+} : Key extends '*' ? {
+    [K in StringKeys<O>]: NonNullable<O[K]> extends XataRecord ? ForwardNullable<O[K], Link<NonNullable<O[K]>>> : O[K];
+} : unknown;
+declare type ForwardNullable<T, R> = T extends NonNullable<T> ? R : R | null;
 export {};

package/dist/schema/selection.spec.d.ts

@@ -0,0 +1 @@
+export {};