@friggframework/core 2.0.0--canary.490.7426521.0 → 2.0.0--canary.490.9a325b5.0

package/credential/repositories/credential-repository-documentdb.js

@@ -1,29 +1,15 @@
 const { ObjectId } = require('mongodb');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { CredentialRepositoryInterface } = require('./credential-repository-interface');
 
-/**
- * DocumentDB Credential Repository Adapter
- * Uses native MongoDB driver to avoid Prisma's $$REMOVE operator
- */
 class CredentialRepositoryDocumentDB extends CredentialRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('Credential');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'Credential');
+        this._base = new BaseRepositoryDocumentDB('Credential', 'Credential');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async findCredentialById(id) {

package/database/connect-database.js

@@ -4,24 +4,23 @@
  */
 
 const { isDocumentDB } = require('./utils/documentdb-compatibility');
-const { logger } = require('../logs');
 
 async function connectDatabase() {
     if (isDocumentDB()) {
-        logger.info('Connecting to DocumentDB using native MongoDB driver...');
+        console.log('Connecting to DocumentDB using native MongoDB driver...');
         const { getNativeMongoClient } = require('./mongodb-native-client');
         const nativeClient = getNativeMongoClient();
         await nativeClient.connect();
-        logger.info('✓ Native MongoDB client connected');
+        console.log('✓ Native MongoDB client connected');
         
         const { initializeMongoDBSchema } = require('./utils/mongodb-schema-init');
         await initializeMongoDBSchema();
-        logger.info('✓ MongoDB schema initialized');
+        console.log('✓ MongoDB schema initialized');
     } else {
-        logger.info('Connecting using Prisma...');
+        console.log('Connecting using Prisma...');
         const { connectPrisma } = require('./prisma');
         await connectPrisma();
-        logger.info('✓ Prisma client connected');
+        console.log('✓ Prisma client connected');
     }
 }
 

package/database/repositories/base-repository-documentdb.js

@@ -0,0 +1,50 @@
+/**
+ * Base DocumentDB Repository
+ * Provides lazy-loaded encrypted collections for DocumentDB adapters
+ * 
+ * Lazy loading pattern ensures:
+ * - Repositories can be instantiated at module load time
+ * - Collections are only accessed after native client connects
+ * - Compatible with eager loading pattern in routers
+ */
+
+const { getNativeMongoClient } = require('../mongodb-native-client');
+const { EncryptedCollection } = require('../encrypted-collection-wrapper');
+const { FieldEncryptionService } = require('../encryption/field-encryption-service');
+const { getEncryptedFields } = require('../encryption/encryption-schema-registry');
+const { Cryptor } = require('../../encrypt/Cryptor');
+
+class BaseRepositoryDocumentDB {
+    constructor(collectionName, modelName) {
+        this.collectionName = collectionName;
+        this.modelName = modelName;
+        this.nativeClient = getNativeMongoClient();
+        this._collection = null;
+    }
+
+    _getCollection() {
+        if (!this._collection) {
+            const collection = this.nativeClient.collection(this.collectionName);
+            
+            const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
+            const encryptionService = new FieldEncryptionService({
+                cryptor,
+                schema: { getEncryptedFields },
+            });
+            
+            this._collection = new EncryptedCollection(
+                collection,
+                encryptionService,
+                this.modelName
+            );
+        }
+        return this._collection;
+    }
+
+    get collection() {
+        return this._getCollection();
+    }
+}
+
+module.exports = { BaseRepositoryDocumentDB };
+

package/database/repositories/health-check-repository-documentdb.js

@@ -1,27 +1,17 @@
 const { ObjectId } = require('mongodb');
 const { getNativeMongoClient } = require('../mongodb-native-client');
-const { EncryptedCollection } = require('../encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../encryption/field-encryption-service');
-const { getEncryptedFields } = require('../encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('./base-repository-documentdb');
 const { HealthCheckRepositoryInterface } = require('./health-check-repository-interface');
 
 class HealthCheckRepositoryDocumentDB extends HealthCheckRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        this.nativeClient = nativeClient;
-        
-        const collection = nativeClient.collection('Credential');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'Credential');
+        this._base = new BaseRepositoryDocumentDB('Credential', 'Credential');
+        this.nativeClient = getNativeMongoClient();
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async getDatabaseConnectionState() {

package/handlers/routers/health.js

@@ -32,7 +32,14 @@ const {
 } = require('../use-cases/check-integrations-health-use-case');
 
 const router = Router();
-const healthCheckRepository = createHealthCheckRepository({ prismaClient: prisma });
+
+let healthCheckRepository;
+function getHealthCheckRepository() {
+    if (!healthCheckRepository) {
+        healthCheckRepository = createHealthCheckRepository({ prismaClient: prisma });
+    }
+    return healthCheckRepository;
+}
 
 // Load integrations and create factories just like auth router does
 // This verifies the system can properly load integrations
@@ -55,15 +62,23 @@ try {
     integrationClasses = [];
 }
 
-const testEncryptionUseCase = new TestEncryptionUseCase({
-    healthCheckRepository,
-});
-const checkDatabaseHealthUseCase = new CheckDatabaseHealthUseCase({
-    healthCheckRepository,
-});
-const checkEncryptionHealthUseCase = new CheckEncryptionHealthUseCase({
-    testEncryptionUseCase,
-});
+function getTestEncryptionUseCase() {
+    return new TestEncryptionUseCase({
+        healthCheckRepository: getHealthCheckRepository(),
+    });
+}
+
+function getCheckDatabaseHealthUseCase() {
+    return new CheckDatabaseHealthUseCase({
+        healthCheckRepository: getHealthCheckRepository(),
+    });
+}
+
+function getCheckEncryptionHealthUseCase() {
+    return new CheckEncryptionHealthUseCase({
+        testEncryptionUseCase: getTestEncryptionUseCase(),
+    });
+}
 const checkExternalApisHealthUseCase = new CheckExternalApisHealthUseCase();
 const checkIntegrationsHealthUseCase = new CheckIntegrationsHealthUseCase({
     moduleFactory,
@@ -418,7 +433,7 @@ router.get('/health/detailed', async (_req, res) => {
     }
 
     try {
-        response.checks.database = await checkDatabaseHealthUseCase.execute();
+        response.checks.database = await getCheckDatabaseHealthUseCase().execute();
         if (response.checks.database.status === 'unhealthy') {
             response.status = 'unhealthy';
         }
@@ -433,7 +448,7 @@ router.get('/health/detailed', async (_req, res) => {
     }
 
     try {
-        response.checks.encryption = await checkEncryptionHealthUseCase.execute();
+        response.checks.encryption = await getCheckEncryptionHealthUseCase().execute();
         if (response.checks.encryption.status === 'unhealthy') {
             response.status = 'unhealthy';
         }
@@ -496,7 +511,7 @@ router.get('/health/live', (_req, res) => {
 });
 
 router.get('/health/ready', async (_req, res) => {
-    const dbHealth = await checkDatabaseHealthUseCase.execute();
+    const dbHealth = await getCheckDatabaseHealthUseCase().execute();
     const isDbReady = dbHealth.status === 'healthy';
 
     const integrationsHealth = checkIntegrationsHealthUseCase.execute();

package/integrations/repositories/integration-mapping-repository-documentdb.js

@@ -1,25 +1,15 @@
 const { ObjectId } = require('mongodb');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { IntegrationMappingRepositoryInterface } = require('./integration-mapping-repository-interface');
 
 class IntegrationMappingRepositoryDocumentDB extends IntegrationMappingRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('IntegrationMapping');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'IntegrationMapping');
+        this._base = new BaseRepositoryDocumentDB('IntegrationMapping', 'IntegrationMapping');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async findMappingById(mappingId) {

package/integrations/repositories/integration-repository-documentdb.js

@@ -1,25 +1,15 @@
 const { ObjectId } = require('mongodb');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { IntegrationRepositoryInterface } = require('./integration-repository-interface');
 
 class IntegrationRepositoryDocumentDB extends IntegrationRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('Integration');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'Integration');
+        this._base = new BaseRepositoryDocumentDB('Integration', 'Integration');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async findIntegrationById(id) {

package/integrations/repositories/process-repository-documentdb.js

@@ -1,25 +1,15 @@
 const { ObjectId } = require('mongodb');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { ProcessRepositoryInterface } = require('./process-repository-interface');
 
 class ProcessRepositoryDocumentDB extends ProcessRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('Process');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'Process');
+        this._base = new BaseRepositoryDocumentDB('Process', 'Process');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async createProcess(processData) {

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

@@ -1,25 +1,15 @@
 const { ObjectId } = require('mongodb');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { ModuleRepositoryInterface } = require('./module-repository-interface');
 
 class ModuleRepositoryDocumentDB extends ModuleRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('Entity');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'Entity');
+        this._base = new BaseRepositoryDocumentDB('Entity', 'Entity');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async findEntityById(entityId) {

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@friggframework/core",
     "prettier": "@friggframework/prettier-config",
-    "version": "2.0.0--canary.490.7426521.0",
+    "version": "2.0.0--canary.490.9a325b5.0",
     "dependencies": {
         "@aws-sdk/client-apigatewaymanagementapi": "^3.588.0",
         "@aws-sdk/client-kms": "^3.588.0",
@@ -38,9 +38,9 @@
         }
     },
     "devDependencies": {
-        "@friggframework/eslint-config": "2.0.0--canary.490.7426521.0",
-        "@friggframework/prettier-config": "2.0.0--canary.490.7426521.0",
-        "@friggframework/test": "2.0.0--canary.490.7426521.0",
+        "@friggframework/eslint-config": "2.0.0--canary.490.9a325b5.0",
+        "@friggframework/prettier-config": "2.0.0--canary.490.9a325b5.0",
+        "@friggframework/test": "2.0.0--canary.490.9a325b5.0",
         "@prisma/client": "^6.17.0",
         "@types/lodash": "4.17.15",
         "@typescript-eslint/eslint-plugin": "^8.0.0",
@@ -80,5 +80,5 @@
     "publishConfig": {
         "access": "public"
     },
-    "gitHead": "742652114f13c2cc94d726ccdb65a5cc3609e943"
+    "gitHead": "9a325b566c69480371bc7cbf22e4a5c8cb173f13"
 }

package/syncs/repositories/sync-repository-documentdb.js

@@ -1,25 +1,15 @@
 const { ObjectId } = require('mongodb');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { SyncRepositoryInterface } = require('./sync-repository-interface');
 
 class SyncRepositoryDocumentDB extends SyncRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('Sync');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'Sync');
+        this._base = new BaseRepositoryDocumentDB('Sync', 'Sync');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async getSyncObject(name, dataIdentifier, entity) {

package/token/repositories/token-repository-documentdb.js

@@ -1,10 +1,6 @@
 const { ObjectId } = require('mongodb');
 const bcrypt = require('bcryptjs');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { TokenRepositoryInterface } = require('./token-repository-interface');
 
 const BCRYPT_ROUNDS = 10;
@@ -12,17 +8,11 @@ const BCRYPT_ROUNDS = 10;
 class TokenRepositoryDocumentDB extends TokenRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('Token');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'Token');
+        this._base = new BaseRepositoryDocumentDB('Token', 'Token');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async createTokenWithExpire(userId, rawToken, minutes) {

package/user/repositories/user-repository-documentdb.js

@@ -1,10 +1,6 @@
 const { ObjectId } = require('mongodb');
 const bcrypt = require('bcryptjs');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { UserRepositoryInterface } = require('./user-repository-interface');
 const { createTokenRepository } = require('../../token/repositories/token-repository-factory');
 
@@ -13,20 +9,14 @@ const BCRYPT_ROUNDS = 10;
 class UserRepositoryDocumentDB extends UserRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('User');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'User');
+        this._base = new BaseRepositoryDocumentDB('User', 'User');
         this.tokenRepository = createTokenRepository();
     }
 
+    get collection() {
+        return this._base.collection;
+    }
+
     async getSessionToken(token) {
         return await this.tokenRepository.getSessionToken(token);
     }

package/websocket/repositories/websocket-connection-repository-documentdb.js

@@ -1,25 +1,15 @@
 const { ObjectId } = require('mongodb');
-const { getNativeMongoClient } = require('../../database/mongodb-native-client');
-const { EncryptedCollection } = require('../../database/encrypted-collection-wrapper');
-const { FieldEncryptionService } = require('../../database/encryption/field-encryption-service');
-const { getEncryptedFields } = require('../../database/encryption/encryption-schema-registry');
-const { Cryptor } = require('../../encrypt/Cryptor');
+const { BaseRepositoryDocumentDB } = require('../../database/repositories/base-repository-documentdb');
 const { WebsocketConnectionRepositoryInterface } = require('./websocket-connection-repository-interface');
 
 class WebsocketConnectionRepositoryDocumentDB extends WebsocketConnectionRepositoryInterface {
     constructor() {
         super();
-        
-        const nativeClient = getNativeMongoClient();
-        const collection = nativeClient.collection('WebsocketConnection');
-        
-        const cryptor = new Cryptor({ shouldUseAws: !!process.env.KMS_KEY_ARN });
-        const encryptionService = new FieldEncryptionService({
-            cryptor,
-            schema: { getEncryptedFields },
-        });
-        
-        this.collection = new EncryptedCollection(collection, encryptionService, 'WebsocketConnection');
+        this._base = new BaseRepositoryDocumentDB('WebsocketConnection', 'WebsocketConnection');
+    }
+
+    get collection() {
+        return this._base.collection;
     }
 
     async createConnection(connectionData) {