scheduled_check.js

import { Telegraf } from 'telegraf';
import dotenv from 'dotenv';
import Logger from './src/utils/logger.js';
import { forceCheckAllMonitors } from './src/utils/force_check.js';
import crud from './src/crud.js';
import mongoose from 'mongoose';

// Load environment variables
dotenv.config();

// Get MongoDB URI from environment variables
const mongoUri = process.env.MONGODB_URI;

if (!mongoUri) {
    console.error('MongoDB URI not found in environment variables. Please set MONGODB_URI in .env file.');
    process.exit(1);
}

// Connect to MongoDB
console.log('Connecting to MongoDB...');
mongoose.connect(mongoUri)
    .then(() => {
        console.log('Connected to MongoDB');
        startScheduler();
    })
    .catch(err => {
        console.error('Failed to connect to MongoDB:', err);
        process.exit(1);
    });

// Initialize the bot
const bot = new Telegraf(process.env.TELEGRAM_BOT_TOKEN);

// Set the interval in milliseconds (default: 5 minutes)
const CHECK_INTERVAL = process.env.CHECK_INTERVAL ? parseInt(process.env.CHECK_INTERVAL) : 5 * 60 * 1000;

// Function to start the scheduler
function startScheduler() {
    // Run the force check immediately on startup
    console.log(`[SCHEDULED] Starting scheduled checks every ${CHECK_INTERVAL / 1000} seconds`);
    runForceCheck().catch(error => {
        Logger.error(`[SCHEDULED] Initial force check failed: ${error.message}`);
        console.error('[SCHEDULED] Initial force check failed:', error);
    });

    // Set up the interval to run the force check
    setInterval(runForceCheck, CHECK_INTERVAL);
}

// Function to run the force check
async function runForceCheck() {
    try {
        Logger.info('[SCHEDULED] Starting scheduled force check...');
        console.log('[SCHEDULED] Starting scheduled force check...');
        
        // Get admin user ID from environment variable or use the first admin from the database
        let adminUserId = process.env.ADMIN_USER_IDS ? process.env.ADMIN_USER_IDS.split(',')[0].trim() : null;
        
        if (!adminUserId) {
            try {
                // Get all users from the database
                const users = await crud.getAllUsers();
                
                if (users && users.length > 0) {
                    adminUserId = users[0].userId;
                    Logger.info(`[SCHEDULED] Using user ID from database: ${adminUserId}`);
                    console.log(`[SCHEDULED] Using user ID from database: ${adminUserId}`);
                } else {
                    Logger.error('[SCHEDULED] No users found in database');
                    console.error('[SCHEDULED] No users found in database');
                    return;
                }
            } catch (dbError) {
                Logger.error(`[SCHEDULED] Error getting users from database: ${dbError.message}`);
                console.error('[SCHEDULED] Error getting users from database:', dbError);
                return;
            }
        } else {
            Logger.info(`[SCHEDULED] Using admin user ID from environment: ${adminUserId}`);
            console.log(`[SCHEDULED] Using admin user ID from environment: ${adminUserId}`);
        }
        
        // Create a mock context object
        const ctx = {
            from: { id: adminUserId },
            reply: async (text) => {
                Logger.info(`[SCHEDULED] Would reply to user: ${text}`);
                console.log(`[SCHEDULED] Would reply to user: ${text}`);
            },
            replyWithHTML: async (text) => {
                Logger.info(`[SCHEDULED] Would reply to user with HTML: ${text}`);
                console.log(`[SCHEDULED] Would reply to user with HTML: ${text}`);
            }
        };
        
        // Run the force check
        await forceCheckAllMonitors(ctx);
        
        Logger.info('[SCHEDULED] Completed scheduled force check');
        console.log('[SCHEDULED] Completed scheduled force check');
    } catch (error) {
        Logger.error(`[SCHEDULED] Error in scheduled force check: ${error.message}`);
        console.error('[SCHEDULED] Error in scheduled force check:', error);
    }
}

// Handle graceful shutdown
process.on('SIGINT', async () => {
    Logger.info('[SCHEDULED] Shutting down scheduled checks...');
    console.log('[SCHEDULED] Shutting down scheduled checks...');
    
    try {
        await mongoose.connection.close();
        console.log('Closed MongoDB connection');
    } catch (err) {
        console.error('Error closing MongoDB connection:', err);
    }
    
    process.exit(0);
});

process.on('SIGTERM', async () => {
    Logger.info('[SCHEDULED] Shutting down scheduled checks...');
    console.log('[SCHEDULED] Shutting down scheduled checks...');
    
    try {
        await mongoose.connection.close();
        console.log('Closed MongoDB connection');
    } catch (err) {
        console.error('Error closing MongoDB connection:', err);
    }
    
    process.exit(0);
});