Spaces:

AUXteam
/

wppconnect-api

Paused

App Files Files Community

AUXteam commited on Apr 1

Commit

bdf4689

verified ·

1 Parent(s): f7a8ac1

Upload folder using huggingface_hub

Browse files

Files changed (1) hide show

src/app.ts +46 -147

src/app.ts CHANGED Viewed

@@ -5,18 +5,12 @@ import fs from 'fs';
 import path from 'path';
 import { exec } from 'child_process';
-// Set public DNS to avoid resolution issues in some environments
-try {
-  dns.setServers(['8.8.8.8', '1.1.1.1']);
-} catch (e) {
-  console.warn('Could not set custom DNS servers:', e.message);
-}
 const app = express();
 const port = process.env.PORT || 7860;
 let whatsappClient: wppconnect.Whatsapp | null = null;
 let isInitializing = false;
 let lastQR: { base64?: string, ascii?: string, timestamp?: number } | null = null;
 let initError: string | null = null;
 app.use(express.json());
@@ -36,96 +30,40 @@ const checkDNS = (host: string): Promise<boolean> => {
   });
 };
-const cleanupProcesses = () => {
-  return new Promise((resolve) => {
-    console.log('Cleaning up any existing Chromium processes...');
-    exec('pkill -f chromium', (err) => {
-      if (err) {
-        // pkill exits with 1 if no processes were found, which is fine
-        console.log('Cleanup: No existing processes or pkill not found.');
-      } else {
-        console.log('Cleanup complete.');
-      }
-      resolve(true);
-    });
-  });
-};
 app.get('/health', (req, res) => {
   res.status(200).json({
     status: 'UP',
     initializing: isInitializing,
     connected: !!whatsappClient,
     hasQR: !!lastQR,
     initError: initError,
     uptime: process.uptime()
   });
 });
-app.get('/api-docs', (req, res) => {
-  res.status(200).json({
-    message: 'WPPConnect API Documentation',
-    endpoints: {
-      '/health': 'GET: Health check endpoint',
-      '/api-docs': 'GET: API documentation',
-      '/qr': 'GET: Get the latest QR code as an image',
-      '/qr-ascii': 'GET: Get the latest QR code as ASCII text',
-      '/send-message': 'POST: Send a text message (telnumber, message)',
-      '/send-poll': 'POST: Send a poll message (telnumber, name, choices)',
-      '/get-connection-status': 'GET: Get the current WhatsApp connection status',
-      '/restart': 'POST: Restart the WhatsApp instance'
-    },
-  });
-});
 app.get('/qr', (req, res) => {
   if (lastQR && lastQR.base64) {
     const base64Data = lastQR.base64.replace(/^data:image\/png;base64,/, '');
     const img = Buffer.from(base64Data, 'base64');
-    res.writeHead(200, {
-      'Content-Type': 'image/png',
-      'Content-Length': img.length
-    });
     res.end(img);
   } else {
-    res.status(404).json({ status: false, message: 'QR code not available yet.' });
   }
 });
 app.get('/qr-ascii', (req, res) => {
   if (lastQR && lastQR.ascii) {
-    res.send(`<html><body><pre style="font-family: monospace; white-space: pre;">${lastQR.ascii}</pre></body></html>`);
   } else {
-    res.status(404).send('QR code not available yet.');
-  }
-});
-app.post('/restart', async (req, res) => {
-  if (isInitializing) {
-    return res.status(409).json({ status: false, message: 'Already initializing' });
-  }
-  res.json({ status: true, message: 'Restarting WhatsApp instance...' });
-  startWPP();
-});
-app.get('/get-connection-status', async (req, res) => {
-  if (whatsappClient) {
-    try {
-      const state = await whatsappClient.getConnectionState();
-      res.json({ status: true, message: state });
-    } catch (error) {
-      res.status(500).json({ status: false, message: error.message });
-    }
-  } else {
-    res.status(503).json({ status: false, message: 'WhatsApp instance not initialized' });
   }
 });
 app.post('/send-message', async (req, res) => {
   const { telnumber, message } = req.body;
-  if (!whatsappClient) {
-    return res.status(503).json({ status: false, message: 'WhatsApp instance not initialized' });
-  }
   try {
     const result = await whatsappClient.sendText(`${telnumber}@c.us`, message);
     res.json({ status: true, message: 'Message sent', result });
@@ -136,13 +74,8 @@ app.post('/send-message', async (req, res) => {
 app.post('/send-poll', async (req, res) => {
   const { telnumber, name, choices } = req.body;
-  if (!whatsappClient) {
-    return res.status(503).json({ status: false, message: 'WhatsApp instance not initialized' });
-  }
   try {
-    if (!telnumber || !name || !Array.isArray(choices)) {
-      return res.status(400).json({ status: false, message: 'Invalid parameters.' });
-    }
     const result = await whatsappClient.sendPollMessage(`${telnumber}@c.us`, name, choices);
     res.json({ status: true, message: 'Poll sent', result });
   } catch (error) {
@@ -150,83 +83,49 @@ app.post('/send-poll', async (req, res) => {
   }
 });
-async function startWPP(retries = 20, delay = 30000) {
   if (isInitializing) return;
   isInitializing = true;
-  initError = null;
-  console.log('Starting WPPConnect initialization sequence...');
-  const sessionName = 'hf-space-session';
-  const tokenPath = path.join(process.cwd(), 'tokens', sessionName);
-  for (let i = 1; i <= retries; i++) {
-    console.log(`Attempt ${i} to start WPPConnect...`);
-    // Check network
-    const dnsOk = await checkDNS('web.whatsapp.com');
-    if (!dnsOk && i < 5) {
-      console.log('Network/DNS not ready, waiting...');
-      await new Promise(r => setTimeout(r, delay));
-      continue;
-    }
-    try {
-      await cleanupProcesses();
-      const lockPath = path.join(tokenPath, 'SingletonLock');
-      if (fs.existsSync(lockPath)) {
-        try { fs.unlinkSync(lockPath); console.log('Removed SingletonLock'); } catch (e) {}
-      }
-      whatsappClient = await wppconnect.create({
-        session: sessionName,
-        catchQR: (base64Qr, asciiQR) => {
-          console.log('QR Code generated.');
-          lastQR = { base64: base64Qr, ascii: asciiQR, timestamp: Date.now() };
-        },
-        statusFind: (statusSession, session) => {
-          console.log('Status Session: ', statusSession);
-          if (statusSession === 'isLogged' || statusSession === 'inChat') {
-             lastQR = null;
-          }
-        },
-        headless: true,
-        useChrome: false,
-        puppeteerOptions: {
-          executablePath: process.env.PUPPETEER_EXECUTABLE_PATH || '/usr/bin/chromium',
-          args: [
-            '--no-sandbox',
-            '--disable-setuid-sandbox',
-            '--disable-dev-shm-usage',
-            '--disable-gpu',
-            '--no-zygote',
-          ],
-        },
-        logQR: true,
-        tokenStore: 'file',
-        folderNameToken: './tokens',
-        autoClose: 0,
-      });
-      console.log('WPPConnect initialized successfully.');
-      isInitializing = false;
-      return;
-    } catch (error) {
-      console.error(`Attempt ${i} failed: ${error.message}`);
-      initError = error.message;
-      if (i === retries) {
-        console.error('All retries exhausted.');
-        isInitializing = false;
-      } else {
-        console.log(`Retrying in ${delay / 1000} seconds...`);
-        await new Promise(r => setTimeout(r, delay));
-      }
-    }
   }
 }
 app.listen(port, () => {
-  console.log(`Server listening on port ${port}`);
-  // Initialization in background
-  startWPP().catch(e => console.error('Background init failed:', e));
 });

 import path from 'path';
 import { exec } from 'child_process';
 const app = express();
 const port = process.env.PORT || 7860;
 let whatsappClient: wppconnect.Whatsapp | null = null;
 let isInitializing = false;
 let lastQR: { base64?: string, ascii?: string, timestamp?: number } | null = null;
+let lastStatus: string = 'Starting';
 let initError: string | null = null;
 app.use(express.json());
   });
 };
 app.get('/health', (req, res) => {
   res.status(200).json({
     status: 'UP',
     initializing: isInitializing,
     connected: !!whatsappClient,
+    whatsappStatus: lastStatus,
     hasQR: !!lastQR,
     initError: initError,
     uptime: process.uptime()
   });
 });
 app.get('/qr', (req, res) => {
   if (lastQR && lastQR.base64) {
     const base64Data = lastQR.base64.replace(/^data:image\/png;base64,/, '');
     const img = Buffer.from(base64Data, 'base64');
+    res.writeHead(200, { 'Content-Type': 'image/png', 'Content-Length': img.length });
     res.end(img);
   } else {
+    res.status(404).json({ status: false, message: 'QR code not available.' });
   }
 });
 app.get('/qr-ascii', (req, res) => {
   if (lastQR && lastQR.ascii) {
+    res.send(`<html><body style="background: #000; color: #fff;"><pre style="font-family: monospace; white-space: pre;">${lastQR.ascii}</pre></body></html>`);
   } else {
+    res.status(404).send('QR code not available.');
   }
 });
 app.post('/send-message', async (req, res) => {
   const { telnumber, message } = req.body;
+  if (!whatsappClient) return res.status(503).json({ status: false, message: 'WhatsApp instance not initialized' });
   try {
     const result = await whatsappClient.sendText(`${telnumber}@c.us`, message);
     res.json({ status: true, message: 'Message sent', result });
 app.post('/send-poll', async (req, res) => {
   const { telnumber, name, choices } = req.body;
+  if (!whatsappClient) return res.status(503).json({ status: false, message: 'WhatsApp instance not initialized' });
   try {
     const result = await whatsappClient.sendPollMessage(`${telnumber}@c.us`, name, choices);
     res.json({ status: true, message: 'Poll sent', result });
   } catch (error) {
   }
 });
+async function startWPP() {
   if (isInitializing) return;
   isInitializing = true;
+  lastStatus = 'Initializing';
+  console.log('Starting WPPConnect...');
+  const sessionName = 'hf-space-session';
+  // Minimal config to avoid DNS/Network issues at startup
+  try {
+    whatsappClient = await wppconnect.create({
+      session: sessionName,
+      catchQR: (base64Qr, asciiQR) => {
+        lastQR = { base64: base64Qr, ascii: asciiQR, timestamp: Date.now() };
+        console.log('QR Generated');
+      },
+      statusFind: (statusSession) => {
+        lastStatus = statusSession;
+        if (statusSession === 'isLogged' || statusSession === 'inChat') lastQR = null;
+      },
+      headless: true,
+      useChrome: false,
+      puppeteerOptions: {
+        executablePath: '/usr/bin/chromium',
+        args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'],
+      },
+      logQR: true,
+      tokenStore: 'file',
+      folderNameToken: './tokens',
+      autoClose: 0,
+    });
+    console.log('WPPConnect Ready');
+    isInitializing = false;
+  } catch (error) {
+    console.error('WPPConnect Failed:', error.message);
+    initError = error.message;
+    isInitializing = false;
+    // Retry after a delay
+    setTimeout(startWPP, 30000);
   }
 }
 app.listen(port, () => {
+  console.log(`Server port ${port}`);
+  startWPP();
 });