add: added image attachments to messages

back/api/attachments.js

@@ -0,0 +1,22 @@
+const express = require('express');
+const { getConnection, getAttachment } = require('../libs/mysql');
+const fs = require('node:fs');
+const path = require('node:path');
+
+const router = express.Router();
+
+router.get('/:file_name', async (req, res) => {
+    const { file_name } = req.params;
+
+    const connection = await getConnection();
+    const attachment = await getAttachment(connection, file_name);
+    connection.end();
+
+    if (!attachment[0]) {
+        return res.status(404).send({ error: 'File not found' });
+    }
+
+    res.sendFile(path.join(__dirname, `../data/attachments/${attachment[0].file_name}`), { headers: { 'Content-Type': 'image' } });
+})
+
+module.exports = router;

back/api/channels.js

@@ -11,14 +11,19 @@ const {
     getUserByUsername,
     deleteChannelMessages,
     deleteChannel,
-    getMessageReplies
+    getMessageReplies,
+    addAttachment,
+    getMessageAttachments,
+    getUnusedAttachments,
+    deleteUnusedAttachments
 } = require('../libs/mysql');
 const rateLimit = require("express-rate-limit");
 const slowDown = require("express-slow-down");
 const { checkAuth } = require('../libs/middlewares');
 const multer = require('multer');
+const fs = require('node:fs');
 
-const upload = multer({ dest: 'data/attachements/' })
+const upload = multer({ dest: 'data/attachments/' })
 upload.limits = {
     fileSize: 1024 * 1024 * 5,
     files: 1,
@@ -104,6 +109,15 @@ router.post('/:name/delete', async (req, res) => {
     }
 
     await deleteChannel(connection, channel[0].id);
+
+    const attachments = await getUnusedAttachments(connection);
+    for (const attachment of attachments) {
+        if (fs.existsSync(`data/attachments/${attachment.file_name}`)) {
+            fs.unlinkSync(`data/attachments/${attachment.file_name}`);
+        }
+    }
+    await deleteUnusedAttachments(connection);
+
     connection.end();
 
     req.sockets.emit({
@@ -143,7 +157,7 @@ router.get('/:name/messages', async (req, res) => {
     const messages = await getMessages(connection, channel[0].id, limit);
 
     for (const message of messages) {
-        if (message.content.includes('@')) {
+        if (message.has_mentions) {
             const mentions = await getMentions(connection, message.id);
             message.mentions = mentions;
         } else {
@@ -157,18 +171,28 @@ router.get('/:name/messages', async (req, res) => {
         else {
             message.replies = [];
         }
+
+        if (message.has_attachments) {
+            const attachments = await getMessageAttachments(connection, message.id);
+            message.attachments = attachments;
+        }
+        else {
+            message.attachments = [];
+        }
     }
     
     connection.end();
     res.send(messages);
 });
 
-router.post('/:name/messages/send', speedLimiter, limiter, upload.single("attachement"), checkAuth, async (req, res) => {
+router.post('/:name/messages/send', speedLimiter, limiter, upload.single("attachment"), checkAuth, async (req, res) => {
     const { message } = req.body;
     const name = req.params.name;
     const user = req.user;
+    const attachement = req.file;
 
     if (!message) {
+        if (attachement) fs.unlinkSync(`data/attachements/${attachement.filename}`);
         return res.status(400).send({ error: 'Missing parameters' });
     }
 
@@ -177,12 +201,17 @@ router.post('/:name/messages/send', speedLimiter, limiter, upload.single("attach
     const channel = await getChannel(connection, name);
     if (!channel[0]) {
         connection.end();
+        if (attachement) fs.unlinkSync(`data/attachements/${attachement.filename}`);
         return res.send('No channel found');
     }
 
     const sent_message = await addMessage(connection, channel[0].id, user.id, message.replace("\"", "'"));
     const message_id = sent_message.insertId;
 
+    if (attachement) {
+        await addAttachment(connection, message_id, attachement.filename);
+    }
+
     for (const word of message.split(' ')) {
         if (word.startsWith('@')) {
             const username = word.substring(1);

back/api/lastmessages.js

@@ -1,5 +1,5 @@
 const express = require('express');
-const { getConnection, getLastMessages, getMentions } = require('../libs/mysql');
+const { getConnection, getLastMessages, getMentions, getMessageAttachments } = require('../libs/mysql');
 
 const router = express.Router();
 
@@ -15,6 +15,12 @@ router.get('/', async (req, res) => {
             message.mentions = [];
         }
         message.replies = [];
+        if (message.has_attachments) {
+            const attachments = await getMessageAttachments(connection, message.id);
+            message.attachments = attachments;
+        } else {
+            message.attachments = [];
+        }
     }
 
     connection.end();

back/api/messages.js

@@ -8,11 +8,14 @@ const {
     getMentions,
     getUserByUsername,
     addReply,
-    getMessageReplies
+    getMessageReplies,
+    getUnusedAttachments,
+    deleteUnusedAttachments
 } = require('../libs/mysql');
 const rateLimit = require("express-rate-limit");
 const slowDown = require("express-slow-down");
 const { checkAuth } = require('../libs/middlewares');
+const fs = require('node:fs');
 
 const limiter = rateLimit({
     windowMs: 1 * 1000,
@@ -132,6 +135,15 @@ router.post('/:message_id/delete', checkAuth, async (req, res) => {
     }
 
     await deleteMessage(connection, message_id);
+
+    const attachments = await getUnusedAttachments(connection);
+    for (const attachment of attachments) {
+        if (fs.existsSync(`data/attachments/${attachment.file_name}`)) {
+            fs.unlinkSync(`data/attachments/${attachment.file_name}`);
+        }
+    }
+    await deleteUnusedAttachments(connection);
+
     connection.end();
 
     req.sockets.emit({

back/api/users.js

@@ -1,5 +1,5 @@
 const express = require('express');
-const { getConnection, getUsers, getUserByUsername, getUserLastMessages, getMentions, deleteUser, deleteUserMessages, deleteUserMentions, setUserPfp } = require('../libs/mysql');
+const { getConnection, getUsers, getUserByUsername, getUserLastMessages, getMentions, deleteUser, setUserPfp, getMessageAttachments, getUnusedAttachments, deleteUnusedAttachments } = require('../libs/mysql');
 const { checkAuth } = require("../libs/middlewares")
 const path = require('path');
 const fs = require('node:fs');
@@ -38,6 +38,12 @@ router.get('/:username/lastmessages', async (req, res) => {
             message.mentions = [];
         }
         message.replies = [];
+        if (message.has_attachments) {
+            const attachments = await getMessageAttachments(connection, message.id);
+            message.attachments = attachments;
+        } else {
+            message.attachments = [];
+        }
     }
 
     connection.end();
@@ -87,6 +93,14 @@ router.post('/:username/delete', checkAuth, async (req, res) => {
 
     await deleteUser(connection, userToDelete[0].id);
 
+    const attachments = await getUnusedAttachments(connection);
+    for (const attachment of attachments) {
+        if (fs.existsSync(`data/attachments/${attachment.file_name}`)) {
+            fs.unlinkSync(`data/attachments/${attachment.file_name}`);
+        }
+    }
+    await deleteUnusedAttachments(connection);
+
     connection.end();
 
     req.sockets.emit({