@friggframework/core 2.0.0--canary.482.bda352c.0 → 2.0.0--canary.487.5a8e56e.0

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

@@ -1,27 +1,22 @@
 const { HealthCheckRepositoryMongoDB } = require('./health-check-repository-mongodb');
 const { HealthCheckRepositoryPostgreSQL } = require('./health-check-repository-postgres');
+const { prisma } = require('../prisma');
 const config = require('../config');
 
 /**
- * Health Check Repository Factory
- * Creates the appropriate repository adapter based on database type
- *
- * Usage:
- * ```javascript
- * const repository = createHealthCheckRepository();
- * ```
- *
- * @returns {HealthCheckRepositoryInterface} Configured repository adapter
+ * @param {Object} [options]
+ * @param {Object} [options.prismaClient] - Prisma client (defaults to singleton)
+ * @returns {HealthCheckRepositoryInterface}
  */
-function createHealthCheckRepository() {
+function createHealthCheckRepository({ prismaClient = prisma } = {}) {
     const dbType = config.DB_TYPE;
 
     switch (dbType) {
         case 'mongodb':
-            return new HealthCheckRepositoryMongoDB();
+            return new HealthCheckRepositoryMongoDB({ prismaClient });
 
         case 'postgresql':
-            return new HealthCheckRepositoryPostgreSQL();
+            return new HealthCheckRepositoryPostgreSQL({ prismaClient });
 
         default:
             throw new Error(
@@ -32,7 +27,6 @@ function createHealthCheckRepository() {
 
 module.exports = {
     createHealthCheckRepository,
-    // Export adapters for direct testing
     HealthCheckRepositoryMongoDB,
     HealthCheckRepositoryPostgreSQL,
 };

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

@@ -76,9 +76,10 @@ class HealthCheckRepositoryInterface {
     /**
      * Get database connection state
      *
-     * @returns {Object} Connection state info
+     * @returns {Promise<Object>} Connection state info
+     * @abstract
      */
-    getDatabaseConnectionState() {
+    async getDatabaseConnectionState() {
         throw new Error('Method getDatabaseConnectionState must be implemented by subclass');
     }
 }

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

@@ -1,62 +1,67 @@
-const { prisma } = require('../prisma');
 const { mongoose } = require('../mongoose');
 const {
     HealthCheckRepositoryInterface,
 } = require('./health-check-repository-interface');
 
-/**
- * MongoDB-specific Health Check Repository
- *
- * Provides MongoDB-specific database operations for health testing.
- * Uses Mongoose for MongoDB-specific operations (raw access, ping).
- */
 class HealthCheckRepositoryMongoDB extends HealthCheckRepositoryInterface {
-    constructor() {
+    /**
+     * @param {Object} params
+     * @param {Object} params.prismaClient - Prisma client instance
+     */
+    constructor({ prismaClient }) {
         super();
+        this.prisma = prismaClient;
     }
 
-    getDatabaseConnectionState() {
-        const stateMap = {
-            0: 'disconnected',
-            1: 'connected',
-            2: 'connecting',
-            3: 'disconnecting',
-        };
-        const readyState = mongoose.connection.readyState;
+    /**
+     * @returns {Promise<{readyState: number, stateName: string, isConnected: boolean}>}
+     */
+    async getDatabaseConnectionState() {
+        let isConnected = false;
+        let stateName = 'unknown';
+        
+        try {
+            await this.prisma.$runCommandRaw({ ping: 1 });
+            isConnected = true;
+            stateName = 'connected';
+        } catch (error) {
+            stateName = 'disconnected';
+        }
 
         return {
-            readyState,
-            stateName: stateMap[readyState],
-            isConnected: readyState === 1,
+            readyState: isConnected ? 1 : 0,
+            stateName,
+            isConnected,
         };
     }
 
+    /**
+     * @param {number} maxTimeMS
+     * @returns {Promise<number>} Response time in milliseconds
+     */
     async pingDatabase(maxTimeMS = 2000) {
         const pingStart = Date.now();
-        await mongoose.connection.db.admin().ping({ maxTimeMS });
+        await this.prisma.$queryRaw`SELECT 1`.catch(() => {
+            return this.prisma.$runCommandRaw({ ping: 1 });
+        });
         return Date.now() - pingStart;
     }
 
     async createCredential(credentialData) {
-        // Note: Collection existence is ensured at application startup via
-        // initializeMongoDBSchema() in database/utils/mongodb-schema-init.js
-        // This prevents "Cannot create namespace in multi-document transaction" errors
-        return await prisma.credential.create({
+        return await this.prisma.credential.create({
             data: credentialData,
         });
     }
 
     async findCredentialById(id) {
-        return await prisma.credential.findUnique({
+        return await this.prisma.credential.findUnique({
             where: { id },
         });
     }
 
     /**
-     * Get raw credential from MongoDB bypassing Prisma encryption extension
-     * Uses Mongoose to access raw MongoDB collection
-     * @param {string} id - Credential ID
-     * @returns {Promise<Object|null>} Raw credential from database
+     * @param {string} id
+     * @returns {Promise<Object|null>}
      */
     async getRawCredentialById(id) {
         const { ObjectId } = require('mongodb');
@@ -66,7 +71,7 @@ class HealthCheckRepositoryMongoDB extends HealthCheckRepositoryInterface {
     }
 
     async deleteCredential(id) {
-        await prisma.credential.delete({
+        await this.prisma.credential.delete({
             where: { id },
         });
     }

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

@@ -1,59 +1,67 @@
-const { prisma } = require('../prisma');
 const {
     HealthCheckRepositoryInterface,
 } = require('./health-check-repository-interface');
 
-/**
- * PostgreSQL-specific Health Check Repository
- *
- * Provides PostgreSQL-specific database operations for health testing.
- * Uses Prisma raw queries for PostgreSQL-specific operations.
- */
 class HealthCheckRepositoryPostgreSQL extends HealthCheckRepositoryInterface {
-    constructor() {
+    /**
+     * @param {Object} params
+     * @param {Object} params.prismaClient - Prisma client instance
+     */
+    constructor({ prismaClient }) {
         super();
+        this.prisma = prismaClient;
     }
 
-    getDatabaseConnectionState() {
-        // PostgreSQL connection state via Prisma
-        // Note: Prisma doesn't expose connection state like Mongoose
-        // We check if prisma is connected by attempting a query
+    /**
+     * @returns {Promise<{readyState: number, stateName: string, isConnected: boolean}>}
+     */
+    async getDatabaseConnectionState() {
+        let isConnected = false;
+        let stateName = 'unknown';
+        
+        try {
+            await this.prisma.$queryRaw`SELECT 1`;
+            isConnected = true;
+            stateName = 'connected';
+        } catch (error) {
+            stateName = 'disconnected';
+        }
+
         return {
-            readyState: 1, // Assume connected if Prisma instance exists
-            stateName: 'connected',
-            isConnected: true,
+            readyState: isConnected ? 1 : 0,
+            stateName,
+            isConnected,
         };
     }
 
+    /**
+     * @param {number} maxTimeMS
+     * @returns {Promise<number>} Response time in milliseconds
+     */
     async pingDatabase(maxTimeMS = 2000) {
         const pingStart = Date.now();
-
-        // PostgreSQL ping using SELECT 1
-        await prisma.$queryRaw`SELECT 1`;
-
+        await this.prisma.$queryRaw`SELECT 1`;
         return Date.now() - pingStart;
     }
 
     async createCredential(credentialData) {
-        return await prisma.credential.create({
+        return await this.prisma.credential.create({
             data: credentialData,
         });
     }
 
     async findCredentialById(id) {
-        return await prisma.credential.findUnique({
+        return await this.prisma.credential.findUnique({
             where: { id },
         });
     }
 
     /**
-     * Get raw credential from PostgreSQL bypassing Prisma encryption extension
-     * Uses $queryRaw to access raw PostgreSQL table
-     * @param {string} id - Credential ID
-     * @returns {Promise<Object|null>} Raw credential from database
+     * @param {string} id
+     * @returns {Promise<Object|null>}
      */
     async getRawCredentialById(id) {
-        const results = await prisma.$queryRaw`
+        const results = await this.prisma.$queryRaw`
             SELECT * FROM "Credential" WHERE id = ${id}
         `;
 
@@ -61,12 +69,11 @@ class HealthCheckRepositoryPostgreSQL extends HealthCheckRepositoryInterface {
             return null;
         }
 
-        // Return first result
         return results[0];
     }
 
     async deleteCredential(id) {
-        await prisma.credential.delete({
+        await this.prisma.credential.delete({
             where: { id },
         });
     }

package/database/use-cases/check-database-health-use-case.js

@@ -1,22 +1,17 @@
-/**
- * Use Case for checking database health.
- * Contains business logic for determining database connectivity and health status.
- */
 class CheckDatabaseHealthUseCase {
     /**
      * @param {Object} params
-     * @param {import('../health-check-repository-interface').HealthCheckRepositoryInterface} params.healthCheckRepository
+     * @param {import('../repositories/health-check-repository-interface').HealthCheckRepositoryInterface} params.healthCheckRepository
      */
     constructor({ healthCheckRepository }) {
         this.repository = healthCheckRepository;
     }
 
     /**
-     * Execute database health check
-     * @returns {Promise<Object>} Health check result with status, state, and response time
+     * @returns {Promise<{status: string, state: string, responseTime?: number}>}
      */
     async execute() {
-        const { stateName, isConnected } = this.repository.getDatabaseConnectionState();
+        const { stateName, isConnected } = await this.repository.getDatabaseConnectionState();
 
         const result = {
             status: isConnected ? 'healthy' : 'unhealthy',

package/package.json

@@ -1,7 +1,7 @@
 {
     "name": "@friggframework/core",
     "prettier": "@friggframework/prettier-config",
-    "version": "2.0.0--canary.482.bda352c.0",
+    "version": "2.0.0--canary.487.5a8e56e.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.482.bda352c.0",
-        "@friggframework/prettier-config": "2.0.0--canary.482.bda352c.0",
-        "@friggframework/test": "2.0.0--canary.482.bda352c.0",
+        "@friggframework/eslint-config": "2.0.0--canary.487.5a8e56e.0",
+        "@friggframework/prettier-config": "2.0.0--canary.487.5a8e56e.0",
+        "@friggframework/test": "2.0.0--canary.487.5a8e56e.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": "bda352ca5630e5880fc4d75a1e84bb1157a6caf3"
+    "gitHead": "5a8e56e0a0507ac7b3ee9f9d2b88448569b44cd4"
 }