refactor(app.ts): to handle errors and shutdowns, and update bot/index.ts to handle webhook requests and errors

Author: m-mdy-m

src/app.ts

@@ -1,31 +1,57 @@
 import { CopBot } from './bot';
 import { ServiceProvider } from './service/database/ServiceProvider';
 import logger from './utils/logger';
+
 async function main() {
   const cop = CopBot.getInstance();
-  await ServiceProvider.initialize();
-  logger.info('initialize Database');
-  await cop.initial();
-  logger.info('initial bot');
+  try {
+    await ServiceProvider.initialize();
+    logger.info('Initialized Database');
+
+    await cop.initial();
+    logger.info('Bot initialized');
+  } catch (error: any) {
+    logger.error('Error during initialization:', error);
+    process.exit(1); // Exit if initialization fails
+  }
+
+  // Handle graceful shutdown on SIGTERM
   process.on('SIGTERM', async () => {
     logger.info('SIGTERM signal received. Closing bot...');
     try {
       const db = ServiceProvider.getInstance();
-      await db.close();
+      await db.close(); // Ensure DB is closed before exiting
       logger.info('Database closed.');
     } catch (error: any) {
       logger.error('Error during shutdown:', error);
     } finally {
+      logger.info('Shutting down bot process');
       process.exit(0);
     }
   });
+
+  // Handle unhandled promise rejections
   process.on('unhandledRejection', (reason, promise) => {
     console.error('Unhandled Rejection at:', promise, 'reason:', reason);
+    // Consider gracefully shutting down the app here if necessary
   });
 
-  process.on('uncaughtException', (error) => {
-    console.error('Uncaught Exception:', error);
-    process.exit(1);
+  // Handle uncaught exceptions
+  process.on('uncaughtException', async (error: any) => {
+    logger.error('Uncaught Exception:', error);
+    try {
+      const db = ServiceProvider.getInstance();
+      await db.close(); // Close DB before exiting on uncaught exception
+      logger.info('Database closed due to uncaught exception.');
+    } catch (shutdownError: any) {
+      logger.error('Error during shutdown:', shutdownError);
+    } finally {
+      process.exit(1); // Exit the process on uncaught exception
+    }
   });
 }
-main().catch((error) => console.error('Application error:', error));
+
+main().catch((error) => {
+  logger.error('Application error during startup:', error);
+  process.exit(1); // Exit if the main function fails
+});

src/bot/index.ts

@@ -31,7 +31,46 @@ export class CopBot {
     const web_hook = Config.web_hook;
     const isProduction = Config.environment === 'production';
     if (isProduction) {
-      const server = http.createServer(webhookCallback(this._bot, 'http', { timeoutMilliseconds: 30000 }));
+      const server = http.createServer(async (req, res) => {
+        if (req.method === 'POST' && req.url === '/webhook') {
+          let body = '';
+
+          // Collect data chunks
+          req.on('data', (chunk) => {
+            body += chunk;
+          });
+
+          // Handle incoming request once the data is complete
+          req.on('end', async () => {
+            try {
+              // Parse the incoming JSON body
+              const update = JSON.parse(body);
+              if (!update) {
+                console.error('Received empty body or malformed JSON.');
+                return (res.statusCode = 400);
+              }
+
+              console.log('Received webhook body:', update);
+              await this._bot.handleUpdate(update); // Handle the update
+              res.statusCode = 200; // Success response to Telegram
+              res.end();
+            } catch (error: any) {
+              console.error('Error parsing JSON in webhook request:', error.message || error.stack);
+              res.statusCode = 500; // Internal Server Error
+              res.end('Internal Server Error');
+            }
+          });
+
+          req.on('error', (err) => {
+            console.error('Request error:', err);
+            res.statusCode = 400;
+            res.end('Bad Request');
+          });
+        } else {
+          res.statusCode = 404; // Not Found
+          res.end();
+        }
+      });
       server.listen(port, '0.0.0.0', () => {
         console.log(`Bot started on port ${port}`);
       });
@@ -44,7 +83,7 @@ export class CopBot {
         } else {
           console.log('No webhook set.');
         }
-      } catch (error:any) {
+      } catch (error: any) {
         console.error('Error setting webhook:', error.message || error.stack);
         process.exit(1); // Exit if critical error
       }