feat: Enhance backend with request/SQL logging via morgan/winston and fix frontend by correctly rendering email Markdown content using the 'marked' library.

backend/app.js

@@ -1,13 +1,21 @@
 const express = require('express');
 const cors = require('cors');
+const http = require('http');
+const { WebSocketServer } = require('ws');
 const db = require('./db');
 const { SMTPServer } = require('smtp-server');
 const { saveEmail } = require('./saveEmail');
+const emitter = require('./eventEmitter');
+const logger = require('./logger');
+const morgan = require('morgan');
 
 const app = express();
 const apiPort = 5182;
 const smtpPort = 25;
 
+// Setup morgan to log HTTP requests to winston
+app.use(morgan('combined', { stream: { write: message => logger.info(message.trim()) } }));
+
 app.use(cors());
 app.use(express.json());
 
@@ -15,24 +23,136 @@ app.use(express.json());
 app.get('/api/messages', async (req, res) => {
   const { recipient } = req.query;
   if (!recipient) {
+    logger.warn('Attempted to get messages without a recipient');
     return res.status(400).send('Recipient is required');
   }
 
   try {
     const [rows] = await db.execute(
-      'SELECT id, sender, recipient, subject, body, received_at FROM emails WHERE recipient = ? ORDER BY received_at DESC',
-      [recipient]
+      'SELECT id, sender, recipient, subject, body, received_at FROM emails WHERE recipient LIKE ? ORDER BY received_at DESC',
+      [`%${recipient}%`]
     );
     res.json(rows);
   } catch (error) {
-    console.error('Failed to fetch emails:', error);
+    logger.error('Failed to fetch emails:', error);
     res.status(500).send('Failed to fetch emails');
   }
 });
 
+// API to get attachments for a message
+app.get('/api/messages/:id/attachments', async (req, res) => {
+  const { id } = req.params;
+  try {
+    const [rows] = await db.execute(
+      'SELECT id, filename, content_type FROM email_attachments WHERE email_id = ?',
+      [id]
+    );
+    res.json(rows);
+  } catch (error) {
+    logger.error('Failed to fetch attachments:', { error, emailId: id });
+    res.status(500).send('Failed to fetch attachments');
+  }
+});
+
+// API to download an attachment
+app.get('/api/attachments/:attachmentId', async (req, res) => {
+  const { attachmentId } = req.params;
+  try {
+    const [rows] = await db.execute(
+      'SELECT filename, content_type, content FROM email_attachments WHERE id = ?',
+      [attachmentId]
+    );
+    if (rows.length === 0) {
+      return res.status(404).send('Attachment not found');
+    }
+    const attachment = rows[0];
+    res.setHeader('Content-Disposition', `attachment; filename="${attachment.filename}"`);
+    res.setHeader('Content-Type', attachment.content_type);
+    res.send(attachment.content);
+  } catch (error) {
+    logger.error('Failed to download attachment:', { error, attachmentId });
+    res.status(500).send('Failed to download attachment');
+  }
+});
+
+// API to delete a message
+app.delete('/api/messages/:id', async (req, res) => {
+  const { id } = req.params;
+  try {
+    const [result] = await db.execute('DELETE FROM emails WHERE id = ?', [id]);
+    if (result.affectedRows === 0) {
+      logger.warn('Attempted to delete a message that was not found', { id });
+      return res.status(404).send('Message not found');
+    }
+    logger.info('Message deleted successfully', { id });
+    res.status(204).send(); // No Content
+  } catch (error) {
+    logger.error('Failed to delete message:', { error, id });
+    res.status(500).send('Failed to delete message');
+  }
+});
+
+// API to delete all messages for a recipient
+app.delete('/api/messages', async (req, res) => {
+  const { recipient } = req.query;
+  if (!recipient) {
+    logger.warn('Attempted to delete all messages without a recipient');
+    return res.status(400).send('Recipient is required');
+  }
+  try {
+    await db.execute('DELETE FROM emails WHERE recipient = ?', [recipient]);
+    logger.info('All messages for recipient deleted successfully', { recipient });
+    res.status(204).send(); // No Content
+  } catch (error) {
+    logger.error('Failed to delete all messages:', { error, recipient });
+    res.status(500).send('Failed to delete all messages');
+  }
+});
+
+// Create HTTP server
+const server = http.createServer(app);
+
+// Create WebSocket server
+const wss = new WebSocketServer({ server });
+
+// Map to store connections for each recipient
+const clients = new Map();
+
+wss.on('connection', (ws, req) => {
+  const url = new URL(req.url, `http://${req.headers.host}`);
+  const recipient = url.searchParams.get('recipient');
+
+  if (recipient) {
+    logger.info(`WebSocket client connected`, { recipient });
+    clients.set(recipient, ws);
+
+    ws.on('close', () => {
+      logger.info(`WebSocket client disconnected`, { recipient });
+      clients.delete(recipient);
+    });
+
+    ws.on('error', (error) => {
+      logger.error(`WebSocket error`, { recipient, error });
+      clients.delete(recipient);
+    });
+  } else {
+    logger.warn('WebSocket client connected without recipient, closing.');
+    ws.close();
+  }
+});
+
+emitter.on('newEmail', (email) => {
+  const recipient = email.recipient;
+  if (clients.has(recipient)) {
+    const ws = clients.get(recipient);
+    logger.info(`Sending new email notification to`, { recipient });
+    ws.send(JSON.stringify(email));
+  }
+});
+
 // Start API server
-app.listen(apiPort, () => {
-  console.log(`Backend API server listening at http://localhost:${apiPort}`);
+server.listen(apiPort, () => {
+  logger.info(`Backend API and WebSocket server listening at http://localhost:${apiPort}`);
 });
 
 // Configure and start SMTP server
@@ -40,23 +160,23 @@ const smtpServer = new SMTPServer({
   authOptional: true,
   disabledCommands: ['AUTH'],
   onData(stream, session, callback) {
-    console.log('Receiving email...');
+    logger.info('Receiving email...', { session });
     saveEmail(stream)
       .then(() => {
-        console.log('Email processed and saved successfully.');
+        logger.info('Email processed and saved successfully.');
         callback(); // Accept the message
       })
       .catch(err => {
-        console.error('Error processing email:', err);
+        logger.error('Error processing email:', err);
         callback(new Error('Failed to process email.'));
       });
   },
 });
 
 smtpServer.on('error', err => {
-  console.error('SMTP Server Error:', err.message);
+  logger.error('SMTP Server Error:', err);
 });
 
 smtpServer.listen(smtpPort, () => {
-  console.log(`SMTP server listening on port ${smtpPort}`);
+  logger.info(`SMTP server listening on port ${smtpPort}`);
 });