@friggframework/core 2.0.0-next.5 → 2.0.0-next.51

package/modules/repositories/module-repository-postgres.js

@@ -0,0 +1,426 @@
+const { prisma } = require('../../database/prisma');
+const { ModuleRepositoryInterface } = require('./module-repository-interface');
+
+/**
+ * PostgreSQL Module Repository Adapter
+ * Handles Entity model operations for external service entities with PostgreSQL
+ *
+ * PostgreSQL-specific characteristics:
+ * - Uses Int IDs with autoincrement
+ * - Requires ID conversion: String (app layer) ↔ Int (database)
+ * - All returned IDs are converted to strings for application layer consistency
+ */
+class ModuleRepositoryPostgres extends ModuleRepositoryInterface {
+    constructor() {
+        super();
+        this.prisma = prisma;
+    }
+
+    /**
+     * Convert string ID to integer for PostgreSQL queries
+     * @private
+     * @param {string|number|null|undefined} id - ID to convert
+     * @returns {number|null|undefined} Integer ID or null/undefined
+     * @throws {Error} If ID cannot be converted to integer
+     */
+    _convertId(id) {
+        if (id === null || id === undefined) return id;
+        const parsed = parseInt(id, 10);
+        if (isNaN(parsed)) {
+            throw new Error(`Invalid ID: ${id} cannot be converted to integer`);
+        }
+        return parsed;
+    }
+
+    /**
+     * Convert any value to string (handles null/undefined)
+     * @private
+     * @param {*} value - Value to convert
+     * @returns {string|null|undefined} String value or null/undefined
+     */
+    _toString(value) {
+        if (value === null || value === undefined) return value;
+        return String(value);
+    }
+
+    /**
+     * Convert credential object IDs to strings
+     * @private
+     * @param {Object|null} credential - Credential object from database
+     * @returns {Object|null} Credential with string IDs
+     */
+    _convertCredentialIds(credential) {
+        if (!credential) return credential;
+        return {
+            ...credential,
+            id: credential.id?.toString(),
+            userId: credential.userId?.toString(),
+        };
+    }
+
+    /**
+     * Fetch credential by ID separately to ensure encryption extension processes it
+     * This fixes the bug where credentials fetched via include bypass decryption
+     * @private
+     * @param {number|null|undefined} credentialId - Credential ID (integer for PostgreSQL)
+     * @returns {Promise<Object|null>} Decrypted credential with string IDs or null
+     */
+    async _fetchCredential(credentialId) {
+        if (!credentialId) return null;
+
+        const credential = await this.prisma.credential.findUnique({
+            where: { id: credentialId },
+        });
+
+        return this._convertCredentialIds(credential);
+    }
+
+    /**
+     * Fetch multiple credentials in bulk separately to ensure decryption
+     * More efficient than fetching one-by-one for arrays of entities
+     * @private
+     * @param {Array<number>} credentialIds - Array of credential IDs (integers for PostgreSQL)
+     * @returns {Promise<Map<number, Object>>} Map of credentialId -> credential object
+     */
+    async _fetchCredentialsBulk(credentialIds) {
+        if (!credentialIds || credentialIds.length === 0) {
+            return new Map();
+        }
+
+        const validIds = credentialIds.filter(id => id !== null && id !== undefined);
+
+        if (validIds.length === 0) {
+            return new Map();
+        }
+
+        const credentials = await this.prisma.credential.findMany({
+            where: { id: { in: validIds } },
+        });
+
+        const credentialMap = new Map();
+        for (const credential of credentials) {
+            credentialMap.set(
+                credential.id,
+                this._convertCredentialIds(credential)
+            );
+        }
+
+        return credentialMap;
+    }
+
+    /**
+     * Find entity by ID with credential
+     * Replaces: Entity.findById(entityId).populate('credential')
+     *
+     * @param {string} entityId - Entity ID (string from application layer)
+     * @returns {Promise<Object>} Entity object with string IDs
+     * @throws {Error} If entity not found
+     */
+    async findEntityById(entityId) {
+        const intId = this._convertId(entityId);
+
+        const entity = await this.prisma.entity.findUnique({
+            where: { id: intId },
+        });
+
+        if (!entity) {
+            throw new Error(`Entity ${entityId} not found`);
+        }
+
+        const credential = await this._fetchCredential(entity.credentialId);
+
+        return {
+            id: entity.id.toString(),
+            accountId: entity.accountId,
+            credential,
+            userId: entity.userId?.toString(),
+            name: entity.name,
+            externalId: entity.externalId,
+            moduleName: entity.moduleName,
+        };
+    }
+
+    /**
+     * Find all entities for a user
+     * Replaces: Entity.find({ user: userId }).populate('credential')
+     *
+     * @param {string} userId - User ID (string from application layer)
+     * @returns {Promise<Array>} Array of entity objects with string IDs
+     */
+    async findEntitiesByUserId(userId) {
+        const intUserId = this._convertId(userId);
+
+        const entities = await this.prisma.entity.findMany({
+            where: { userId: intUserId },
+        });
+
+        const credentialIds = entities.map(e => e.credentialId).filter(Boolean);
+        const credentialMap = await this._fetchCredentialsBulk(credentialIds);
+
+        return entities.map((e) => ({
+            id: e.id.toString(),
+            accountId: e.accountId,
+            credential: credentialMap.get(e.credentialId) || null,
+            userId: e.userId?.toString(),
+            name: e.name,
+            externalId: e.externalId,
+            moduleName: e.moduleName,
+        }));
+    }
+
+    /**
+     * Find entities by array of IDs
+     * Replaces: Entity.find({ _id: { $in: entitiesIds } }).populate('credential')
+     *
+     * @param {Array<string>} entitiesIds - Array of entity IDs (strings from application layer)
+     * @returns {Promise<Array>} Array of entity objects with string IDs
+     */
+    async findEntitiesByIds(entitiesIds) {
+        const intIds = entitiesIds.map((id) => this._convertId(id));
+
+        const entities = await this.prisma.entity.findMany({
+            where: { id: { in: intIds } },
+        });
+
+        const credentialIds = entities.map(e => e.credentialId).filter(Boolean);
+        const credentialMap = await this._fetchCredentialsBulk(credentialIds);
+
+        return entities.map((e) => ({
+            id: e.id.toString(),
+            accountId: e.accountId,
+            credential: credentialMap.get(e.credentialId) || null,
+            userId: e.userId?.toString(),
+            name: e.name,
+            externalId: e.externalId,
+            moduleName: e.moduleName,
+        }));
+    }
+
+    /**
+     * Find entities by user ID and module name
+     * Replaces: Entity.find({ user: userId, moduleName: moduleName }).populate('credential')
+     *
+     * @param {string} userId - User ID (string from application layer)
+     * @param {string} moduleName - Module name
+     * @returns {Promise<Array>} Array of entity objects with string IDs
+     */
+    async findEntitiesByUserIdAndModuleName(userId, moduleName) {
+        const intUserId = this._convertId(userId);
+
+        const entities = await this.prisma.entity.findMany({
+            where: {
+                userId: intUserId,
+                moduleName,
+            },
+        });
+
+        const credentialIds = entities.map(e => e.credentialId).filter(Boolean);
+        const credentialMap = await this._fetchCredentialsBulk(credentialIds);
+
+        return entities.map((e) => ({
+            id: e.id.toString(),
+            accountId: e.accountId,
+            credential: credentialMap.get(e.credentialId) || null,
+            userId: e.userId?.toString(),
+            name: e.name,
+            externalId: e.externalId,
+            moduleName: e.moduleName,
+        }));
+    }
+
+    /**
+     * Remove credential reference from entity
+     * Replaces: Entity.updateOne({ _id: entityId }, { $unset: { credential: "" } })
+     *
+     * @param {string} entityId - Entity ID (string from application layer)
+     * @returns {Promise<boolean>} Success indicator
+     */
+    async unsetCredential(entityId) {
+        const intId = this._convertId(entityId);
+        await this.prisma.entity.update({
+            where: { id: intId },
+            data: { credentialId: null },
+        });
+
+        return true;
+    }
+
+    /**
+     * Find entity by filter criteria
+     *
+     * @param {Object} filter - Filter criteria
+     * @returns {Promise<Object|null>} Entity object with string IDs or null
+     */
+    async findEntity(filter) {
+        const where = this._convertFilterToWhere(filter);
+
+        const entity = await this.prisma.entity.findFirst({
+            where,
+        });
+
+        if (!entity) {
+            return null;
+        }
+
+        const credential = await this._fetchCredential(entity.credentialId);
+
+        return {
+            id: entity.id.toString(),
+            accountId: entity.accountId,
+            credential,
+            userId: entity.userId?.toString(),
+            name: entity.name,
+            externalId: entity.externalId,
+            moduleName: entity.moduleName,
+        };
+    }
+
+    /**
+     * Create a new entity
+     * Replaces: Entity.create(entityData)
+     *
+     * @param {Object} entityData - Entity data (with string IDs from application layer)
+     * @returns {Promise<Object>} Created entity object with string IDs
+     */
+    async createEntity(entityData) {
+        const data = {
+            userId: this._convertId(entityData.user || entityData.userId),
+            credentialId: this._convertId(
+                entityData.credential || entityData.credentialId
+            ),
+            name: entityData.name,
+            moduleName: entityData.moduleName,
+            externalId: entityData.externalId,
+            accountId: entityData.accountId,
+        };
+
+        const entity = await this.prisma.entity.create({
+            data,
+        });
+
+        const credential = await this._fetchCredential(entity.credentialId);
+
+        return {
+            id: entity.id.toString(),
+            accountId: entity.accountId,
+            credential,
+            userId: entity.userId?.toString(),
+            name: entity.name,
+            externalId: entity.externalId,
+            moduleName: entity.moduleName,
+        };
+    }
+
+    /**
+     * Update an entity by ID
+     * Replaces: Entity.findByIdAndUpdate(entityId, updates, { new: true })
+     *
+     * @param {string} entityId - Entity ID to update (string from application layer)
+     * @param {Object} updates - Fields to update (with string IDs from application layer)
+     * @returns {Promise<Object|null>} Updated entity object with string IDs or null if not found
+     */
+    async updateEntity(entityId, updates) {
+        const data = {};
+        if (updates.user !== undefined)
+            data.userId = this._convertId(updates.user);
+        if (updates.userId !== undefined)
+            data.userId = this._convertId(updates.userId);
+        if (updates.credential !== undefined)
+            data.credentialId = this._convertId(updates.credential);
+        if (updates.credentialId !== undefined)
+            data.credentialId = this._convertId(updates.credentialId);
+        if (updates.name !== undefined) data.name = updates.name;
+        if (updates.moduleName !== undefined)
+            data.moduleName = updates.moduleName;
+        if (updates.externalId !== undefined)
+            data.externalId = updates.externalId;
+        if (updates.accountId !== undefined) data.accountId = updates.accountId;
+
+        try {
+            const intId = this._convertId(entityId);
+
+            const entity = await this.prisma.entity.update({
+                where: { id: intId },
+                data,
+            });
+
+            const credential = await this._fetchCredential(entity.credentialId);
+
+            return {
+                id: entity.id.toString(),
+                accountId: entity.accountId,
+                credential,
+                userId: entity.userId?.toString(),
+                name: entity.name,
+                externalId: entity.externalId,
+                moduleName: entity.moduleName,
+            };
+        } catch (error) {
+            if (error.code === 'P2025') {
+                return null;
+            }
+            throw error;
+        }
+    }
+
+    /**
+     * Delete an entity by ID
+     * Replaces: Entity.deleteOne({ _id: entityId })
+     *
+     * @param {string} entityId - Entity ID to delete (string from application layer)
+     * @returns {Promise<boolean>} True if deleted successfully
+     */
+    async deleteEntity(entityId) {
+        try {
+            const intId = this._convertId(entityId);
+            await this.prisma.entity.delete({
+                where: { id: intId },
+            });
+            return true;
+        } catch (error) {
+            if (error.code === 'P2025') {
+                // Record not found
+                return false;
+            }
+            throw error;
+        }
+    }
+
+    /**
+     * Convert Mongoose-style filter to Prisma where clause (converting IDs to Int)
+     * @private
+     * @param {Object} filter - Mongoose filter (with string IDs from application layer)
+     * @returns {Object} Prisma where clause (with Int IDs for PostgreSQL)
+     */
+    _convertFilterToWhere(filter) {
+        const where = {};
+
+        // Handle _id field (Mongoose uses _id, Prisma uses id)
+        if (filter._id) {
+            where.id = this._convertId(filter._id);
+        }
+
+        // Handle user field (Mongoose uses user, Prisma uses userId)
+        if (filter.user) {
+            where.userId = this._convertId(filter.user);
+        }
+
+        // Handle credential field (Mongoose uses credential, Prisma uses credentialId)
+        if (filter.credential) {
+            where.credentialId = this._convertId(filter.credential);
+        }
+
+        // Copy other fields directly (converting IDs)
+        if (filter.id) where.id = this._convertId(filter.id);
+        if (filter.userId) where.userId = this._convertId(filter.userId);
+        if (filter.credentialId)
+            where.credentialId = this._convertId(filter.credentialId);
+        if (filter.name) where.name = filter.name;
+        if (filter.moduleName) where.moduleName = filter.moduleName;
+        if (filter.externalId) where.externalId = this._toString(filter.externalId);
+
+        return where;
+    }
+}
+
+module.exports = { ModuleRepositoryPostgres };