Update index.js

index.js

@@ -1,70 +1,64 @@
 const express = require('express');
 const fs = require('fs');
 const path = require('path');
-const app = express();
+const qr = require('qr-image');
+const makeWASocket = require('@whiskeysockets/baileys').default;
+const Boom = require('@hapi/boom');
 
+const app = express();
 app.use(express.json());
 
+const sessionDir = path.join(__dirname, 'sessions');
+
+// Function to generate API key
 function generateApiKey() {
     return Math.random().toString(36).substr(2, 16);
 }
 
 // Function to validate IP address format and ensure it's not a private or reserved IP
 function isValidIp(ip) {
-    // Regular expression to validate IPv4 format
     const ipv4Regex = /^(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)\.(25[0-5]|2[0-4][0-9]|[01]?[0-9][0-9]?)$/;
+    if (!ipv4Regex.test(ip)) return false;
 
-    if (!ipv4Regex.test(ip)) {
-        return false; // Invalid IP format
-    }
-
-    // Split the IP into octets
     const octets = ip.split('.').map(Number);
-
-    // Check for private IP ranges
-    const isPrivateIp =
+    return !(
         (octets[0] === 10) ||
         (octets[0] === 172 && octets[1] >= 16 && octets[1] <= 31) ||
         (octets[0] === 192 && octets[1] === 168) ||
-        (ip === '127.0.0.1'); // localhost
-
-    return !isPrivateIp; // Return true if IP is not private
+        (ip === '127.0.0.1')
+    );
 }
 
-app.get('/checkState', (req, res) => {
+// Handle checkState request
+app.get('/checkState', async (req, res) => {
     const { ip } = req.query;
 
     if (ip && isValidIp(ip)) {
-        const sessionDir = path.join(__dirname, 'db', ip);
-        const sessionFile = path.join(sessionDir, 'info.json');
+        const sessionFile = path.join(sessionDir, `${ip}.json`);
 
         let apiKey;
 
-        // Check if the session file exists
         if (fs.existsSync(sessionFile)) {
-            // Read existing API key from file
             const sessionData = JSON.parse(fs.readFileSync(sessionFile, 'utf8'));
             apiKey = sessionData.apiKey;
         } else {
-            // Generate a new API key
             apiKey = generateApiKey();
 
-            // Create directory for user session
             if (!fs.existsSync(sessionDir)) {
                 fs.mkdirSync(sessionDir, { recursive: true });
             }
 
-            // Save API key and IP in info.json
             const sessionData = { ip, apiKey };
             fs.writeFileSync(sessionFile, JSON.stringify(sessionData, null, 2));
+
+            connectToWhatsApp(apiKey, ip); // Start WhatsApp connection
         }
 
-        // Set cookie and respond with API key
         res.cookie('apiKey', apiKey, { 
-          maxAge: 365 * 24 * 60 * 60 * 1000, 
-          httpOnly: false, 
-          sameSite: 'None', 
-          secure: true // Ensure this is set for HTTPS
+            maxAge: 365 * 24 * 60 * 60 * 1000, 
+            httpOnly: false, 
+            sameSite: 'None', 
+            secure: true
         });
 
         res.status(200).json({
@@ -77,10 +71,72 @@ app.get('/checkState', (req, res) => {
     }
 });
 
-app.get('/', (req, res) => {
-    res.send('Hello World');
+// Serve QR code for login
+app.get('/qrCode', (req, res) => {
+    const { apiKey } = req.query;
+
+    if (apiKey) {
+        const qrData = `https://example.com/login?apiKey=${apiKey}`; // Replace with your actual URL
+        const qrImage = qr.imageSync(qrData, { type: 'svg' });
+        res.setHeader('Content-Type', 'image/svg+xml');
+        res.send(qrImage);
+    } else {
+        res.status(400).send('API key is required');
+    }
+});
+
+// Handle user credentials storage
+app.post('/storeCredentials', (req, res) => {
+    const { apiKey, credentials } = req.body;
+
+    if (!apiKey || !credentials) {
+        return res.status(400).json({ message: 'API key and credentials are required' });
+    }
+
+    const credentialsDir = path.join(sessionDir, apiKey);
+    const credentialsFile = path.join(credentialsDir, 'creds.json');
+
+    if (!fs.existsSync(credentialsDir)) {
+        fs.mkdirSync(credentialsDir, { recursive: true });
+    }
+
+    fs.writeFileSync(credentialsFile, JSON.stringify(credentials, null, 2));
+
+    res.status(200).json({ message: 'Credentials stored successfully' });
 });
 
+// Serve login page
+app.use(express.static(path.join(__dirname, 'public')));
+
+async function connectToWhatsApp(apiKey, ip) {
+    const sock = makeWASocket({
+        printQRInTerminal: true
+    });
+
+    sock.ev.on('connection.update', (update) => {
+        const { connection, lastDisconnect } = update;
+        if (connection === 'close') {
+            const shouldReconnect = (lastDisconnect.error instanceof Boom) ? lastDisconnect.error.output.statusCode !== DisconnectReason.loggedOut : true;
+            console.log('Connection closed due to ', lastDisconnect.error, ', reconnecting ', shouldReconnect);
+            if (shouldReconnect) {
+                connectToWhatsApp(apiKey, ip);
+            }
+        } else if (connection === 'open') {
+            console.log('Opened connection');
+        }
+    });
+
+    sock.ev.on('messages.upsert', async m => {
+        console.log(JSON.stringify(m, undefined, 2));
+        console.log('Replying to', m.messages[0].key.remoteJid);
+        await sock.sendMessage(m.messages[0].key.remoteJid, { text: 'Hello there!' });
+    });
+
+    // Save session to file
+    const sessionFile = path.join(sessionDir, `${ip}.json`);
+    fs.writeFileSync(sessionFile, JSON.stringify({ apiKey, ip, session: sock.state }, null, 2));
+}
+
 const PORT = 7860;
 app.listen(PORT, () => {
     console.log(`Server is running on port ${PORT}`);