Update app.js

app.js

@@ -1,290 +1,255 @@
 const express = require('express');
 const admin = require('firebase-admin');
-const crypto = require('crypto'); // Telegram Security-এর জন্য ইনবিল্ট মডিউল
-
-const app = express();
-app.use(express.json());
-
-// ==========================================
-// ০. রিকোয়েস্ট এবং রেসপন্স লগার মিডলওয়্যার (Terminal Console Log)
-// ==========================================
-app.use((req, res, next) => {
-    const startTime = Date.now();
-    
-    // original res.json এবং res.send মেথড ব্যাকআপ রাখা হচ্ছে রেসপন্স বডি ট্র্যাক করার জন্য
-    const oldJson = res.json;
-    const oldSend = res.send;
-    let responseBody;
-
-    res.json = function(data) {
-        responseBody = data;
-        return oldJson.apply(res, arguments);
-    };
+const crypto = require('crypto');
+const cluster = require('cluster');
+const os = require('os');
 
-    res.send = function(data) {
-        responseBody = data;
-        return oldSend.apply(res, arguments);
-    };
+// সার্ভারের মোট সিপিইউ কোর সংখ্যা স্বয়ংক্রিয়ভাবে ডিটেক্ট করা (১২ কোর হলে ১২টি প্রসেস তৈরি হবে)
+const numCPUs = os.cpus().length;
+
+if (cluster.isMaster) {
+    console.log(`Master Server Process ${process.pid} is running. Detecting ${numCPUs} CPU cores...`);
 
-    // রিকোয়েস্ট শেষ হলে টার্মিনালে বা Docker লগ-এ প্রিন্ট হবে
-    res.on('finish', () => {
-        const duration = Date.now() - startTime;
-        console.log(`\n=========================================`);
-        console.log(`[${new Date().toISOString()}] ${req.method} -> ${req.originalUrl}`);
-        console.log(`Status: ${res.statusCode} (${res.statusMessage}) | Time: ${duration}ms`);
-        
-        // রিকোয়েস্ট বডি বা কুয়েরি ডেটা দেখা
-        if (Object.keys(req.body).length > 0) {
-            // সিকিউরিটির জন্য এবং টার্মিনাল ক্লিন রাখতে বড় init_data হাইড করা হয়েছে
-            const cleanBody = { ...req.body };
-            if (cleanBody.init_data) cleanBody.init_data = "[TELEGRAM_INIT_DATA_HIDDEN]";
-            console.log(`Request Body:`, JSON.stringify(cleanBody, null, 2));
-        } else if (Object.keys(req.query).length > 0) {
-            console.log(`Request Query:`, JSON.stringify(req.query, null, 2));
+    // প্রতিটি কোরের জন্য একটি করে ওয়ার্কার সার্ভার তৈরি করা হচ্ছে
+    for (let i = 0; i < numCPUs; i++) {
+        cluster.fork();
+    }
+
+    // কোনো ওয়ার্কার কোনো কারণে ক্র্যাশ করলে স্বয়ংক্রিয়ভাবে নতুন ওয়ার্কার চালু হবে
+    cluster.on('exit', (worker, code, signal) => {
+        cluster.fork();
+    });
+
+} else {
+    // এটি হলো ওয়ার্কার প্রসেস - যা আসল ট্রাফিক হ্যান্ডেল করবে
+    const app = express();
+    app.use(express.json());
+
+    // ==========================================
+    // ১. ফায়ারবেজ মাল্টিপল অ্যাডমিন সেটআপ
+    // ==========================================
+    const initFirebase = (envAdmin, envUrl, appName) => {
+        if (!process.env[envAdmin] || !process.env[envUrl]) {
+            return null;
         }
+        return admin.initializeApp({
+            credential: admin.credential.cert(JSON.parse(process.env[envAdmin])),
+            databaseURL: process.env[envUrl]
+        }, appName);
+    };
+
+    const app1 = initFirebase('FIREBASE_ADMIN_1', 'FIREBASE_DB_URL_1', 'Admin1');
+    const app3 = initFirebase('FIREBASE_ADMIN_3', 'FIREBASE_DB_URL_3', 'Admin3');
+    const app4 = initFirebase('FIREBASE_ADMIN_4', 'FIREBASE_DB_URL_4', 'Admin4');
 
-        // সার্ভার ক্লায়েন্টকে কী রিপ্লাই দিল তা প্রিন্ট করা
-        if (responseBody) {
-            console.log(`Server Reply:`, typeof responseBody === 'object' ? JSON.stringify(responseBody, null, 2) : responseBody);
+    const db1 = app1 ? app1.database() : null;
+    const db3 = app3 ? app3.database() : null;
+    const db4 = app4 ? app4.database() : null;
+
+    // ==========================================
+    // ২. হেলথ চেক রাউটস (যেকোনো রিকোয়েস্ট)
+    // ==========================================
+    const healthHandler = (req, res) => {
+        res.send('Ok, yes, done');
+    };
+    app.all('/health', healthHandler);
+    app.all('/ch', healthHandler);
+    app.all('/unps', healthHandler);
+
+    // ==========================================
+    // ৩. মিডলওয়্যার: শুধুমাত্র POST রিকোয়েস্ট পারমিট করা
+    // ==========================================
+    const checkPostMethod = (req, res, next) => {
+        if (req.method !== 'POST') {
+            return res.status(405).json({ error: 'Method Not Allowed. Only POST is accepted.' });
         }
-        console.log(`=========================================`);
-    });
+        next();
+    };
 
-    next();
-});
+    // ==========================================
+    // ৪. Telegram Security ভ্যালিডেশন মিডলওয়্যার
+    // ==========================================
+    const verifyTelegramAuth = (req, res, next) => {
+        const { init_data, user_id, name } = req.body;
 
-// ==========================================
-// ১. ফায়ারবেজ মাল্টিপল অ্যাডমিন সেটআপ
-// ==========================================
-const initFirebase = (envAdmin, envUrl, appName) => {
-    if (!process.env[envAdmin] || !process.env[envUrl]) {
-        console.warn(`Warning: ${envAdmin} or ${envUrl} is missing.`);
-        return null;
-    }
-    return admin.initializeApp({
-        credential: admin.credential.cert(JSON.parse(process.env[envAdmin])),
-        databaseURL: process.env[envUrl]
-    }, appName);
-};
-
-const app1 = initFirebase('FIREBASE_ADMIN_1', 'FIREBASE_DB_URL_1', 'Admin1');
-const app3 = initFirebase('FIREBASE_ADMIN_3', 'FIREBASE_DB_URL_3', 'Admin3');
-const app4 = initFirebase('FIREBASE_ADMIN_4', 'FIREBASE_DB_URL_4', 'Admin4');
-
-const db1 = app1 ? app1.database() : null;
-const db3 = app3 ? app3.database() : null;
-const db4 = app4 ? app4.database() : null;
-
-// ==========================================
-// ২. হেলথ চেক রাউটস (যেকোনো রিকোয়েস্ট)
-// ==========================================
-const healthHandler = (req, res) => {
-    res.send('Ok, yes, done');
-};
-app.all('/health', healthHandler);
-app.all('/ch', healthHandler);
-app.all('/unps', healthHandler);
-
-// ==========================================
-// ৩. মিডলওয়্যার: শুধুমাত্র POST রিকোয়েস্ট পারমিট করা
-// ==========================================
-const checkPostMethod = (req, res, next) => {
-    if (req.method !== 'POST') {
-        return res.status(405).json({ error: 'Method Not Allowed. Only POST is accepted.' });
-    }
-    next();
-};
+        if (!init_data) return res.status(401).json({ error: 'Unauthorized: init_data is missing' });
 
-// ==========================================
-// ৪. Telegram Security ভ্যালিডেশন মিডলওয়্যার
-// ==========================================
-const verifyTelegramAuth = (req, res, next) => {
-    const { init_data, user_id, name } = req.body;
+        try {
+            const botToken = process.env.BOT_TOKEN;
+            if (!botToken) return res.status(500).json({ error: 'Server configuration error' });
 
-    if (!init_data) return res.status(401).json({ error: 'Unauthorized: init_data is missing' });
+            const urlParams = new URLSearchParams(init_data);
+            const hash = urlParams.get('hash');
+            urlParams.delete('hash');
 
-    try {
-        const botToken = process.env.BOT_TOKEN;
-        if (!botToken) return res.status(500).json({ error: 'Server configuration error' });
+            const dataToCheck = [...urlParams.entries()]
+                .map(([key, value]) => `${key}=${value}`)
+                .sort()
+                .join('\n');
 
-        const urlParams = new URLSearchParams(init_data);
-        const hash = urlParams.get('hash');
-        urlParams.delete('hash');
+            const secretKey = crypto.createHmac('sha256', 'WebAppData').update(botToken).digest();
+            const calculatedHash = crypto.createHmac('sha256', secretKey).update(dataToCheck).digest('hex');
 
-        const dataToCheck = [...urlParams.entries()]
-            .map(([key, value]) => `${key}=${value}`)
-            .sort()
-            .join('\n');
+            if (calculatedHash !== hash) {
+                return res.status(403).json({ error: 'Forbidden: Invalid Telegram signature' });
+            }
 
-        const secretKey = crypto.createHmac('sha256', 'WebAppData').update(botToken).digest();
-        const calculatedHash = crypto.createHmac('sha256', secretKey).update(dataToCheck).digest('hex');
+            const tgUser = JSON.parse(urlParams.get('user'));
 
-        if (calculatedHash !== hash) {
-            return res.status(403).json({ error: 'Forbidden: Invalid Telegram signature' });
-        }
+            if (String(tgUser.id) !== String(user_id)) {
+                return res.status(403).json({ error: 'Forbidden: User ID mismatch' });
+            }
 
-        const tgUser = JSON.parse(urlParams.get('user'));
+            const tgFullName = tgUser.last_name ? `${tgUser.first_name} ${tgUser.last_name}`.trim() : tgUser.first_name;
+            if (String(tgFullName) !== String(name) && String(tgUser.first_name) !== String(name)) {
+                return res.status(403).json({ error: 'Forbidden: Name mismatch' });
+            }
 
-        // আইডি এবং নাম হুবহু সেম কি না তা চেক করা
-        if (String(tgUser.id) !== String(user_id)) {
-            return res.status(403).json({ error: 'Forbidden: User ID mismatch' });
+            next();
+        } catch (error) {
+            return res.status(500).json({ error: 'Internal server error during validation' });
         }
+    };
 
-        const tgFullName = tgUser.last_name ? `${tgUser.first_name} ${tgUser.last_name}`.trim() : tgUser.first_name;
-        if (String(tgFullName) !== String(name) && String(tgUser.first_name) !== String(name)) {
-            return res.status(403).json({ error: 'Forbidden: Name mismatch' });
-        }
+    // ==========================================
+    // ৫. API: ক্রিয়েট অ্যাকাউন্ট (/creataccoumt)
+    // ==========================================
+    app.post('/creataccoumt', checkPostMethod, verifyTelegramAuth, async (req, res) => {
+        const { user_id, refer_id, name, username, profile_pic_url } = req.body;
 
-        next(); // সব ঠিক থাকলে মূল লজিকে যাবে
-    } catch (error) {
-        console.error('Telegram Validation Error:', error);
-        return res.status(500).json({ error: 'Internal server error during validation' });
-    }
-};
+        try {
+            const userRef = db1.ref('user').child(user_id);
+            const snapshot = await userRef.once('value');
 
-// ==========================================
-// ৫. API: ক্রিয়েট অ্যাকাউন্ট (/creataccoumt)
-// ==========================================
-app.post('/creataccoumt', checkPostMethod, verifyTelegramAuth, async (req, res) => {
-    const { user_id, refer_id, name, username, profile_pic_url } = req.body;
+            if (snapshot.exists()) {
+                return res.status(200).json({ message: 'Account already exists. No action taken.' });
+            }
 
-    try {
-        const userRef = db1.ref('user').child(user_id);
-        const snapshot = await userRef.once('value');
+            const newUserData = {
+                user_id,
+                refer_id: refer_id || null,
+                name: name || null,
+                username: username || null,
+                profile_pic_url: profile_pic_url || null,
+                balance: 0,
+                total_withdraw: 0,
+                total_withdraw_amount: 0,
+                join_date: admin.database.ServerValue.TIMESTAMP,
+                total_refer: 0
+            };
+
+            await userRef.set(newUserData);
+
+            if (refer_id) {
+                const referUserRef = db1.ref('user').child(refer_id);
+                const referUserSnap = await referUserRef.once('value');
+
+                if (referUserSnap.exists()) {
+                    await referUserRef.child('total_refer').transaction((currentValue) => {
+                        return (currentValue || 0) + 1;
+                    });
+
+                    const referListRef = db3.ref('refer').child(refer_id);
+                    await referListRef.push(user_id);
+                }
+            }
 
-        if (snapshot.exists()) {
-            return res.status(200).json({ message: 'Account already exists. No action taken.' });
+            res.status(201).json({ message: 'Account created successfully!' });
+        } catch (error) {
+            res.status(500).json({ error: 'Internal Server Error' });
         }
+    });
 
-        const newUserData = {
-            user_id,
-            refer_id: refer_id || null,
-            name: name || null,
-            username: username || null,
-            profile_pic_url: profile_pic_url || null,
-            balance: 0,
-            total_withdraw: 0,
-            total_withdraw_amount: 0,
-            join_date: admin.database.ServerValue.TIMESTAMP,
-            total_refer: 0
-        };
-
-        await userRef.set(newUserData);
-
-        if (refer_id) {
-            const referUserRef = db1.ref('user').child(refer_id);
-            const referUserSnap = await referUserRef.once('value');
-
-            if (referUserSnap.exists()) {
-                await referUserRef.child('total_refer').transaction((currentValue) => {
-                    return (currentValue || 0) + 1;
-                });
-
-                const referListRef = db3.ref('refer').child(refer_id);
-                await referListRef.push(user_id);
-            }
-        }
+    // ==========================================
+    // ৬. API: উইথড্র মেইন ব্যালেন্স (/withdraw-main)
+    // ==========================================
+    app.post('/withdraw-main', checkPostMethod, verifyTelegramAuth, async (req, res) => {
+        const { user_id, method, address, time, amount } = req.body;
+        const allowedMethods = ['bkash', 'mobile-gp', 'mobile-ar', 'mobile-ro', 'mobile-bl', 'mobile-tk', 'nagad', 'binance', 'usdt', 'ton'];
 
-        res.status(201).json({ message: 'Account created successfully!' });
-    } catch (error) {
-        console.error('Account Creation Error:', error);
-        res.status(500).json({ error: 'Internal Server Error' });
-    }
-});
+        if (!method || !address || !amount) return res.status(400).json({ error: 'Missing required fields' });
+        if (!allowedMethods.includes(method)) return res.status(400).json({ error: 'Invalid withdrawal method' });
 
-// ==========================================
-// ৬. API: উইথড্র মেইন ব্যালেন্স (/withdraw-main)
-// ==========================================
-app.post('/withdraw-main', checkPostMethod, verifyTelegramAuth, async (req, res) => {
-    const { user_id, method, address, time, amount } = req.body;
-    const allowedMethods = ['bkash', 'mobile-gp', 'mobile-ar', 'mobile-ro', 'mobile-bl', 'mobile-tk', 'nagad', 'binance', 'usdt', 'ton'];
+        try {
+            const userRef = db1.ref('user').child(user_id);
+            const snapshot = await userRef.once('value');
 
-    if (!method || !address || !amount) return res.status(400).json({ error: 'Missing required fields' });
-    if (!allowedMethods.includes(method)) return res.status(400).json({ error: 'Invalid withdrawal method' });
+            if (!snapshot.exists()) return res.status(404).json({ error: 'User not found' });
 
-    try {
-        const userRef = db1.ref('user').child(user_id);
-        const snapshot = await userRef.once('value');
+            const userData = snapshot.val();
+            const currentBalance = Number(userData.balance || 0);
+            const requestAmount = Number(amount);
 
-        if (!snapshot.exists()) return res.status(404).json({ error: 'User not found' });
+            if (currentBalance < requestAmount) {
+                return res.status(400).json({ error: 'Insufficient balance' });
+            }
 
-        const userData = snapshot.val();
-        const currentBalance = Number(userData.balance || 0);
-        const requestAmount = Number(amount);
+            await userRef.child('balance').set(currentBalance - requestAmount);
+
+            const pendingRef = db3.ref('withdrow/painding').child(user_id);
+            await pendingRef.push({
+                method,
+                address,
+                time: time || admin.database.ServerValue.TIMESTAMP,
+                amount: requestAmount,
+                status: 'pending'
+            });
 
-        if (currentBalance < requestAmount) {
-            return res.status(400).json({ error: 'Insufficient balance' });
+            res.status(200).json({ message: 'Withdrawal request successful' });
+        } catch (error) {
+            res.status(500).json({ error: 'Internal Server Error' });
         }
+    });
 
-        await userRef.child('balance').set(currentBalance - requestAmount);
-
-        const pendingRef = db3.ref('withdrow/painding').child(user_id);
-        await pendingRef.push({
-            method,
-            address,
-            time: time || admin.database.ServerValue.TIMESTAMP,
-            amount: requestAmount,
-            status: 'pending'
-        });
-
-        res.status(200).json({ message: 'Withdrawal request successful' });
-    } catch (error) {
-        console.error('Withdraw Main Error:', error);
-        res.status(500).json({ error: 'Internal Server Error' });
-    }
-});
-
-// ==========================================
-// ৭. API: অ্যাড ব্যালেন্স উইথড্র (Admin 4 -> Admin 1)
-// ==========================================
-const adWithdrawPaths = [
-    '/withdraw-link-visit',
-    '/withdraw-adsgram-1',
-    '/withdraw-adsgram-2',
-    '/withdraw-gigapub-1',
-    '/withdraw-gigapub-2',
-    '/withdraw-monetag-1',
-    '/withdraw-monetag-2'
-];
-
-app.post(adWithdrawPaths, checkPostMethod, async (req, res) => {
-    const { user_id, amount, method } = req.body;
-
-    if (!user_id || !amount || method !== 'wallet') {
-        return res.status(400).json({ error: 'Invalid request. Method must be wallet and amount is required.' });
-    }
+    // ==========================================
+    // ৭. API: অ্যাড ব্যালেন্স উইথড্র (Admin 4 -> Admin 1)
+    // ==========================================
+    const adWithdrawPaths = [
+        '/withdraw-link-visit',
+        '/withdraw-adsgram-1',
+        '/withdraw-adsgram-2',
+        '/withdraw-gigapub-1',
+        '/withdraw-gigapub-2',
+        '/withdraw-monetag-1',
+        '/withdraw-monetag-2'
+    ];
+
+    app.post(adWithdrawPaths, checkPostMethod, async (req, res) => {
+        const { user_id, amount, method } = req.body;
+
+        if (!user_id || !amount || method !== 'wallet') {
+            return res.status(400).json({ error: 'Invalid request. Method must be wallet and amount is required.' });
+        }
 
-    const adNetworkName = req.path.replace('/withdraw-', '');
-    const requestAmount = Number(amount);
+        const adNetworkName = req.path.replace('/withdraw-', '');
+        const requestAmount = Number(amount);
 
-    try {
-        const adRef = db4.ref(adNetworkName).child(user_id);
-        const adSnapshot = await adRef.once('value');
+        try {
+            const adRef = db4.ref(adNetworkName).child(user_id);
+            const adSnapshot = await adRef.once('value');
 
-        if (!adSnapshot.exists()) return res.status(404).json({ error: 'No ad earnings found for this user' });
+            if (!adSnapshot.exists()) return res.status(404).json({ error: 'No ad earnings found for this user' });
 
-        let adCurrentBalance = Number(adSnapshot.val().balance || 0);
+            let adCurrentBalance = Number(adSnapshot.val().balance || 0);
 
-        if (adCurrentBalance < requestAmount) {
-            return res.status(400).json({ error: 'Insufficient ad balance' });
-        }
+            if (adCurrentBalance < requestAmount) {
+                return res.status(400).json({ error: 'Insufficient ad balance' });
+            }
 
-        await adRef.child('balance').set(adCurrentBalance - requestAmount);
+            await adRef.child('balance').set(adCurrentBalance - requestAmount);
 
-        const mainUserRef = db1.ref('user').child(user_id);
-        await mainUserRef.child('balance').transaction((currentMainBalance) => {
-            return (currentMainBalance || 0) + requestAmount;
-        });
+            const mainUserRef = db1.ref('user').child(user_id);
+            await mainUserRef.child('balance').transaction((currentMainBalance) => {
+                return (currentMainBalance || 0) + requestAmount;
+            });
 
-        res.status(200).json({ message: `Success! ${requestAmount} transferred to main balance from ${adNetworkName}` });
-    } catch (error) {
-        console.error('Ad Withdraw Error:', error);
-        res.status(500).json({ error: 'Internal Server Error' });
-    }
-});
+            res.status(200).json({ message: `Success! ${requestAmount} transferred to main balance from ${adNetworkName}` });
+        } catch (error) {
+            res.status(500).json({ error: 'Internal Server Error' });
+        }
+    });
 
-// সার্ভার স্টার্ট
-const PORT = process.env.PORT || 7860;
-app.listen(PORT, () => {
-    console.log(`Server is running on port ${PORT}`);
-});
+    // সার্ভার স্টার্ট (প্রতিটি ওয়ার্কার এই পোর্টে রিকোয়েস্ট শেয়ার করবে)
+    const PORT = process.env.PORT || 7860;
+    app.listen(PORT);
+}