server.js

const express = require('express');
const http = require('http');
const { Server } = require('socket.io');
const multer = require('multer');
const path = require('path');
const fs = require('fs');
const os = require('os');

const app = express();
const server = http.createServer(app);
const io = new Server(server);

const PORT = 3002;

// In-memory storage for messages per room
const rooms = {};

// Create uploads directory in the user's home directory
const uploadsDir = path.join(os.homedir(), 'messager-uploads');
if (!fs.existsSync(uploadsDir)) {
  fs.mkdirSync(uploadsDir, { recursive: true });
}

// Multer setup for image uploads
const upload = multer({ dest: uploadsDir });

// Serve static files from the public directory
app.use(express.static(path.join(__dirname, 'public')));

// Serve uploaded images statically
app.use('/uploads', express.static(uploadsDir));

// Endpoint for image uploads
app.post('/upload', upload.single('image'), (req, res) => {
  if (!req.file) return res.status(400).send('No file uploaded.');
  res.json({ url: `/uploads/${req.file.filename}` });
});

// Socket.io logic
io.on('connection', (socket) => {
  let currentRoom = null;
  let userName = null;

  socket.on('joinRoom', ({ room, name }) => {
    currentRoom = room;
    userName = name;
    socket.join(room);
    if (!rooms[room]) rooms[room] = [];
    // Send existing messages to the new user
    socket.emit('chatHistory', rooms[room]);
  });

  socket.on('chatMessage', (msg) => {
    if (!currentRoom || !userName) return;
    const message = { name: userName, text: msg, type: 'text', time: Date.now() };
    rooms[currentRoom].push(message);
    io.to(currentRoom).emit('chatMessage', message);
  });

  socket.on('imageMessage', (imgUrl) => {
    if (!currentRoom || !userName) return;
    const message = { name: userName, image: imgUrl, type: 'image', time: Date.now() };
    rooms[currentRoom].push(message);
    io.to(currentRoom).emit('chatMessage', message);
  });
});

// Serve room.html for /room/:code
app.get('/room/:code', (req, res) => {
  res.sendFile(path.join(__dirname, 'public', 'room.html'));
});

server.listen(PORT, '0.0.0.0', () => {
  console.log(`Server running at http://localhost:${PORT}`);
});