payservedb 7.0.7 → 7.0.8

package/index.js

@@ -1,283 +1,283 @@
-const mongoose = require("mongoose");
-require("dotenv").config();
-
-// Maintain a record of open connections for each database
-const connections = {};
-
-// Utility function to connect to MongoDB
-async function connectToMongoDB(
-    dbName,
-    secured,
-    username,
-    password,
-    url,
-    port,
-) {
-    try {
-        if (secured === false) {
-            const url_ = url + ":" + port;
-            const connectionString = `mongodb://${url_}/${dbName}`;
-            await mongoose.connect(connectionString, {
-                useNewUrlParser: true,
-            });
-            console.log("Connected to MongoDB");
-        } else {
-            const url_ = `${username}:${password}@${url}:${port}`;
-            const source = "?authSource=admin";
-            const connectionString = `mongodb://${url_}/${dbName}${source}`;
-            await mongoose.connect(connectionString, {
-                useNewUrlParser: true,
-            });
-            console.log("Connected to MongoDB");
-        }
-    } catch (err) {
-        console.error("Error connecting to MongoDB:", err);
-        throw err;
-    }
-}
-
-// Function to switch databases dynamically
-async function switchDB(dbName) {
-    // Check if there's already a connection to the desired database
-    if (connections[dbName]) {
-        return connections[dbName]; // Return existing connection
-    }
-
-    try {
-        // Use environment variables for connection parameters - same as in connectToMongoDB
-        const username = process.env.MONGODB_USER;
-        const password = process.env.MONGODB_PASSWORD;
-        const url = process.env.MONGODB_HOST;
-        const port = process.env.MONGODB_PORT;
-        const source = "?authSource=admin";
-
-        const connectionString = `mongodb://${username}:${password}@${url}:${port}/${dbName}${source}`;
-        const dbConnection = await mongoose.createConnection(connectionString, {
-            useNewUrlParser: true,
-        });
-
-        connections[dbName] = dbConnection; // Store the connection for reuse
-        console.log(`Switched to database: ${dbName}`);
-        return dbConnection;
-    } catch (err) {
-        console.error(`Error switching to database: ${dbName}`, err);
-        throw err;
-    }
-}
-
-// Importing all model files (Note: models may need to be re-registered for different connections)
-// This maintains backward compatibility for existing services
-const models = {
-    User: require("./src/models/user"),
-    Company: require("./src/models/company"),
-    Customer: require("./src/models/customer"),
-    FacilityEmailDetails: require("./src/models/email"),
-    Module: require("./src/models/module"),
-    Facility: require("./src/models/facility"),
-    RefreshToken: require("./src/models/refresh_token"),
-    ApiLog: require("./src/models/apilog"),
-    ArchivedApiLog: require("./src/models/archivedapilog"),
-    FacilityAsset: require("./src/models/facilityasset"),
-    Unit: require("./src/models/units"),
-    FAQ: require("./src/models/faq"),
-    UnitAsset: require("./src/models/unitasset"),
-    CombinedUnit: require("./src/models/combinedUnits"),
-    Message: require("./src/models/message"),
-    SMSAfricastalking: require("./src/models/sms_africastalking"),
-    SMSMeliora: require("./src/models/sms_meliora"),
-    EntryExit: require("./src/models/entry_exit"),
-    Guard: require("./src/models/guard"),
-    Visitor: require("./src/models/visitor"),
-    VisitLog: require("./src/models/visitLog"),
-    Settings: require("./src/models/settings"),
-    Levy: require("./src/models/levy"),
-    LevyType: require("./src/models/levytype"),
-    LevyContract: require("./src/models/levycontract"),
-    Invoice: require("./src/models/invoice"),
-    Reminder: require("./src/models/reminder"),
-    Penalty: require("./src/models/penalty"),
-    Notifiaction: require("./src/models/notification"),
-    Resident: require("./src/models/resident"),
-    Asset: require("./src/models/asset"),
-    DutyRoster: require("./src/models/dutyroster"),
-    LeaseTemplate: require("./src/models/leasetemplate"),
-    Report: require("./src/models/report"),
-    Ticket: require("./src/models/tickets"),
-    Stocksandspare: require("./src/models/stocksandspare"),
-    ServiceVendor: require("./src/models/maintenance_service_vendor"),
-    MaintenanceServices: require("./src/models/maintenance_services"),
-    StockRequisition: require("./src/models/maintenancerequisition"),
-    LeaseAgreement: require("./src/models/leaseagreement"),
-    WorkOrder: require("./src/models/workorder"),
-    ValueAddedService: require("./src/models/valueaddedservices"),
-    VasInvoice: require("./src/models/vasinvoice"),
-    VasVendor: require("./src/models/vasvendor"),
-    Staff: require("./src/models/staff"),
-    ServiceRequest: require("./src/models/servicerequest"),
-    CountryTaxRate: require("./src/models/country_tax"),
-    WaterMeter: require("./src/models/water_meters"),
-    DailyConsumption: require("./src/models/smart_meter_daily_consumption"),
-    WaterMeterSettings: require("./src/models/water_meter_settings"),
-    AnalogBilling: require("./src/models/water_meter_billing"),
-    MeterSize: require("./src/models/water_meter_size"),
-    WaterInvoice: require("./src/models/water_invoice"),
-    MeterProtocol: require("./src/models/water_meter_communication"),
-    MeterManufacturer: require("./src/models/water_meter_manufacturer"),
-    MeterIotCard: require("./src/models/water_meter_iot_cards"),
-    MetersDelivery: require("./src/models/water_meters_delivery"),
-    Notification: require("./src/models/notification"),
-    Concentrator: require("./src/models/water_meter_concentrator"),
-    Handover: require("./src/models/handover"),
-    Budget: require("./src/models/budget"),
-    BudgetCategory: require("./src/models/budgetCategory"),
-    Expense: require("./src/models/expense"),
-    ExpenseCategory: require("./src/models/expense_category"),
-    InvoiceSettings: require("./src/models/levy_invoice_settings"),
-    Account: require("./src/models/account"),
-    FacilityPaymentDetails: require("./src/models/facility_payment_details"),
-    DefaultPaymentDetails: require("./src/models/default_payment_details"),
-    Currency: require("./src/models/currency_settings"),
-    WaterMeterAccount: require("./src/models/water_meter_account"),
-    SingleDayWaterMeterHistory: require("./src/models/water_meter_single_day_history"),
-    DailyWaterMeterHistory: require("./src/models/water_meter_daily_history"),
-    MonthlyWaterMeterHistory: require("./src/models/water_meter_monthly_history"),
-    WaterPrepaidCredit: require("./src/models/water_prepaid_credit"),
-    WaterPrepaidDebit: require("./src/models/water_prepaid_debit"),
-    MeterLog: require("./src/models/water_meter_communication_logs"),
-    CashPayment: require("./src/models/cashpayment"),
-    VasPayment: require("./src/models/vas_payments"),
-    VasInvoicesQuickBooks: require("./src/models/vas_invoices_upload"),
-    ServiceChargePayment: require("./src/models/service_charge_payments"),
-    ServiceChargeInvoiceUpload: require("./src/models/service_charge_invoice_upload"),
-    Campaign: require("./src/models/campaigns"),
-    InspectionItem: require("./src/models/item_inspection"),
-    Supplier: require("./src/models/suppliers"),
-    PurchaseRequest: require("./src/models/purchase_request"),
-    PurchaseOrder: require("./src/models/purchase_order"),
-    PaymentTermMark: require("./src/models/paymentTermsMarks"),
-    DeliveryTimeMark: require("./src/models/deliveryTimeMarks"),
-    RFQDetails: require("./src/models/rfq_details"),
-    RFQResponse: require("./src/models/rfq_response"),
-    ApprovalWorkflow: require("./src/models/approvalsWorkflows"),
-    CommonAreaElectricityReading: require("./src/models/common_area_electricity"),
-    CommonAreaWaterReading: require("./src/models/common_area_water"),
-    CommonAreaGeneratorReading: require("./src/models/common_area_generator"),
-    CommonAreaUtilityAlert: require("./src/models/common_area_utility_alert"),
-    BookingProperty: require("./src/models/bookingproperty"),
-    BookingReservation: require("./src/models/bookingreservation"),
-    BookingConfig: require("./src/models/bookingconfig"),
-    BookingAnalytics: require("./src/models/bookinganalytics"),
-    BookingInvoice: require("./src/models/booking_invoice"),
-    BankDetails: require("./src/models/bankdetails"),
-    RevenueRecord: require("./src/models/bookingrevenuerecord"),
-    GLAccount: require("./src/models/gl_accounts"),
-    GLEntry: require("./src/models/gl_entries"),
-    GLAccountDoubleEntries: require("./src/models/gl_account_double_entries"),
-    PendingCredential: require("./src/models/pendingCredentials"),
-    UnitManagementTemplate: require("./src/models/unitManagementTemplate"),
-    PropertyManagerRevenue: require("./src/models/propertyManagerRevenue"),
-    PropertyManagerContract: require("./src/models/propertyManagerContract"),
-    BillerAddress: require("./src/models/billerAddress"),
-    AssetAssignment: require("./src/models/assetsAssignment"),
-    Wallet: require("./src/models/wallet"),
-    WalletTransaction: require("./src/models/wallet_transactions"),
-    GoodsReceivedNote: require("./src/models/goodsReceivedNotes"),
-    CommunicationStatus: require("./src/models/communication_status"),
-    MeterCommandQueue: require("./src/models/water_meter_Command_Queue"),
-    FacilityDepartment: require("./src/models/facility_departements"),
-    EmailSmsQueue: require("./src/models/email_sms_queue"),
-    PurchaseOrderInvoice: require("./src/models/purchaseOrderInvoice"),
-    PowerMeterCustomerBand: require("./src/models/powerMeterCustomerBand"),
-    PowerMeterCommunicationProtocol: require("./src/models/powerMeterCommunicationProtocol"),
-    PowerMeterDailyReading: require("./src/models/powerMeterDailyReading"),
-    PowerMeterPowerCharge: require("./src/models/powerMeterPowerCharges"),
-    PowerMeterGateway: require("./src/models/powerMeterGateways"),
-    PowerMeters: require("./src/models/powerMeters"),
-    PowerMeterSingleDayReading: require("./src/models/powerMeterSingleDayReading"),
-    PowerMeterAccount: require("./src/models/power_meter_account"),
-    PowerMeterLog: require("./src/models/power_meter_command_logs"),
-    PowerMeterManufacturer: require("./src/models/powerMetersManufacturer"),
-    PowerMeterMonthlyReading: require("./src/models/powerMeterMonthlyReading"),
-    PowerMeterSettings: require("./src/models/powerMeterSettings"),
-    FacilityWalletTransactionsMetadata: require("./src/models/facilityWalletTransactionsMetadata"),
-    PowerMeterCustomerAccount: require("./src/models/powerMeterCustomerAccount"),
-    SmsNotification: require("./src/models/sms_balance_notification"),
-    NegativeBalance: require("./src/models/water_meter_negative_amounts"),
-    MasterWorkplan: require("./src/models/master_workplan"),
-    MasterWorkplanChild: require("./src/models/master_workplan_child"),
-    DutyRosterChecklist: require("./src/models/dutyRosterChecklist"),
-    AuditTrail: require("./src/models/auditTrail"),
-    DailyChecklist: require("./src/models/dailyChecklist"),
-    CoreInvoiceSettings: require("./src/models/coreInvoiceSettings"),
-    FacilityInvoice: require("./src/models/facilityInvoice"),
-    CoreBaseSettings: require("./src/models/coreBaseSettings"),
-    Recipient: require("./src/models/facilityInvoiceRecipient")
-};
-
-// Function to get models dynamically from a specific database connection
-async function getModelFromDB(dbConnection, modelName, schema) {
-    if (!dbConnection.models[modelName]) {
-        return dbConnection.model(modelName, schema); // Register the model in the db connection
-    }
-    return dbConnection.models[modelName]; // Return existing model if already registered
-}
-
-// Function to initialize service with specific models - NEW FEATURE
-function initializeService(modelNames = []) {
-    let currentConnection = null;
-
-    // Enhanced connect function that only registers specified models
-    async function connectWithModels(
-        dbName,
-        secured,
-        username,
-        password,
-        url,
-        port,
-    ) {
-        await connectToMongoDB(dbName, secured, username, password, url, port);
-        currentConnection = mongoose.connection;
-
-        // Only register the models specified in modelNames
-        modelNames.forEach((modelName) => {
-            if (models[modelName]) {
-                let schema;
-
-                // Extract schema from the imported model
-                if (models[modelName].schema) {
-                    schema = models[modelName].schema;
-                } else {
-                    console.warn(`Unable to extract schema from model ${modelName}`);
-                    return;
-                }
-
-                if (!currentConnection.models[modelName]) {
-                    const modelInstance = currentConnection.model(modelName, schema);
-                    // Make model globally accessible through payservedb
-                    module.exports[modelName] = modelInstance;
-                } else {
-                    module.exports[modelName] = currentConnection.models[modelName];
-                }
-            } else {
-                console.warn(`Model ${modelName} not found in models registry`);
-            }
-        });
-
-        return module.exports;
-    }
-
-    return {
-        connectToMongoDB: connectWithModels,
-        switchDB,
-        getModelFromDB,
-        getCurrentConnection: () => currentConnection,
-    };
-}
-
-module.exports = {
-    connectToMongoDB,
-    switchDB,
-    getModelFromDB,
-    initializeService,
-    ...models,
-};
+const mongoose = require("mongoose");
+require("dotenv").config();
+
+// Maintain a record of open connections for each database
+const connections = {};
+
+// Utility function to connect to MongoDB
+async function connectToMongoDB(
+    dbName,
+    secured,
+    username,
+    password,
+    url,
+    port,
+) {
+    try {
+        if (secured === false) {
+            const url_ = url + ":" + port;
+            const connectionString = `mongodb://${url_}/${dbName}`;
+            await mongoose.connect(connectionString, {
+                useNewUrlParser: true,
+            });
+            console.log("Connected to MongoDB");
+        } else {
+            const url_ = `${username}:${password}@${url}:${port}`;
+            const source = "?authSource=admin";
+            const connectionString = `mongodb://${url_}/${dbName}${source}`;
+            await mongoose.connect(connectionString, {
+                useNewUrlParser: true,
+            });
+            console.log("Connected to MongoDB");
+        }
+    } catch (err) {
+        console.error("Error connecting to MongoDB:", err);
+        throw err;
+    }
+}
+
+// Function to switch databases dynamically
+async function switchDB(dbName) {
+    // Check if there's already a connection to the desired database
+    if (connections[dbName]) {
+        return connections[dbName]; // Return existing connection
+    }
+
+    try {
+        // Use environment variables for connection parameters - same as in connectToMongoDB
+        const username = process.env.MONGODB_USER;
+        const password = process.env.MONGODB_PASSWORD;
+        const url = process.env.MONGODB_HOST;
+        const port = process.env.MONGODB_PORT;
+        const source = "?authSource=admin";
+
+        const connectionString = `mongodb://${username}:${password}@${url}:${port}/${dbName}${source}`;
+        const dbConnection = await mongoose.createConnection(connectionString, {
+            useNewUrlParser: true,
+        });
+
+        connections[dbName] = dbConnection; // Store the connection for reuse
+        console.log(`Switched to database: ${dbName}`);
+        return dbConnection;
+    } catch (err) {
+        console.error(`Error switching to database: ${dbName}`, err);
+        throw err;
+    }
+}
+
+// Importing all model files (Note: models may need to be re-registered for different connections)
+// This maintains backward compatibility for existing services
+const models = {
+    User: require("./src/models/user"),
+    Company: require("./src/models/company"),
+    Customer: require("./src/models/customer"),
+    FacilityEmailDetails: require("./src/models/email"),
+    Module: require("./src/models/module"),
+    Facility: require("./src/models/facility"),
+    RefreshToken: require("./src/models/refresh_token"),
+    ApiLog: require("./src/models/apilog"),
+    ArchivedApiLog: require("./src/models/archivedapilog"),
+    FacilityAsset: require("./src/models/facilityasset"),
+    Unit: require("./src/models/units"),
+    FAQ: require("./src/models/faq"),
+    UnitAsset: require("./src/models/unitasset"),
+    CombinedUnit: require("./src/models/combinedUnits"),
+    Message: require("./src/models/message"),
+    SMSAfricastalking: require("./src/models/sms_africastalking"),
+    SMSMeliora: require("./src/models/sms_meliora"),
+    EntryExit: require("./src/models/entry_exit"),
+    Guard: require("./src/models/guard"),
+    Visitor: require("./src/models/visitor"),
+    VisitLog: require("./src/models/visitLog"),
+    Settings: require("./src/models/settings"),
+    Levy: require("./src/models/levy"),
+    LevyType: require("./src/models/levytype"),
+    LevyContract: require("./src/models/levycontract"),
+    Invoice: require("./src/models/invoice"),
+    Reminder: require("./src/models/reminder"),
+    Penalty: require("./src/models/penalty"),
+    Notifiaction: require("./src/models/notification"),
+    Resident: require("./src/models/resident"),
+    Asset: require("./src/models/asset"),
+    DutyRoster: require("./src/models/dutyroster"),
+    LeaseTemplate: require("./src/models/leasetemplate"),
+    Report: require("./src/models/report"),
+    Ticket: require("./src/models/tickets"),
+    Stocksandspare: require("./src/models/stocksandspare"),
+    ServiceVendor: require("./src/models/maintenance_service_vendor"),
+    MaintenanceServices: require("./src/models/maintenance_services"),
+    StockRequisition: require("./src/models/maintenancerequisition"),
+    LeaseAgreement: require("./src/models/leaseagreement"),
+    WorkOrder: require("./src/models/workorder"),
+    ValueAddedService: require("./src/models/valueaddedservices"),
+    VasInvoice: require("./src/models/vasinvoice"),
+    VasVendor: require("./src/models/vasvendor"),
+    Staff: require("./src/models/staff"),
+    ServiceRequest: require("./src/models/servicerequest"),
+    CountryTaxRate: require("./src/models/country_tax"),
+    WaterMeter: require("./src/models/water_meters"),
+    DailyConsumption: require("./src/models/smart_meter_daily_consumption"),
+    WaterMeterSettings: require("./src/models/water_meter_settings"),
+    AnalogBilling: require("./src/models/water_meter_billing"),
+    MeterSize: require("./src/models/water_meter_size"),
+    WaterInvoice: require("./src/models/water_invoice"),
+    MeterProtocol: require("./src/models/water_meter_communication"),
+    MeterManufacturer: require("./src/models/water_meter_manufacturer"),
+    MeterIotCard: require("./src/models/water_meter_iot_cards"),
+    MetersDelivery: require("./src/models/water_meters_delivery"),
+    Notification: require("./src/models/notification"),
+    Concentrator: require("./src/models/water_meter_concentrator"),
+    Handover: require("./src/models/handover"),
+    Budget: require("./src/models/budget"),
+    BudgetCategory: require("./src/models/budgetCategory"),
+    Expense: require("./src/models/expense"),
+    ExpenseCategory: require("./src/models/expense_category"),
+    InvoiceSettings: require("./src/models/levy_invoice_settings"),
+    Account: require("./src/models/account"),
+    FacilityPaymentDetails: require("./src/models/facility_payment_details"),
+    DefaultPaymentDetails: require("./src/models/default_payment_details"),
+    Currency: require("./src/models/currency_settings"),
+    WaterMeterAccount: require("./src/models/water_meter_account"),
+    SingleDayWaterMeterHistory: require("./src/models/water_meter_single_day_history"),
+    DailyWaterMeterHistory: require("./src/models/water_meter_daily_history"),
+    MonthlyWaterMeterHistory: require("./src/models/water_meter_monthly_history"),
+    WaterPrepaidCredit: require("./src/models/water_prepaid_credit"),
+    WaterPrepaidDebit: require("./src/models/water_prepaid_debit"),
+    MeterLog: require("./src/models/water_meter_communication_logs"),
+    CashPayment: require("./src/models/cashpayment"),
+    VasPayment: require("./src/models/vas_payments"),
+    VasInvoicesQuickBooks: require("./src/models/vas_invoices_upload"),
+    ServiceChargePayment: require("./src/models/service_charge_payments"),
+    ServiceChargeInvoiceUpload: require("./src/models/service_charge_invoice_upload"),
+    Campaign: require("./src/models/campaigns"),
+    InspectionItem: require("./src/models/item_inspection"),
+    Supplier: require("./src/models/suppliers"),
+    PurchaseRequest: require("./src/models/purchase_request"),
+    PurchaseOrder: require("./src/models/purchase_order"),
+    PaymentTermMark: require("./src/models/paymentTermsMarks"),
+    DeliveryTimeMark: require("./src/models/deliveryTimeMarks"),
+    RFQDetails: require("./src/models/rfq_details"),
+    RFQResponse: require("./src/models/rfq_response"),
+    ApprovalWorkflow: require("./src/models/approvalsWorkflows"),
+    CommonAreaElectricityReading: require("./src/models/common_area_electricity"),
+    CommonAreaWaterReading: require("./src/models/common_area_water"),
+    CommonAreaGeneratorReading: require("./src/models/common_area_generator"),
+    CommonAreaUtilityAlert: require("./src/models/common_area_utility_alert"),
+    BookingProperty: require("./src/models/bookingproperty"),
+    BookingReservation: require("./src/models/bookingreservation"),
+    BookingConfig: require("./src/models/bookingconfig"),
+    BookingAnalytics: require("./src/models/bookinganalytics"),
+    BookingInvoice: require("./src/models/booking_invoice"),
+    BankDetails: require("./src/models/bankdetails"),
+    RevenueRecord: require("./src/models/bookingrevenuerecord"),
+    GLAccount: require("./src/models/gl_accounts"),
+    GLEntry: require("./src/models/gl_entries"),
+    GLAccountDoubleEntries: require("./src/models/gl_account_double_entries"),
+    PendingCredential: require("./src/models/pendingCredentials"),
+    UnitManagementTemplate: require("./src/models/unitManagementTemplate"),
+    PropertyManagerRevenue: require("./src/models/propertyManagerRevenue"),
+    PropertyManagerContract: require("./src/models/propertyManagerContract"),
+    BillerAddress: require("./src/models/billerAddress"),
+    AssetAssignment: require("./src/models/assetsAssignment"),
+    Wallet: require("./src/models/wallet"),
+    WalletTransaction: require("./src/models/wallet_transactions"),
+    GoodsReceivedNote: require("./src/models/goodsReceivedNotes"),
+    CommunicationStatus: require("./src/models/communication_status"),
+    MeterCommandQueue: require("./src/models/water_meter_Command_Queue"),
+    FacilityDepartment: require("./src/models/facility_departements"),
+    EmailSmsQueue: require("./src/models/email_sms_queue"),
+    PurchaseOrderInvoice: require("./src/models/purchaseOrderInvoice"),
+    PowerMeterCustomerBand: require("./src/models/powerMeterCustomerBand"),
+    PowerMeterCommunicationProtocol: require("./src/models/powerMeterCommunicationProtocol"),
+    PowerMeterDailyReading: require("./src/models/powerMeterDailyReading"),
+    PowerMeterPowerCharge: require("./src/models/powerMeterPowerCharges"),
+    PowerMeterGateway: require("./src/models/powerMeterGateways"),
+    PowerMeters: require("./src/models/powerMeters"),
+    PowerMeterSingleDayReading: require("./src/models/powerMeterSingleDayReading"),
+    PowerMeterAccount: require("./src/models/power_meter_account"),
+    PowerMeterLog: require("./src/models/power_meter_command_logs"),
+    PowerMeterManufacturer: require("./src/models/powerMetersManufacturer"),
+    PowerMeterMonthlyReading: require("./src/models/powerMeterMonthlyReading"),
+    PowerMeterSettings: require("./src/models/powerMeterSettings"),
+    FacilityWalletTransactionsMetadata: require("./src/models/facilityWalletTransactionsMetadata"),
+    PowerMeterCustomerAccount: require("./src/models/powerMeterCustomerAccount"),
+    SmsNotification: require("./src/models/sms_balance_notification"),
+    NegativeBalance: require("./src/models/water_meter_negative_amounts"),
+    MasterWorkplan: require("./src/models/master_workplan"),
+    MasterWorkplanChild: require("./src/models/master_workplan_child"),
+    DutyRosterChecklist: require("./src/models/dutyRosterChecklist"),
+    AuditTrail: require("./src/models/auditTrail"),
+    DailyChecklist: require("./src/models/dailyChecklist"),
+    CoreInvoiceSettings: require("./src/models/coreInvoiceSettings"),
+    FacilityInvoice: require("./src/models/facilityInvoice"),
+    CoreBaseSettings: require("./src/models/coreBaseSettings"),
+    Recipient: require("./src/models/facilityInvoiceRecipient")
+};
+
+// Function to get models dynamically from a specific database connection
+async function getModelFromDB(dbConnection, modelName, schema) {
+    if (!dbConnection.models[modelName]) {
+        return dbConnection.model(modelName, schema); // Register the model in the db connection
+    }
+    return dbConnection.models[modelName]; // Return existing model if already registered
+}
+
+// Function to initialize service with specific models - NEW FEATURE
+function initializeService(modelNames = []) {
+    let currentConnection = null;
+
+    // Enhanced connect function that only registers specified models
+    async function connectWithModels(
+        dbName,
+        secured,
+        username,
+        password,
+        url,
+        port,
+    ) {
+        await connectToMongoDB(dbName, secured, username, password, url, port);
+        currentConnection = mongoose.connection;
+
+        // Only register the models specified in modelNames
+        modelNames.forEach((modelName) => {
+            if (models[modelName]) {
+                let schema;
+
+                // Extract schema from the imported model
+                if (models[modelName].schema) {
+                    schema = models[modelName].schema;
+                } else {
+                    console.warn(`Unable to extract schema from model ${modelName}`);
+                    return;
+                }
+
+                if (!currentConnection.models[modelName]) {
+                    const modelInstance = currentConnection.model(modelName, schema);
+                    // Make model globally accessible through payservedb
+                    module.exports[modelName] = modelInstance;
+                } else {
+                    module.exports[modelName] = currentConnection.models[modelName];
+                }
+            } else {
+                console.warn(`Model ${modelName} not found in models registry`);
+            }
+        });
+
+        return module.exports;
+    }
+
+    return {
+        connectToMongoDB: connectWithModels,
+        switchDB,
+        getModelFromDB,
+        getCurrentConnection: () => currentConnection,
+    };
+}
+
+module.exports = {
+    connectToMongoDB,
+    switchDB,
+    getModelFromDB,
+    initializeService,
+    ...models,
+};

package/package.json

@@ -1,17 +1,17 @@
-{
-  "name": "payservedb",
-  "version": "7.0.7",
-  "main": "index.js",
-  "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
-  },
-  "keywords": [],
-  "author": "",
-  "license": "ISC",
-  "description": "",
-  "dependencies": {
-    "dotenv": "^16.4.5",
-    "moment-timezone": "^0.5.47",
-    "mongoose": "^8.3.5"
-  }
-}
+{
+  "name": "payservedb",
+  "version": "7.0.8",
+  "main": "index.js",
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
+  "keywords": [],
+  "author": "",
+  "license": "ISC",
+  "description": "",
+  "dependencies": {
+    "dotenv": "^16.4.5",
+    "moment-timezone": "^0.5.47",
+    "mongoose": "^8.3.5"
+  }
+}