@rebasepro/server-postgresql 0.0.1-canary.4d4fb3e

package/src/services/EntityFetchService.ts

@@ -0,0 +1,1447 @@
+import { and, asc, count, desc, eq, getTableName, gt, lt, or, SQL, TableRelationalConfig, TablesRelationalConfig } from "drizzle-orm";
+import { AnyPgColumn, PgTable } from "drizzle-orm/pg-core";
+import { Entity, EntityCollection, FilterValues, Relation } from "@rebasepro/types";
+import { resolveCollectionRelations } from "@rebasepro/common";
+import { DrizzleConditionBuilder } from "../utils/drizzle-conditions";
+import {
+    getCollectionByPath,
+    getTableForCollection,
+    getPrimaryKeys,
+    parseIdValues,
+    buildCompositeId
+} from "./entity-helpers";
+import { parseDataFromServer, normalizeDbValues } from "../data-transformer";
+import { RelationService } from "./RelationService";
+import { RelationalQueryBuilder } from "drizzle-orm/pg-core/query-builders/query";
+import { DrizzleClient } from "../interfaces";
+import { PostgresCollectionRegistry } from "../collections/PostgresCollectionRegistry";
+
+/** Type-safe accessor for Drizzle's relational query API via dynamic table name */
+type DbQueryAccessor = Record<string, RelationalQueryBuilder<any, any>> | undefined;
+
+/**
+ * Service for handling all entity read operations.
+ * Handles fetching, searching, counting, and filtering entities.
+ */
+export class EntityFetchService {
+    private relationService: RelationService;
+
+    constructor(private db: DrizzleClient, private registry: PostgresCollectionRegistry) {
+        this.relationService = new RelationService(db, registry);
+    }
+
+    /**
+     * Get the relational query builder for a given table name.
+     * Safely narrows the DrizzleClient union type to access db.query[tableName].
+     */
+    private getQueryBuilder(tableName: string): RelationalQueryBuilder<TablesRelationalConfig, TableRelationalConfig> | undefined {
+        const query = (this.db as { query?: DbQueryAccessor }).query;
+        return query?.[tableName] as RelationalQueryBuilder<TablesRelationalConfig, TableRelationalConfig> | undefined;
+    }
+
+    /**
+     * Build filter conditions from FilterValues
+     * Delegates to DrizzleConditionBuilder.buildFilterConditions
+     */
+    buildFilterConditions<M extends Record<string, any>>(
+        filter: FilterValues<Extract<keyof M, string>>,
+        table: PgTable<any>,
+        collectionPath: string
+    ): SQL[] {
+        return DrizzleConditionBuilder.buildFilterConditions(filter, table, collectionPath);
+    }
+
+    // =============================================================
+    // DRIZZLE QUERY HELPERS
+    // =============================================================
+
+    /**
+     * Build the `with` config for Drizzle's relational query API.
+     * Converts collection relations to a Drizzle-compatible `with` object.
+     *
+     * When `include` is provided, only those relations are loaded.
+     * When `include` is absent, ALL relations are loaded (CMS path).
+     *
+     * Automatically detects many-to-many junction tables and nests
+     * the target relation so actual entity data is returned.
+     */
+    private buildWithConfig(
+        collection: EntityCollection,
+        include?: string[]
+    ): Record<string, boolean | { with: Record<string, boolean> }> {
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+        const propertyKeys = new Set(Object.keys(collection.properties || {}));
+        const withConfig: Record<string, boolean | { with: Record<string, boolean> }> = {};
+
+        const shouldInclude = (key: string) =>
+            !include || include.length === 0 || include[0] === "*" || include.includes(key);
+
+        for (const [key, relation] of Object.entries(resolvedRelations)) {
+            if (!shouldInclude(key)) continue;
+            // Only include relations that map to actual properties (or include all for REST)
+            if (!include && !propertyKeys.has(key)) continue;
+
+            const drizzleRelName = relation.relationName || key;
+
+            // Skip relations that use joinPath as they are not mapped in Drizzle schemas
+            if (relation.joinPath && relation.joinPath.length > 0) {
+                continue;
+            }
+
+            // Detect many-to-many junction tables:
+            // If the relation goes through a junction table (relation.through exists or
+            // the Drizzle schema maps to a junction table), we need two-level with.
+            if (relation.cardinality === "many" && this.isJunctionRelation(relation, collection)) {
+                // The Drizzle relation points to the junction table.
+                // We need: { [junctionRelName]: { with: { [targetFkName]: true } } }
+                // The target FK name is the relation on the junction table that points to the actual target.
+                const targetFkName = this.getJunctionTargetRelationName(relation, collection);
+                if (targetFkName) {
+                    withConfig[drizzleRelName] = { with: { [targetFkName]: true } };
+                } else {
+                    withConfig[drizzleRelName] = true;
+                }
+            } else {
+                withConfig[drizzleRelName] = true;
+            }
+        }
+
+        return withConfig;
+    }
+
+    /**
+     * Detect if a many-to-many relation uses a junction table in the Drizzle schema.
+     */
+    private isJunctionRelation(relation: Relation, _collection: EntityCollection): boolean {
+        // If `through` is defined, it's explicitly a junction relation
+        if (relation.through) return true;
+        // If joinPath has an intermediate table, it's likely junction-based
+        if (relation.joinPath && relation.joinPath.length > 1) return true;
+        return false;
+    }
+
+    /**
+     * Get the Drizzle relation name on the junction table that points to the actual target entity.
+     * For example, for posts_tags junction, this returns "tag_id" (the relation pointing to tags).
+     */
+    private getJunctionTargetRelationName(relation: Relation, _collection: EntityCollection): string | null {
+        if (relation.through) {
+            // The junction relation on the junction table pointing to the target
+            // uses the targetColumn name as the Drizzle relation name
+            return relation.through.targetColumn.replace(/_id$/, "_id");
+        }
+        return null;
+    }
+
+    /**
+     * Convert a db.query result row (with nested relation objects) to an Entity<M>.
+     * Handles:
+     * - The { id, path, values } wrapping
+     * - Type normalization (dates, numbers, NaN) via normalizeDbValues
+     * - Converting nested relation objects to { id, path, __type: "relation" } for CMS
+     * - Flattening junction-table many-to-many results
+     */
+    private drizzleResultToEntity<M extends Record<string, any>>(
+        row: Record<string, unknown>,
+        collection: EntityCollection,
+        collectionPath: string,
+        idInfo: { fieldName: string; type: "string" | "number" },
+        databaseId?: string,
+        idInfoArray?: { fieldName: string; type: "string" | "number" }[]
+    ): Entity<M> {
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+        const propertyKeys = new Set(Object.keys(collection.properties || {}));
+
+        // Normalize non-relation values (dates, numbers, etc.)
+        const normalizedValues = normalizeDbValues(row as M, collection);
+
+        // Convert nested relation objects to CMS-style { id, path, __type: "relation" }
+        for (const [key, relation] of Object.entries(resolvedRelations)) {
+            if (!propertyKeys.has(key)) continue;
+            const drizzleRelName = relation.relationName || key;
+            const relData = row[drizzleRelName];
+
+            if (relData === undefined || relData === null) continue;
+
+            if (relation.cardinality === "many" && Array.isArray(relData)) {
+                const targetCollection = relation.target();
+                const targetPath = targetCollection.slug;
+                const targetPks = getPrimaryKeys(targetCollection, this.registry);
+                const targetIdField = targetPks[0].fieldName;
+
+                (normalizedValues as Record<string, unknown>)[key] = relData.map((item: Record<string, unknown>) => {
+                    // Handle junction table flattening:
+                    // Junction rows look like { post_id: 1, tag_id: { id: 5, name: "ts" } }
+                    let targetEntity = item;
+                    if (this.isJunctionRelation(relation, collection)) {
+                        // Find the nested target object in the junction row
+                        const nestedKey = Object.keys(item).find(
+                            nk => typeof item[nk] === "object" && item[nk] !== null && !Array.isArray(item[nk])
+                        );
+                        if (nestedKey) {
+                            targetEntity = item[nestedKey] as Record<string, unknown>;
+                        }
+                    }
+
+                    const relId = String(targetEntity[targetIdField] ?? targetEntity.id ?? targetEntity[Object.keys(targetEntity)[0]]);
+                    const targetValues = normalizeDbValues(targetEntity, targetCollection);
+
+                    return {
+                        id: relId,
+                        path: targetPath,
+                        __type: "relation" as const,
+                        data: {
+                            id: relId,
+                            path: targetPath,
+                            values: targetValues,
+                            databaseId
+                        }
+                    };
+                });
+            } else if (relation.cardinality === "one" && typeof relData === "object" && !Array.isArray(relData)) {
+                const targetCollection = relation.target();
+                const targetPath = targetCollection.slug;
+                const targetPks = getPrimaryKeys(targetCollection, this.registry);
+                const targetIdField = targetPks[0].fieldName;
+                const relObj = relData as Record<string, unknown>;
+
+                const relId = String(relObj[targetIdField] ?? relObj.id ?? relObj[Object.keys(relObj)[0]]);
+                const targetValues = normalizeDbValues(relObj, targetCollection);
+
+                (normalizedValues as Record<string, unknown>)[key] = {
+                    id: relId,
+                    path: targetPath,
+                    __type: "relation" as const,
+                    data: {
+                        id: relId,
+                        path: targetPath,
+                        values: targetValues,
+                        databaseId
+                    }
+                };
+            }
+        }
+
+        return {
+            id: (idInfoArray && idInfoArray.length > 1) ? buildCompositeId(row as Record<string, any>, idInfoArray) : String(row[idInfo.fieldName]),
+            path: collectionPath,
+            values: normalizedValues as M,
+            databaseId
+        };
+    }
+
+    /**
+     * Post-fetch joinPath relations for a single entity.
+     * joinPath relations cannot be expressed via Drizzle's `with` config,
+     * so they must be loaded separately after the primary query.
+     */
+    private async resolveJoinPathRelations<M extends Record<string, any>>(
+        entity: Entity<M>,
+        collection: EntityCollection,
+        collectionPath: string,
+        parsedId: string | number,
+        databaseId?: string
+    ): Promise<void> {
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+        const propertyKeys = new Set(Object.keys(collection.properties || {}));
+
+        const promises = Object.entries(resolvedRelations)
+            .filter(([key, relation]) => propertyKeys.has(key) && relation.joinPath && relation.joinPath.length > 0)
+            .map(async ([key, relation]) => {
+                try {
+                    const relatedEntities = await this.relationService.fetchRelatedEntities(
+                        collectionPath,
+                        parsedId,
+                        key,
+                        { limit: relation.cardinality === "one" ? 1 : undefined }
+                    );
+
+                    if (relation.cardinality === "one" && relatedEntities.length > 0) {
+                        const e = relatedEntities[0];
+                        (entity.values as Record<string, unknown>)[key] = {
+                            id: e.id,
+                            path: e.path,
+                            __type: "relation" as const,
+                            data: e
+                        };
+                    } else if (relation.cardinality === "many") {
+                        (entity.values as Record<string, unknown>)[key] = relatedEntities.map(e => ({
+                            id: e.id,
+                            path: e.path,
+                            __type: "relation" as const,
+                            data: e
+                        }));
+                    }
+                } catch (e) {
+                    console.warn(`Could not resolve joinPath relation '${key}':`, e);
+                }
+            });
+
+        await Promise.all(promises);
+    }
+
+    /**
+     * Post-fetch joinPath relations for a batch of entities.
+     * Uses batch fetching to avoid N+1 queries for list views.
+     */
+    private async resolveJoinPathRelationsBatch<M extends Record<string, any>>(
+        entities: Entity<M>[],
+        collection: EntityCollection,
+        collectionPath: string,
+        idInfo: { fieldName: string; type: "string" | "number" },
+        databaseId?: string
+    ): Promise<void> {
+        if (entities.length === 0) return;
+
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+        const propertyKeys = new Set(Object.keys(collection.properties || {}));
+
+        const joinPathRelations = Object.entries(resolvedRelations)
+            .filter(([key, relation]) => propertyKeys.has(key) && relation.joinPath && relation.joinPath.length > 0);
+
+        if (joinPathRelations.length === 0) return;
+
+        for (const [key, relation] of joinPathRelations) {
+            try {
+                const entityIds = entities.map(e => {
+                    const parsed = parseIdValues(e.id, [idInfo]);
+                    return parsed[idInfo.fieldName] as string | number;
+                });
+
+                const resultMap = await this.relationService.batchFetchRelatedEntities(
+                    collectionPath,
+                    entityIds,
+                    key,
+                    relation
+                );
+
+                for (const entity of entities) {
+                    const parsed = parseIdValues(entity.id, [idInfo]);
+                    const entityId = parsed[idInfo.fieldName] as string | number;
+                    const relatedEntity = resultMap.get(entityId);
+
+                    if (relatedEntity) {
+                        if (relation.cardinality === "one") {
+                            (entity.values as Record<string, unknown>)[key] = {
+                                id: relatedEntity.id,
+                                path: relatedEntity.path,
+                                __type: "relation" as const,
+                                data: relatedEntity
+                            };
+                        }
+                    }
+                }
+            } catch (e) {
+                console.warn(`Could not batch resolve joinPath relation '${key}':`, e);
+            }
+        }
+    }
+
+    /**
+     * Convert a db.query result row to a flat REST-style object with populated relations.
+     */
+    private drizzleResultToRestRow(
+        row: Record<string, unknown>,
+        collection: EntityCollection,
+        idInfo: { fieldName: string; type: "string" | "number" },
+        idInfoArray?: { fieldName: string; type: "string" | "number" }[]
+    ): Record<string, unknown> {
+        const flat: Record<string, unknown> = { id: (idInfoArray && idInfoArray.length > 1) ? buildCompositeId(row as Record<string, any>, idInfoArray) : String(row[idInfo.fieldName]) };
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+
+        for (const [k, v] of Object.entries(row)) {
+            if (k === idInfo.fieldName) continue;
+
+            const relation = resolvedRelations[k];
+            if (Array.isArray(v) && relation) {
+                // Many relation — flatten each nested entity, handling junction tables
+                flat[k] = v.map((item: Record<string, unknown>) => {
+                    if (this.isJunctionRelation(relation, collection)) {
+                        const nestedKey = Object.keys(item).find(
+                            nk => typeof item[nk] === "object" && item[nk] !== null && !Array.isArray(item[nk])
+                        );
+                        if (nestedKey) {
+                            const nested = item[nestedKey] as Record<string, unknown>;
+                            return { id: String(nested.id ?? nested[Object.keys(nested)[0]]), ...nested };
+                        }
+                    }
+                    return { id: String(item.id ?? item[Object.keys(item)[0]]), ...item };
+                });
+            } else if (typeof v === "object" && v !== null && !Array.isArray(v) && relation) {
+                // One-to-one relation — inline the object
+                const relObj = v as Record<string, unknown>;
+                flat[k] = { id: String(relObj.id ?? relObj[Object.keys(relObj)[0]]), ...relObj };
+            } else {
+                flat[k] = v;
+            }
+        }
+        return flat;
+    }
+
+    /**
+     * Build db.query-compatible options from standard fetch options.
+     * Handles filter, search, orderBy, limit, and cursor-based pagination.
+     */
+    private buildDrizzleQueryOptions<M extends Record<string, any>>(
+        table: PgTable<any>,
+        idField: AnyPgColumn,
+        idInfo: { fieldName: string; type: "string" | "number" },
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+            startAfter?: Record<string, unknown>;
+            searchString?: string;
+        },
+        collectionPath: string,
+        withConfig?: Record<string, any>
+    ): Record<string, unknown> {
+        const queryOpts: Record<string, unknown> = {};
+
+        if (withConfig) queryOpts.with = withConfig;
+
+        // Build where conditions
+        const allConditions: SQL[] = [];
+
+        if (options.searchString) {
+            const collection = getCollectionByPath(collectionPath, this.registry);
+            const searchConditions = DrizzleConditionBuilder.buildSearchConditions(
+                options.searchString, collection.properties, table
+            );
+            if (searchConditions.length === 0) {
+                // Return options that will produce empty results
+                queryOpts.where = and(eq(idField, -99999999)); // impossible condition
+                return queryOpts;
+            }
+            allConditions.push(DrizzleConditionBuilder.combineConditionsWithOr(searchConditions)!);
+        }
+
+        if (options.filter) {
+            const filterConditions = this.buildFilterConditions(options.filter, table, collectionPath);
+            if (filterConditions.length > 0) allConditions.push(...filterConditions);
+        }
+
+        // Cursor-based pagination (startAfter)
+        if (options.startAfter) {
+            const cursorConditions = this.buildCursorConditions(table, idField, idInfo, options);
+            if (cursorConditions.length > 0) allConditions.push(...cursorConditions);
+        }
+
+        if (allConditions.length > 0) {
+            queryOpts.where = and(...allConditions);
+        }
+
+        // OrderBy
+        const orderExpressions: unknown[] = [];
+        if (options.orderBy) {
+            const orderByField = table[options.orderBy as keyof typeof table] as AnyPgColumn;
+            if (orderByField) {
+                orderExpressions.push(options.order === "asc" ? asc(orderByField) : desc(orderByField));
+            }
+        }
+        orderExpressions.push(desc(idField));
+        if (orderExpressions.length > 0) {
+            queryOpts.orderBy = orderExpressions;
+        }
+
+        // Limit
+        const limitValue = options.searchString ? (options.limit || 50) : options.limit;
+        if (limitValue) queryOpts.limit = limitValue;
+
+        return queryOpts;
+    }
+
+    /**
+     * Extract cursor pagination conditions from startAfter options.
+     */
+    private buildCursorConditions(
+        table: PgTable<any>,
+        idField: AnyPgColumn,
+        idInfo: { fieldName: string; type: "string" | "number" },
+        options: { orderBy?: string; order?: "desc" | "asc"; startAfter?: Record<string, unknown> }
+    ): SQL[] {
+        if (!options.startAfter) return [];
+        const cursor = options.startAfter;
+
+        if (options.orderBy) {
+            const orderByField = table[options.orderBy as keyof typeof table] as AnyPgColumn;
+            if (orderByField) {
+                const startAfterOrderValue = (cursor.values as Record<string, unknown> | undefined)?.[options.orderBy] ?? cursor[options.orderBy];
+                const startAfterId = cursor.id ?? cursor[idInfo.fieldName];
+
+                if (startAfterOrderValue !== undefined && startAfterId !== undefined) {
+                    if (options.order === "asc") {
+                        return [or(
+                            gt(orderByField, startAfterOrderValue),
+                            and(eq(orderByField, startAfterOrderValue), gt(idField, startAfterId))
+                        )!];
+                    } else {
+                        return [or(
+                            lt(orderByField, startAfterOrderValue),
+                            and(eq(orderByField, startAfterOrderValue), lt(idField, startAfterId))
+                        )!];
+                    }
+                }
+            }
+        } else {
+            const startAfterId = cursor.id ?? cursor[idInfo.fieldName];
+            if (startAfterId !== undefined && startAfterId !== null) {
+                const idInfoArray = [idInfo] as Array<{ fieldName: string; type: "string" | "number" }>;
+                const parsedStartAfterIdObj = parseIdValues(startAfterId as string | number, idInfoArray);
+                return [lt(idField, parsedStartAfterIdObj[idInfo.fieldName])];
+            }
+        }
+
+        return [];
+    }
+
+    /**
+     * Fetch a single entity by ID
+     */
+    async fetchEntity<M extends Record<string, any>>(
+        collectionPath: string,
+        entityId: string | number,
+        databaseId?: string
+    ): Promise<Entity<M> | undefined> {
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+        const idInfoArray = getPrimaryKeys(collection, this.registry);
+        const idInfo = idInfoArray[0];
+        const idField = table[idInfo.fieldName as keyof typeof table] as AnyPgColumn;
+
+        if (!idField) {
+            throw new Error(`ID field '${idInfo.fieldName}' not found in table for collection '${collectionPath}'`);
+        }
+
+        const parsedIdObj = parseIdValues(entityId, idInfoArray);
+        const parsedId = parsedIdObj[idInfo.fieldName];
+
+        // Primary path: use db.query.findFirst with relation loading
+        
+        const tableName = getTableName(table);
+
+        const qb = this.getQueryBuilder(tableName);
+        if (qb) {
+            try {
+                const withConfig = this.buildWithConfig(collection);
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any -- dynamic query config
+                const row = await qb.findFirst({
+                    where: eq(idField, parsedId),
+                    with: withConfig
+                } as unknown as Parameters<NonNullable<typeof qb>["findFirst"]>[0]);
+
+                if (!row) return undefined;
+
+                const entity = this.drizzleResultToEntity<M>(row, collection, collectionPath, idInfo, databaseId, idInfoArray);
+
+                // Post-fetch joinPath relations that Drizzle's `with` can't express
+                await this.resolveJoinPathRelations(entity, collection, collectionPath, parsedId, databaseId);
+
+                return entity;
+            } catch (e) {
+                console.warn(`[EntityFetchService] db.query.findFirst failed for ${collectionPath}, falling back to db.select:`, e);
+            }
+        }
+
+        // Fallback: db.select + N+1 relation loading
+        const result = await this.db
+            .select()
+            .from(table)
+            .where(eq(idField, parsedId))
+            .limit(1);
+
+        if (result.length === 0) return undefined;
+
+        const raw = result[0] as M;
+        const values = await parseDataFromServer(raw, collection, this.db, this.registry);
+
+        // Load relations based on cardinality (N+1 — only used in fallback)
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+        const propertyKeys = new Set(Object.keys(collection.properties));
+
+        const relationPromises = Object.entries(resolvedRelations)
+            .filter(([key]) => propertyKeys.has(key))
+            .map(async ([key, relation]) => {
+                if (relation.cardinality === "many") {
+                    const relatedEntities = await this.relationService.fetchRelatedEntities(
+                        collectionPath,
+                        parsedId,
+                        key,
+                        {}
+                    );
+                    (values as Record<string, unknown>)[key] = relatedEntities.map(e => ({
+                        id: e.id,
+                        path: e.path,
+                        __type: "relation"
+                    }));
+                } else if (relation.cardinality === "one") {
+                    if ((values as Record<string, unknown>)[key] == null) {
+                        try {
+                            const relatedEntities = await this.relationService.fetchRelatedEntities(
+                                collectionPath,
+                                parsedId,
+                                key,
+                                { limit: 1 }
+                            );
+                            if (relatedEntities.length > 0) {
+                                const e = relatedEntities[0];
+                                (values as Record<string, unknown>)[key] = {
+                                    id: e.id,
+                                    path: e.path,
+                                    __type: "relation"
+                                };
+                            }
+                        } catch (e) {
+                            console.warn(`Could not resolve one-to-one relation property: ${key}`, e);
+                        }
+                    }
+                }
+            });
+
+        await Promise.all(relationPromises);
+
+        return {
+            id: entityId.toString(),
+            path: collectionPath,
+            values: values as M,
+            databaseId
+        };
+    }
+
+    /**
+     * Unified method to fetch entities with optional search functionality
+     */
+    async fetchEntitiesWithConditions<M extends Record<string, any>>(
+        collectionPath: string,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+            startAfter?: Record<string, unknown>;
+            searchString?: string;
+            databaseId?: string;
+        } = {}
+    ): Promise<Entity<M>[]> {
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+        const idInfoArray = getPrimaryKeys(collection, this.registry);
+        const idInfo = idInfoArray[0];
+        const idField = table[idInfo.fieldName as keyof typeof table] as AnyPgColumn;
+
+        if (!idField) {
+            throw new Error(`ID field '${idInfo.fieldName}' not found in table for collection '${collectionPath}'`);
+        }
+
+        // Primary path: use db.query.findMany with relation loading
+        
+        const tableName = getTableName(table);
+
+        const qb = this.getQueryBuilder(tableName);
+        if (qb) {
+            try {
+                const withConfig = this.buildWithConfig(collection);
+                const queryOpts = this.buildDrizzleQueryOptions<M>(
+                    table, idField, idInfo, options, collectionPath, withConfig
+                );
+
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any -- dynamic query config
+                const results = await qb.findMany(queryOpts as unknown as Parameters<NonNullable<typeof qb>["findMany"]>[0]);
+
+                const entities = (results as Record<string, unknown>[]).map(row =>
+                    this.drizzleResultToEntity<M>(row, collection, collectionPath, idInfo, options.databaseId, idInfoArray)
+                );
+
+                // Post-fetch joinPath relations that Drizzle's `with` can't express
+                await this.resolveJoinPathRelationsBatch(entities, collection, collectionPath, idInfo, options.databaseId);
+
+                return entities;
+            } catch (e) {
+                console.warn(`[EntityFetchService] db.query.findMany failed for ${collectionPath}, falling back to db.select:`, e);
+            }
+        }
+
+        // Fallback: db.select + processEntityResults (N+1 for relations)
+        let query = this.db.select().from(table).$dynamic();
+        const allConditions: SQL[] = [];
+
+        if (options.searchString) {
+            const searchConditions = DrizzleConditionBuilder.buildSearchConditions(
+                options.searchString, collection.properties, table
+            );
+            if (searchConditions.length === 0) return [];
+            allConditions.push(DrizzleConditionBuilder.combineConditionsWithOr(searchConditions)!);
+        }
+
+        if (options.filter) {
+            const filterConditions = this.buildFilterConditions(options.filter, table, collectionPath);
+            if (filterConditions.length > 0) allConditions.push(...filterConditions);
+        }
+
+        if (allConditions.length > 0) {
+            const finalCondition = DrizzleConditionBuilder.combineConditionsWithAnd(allConditions);
+            if (finalCondition) query = query.where(finalCondition);
+        }
+
+        const orderExpressions = [];
+        if (options.orderBy) {
+            const orderByField = table[options.orderBy as keyof typeof table] as AnyPgColumn;
+            if (orderByField) {
+                orderExpressions.push(options.order === "asc" ? asc(orderByField) : desc(orderByField));
+            }
+        }
+        orderExpressions.push(desc(idField));
+        if (orderExpressions.length > 0) query = query.orderBy(...orderExpressions);
+
+        if (options.startAfter) {
+            const cursorConditions = this.buildCursorConditions(table, idField, idInfo, options);
+            if (cursorConditions.length > 0) {
+                allConditions.push(...cursorConditions);
+                const finalCondition = DrizzleConditionBuilder.combineConditionsWithAnd(allConditions);
+                if (finalCondition) query = query.where(finalCondition);
+            }
+        }
+
+        const limitValue = options.searchString ? (options.limit || 50) : options.limit;
+        if (limitValue) query = query.limit(limitValue);
+
+        const results = await query;
+
+        return this.processEntityResults<M>(results, collection, collectionPath, idInfo, options.databaseId, false, idInfoArray);
+    }
+
+    /**
+     * Fallback path used when db.query is unavailable.
+     * The primary path uses drizzleResultToEntity which handles relation
+     * mapping without N+1 queries.
+     *
+     * Process raw database results into Entity objects with relations.
+     */
+    private async processEntityResults<M extends Record<string, any>>(
+        results: Record<string, unknown>[],
+        collection: EntityCollection,
+        collectionPath: string,
+        idInfo: { fieldName: string; type: "string" | "number" },
+        databaseId?: string,
+        skipRelations: boolean = false,
+        idInfoArray?: { fieldName: string; type: "string" | "number" }[]
+    ): Promise<Entity<M>[]> {
+        if (results.length === 0) return [];
+
+        // First pass: parse all entities
+        const entitiesWithValues = await Promise.all(results.map(async (entity: Record<string, unknown>) => {
+            const values = await parseDataFromServer(entity as M, collection, this.db, this.registry);
+            return {
+                entity,
+                values,
+                id: (idInfoArray && idInfoArray.length > 1) ? buildCompositeId(entity as Record<string, any>, idInfoArray!) : String(entity[idInfo.fieldName]),
+                path: collectionPath
+            };
+        }));
+
+        if (!skipRelations) {
+            // Second pass: batch load missing one-to-one relations
+            const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+            const propertyKeys = new Set(Object.keys(collection.properties));
+
+            for (const [key, relation] of Object.entries(resolvedRelations)) {
+                if (!propertyKeys.has(key) || relation.cardinality !== "one") continue;
+
+                const entitiesMissingRelation = entitiesWithValues.filter(item => {
+                    const val = (item.values as Record<string, unknown>)[key];
+                    if (val == null) return true;
+                    if (typeof val === "object" && !Array.isArray(val) && (val as Record<string, unknown>).__type === "relation" && (val as Record<string, unknown>).data == null) return true;
+                    return false;
+                });
+
+                if (entitiesMissingRelation.length === 0) continue;
+
+                try {
+                    const entityIds = entitiesMissingRelation.map(item => item.entity[idInfo.fieldName] as string | number);
+                    const relationResults = await this.relationService.batchFetchRelatedEntities(
+                        collectionPath,
+                        entityIds,
+                        key,
+                        relation
+                    );
+
+                    entitiesMissingRelation.forEach(item => {
+                        const entityId = item.entity[idInfo.fieldName] as string | number;
+                        const relatedEntity = relationResults.get(entityId);
+                        if (relatedEntity) {
+                            (item.values as Record<string, unknown>)[key] = {
+                                id: relatedEntity.id,
+                                path: relatedEntity.path,
+                                __type: "relation",
+                                data: relatedEntity
+                            };
+                        }
+                    });
+                } catch (e) {
+                    console.warn(`Could not batch load one-to-one relation property: ${key}`, e);
+                }
+            }
+
+            // Handle many relations
+            const manyRelationPromises = entitiesWithValues.map(async (item) => {
+                const manyRelationQueries = Object.entries(resolvedRelations)
+                    .filter(([key, relation]) => propertyKeys.has(key) && relation.cardinality === "many")
+                    .map(async ([key]) => {
+                        try {
+                            const relatedEntities = await this.relationService.fetchRelatedEntities(
+                                collectionPath,
+                                item.entity[idInfo.fieldName] as string | number,
+                                key,
+                                {}
+                            );
+                            (item.values as Record<string, unknown>)[key] = relatedEntities.map(e => ({
+                                id: e.id,
+                                path: e.path,
+                                __type: "relation",
+                                data: e
+                            }));
+                        } catch (e) {
+                            console.warn(`Could not resolve many relation property: ${key}`, e);
+                        }
+                    });
+                await Promise.all(manyRelationQueries);
+            });
+
+            await Promise.all(manyRelationPromises);
+        }
+
+        return entitiesWithValues.map(item => ({
+            id: item.id,
+            path: item.path,
+            values: item.values as M,
+            databaseId
+        }));
+    }
+
+    /**
+     * Fetch a collection of entities
+     */
+    async fetchCollection<M extends Record<string, any>>(
+        collectionPath: string,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+            startAfter?: Record<string, unknown>;
+            searchString?: string;
+            databaseId?: string;
+        } = {}
+    ): Promise<Entity<M>[]> {
+        // Handle multi-segment paths by resolving through relations
+        if (collectionPath.includes("/")) {
+            return this.fetchCollectionFromPath<M>(collectionPath, options);
+        }
+
+        return this.fetchEntitiesWithConditions<M>(collectionPath, options);
+    }
+
+    /**
+     * Search entities by text
+     */
+    async searchEntities<M extends Record<string, any>>(
+        collectionPath: string,
+        searchString: string,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+            databaseId?: string;
+        } = {}
+    ): Promise<Entity<M>[]> {
+        return this.fetchEntitiesWithConditions<M>(collectionPath, {
+            ...options,
+            searchString
+        });
+    }
+
+    /**
+     * Fetch collection from multi-segment path
+     */
+    private async fetchCollectionFromPath<M extends Record<string, any>>(
+        path: string,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+            startAfter?: Record<string, unknown>;
+            searchString?: string;
+            databaseId?: string;
+        } = {}
+    ): Promise<Entity<M>[]> {
+        const pathSegments = path.split("/").filter(p => p && p !== "undefined");
+
+        if (pathSegments.length < 3 || pathSegments.length % 2 === 0) {
+            throw new Error(`Invalid relation path: ${path}. Expected format: collection/id/relation`);
+        }
+
+        const rootCollectionPath = pathSegments[0];
+        let currentCollection = getCollectionByPath(rootCollectionPath, this.registry);
+        let currentEntityId: string | number = pathSegments[1];
+
+        for (let i = 2; i < pathSegments.length; i += 2) {
+            const relationKey = pathSegments[i];
+            const resolvedRelations = resolveCollectionRelations(currentCollection as import("@rebasepro/types").PostgresCollection<any, any>);
+            const relation = resolvedRelations[relationKey];
+
+            if (!relation) {
+                throw new Error(`Relation '${relationKey}' not found in collection '${currentCollection.slug}'`);
+            }
+
+            if (i === pathSegments.length - 1) {
+                return this.relationService.fetchRelatedEntities<M>(
+                    currentCollection.slug,
+                    currentEntityId,
+                    relationKey,
+                    options
+                );
+            }
+
+            if (i + 1 < pathSegments.length) {
+                const nextEntityId = pathSegments[i + 1];
+                currentCollection = relation.target();
+                currentEntityId = nextEntityId;
+            }
+        }
+
+        throw new Error(`Unable to resolve path: ${path}`);
+    }
+
+    /**
+     * Count entities in a collection
+     */
+    async countEntities<M extends Record<string, any>>(
+        collectionPath: string,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            databaseId?: string;
+        } = {}
+    ): Promise<number> {
+        if (collectionPath.includes("/")) {
+            return this.countEntitiesFromPath<M>(collectionPath, options);
+        }
+
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+
+        let query = this.db.select({ count: count() }).from(table).$dynamic();
+
+        if (options.filter) {
+            const filterConditions = this.buildFilterConditions(options.filter, table, collectionPath);
+            if (filterConditions.length > 0) {
+                query = query.where(and(...filterConditions));
+            }
+        }
+
+        const result = await query;
+        return Number(result[0]?.count || 0);
+    }
+
+    /**
+     * Count entities from multi-segment path
+     */
+    private async countEntitiesFromPath<M extends Record<string, any>>(
+        path: string,
+        options: { filter?: FilterValues<Extract<keyof M, string>>; databaseId?: string } = {}
+    ): Promise<number> {
+        const pathSegments = path.split("/").filter(p => p && p !== "undefined");
+
+        if (pathSegments.length < 3 || pathSegments.length % 2 === 0) {
+            throw new Error(`Invalid relation path: ${path}`);
+        }
+
+        const rootCollectionPath = pathSegments[0];
+        let currentCollection = getCollectionByPath(rootCollectionPath, this.registry);
+        let currentEntityId: string | number = pathSegments[1];
+
+        for (let i = 2; i < pathSegments.length; i += 2) {
+            const relationKey = pathSegments[i];
+            const resolvedRelations = resolveCollectionRelations(currentCollection as import("@rebasepro/types").PostgresCollection<any, any>);
+            const relation = resolvedRelations[relationKey];
+
+            if (!relation) {
+                throw new Error(`Relation '${relationKey}' not found`);
+            }
+
+            if (i === pathSegments.length - 1) {
+                return this.relationService.countRelatedEntities(
+                    currentCollection.slug,
+                    currentEntityId,
+                    relationKey,
+                    options
+                );
+            }
+
+            if (i + 1 < pathSegments.length) {
+                currentCollection = relation.target();
+                currentEntityId = pathSegments[i + 1];
+            }
+        }
+
+        throw new Error(`Unable to count for path: ${path}`);
+    }
+
+    /**
+     * Check if a field value is unique
+     */
+    async checkUniqueField(
+        collectionPath: string,
+        fieldName: string,
+        value: unknown,
+        excludeEntityId?: string,
+        _databaseId?: string
+    ): Promise<boolean> {
+        if (value === undefined || value === null) return true;
+
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+        const idInfoArray = getPrimaryKeys(collection, this.registry);
+        const idInfo = idInfoArray[0];
+        const idField = table[idInfo.fieldName as keyof typeof table] as AnyPgColumn;
+        const field = table[fieldName as keyof typeof table] as AnyPgColumn;
+
+        if (!field) return true;
+
+        const parsedExcludeId = excludeEntityId ? parseIdValues(excludeEntityId, idInfoArray)[idInfo.fieldName] : undefined;
+        const conditions = DrizzleConditionBuilder.buildUniqueFieldCondition(
+            field,
+            value,
+            idField,
+            parsedExcludeId
+        );
+
+        const result = await this.db
+            .select({ count: count() })
+            .from(table)
+            .where(and(...conditions));
+
+        const countResult = Number(result[0]?.count || 0);
+        return countResult === 0;
+    }
+
+    /**
+     * Get the RelationService instance for external use
+     */
+    getRelationService(): RelationService {
+        return this.relationService;
+    }
+
+    // =============================================================
+    // REST API INCLUDE-AWARE METHODS
+    // =============================================================
+
+    /**
+     * Fetch a collection of entities with optional relation includes.
+     * When `include` is provided, only the specified relations are populated
+     * with full entity data (not just { id, path, __type }).
+     * When `include` is absent, no relation queries are made (fast path).
+     *
+     * @param include - Array of relation keys to populate, or ["*"] for all
+     */
+    async fetchCollectionForRest<M extends Record<string, any>>(
+        collectionPath: string,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+            startAfter?: Record<string, unknown>;
+            searchString?: string;
+            databaseId?: string;
+        } = {},
+        include?: string[]
+    ): Promise<Record<string, unknown>[]> {
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+        const idInfoArray = getPrimaryKeys(collection, this.registry);
+        const idInfo = idInfoArray[0];
+        const idField = table[idInfo.fieldName as keyof typeof table] as AnyPgColumn;
+
+        // Primary path: use db.query.findMany
+        
+        const tableName = getTableName(table);
+
+        const qb = this.getQueryBuilder(tableName);
+        if (qb) {
+            try {
+                const withConfig = (include && include.length > 0)
+                    ? this.buildWithConfig(collection, include)
+                    : undefined;
+
+                const queryOpts = this.buildDrizzleQueryOptions<M>(
+                    table, idField, idInfo, options, collectionPath, withConfig
+                );
+
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any -- dynamic query config
+                const results = await qb.findMany(queryOpts as unknown as Parameters<NonNullable<typeof qb>["findMany"]>[0]);
+
+                return (results as Record<string, unknown>[]).map(row =>
+                    this.drizzleResultToRestRow(row, collection, idInfo, idInfoArray)
+                );
+            } catch (e) {
+                console.warn(`[fetchCollectionForRest] db.query.findMany failed for ${collectionPath}, falling back:`, e);
+            }
+        }
+
+        // Fallback: fetch base entities without relations
+        const entities = await this.fetchEntitiesWithConditionsRaw<M>(collectionPath, options);
+
+        if (!include || include.length === 0) {
+            return entities.map(entity => ({
+                id: (idInfoArray.length > 1) ? buildCompositeId(entity as Record<string, any>, idInfoArray) : String(entity[idInfo.fieldName]),
+                ...entity
+            }));
+        }
+
+        // Fallback relation loading via batch
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+        const propertyKeys = new Set(Object.keys(collection.properties || {}));
+        const shouldInclude = (key: string) =>
+            include[0] === "*" || include.includes(key);
+
+        const entityIds = entities.map(e => e[idInfo.fieldName] as string | number);
+
+        for (const [key, relation] of Object.entries(resolvedRelations)) {
+            if (!propertyKeys.has(key) || !shouldInclude(key) || relation.cardinality !== "one") continue;
+            try {
+                const batchResults = await this.relationService.batchFetchRelatedEntities(
+                    collectionPath, entityIds, key, relation
+                );
+                for (const entity of entities) {
+                    const eid = entity[idInfo.fieldName] as string | number;
+                    const related = batchResults.get(eid);
+                    if (related) {
+                        (entity as Record<string, unknown>)[key] = { id: related.id, ...related.values };
+                    }
+                }
+            } catch (e) {
+                console.warn(`[include] Failed to batch load one-to-one '${key}':`, e);
+            }
+        }
+
+        for (const [key, relation] of Object.entries(resolvedRelations)) {
+            if (!propertyKeys.has(key) || !shouldInclude(key) || relation.cardinality !== "many") continue;
+            try {
+                const batchResults = await this.batchFetchManyRelatedEntities(
+                    collectionPath, entityIds, key
+                );
+                for (const entity of entities) {
+                    const eid = entity[idInfo.fieldName] as string | number;
+                    const relatedList = batchResults.get(String(eid)) || [];
+                    (entity as Record<string, unknown>)[key] = relatedList.map(e => ({
+                        id: e.id, ...e.values
+                    }));
+                }
+            } catch (e) {
+                console.warn(`[include] Failed to batch load many '${key}':`, e);
+            }
+        }
+
+        return entities.map(entity => ({
+            id: (idInfoArray.length > 1) ? buildCompositeId(entity as Record<string, any>, idInfoArray) : String(entity[idInfo.fieldName]),
+            ...entity
+        }));
+    }
+
+    /**
+     * Fetch a single entity with optional relation includes for REST API.
+     */
+    async fetchEntityForRest<M extends Record<string, any>>(
+        collectionPath: string,
+        entityId: string | number,
+        include?: string[],
+        databaseId?: string
+    ): Promise<Record<string, unknown> | null> {
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+        const idInfoArray = getPrimaryKeys(collection, this.registry);
+        const idInfo = idInfoArray[0];
+        const idField = table[idInfo.fieldName as keyof typeof table] as AnyPgColumn;
+
+        const parsedIdObj = parseIdValues(entityId, idInfoArray);
+        const parsedId = parsedIdObj[idInfo.fieldName];
+
+        // Primary path: use db.query.findFirst
+        
+        const tableName = getTableName(table);
+
+        const qb = this.getQueryBuilder(tableName);
+        if (qb) {
+            try {
+                const withConfig = (include && include.length > 0)
+                    ? this.buildWithConfig(collection, include)
+                    : undefined;
+
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any -- dynamic query config
+                const row = await qb.findFirst({
+                    where: eq(idField, parsedId),
+                    ...(withConfig ? { with: withConfig } : {})
+                } as unknown as Parameters<NonNullable<typeof qb>["findFirst"]>[0]);
+
+                if (!row) return null;
+
+                return this.drizzleResultToRestRow(row, collection, idInfo, idInfoArray);
+            } catch (e) {
+                console.warn(`[fetchEntityForRest] db.query.findFirst failed for ${collectionPath}, falling back:`, e);
+            }
+        }
+
+        // Fallback: db.select + N+1 relation loading
+        const result = await this.db
+            .select()
+            .from(table)
+            .where(eq(idField, parsedId))
+            .limit(1);
+
+        if (result.length === 0) return null;
+
+        const raw = result[0] as Record<string, unknown>;
+        const flatEntity: Record<string, unknown> = { id: (idInfoArray.length > 1) ? buildCompositeId(raw as Record<string, any>, idInfoArray) : String(raw[idInfo.fieldName]), ...raw };
+
+        if (!include || include.length === 0) {
+            return flatEntity;
+        }
+
+        // Fallback relation population
+        const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+        const propertyKeys = new Set(Object.keys(collection.properties || {}));
+        const shouldInclude = (key: string) =>
+            include[0] === "*" || include.includes(key);
+
+        for (const [key, relation] of Object.entries(resolvedRelations)) {
+            if (!propertyKeys.has(key) || !shouldInclude(key)) continue;
+
+            try {
+                const relatedEntities = await this.relationService.fetchRelatedEntities(
+                    collectionPath, parsedId, key, {}
+                );
+
+                if (relation.cardinality === "one") {
+                    if (relatedEntities.length > 0) {
+                        const e = relatedEntities[0];
+                        flatEntity[key] = { id: e.id, ...e.values };
+                    }
+                } else {
+                    flatEntity[key] = relatedEntities.map(e => ({
+                        id: e.id, ...e.values
+                    }));
+                }
+            } catch (e) {
+                console.warn(`[include] Failed to load relation '${key}':`, e);
+            }
+        }
+
+        return flatEntity;
+    }
+
+    /**
+     * Fetch raw rows without any relation processing (for REST fast path)
+     */
+    private async fetchEntitiesWithConditionsRaw<M extends Record<string, any>>(
+        collectionPath: string,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+            startAfter?: Record<string, unknown>;
+            searchString?: string;
+        } = {}
+    ): Promise<Record<string, unknown>[]> {
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+        const idInfoArray = getPrimaryKeys(collection, this.registry);
+        const idInfo = idInfoArray[0];
+        const idField = table[idInfo.fieldName as keyof typeof table] as AnyPgColumn;
+
+        let query = this.db.select().from(table).$dynamic();
+        const allConditions: SQL[] = [];
+
+        if (options.searchString) {
+            const searchConditions = DrizzleConditionBuilder.buildSearchConditions(
+                options.searchString, collection.properties, table
+            );
+            if (searchConditions.length === 0) return [];
+            allConditions.push(DrizzleConditionBuilder.combineConditionsWithOr(searchConditions)!);
+        }
+
+        if (options.filter) {
+            const filterConditions = this.buildFilterConditions(options.filter, table, collectionPath);
+            if (filterConditions.length > 0) allConditions.push(...filterConditions);
+        }
+
+        if (allConditions.length > 0) {
+            const finalCondition = DrizzleConditionBuilder.combineConditionsWithAnd(allConditions);
+            if (finalCondition) query = query.where(finalCondition);
+        }
+
+        const orderExpressions = [];
+        if (options.orderBy) {
+            const orderByField = table[options.orderBy as keyof typeof table] as AnyPgColumn;
+            if (orderByField) {
+                orderExpressions.push(options.order === "asc" ? asc(orderByField) : desc(orderByField));
+            }
+        }
+        orderExpressions.push(desc(idField));
+        if (orderExpressions.length > 0) query = query.orderBy(...orderExpressions);
+
+        const limitValue = options.searchString ? (options.limit || 50) : options.limit;
+        if (limitValue) query = query.limit(limitValue);
+
+        return await query as Record<string, unknown>[];
+    }
+
+    /**
+     * Check if the Drizzle instance has the relational query API available
+     * for a given collection path.
+     * Note: Primary path now uses inline `getQueryBuilder()` checks.
+     */
+    private hasDrizzleQueryAPI(collectionPath: string): boolean {
+        
+        const qb = this.getQueryBuilder("__probe__");
+        if (!qb) {
+            // If getQueryBuilder returns undefined even for a probe, query API is not available
+            return false;
+        }
+        const collection = getCollectionByPath(collectionPath, this.registry);
+        const table = getTableForCollection(collection, this.registry);
+        const tableName = getTableName(table);
+        return !!this.getQueryBuilder(tableName);
+    }
+
+    /**
+     * Attempt to use Drizzle's relational query API (db.query.<table>.findMany)
+     * for efficient JOIN-based relation loading.
+     * Returns null if the API is not available or the query fails.
+     * Note: Primary path now uses `buildWithConfig` + `buildDrizzleQueryOptions`.
+     */
+    private async fetchWithDrizzleQuery<M extends Record<string, any>>(
+        collectionPath: string,
+        collection: EntityCollection,
+        options: {
+            filter?: FilterValues<Extract<keyof M, string>>;
+            orderBy?: string;
+            order?: "desc" | "asc";
+            limit?: number;
+        },
+        include: string[],
+        idInfo: { fieldName: string; type: "string" | "number" },
+        idInfoArray?: { fieldName: string; type: "string" | "number" }[]
+    ): Promise<Record<string, unknown>[] | null> {
+        try {
+            
+            const table = getTableForCollection(collection, this.registry);
+            const tableName = getTableName(table);
+            const queryTarget = this.getQueryBuilder(tableName);
+
+            if (!queryTarget?.findMany) return null;
+
+            // Build the `with` config from include array
+            const resolvedRelations = resolveCollectionRelations(collection as import("@rebasepro/types").PostgresCollection<any, any>);
+            const withConfig: Record<string, boolean> = {};
+            for (const [key, relation] of Object.entries(resolvedRelations)) {
+                if (include[0] === "*" || include.includes(key)) {
+                    // Use the Drizzle relation name (from the schema)
+                    const drizzleRelName = relation.relationName || key;
+                    withConfig[drizzleRelName] = true;
+                }
+            }
+
+            // Build query options
+            const queryOpts: Record<string, unknown> = { with: withConfig };
+            if (options.limit) queryOpts.limit = options.limit;
+
+            // Build where clause
+            if (options.filter) {
+                const filterConditions = this.buildFilterConditions(
+                    options.filter, table, collectionPath
+                );
+                if (filterConditions.length > 0) {
+                    queryOpts.where = and(...filterConditions);
+                }
+            }
+
+            // Build orderBy
+            if (options.orderBy) {
+                const orderByField = table[options.orderBy as keyof typeof table] as AnyPgColumn;
+                if (orderByField) {
+                    queryOpts.orderBy = options.order === "asc" ? asc(orderByField) : desc(orderByField);
+                }
+            }
+
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any -- deprecated method, will be removed
+            const results = await queryTarget.findMany(queryOpts as unknown as Parameters<NonNullable<typeof queryTarget>["findMany"]>[0]);
+
+            // Flatten the nested Drizzle results into REST format
+            return results.map((row: Record<string, unknown>) => {
+                const flat: Record<string, unknown> = { id: (idInfoArray && idInfoArray.length > 1) ? buildCompositeId(row as Record<string, any>, idInfoArray) : String(row[idInfo.fieldName]) };
+                for (const [k, v] of Object.entries(row)) {
+                    if (k === idInfo.fieldName) continue;
+                    if (Array.isArray(v)) {
+                        // Many relation — flatten each nested entity
+                        flat[k] = v.map((item: Record<string, unknown>) => {
+                            // Junction table rows may have the target nested, flatten those
+                            const keys = Object.keys(item);
+                            // If it looks like a junction row (only FKs + nested objects), extract nested
+                            const nestedObj = keys.find(nk => typeof item[nk] === "object" && item[nk] !== null && !Array.isArray(item[nk]));
+                            if (nestedObj && keys.length <= 3) {
+                                const nested = item[nestedObj] as Record<string, unknown>;
+                                return { id: String(nested.id ?? nested[Object.keys(nested)[0]]), ...nested };
+                            }
+                            return { id: String(item.id ?? item[Object.keys(item)[0]]), ...item };
+                        });
+                    } else if (typeof v === "object" && v !== null) {
+                        // One-to-one relation — inline the object
+                        const relObj = v as Record<string, unknown>;
+                        flat[k] = { id: String(relObj.id ?? relObj[Object.keys(relObj)[0]]), ...relObj };
+                    } else {
+                        flat[k] = v;
+                    }
+                }
+                return flat;
+            });
+        } catch (e) {
+            console.warn(`[include] Drizzle relational query failed for '${collectionPath}', falling back:`, e);
+            return null;
+        }
+    }
+
+    /**
+     * Fallback path used when db.query is unavailable.
+     * The primary path uses db.query.findMany with `with` config, which
+     * loads all relations in a single query.
+     *
+     * Batch fetch many-to-many related entities for multiple parent IDs.
+     * Groups results by parent ID to avoid N+1.
+     */
+    private async batchFetchManyRelatedEntities(
+        parentCollectionPath: string,
+        parentIds: (string | number)[],
+        relationKey: string
+    ): Promise<Map<string, Entity[]>> {
+        const resultMap = new Map<string, Entity[]>();
+
+        // Fetch for all parents using Promise.all (limited batch)
+        const batchPromises = parentIds.map(async (parentId) => {
+            try {
+                const related = await this.relationService.fetchRelatedEntities(
+                    parentCollectionPath, parentId, relationKey, {}
+                );
+                resultMap.set(String(parentId), related);
+            } catch (e) {
+                resultMap.set(String(parentId), []);
+            }
+        });
+
+        await Promise.all(batchPromises);
+        return resultMap;
+    }
+}