@peopl-health/nexus 3.2.12 → 3.3.0

package/lib/config/mongoConfig.js

@@ -0,0 +1,61 @@
+const mongoose = require('mongoose');
+const { logger } = require('../utils/logger');
+
+let connection = null;
+const databases = new Map();
+const modelDatabases = {};
+
+function setModelDatabase(modelName, dbName) {
+  modelDatabases[modelName] = dbName;
+}
+
+function setModelDatabases(mapping) {
+  Object.assign(modelDatabases, mapping);
+  logger.info('[mongoConfig] Model databases configured', { mapping });
+}
+
+function getModelDatabase(modelName) {
+  return modelDatabases[modelName] || null;
+}
+
+async function connect(uri = process.env.MONGODB_URI) {
+  if (connection || mongoose.connection.readyState === 1) {
+    connection = mongoose.connection;
+    return connection;
+  }
+  if (!uri) throw new Error('MongoDB URI required. Set MONGODB_URI or pass uri parameter.');
+  await mongoose.connect(uri);
+  connection = mongoose.connection;
+  logger.info('MongoDB connected');
+  return connection;
+}
+
+function getDb(dbName) {
+  if (!connection && mongoose.connection.readyState !== 1) throw new Error('MongoDB not connected');
+  connection = connection || mongoose.connection;
+  if (!dbName) return connection;
+  if (!databases.has(dbName)) {
+    databases.set(dbName, connection.useDb(dbName, { useCache: true }));
+  }
+  return databases.get(dbName);
+}
+
+function getConnection() {
+  return connection || mongoose.connection;
+}
+
+async function disconnect() {
+  if (mongoose.connection.readyState !== 0) await mongoose.disconnect();
+  databases.clear();
+  connection = null;
+}
+
+module.exports = { 
+  connect, 
+  disconnect, 
+  getDb, 
+  getConnection, 
+  setModelDatabase, 
+  setModelDatabases, 
+  getModelDatabase 
+};

package/lib/core/NexusMessaging.js

@@ -8,7 +8,7 @@ const { hasPreprocessingHandler, invokePreprocessingHandler } = require('../serv
 const { ensureThreadExists } = require('../helpers/threadHelper');
 const { logger } = require('../utils/logger');
 
-const mongoose = require('mongoose');
+const { connect } = require('../config/mongoConfig');
 const OpenAI = require('openai');
 const EventEmitter = require('events');
 
@@ -60,13 +60,8 @@ class NexusMessaging {
    * @param {string} mongoUri - MongoDB connection string
    */
   async initializeMongoDB(mongoUri = process.env.MONGODB_URI) {
-    if (!mongoUri) {
-      throw new Error('MongoDB URI not provided. Please set MONGODB_URI environment variable or pass mongoUri parameter.');
-    }
-
     try {
-      await mongoose.connect(mongoUri);
-      this.mongodb = mongoose.connection;
+      this.mongodb = await connect(mongoUri);
       logger.info('MongoDB connected successfully');
     } catch (error) {
       logger.error('MongoDB connection failed', { error: error.message });

package/lib/index.js

@@ -28,6 +28,7 @@ const {
   getRequestId
 } = require('./middleware/requestId');
 const { logger } = require('./utils/logger');
+const { setModelDatabases, setModelDatabase, getModelDatabase } = require('./config/mongoConfig');
 
 /**
  * Main Nexus class that orchestrates all components
@@ -347,5 +348,9 @@ module.exports = {
   // Request tracing
   requestIdMiddleware,
   getRequestId,
-  logger
+  logger,
+  // MongoDB config
+  setModelDatabases,
+  setModelDatabase,
+  getModelDatabase
 };

package/lib/models/interactionModel.js

@@ -1,5 +1,7 @@
 const mongoose = require('mongoose');
 const { INTERACTION_QUALITY_VALUES } = require('../config/interactionConfig');
+const { getDb, getModelDatabase } = require('../config/mongoConfig');
+const { logger } = require('../utils/logger');
 
 const interactionSchema = new mongoose.Schema({
   messages: [{ type: mongoose.Schema.Types.ObjectId, ref: 'Message', required: true }],
@@ -13,6 +15,11 @@ const interactionSchema = new mongoose.Schema({
   description: { type: String, default: null }
 }, { timestamps: true });
 
-const Interaction = mongoose.model('Interaction', interactionSchema);
+const getInteraction = () => {
+  const dbName = getModelDatabase('Interaction');
+  const db = dbName ? getDb(dbName) : mongoose;
+  logger.debug('[Interaction] Using database', { dbName: dbName || 'default' });
+  return db.models.Interaction || db.model('Interaction', interactionSchema);
+};
 
-module.exports = { Interaction, interactionSchema };
+module.exports = { getInteraction, interactionSchema, get Interaction() { return getInteraction(); } };

package/lib/models/predictionMetricsModel.js

@@ -1,4 +1,6 @@
 const mongoose = require('mongoose');
+const { getDb, getModelDatabase } = require('../config/mongoConfig');
+const { logger } = require('../utils/logger');
 
 const predictionMetricsSchema = new mongoose.Schema({
   message_id: { type: String, required: true, index: true },
@@ -16,6 +18,11 @@ predictionMetricsSchema.index({ createdAt: -1 });
 predictionMetricsSchema.index({ assistant_id: 1, createdAt: -1 });
 predictionMetricsSchema.index({ numero: 1, createdAt: -1 });
 
-const PredictionMetrics = mongoose.model('PredictionMetrics', predictionMetricsSchema);
+const getPredictionMetrics = () => {
+  const dbName = getModelDatabase('PredictionMetrics');
+  const db = dbName ? getDb(dbName) : mongoose;
+  logger.debug('[PredictionMetrics] Using database', { dbName: dbName || 'default' });
+  return db.models.PredictionMetrics || db.model('PredictionMetrics', predictionMetricsSchema);
+};
 
-module.exports = { PredictionMetrics, predictionMetricsSchema };
+module.exports = { getPredictionMetrics, predictionMetricsSchema, get PredictionMetrics() { return getPredictionMetrics(); } };

package/lib/models/qualityMessageModel.js

@@ -1,4 +1,6 @@
 const mongoose = require('mongoose');
+const { getDb, getModelDatabase } = require('../config/mongoConfig');
+const { logger } = require('../utils/logger');
 
 const qualityMessageSchema = new mongoose.Schema({
   message_id: { type: mongoose.Schema.Types.ObjectId, ref: 'Message', required: true, index: true },
@@ -14,6 +16,11 @@ const qualityMessageSchema = new mongoose.Schema({
 
 qualityMessageSchema.index({ message_id: 1, voter_username: 1 }, { unique: true });
 
-const QualityMessage = mongoose.model('QualityMessage', qualityMessageSchema);
+const getQualityMessage = () => {
+  const dbName = getModelDatabase('QualityMessage');
+  const db = dbName ? getDb(dbName) : mongoose;
+  logger.debug('[QualityMessage] Using database', { dbName: dbName || 'default' });
+  return db.models.QualityMessage || db.model('QualityMessage', qualityMessageSchema);
+};
 
-module.exports = { QualityMessage, qualityMessageSchema };
+module.exports = { getQualityMessage, qualityMessageSchema, get QualityMessage() { return getQualityMessage(); } };

package/lib/storage/MongoStorage.js

@@ -1,4 +1,4 @@
-const mongoose = require('mongoose');
+const { connect: mongoConnect, disconnect: mongoDisconnect, getDb } = require('../config/mongoConfig');
 const runtimeConfig = require('../config/runtimeConfig');
 
 const { Message, insertMessage } = require('../models/messageModel');
@@ -33,15 +33,8 @@ class MongoStorage {
   }
 
   async connect() {
-    try {
-      await mongoose.connect(this.mongoUri, {
-        useNewUrlParser: true,
-        useUnifiedTopology: true
-      });
-      logger.info('MongoDB connected successfully');
-    } catch (error) {
-      throw new Error(`MongoDB connection failed: ${error.message}`);
-    }
+    await mongoConnect(this.mongoUri);
+    if (this.dbName) this.db = getDb(this.dbName);
   }
 
   async saveMessage(messageData) {
@@ -222,10 +215,7 @@ class MongoStorage {
   }
 
   async disconnect() {
-    await mongoose.disconnect();
-    if (this.client) {
-      await this.client.close();
-    }
+    await mongoDisconnect();
   }
 }
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@peopl-health/nexus",
-  "version": "3.2.12",
+  "version": "3.3.0",
   "description": "Core messaging and assistant library for WhatsApp communication platforms",
   "keywords": [
     "whatsapp",