Create server.js

server.js

@@ -0,0 +1,205 @@
+const express = require('express');
+const { spawn } = require('child_process');
+const { MongoClient } = require('mongodb');
+const path = require('path');
+const app = express();
+const http = require('http').createServer(app);
+const io = require('socket.io')(http, {
+  cors: {
+    origin: "*",
+    methods: ["GET", "POST"]
+  }
+});
+const crypto = require('crypto');
+const fetch = require('node-fetch');
+
+// Connect to MongoDB
+const MONGO_URI = process.env.MONGO_URI || "mongodb://localhost:27017";
+let db;
+
+async function connectDB() {
+  try {
+    const client = new MongoClient(MONGO_URI);
+    await client.connect();
+    db = client.db('whatsapp-bots');
+    console.log("✅ Connected to MongoDB");
+    
+    // Create collections if they don't exist
+    const collections = ['users', 'bots', 'settings'];
+    for (const colName of collections) {
+      if (!(await db.listCollections({name: colName}).hasNext())) {
+        await db.createCollection(colName);
+        console.log(`Created collection: ${colName}`);
+      }
+    }
+    
+    // Create admin user if not exists
+    const adminExists = await db.collection('users').findOne({username: 'admin'});
+    if (!adminExists) {
+      await db.collection('users').insertOne({
+        username: 'admin',
+        password: crypto.createHash('sha256').update('admin123').digest('hex'),
+        isAdmin: true,
+        createdAt: new Date()
+      });
+      console.log("👑 Created admin user (password: admin123)");
+    }
+    
+    // Restart all bots that were running
+    const runningBots = await db.collection('bots').find({status: 'running'}).toArray();
+    runningBots.forEach(bot => {
+      startBotProcess(bot.userId, bot.repoUrl, bot.entryFile);
+    });
+  } catch (err) {
+    console.error("❌ MongoDB connection error:", err);
+  }
+}
+
+// Start a bot process
+function startBotProcess(userId, repoUrl, entryFile) {
+  const botDir = `/persistent/storage/${userId}`;
+  
+  // Clone repo if not exists
+  if (!require('fs').existsSync(botDir)) {
+    const clone = spawn('git', ['clone', repoUrl, botDir]);
+    clone.on('close', (code) => {
+      if (code === 0) {
+        console.log(`✅ Cloned ${repoUrl} for user ${userId}`);
+        installDependencies(botDir, userId, repoUrl, entryFile);
+      } else {
+        console.error(`❌ Failed to clone ${repoUrl} for user ${userId}`);
+      }
+    });
+  } else {
+    installDependencies(botDir, userId, repoUrl, entryFile);
+  }
+}
+
+function installDependencies(botDir, userId, repoUrl, entryFile) {
+  const install = spawn('npm', ['install'], { cwd: botDir });
+  install.on('close', (code) => {
+    if (code === 0) {
+      console.log(`✅ Installed dependencies for ${userId}`);
+      runBot(botDir, userId, repoUrl, entryFile);
+    } else {
+      console.error(`❌ Failed to install dependencies for ${userId}`);
+    }
+  });
+}
+
+function runBot(botDir, userId, repoUrl, entryFile) {
+  const botProcess = spawn('node', [entryFile], { 
+    cwd: botDir,
+    stdio: ['pipe', 'pipe', 'pipe', 'ipc'] 
+  });
+  
+  // Save to database
+  db.collection('bots').updateOne(
+    { userId },
+    { $set: { 
+      status: 'running',
+      repoUrl,
+      entryFile,
+      startedAt: new Date(),
+      pid: botProcess.pid 
+    }},
+    { upsert: true }
+  );
+  
+  botProcess.stdout.on('data', (data) => {
+    io.to(userId).emit('bot-output', data.toString());
+  });
+  
+  botProcess.stderr.on('data', (data) => {
+    io.to(userId).emit('bot-error', data.toString());
+  });
+  
+  botProcess.on('close', (code) => {
+    console.log(`Bot ${userId} exited with code ${code}`);
+    db.collection('bots').updateOne(
+      { userId },
+      { $set: { status: 'stopped', exitCode: code } }
+    );
+  });
+}
+
+// Web server setup
+app.use(express.static('public'));
+app.use(express.json());
+
+// API endpoints
+app.post('/api/deploy', async (req, res) => {
+  const { userId, repoUrl, entryFile } = req.body;
+  
+  if (!userId || !repoUrl || !entryFile) {
+    return res.status(400).json({ error: "Missing parameters" });
+  }
+  
+  try {
+    startBotProcess(userId, repoUrl, entryFile);
+    res.json({ success: true, message: "Bot deployment started" });
+  } catch (err) {
+    res.status(500).json({ error: err.message });
+  }
+});
+
+// Socket.IO
+io.on('connection', (socket) => {
+  console.log(`User connected: ${socket.id}`);
+  
+  socket.on('register', async (data) => {
+    const { username, password } = data;
+    const hashedPassword = crypto.createHash('sha256').update(password).digest('hex');
+    
+    try {
+      await db.collection('users').insertOne({
+        username,
+        password: hashedPassword,
+        createdAt: new Date()
+      });
+      socket.emit('register-success');
+    } catch (err) {
+      socket.emit('register-error', 'Username already exists');
+    }
+  });
+  
+  socket.on('login', async (data) => {
+    const { username, password } = data;
+    const hashedPassword = crypto.createHash('sha256').update(password).digest('hex');
+    
+    const user = await db.collection('users').findOne({ 
+      username, 
+      password: hashedPassword 
+    });
+    
+    if (user) {
+      socket.emit('login-success', { userId: user._id.toString(), isAdmin: user.isAdmin });
+      socket.join(user._id.toString());
+    } else {
+      socket.emit('login-error', 'Invalid credentials');
+    }
+  });
+  
+  socket.on('deploy-bot', (data) => {
+    const { userId, repoUrl, entryFile } = data;
+    startBotProcess(userId, repoUrl, entryFile);
+  });
+});
+
+// Start server
+const PORT = process.env.PORT || 7860;
+connectDB().then(() => {
+  http.listen(PORT, () => {
+    console.log(`🚀 Server running on port ${PORT}`);
+    console.log(`🌐 Open: http://localhost:${PORT}`);
+  });
+});
+
+// Error handling
+process.on('uncaughtException', (err) => {
+  console.error('Uncaught Exception:', err);
+});
+
+process.on('unhandledRejection', (reason, promise) => {
+  console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+});