@friggframework/core 2.0.0-next.40 → 2.0.0-next.42

package/database/encryption/prisma-encryption-extension.js

@@ -0,0 +1,222 @@
+/**
+ * Prisma Client Extension for transparent field-level encryption.
+ * Intercepts Prisma queries to encrypt on write and decrypt on read.
+ */
+
+const { getEncryptedFields } = require('./encryption-schema-registry');
+const { FieldEncryptionService } = require('./field-encryption-service');
+
+function createEncryptionExtension({ cryptor, enabled = true }) {
+    if (!enabled) {
+        return (client) => client;
+    }
+
+    if (!cryptor) {
+        throw new Error(
+            'Cryptor instance required for encryption extension'
+        );
+    }
+
+    const encryptionService = new FieldEncryptionService({
+        cryptor,
+        schema: { getEncryptedFields },
+    });
+
+    return {
+        name: 'frigg-field-encryption',
+        query: {
+            $allModels: {
+                async create({ model, args, query }) {
+                    if (args.data) {
+                        args.data = await encryptionService.encryptFields(
+                            model,
+                            args.data
+                        );
+                    }
+
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+
+                async createMany({ model, args, query }) {
+                    if (args.data && Array.isArray(args.data)) {
+                        args.data =
+                            await encryptionService.encryptFieldsInBulk(
+                                model,
+                                args.data
+                            );
+                    } else if (args.data) {
+                        args.data = await encryptionService.encryptFields(
+                            model,
+                            args.data
+                        );
+                    }
+
+                    return await query(args);
+                },
+
+                async update({ model, args, query }) {
+                    if (args.data) {
+                        args.data = await encryptionService.encryptFields(
+                            model,
+                            args.data
+                        );
+                    }
+
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+
+                async updateMany({ model, args, query }) {
+                    if (args.data) {
+                        args.data = await encryptionService.encryptFields(
+                            model,
+                            args.data
+                        );
+                    }
+
+                    return await query(args);
+                },
+
+                async upsert({ model, args, query }) {
+                    if (args.create) {
+                        args.create = await encryptionService.encryptFields(
+                            model,
+                            args.create
+                        );
+                    }
+
+                    if (args.update) {
+                        args.update = await encryptionService.encryptFields(
+                            model,
+                            args.update
+                        );
+                    }
+
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+
+                async findUnique({ model, args, query }) {
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+
+                async findFirst({ model, args, query }) {
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+
+                async findMany({ model, args, query }) {
+                    const results = await query(args);
+
+                    if (results && Array.isArray(results)) {
+                        return await encryptionService.decryptFieldsInBulk(
+                            model,
+                            results
+                        );
+                    }
+
+                    return results;
+                },
+
+                async delete({ model, args, query }) {
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+
+                async deleteMany({ model, args, query }) {
+                    return await query(args);
+                },
+
+                async count({ model, args, query }) {
+                    return await query(args);
+                },
+
+                async aggregate({ model, args, query }) {
+                    return await query(args);
+                },
+
+                async groupBy({ model, args, query }) {
+                    return await query(args);
+                },
+
+                async findFirstOrThrow({ model, args, query }) {
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+
+                async findUniqueOrThrow({ model, args, query }) {
+                    const result = await query(args);
+
+                    if (result) {
+                        return await encryptionService.decryptFields(
+                            model,
+                            result
+                        );
+                    }
+
+                    return result;
+                },
+            },
+        },
+    };
+}
+
+module.exports = { createEncryptionExtension };

package/database/encryption/prisma-encryption-extension.test.js

@@ -0,0 +1,439 @@
+const { createEncryptionExtension } = require('./prisma-encryption-extension');
+
+describe('Prisma Encryption Extension', () => {
+    let mockCryptor;
+    let mockQuery;
+
+    beforeEach(() => {
+        // Mock Cryptor
+        mockCryptor = {
+            encrypt: jest
+                .fn()
+                .mockImplementation(
+                    (value) => `encrypted:${value}:iv:enckey`
+                ),
+            decrypt: jest
+                .fn()
+                .mockImplementation((value) => {
+                    const parts = value.split(':');
+                    return parts[1]; // Extract original value
+                }),
+        };
+
+        // Mock Prisma query function
+        mockQuery = jest.fn().mockImplementation((args) => args.mockResult);
+    });
+
+    describe('createEncryptionExtension', () => {
+        it('should create extension with valid config', () => {
+            const extension = createEncryptionExtension({
+                cryptor: mockCryptor,
+                enabled: true,
+            });
+
+            expect(extension).toBeDefined();
+            expect(extension.name).toBe('frigg-field-encryption');
+            expect(extension.query).toBeDefined();
+            expect(extension.query.$allModels).toBeDefined();
+        });
+
+        it('should return no-op extension when disabled', () => {
+            const extension = createEncryptionExtension({
+                cryptor: mockCryptor,
+                enabled: false,
+            });
+
+            // No-op extension is a function that returns its input
+            const mockClient = { $extends: jest.fn() };
+            const result = extension(mockClient);
+
+            expect(result).toBe(mockClient);
+        });
+
+        it('should throw if cryptor not provided when enabled', () => {
+            expect(() => {
+                createEncryptionExtension({ enabled: true });
+            }).toThrow('Cryptor instance required');
+        });
+
+        it('should not throw if cryptor not provided when disabled', () => {
+            expect(() => {
+                createEncryptionExtension({ enabled: false });
+            }).not.toThrow();
+        });
+    });
+
+    describe('Query Interceptors', () => {
+        let extension;
+        let handlers;
+
+        beforeEach(() => {
+            extension = createEncryptionExtension({
+                cryptor: mockCryptor,
+                enabled: true,
+            });
+            handlers = extension.query.$allModels;
+        });
+
+        describe('create', () => {
+            it('should encrypt data before create', async () => {
+                const args = {
+                    data: {
+                        data: { access_token: 'secret123' },
+                    },
+                    mockResult: {
+                        id: '1',
+                        data: { access_token: 'encrypted:secret123:iv:enckey' },
+                    },
+                };
+
+                await handlers.create({
+                    model: 'Credential',
+                    operation: 'create',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(mockCryptor.encrypt).toHaveBeenCalledWith('secret123');
+                expect(args.data.data.access_token).toBe(
+                    'encrypted:secret123:iv:enckey'
+                );
+            });
+
+            it('should decrypt result after create', async () => {
+                const args = {
+                    data: { data: { access_token: 'secret123' } },
+                    mockResult: {
+                        id: '1',
+                        data: { access_token: 'encrypted:secret123:iv:enckey' },
+                    },
+                };
+
+                const result = await handlers.create({
+                    model: 'Credential',
+                    operation: 'create',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(mockCryptor.decrypt).toHaveBeenCalledWith(
+                    'encrypted:secret123:iv:enckey'
+                );
+                expect(result.data.access_token).toBe('secret123');
+            });
+
+            it('should handle null result', async () => {
+                const args = {
+                    data: { data: { access_token: 'secret123' } },
+                    mockResult: null,
+                };
+
+                const result = await handlers.create({
+                    model: 'Credential',
+                    operation: 'create',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(result).toBeNull();
+            });
+        });
+
+        describe('createMany', () => {
+            it('should encrypt array of data', async () => {
+                const args = {
+                    data: [
+                        { data: { access_token: 'secret1' } },
+                        { data: { access_token: 'secret2' } },
+                    ],
+                    mockResult: { count: 2 },
+                };
+
+                await handlers.createMany({
+                    model: 'Credential',
+                    operation: 'createMany',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(mockCryptor.encrypt).toHaveBeenCalledWith('secret1');
+                expect(mockCryptor.encrypt).toHaveBeenCalledWith('secret2');
+                expect(args.data[0].data.access_token).toBe(
+                    'encrypted:secret1:iv:enckey'
+                );
+                expect(args.data[1].data.access_token).toBe(
+                    'encrypted:secret2:iv:enckey'
+                );
+            });
+
+            it('should handle single object in createMany', async () => {
+                const args = {
+                    data: { data: { access_token: 'secret' } },
+                    mockResult: { count: 1 },
+                };
+
+                await handlers.createMany({
+                    model: 'Credential',
+                    operation: 'createMany',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(mockCryptor.encrypt).toHaveBeenCalledWith('secret');
+            });
+        });
+
+        describe('update', () => {
+            it('should encrypt update data', async () => {
+                const args = {
+                    data: { data: { access_token: 'newsecret' } },
+                    mockResult: {
+                        id: '1',
+                        data: { access_token: 'encrypted:newsecret:iv:enckey' },
+                    },
+                };
+
+                await handlers.update({
+                    model: 'Credential',
+                    operation: 'update',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(mockCryptor.encrypt).toHaveBeenCalledWith('newsecret');
+            });
+
+            it('should decrypt result after update', async () => {
+                const args = {
+                    data: { data: { access_token: 'newsecret' } },
+                    mockResult: {
+                        id: '1',
+                        data: { access_token: 'encrypted:newsecret:iv:enckey' },
+                    },
+                };
+
+                const result = await handlers.update({
+                    model: 'Credential',
+                    operation: 'update',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(result.data.access_token).toBe('newsecret');
+            });
+        });
+
+        describe('upsert', () => {
+            it('should encrypt both create and update data', async () => {
+                const args = {
+                    create: { data: { access_token: 'createsecret' } },
+                    update: { data: { access_token: 'updatesecret' } },
+                    mockResult: {
+                        id: '1',
+                        data: { access_token: 'encrypted:createsecret:iv:enckey' },
+                    },
+                };
+
+                await handlers.upsert({
+                    model: 'Credential',
+                    operation: 'upsert',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(mockCryptor.encrypt).toHaveBeenCalledWith('createsecret');
+                expect(mockCryptor.encrypt).toHaveBeenCalledWith('updatesecret');
+            });
+        });
+
+        describe('findUnique', () => {
+            it('should decrypt result', async () => {
+                const args = {
+                    where: { id: '1' },
+                    mockResult: {
+                        id: '1',
+                        data: { access_token: 'encrypted:secret:iv:enckey' },
+                    },
+                };
+
+                const result = await handlers.findUnique({
+                    model: 'Credential',
+                    operation: 'findUnique',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(mockCryptor.decrypt).toHaveBeenCalledWith(
+                    'encrypted:secret:iv:enckey'
+                );
+                expect(result.data.access_token).toBe('secret');
+            });
+
+            it('should handle null result', async () => {
+                const args = {
+                    where: { id: '999' },
+                    mockResult: null,
+                };
+
+                const result = await handlers.findUnique({
+                    model: 'Credential',
+                    operation: 'findUnique',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(result).toBeNull();
+                expect(mockCryptor.decrypt).not.toHaveBeenCalled();
+            });
+        });
+
+        describe('findMany', () => {
+            it('should decrypt array of results', async () => {
+                const args = {
+                    mockResult: [
+                        {
+                            id: '1',
+                            data: { access_token: 'encrypted:secret1:iv:enckey' },
+                        },
+                        {
+                            id: '2',
+                            data: { access_token: 'encrypted:secret2:iv:enckey' },
+                        },
+                    ],
+                };
+
+                const results = await handlers.findMany({
+                    model: 'Credential',
+                    operation: 'findMany',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(results).toHaveLength(2);
+                expect(results[0].data.access_token).toBe('secret1');
+                expect(results[1].data.access_token).toBe('secret2');
+            });
+
+            it('should handle empty array', async () => {
+                const args = {
+                    mockResult: [],
+                };
+
+                const results = await handlers.findMany({
+                    model: 'Credential',
+                    operation: 'findMany',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(results).toEqual([]);
+            });
+        });
+
+        describe('delete', () => {
+            it('should decrypt deleted record', async () => {
+                const args = {
+                    where: { id: '1' },
+                    mockResult: {
+                        id: '1',
+                        data: { access_token: 'encrypted:secret:iv:enckey' },
+                    },
+                };
+
+                const result = await handlers.delete({
+                    model: 'Credential',
+                    operation: 'delete',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(result.data.access_token).toBe('secret');
+            });
+        });
+
+        describe('count', () => {
+            it('should pass through without encryption', async () => {
+                const args = {
+                    mockResult: { count: 5 },
+                };
+
+                const result = await handlers.count({
+                    model: 'Credential',
+                    operation: 'count',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(result).toEqual({ count: 5 });
+                expect(mockCryptor.encrypt).not.toHaveBeenCalled();
+                expect(mockCryptor.decrypt).not.toHaveBeenCalled();
+            });
+        });
+
+        describe('Model without encrypted fields', () => {
+            it('should pass through State model without encryption', async () => {
+                const args = {
+                    data: { state: { some: 'data' } },
+                    mockResult: { id: '1', state: { some: 'data' } },
+                };
+
+                const result = await handlers.create({
+                    model: 'State',
+                    operation: 'create',
+                    args,
+                    query: mockQuery,
+                });
+
+                expect(result.state).toEqual({ some: 'data' });
+                expect(mockCryptor.encrypt).not.toHaveBeenCalled();
+                expect(mockCryptor.decrypt).not.toHaveBeenCalled();
+            });
+        });
+    });
+
+    describe('Integration with FieldEncryptionService', () => {
+        it('should handle nested JSON paths correctly', async () => {
+            const extension = createEncryptionExtension({
+                cryptor: mockCryptor,
+                enabled: true,
+            });
+            const handlers = extension.query.$allModels;
+
+            const args = {
+                data: {
+                    id: '123',
+                    data: {
+                        access_token: 'secret',
+                        refresh_token: 'refresh',
+                        other: 'public',
+                    },
+                },
+                mockResult: {
+                    id: '123',
+                    data: {
+                        access_token: 'encrypted:secret:iv:enckey',
+                        refresh_token: 'encrypted:refresh:iv:enckey',
+                        other: 'public',
+                    },
+                },
+            };
+
+            const result = await handlers.create({
+                model: 'Credential',
+                operation: 'create',
+                args,
+                query: mockQuery,
+            });
+
+            // Verify encryption was called for encrypted fields
+            expect(mockCryptor.encrypt).toHaveBeenCalledWith('secret');
+            expect(mockCryptor.encrypt).toHaveBeenCalledWith('refresh');
+            // 'other' should not be encrypted
+
+            // Verify decryption in result
+            expect(result.data.access_token).toBe('secret');
+            expect(result.data.refresh_token).toBe('refresh');
+            expect(result.data.other).toBe('public');
+        });
+    });
+});

package/database/index.js

@@ -1,25 +1,38 @@
+//todo: probably most of this file content can be removed
+
+/**
+ * Database Module Index
+ * Exports Mongoose models and connection utilities
+ *
+ * Note: Frigg uses the Repository pattern for data access.
+ * Models are not meant to be used directly - use repositories instead:
+ * - SyncRepository (syncs/sync-repository.js)
+ * - IntegrationRepository (integrations/integration-repository.js)
+ * - CredentialRepository (credential/credential-repository.js)
+ * etc.
+ */
+
 const { mongoose } = require('./mongoose');
-const {
-    connectToDatabase,
-    disconnectFromDatabase,
-    createObjectId,
-} = require('./mongo');
 const { IndividualUser } = require('./models/IndividualUser');
 const { OrganizationUser } = require('./models/OrganizationUser');
-const { State } = require('./models/State');
-const { Token } = require('./models/Token');
 const { UserModel } = require('./models/UserModel');
 const { WebsocketConnection } = require('./models/WebsocketConnection');
 
+// Prisma exports
+const { prisma } = require('./prisma');
+const { TokenRepository } = require('../token/repositories/token-repository');
+const {
+    WebsocketConnectionRepository,
+} = require('../websocket/repositories/websocket-connection-repository');
+
 module.exports = {
     mongoose,
-    connectToDatabase,
-    disconnectFromDatabase,
-    createObjectId,
     IndividualUser,
     OrganizationUser,
-    State,
-    Token,
     UserModel,
     WebsocketConnection,
+    // Prisma
+    prisma,
+    TokenRepository,
+    WebsocketConnectionRepository,
 };