Update project structure

Author: imreyesjorge

package.json

@@ -4,7 +4,7 @@
   "description": "",
   "main": "index.js",
   "scripts": {
-    "dev": "nodemon --watch src/**/*.ts --exec 'ts-node src/index.ts'",
+    "dev": "nodemon src/index.ts",
     "build": "tsc",
     "start": "node dist/index.js"
   },

src/index.ts

@@ -1,41 +1,24 @@
 import express from "express";
 import { createServer } from "http";
 import { Server } from "socket.io";
+import { socketHandler } from "./sockets";
 
 const app = express();
 const server = createServer(app);
-const io = new Server(server, {
+const port = process.env.PORT || 3001;
+
+export const io = new Server(server, {
   cors: {
     origin: "*",
   },
 });
-const port = process.env.PORT || 3001;
 
-let onlineUsers = 0;
+io.on("connection", socketHandler);
 
 app.get("/", (req, res) => {
   res.send({ message: "Server is running." });
 });
 
-io.on("connection", (socket) => {
-  onlineUsers += 1;
-  io.emit("onlineUsers", onlineUsers);
-
-  socket.on("message", (msg) => {
-    io.emit(
-      "newMessage",
-      socket.handshake.auth.username,
-      msg,
-      new Date().toLocaleTimeString()
-    );
-  });
-
-  socket.on("disconnect", () => {
-    onlineUsers -= 1;
-    io.emit("onlineUsers", onlineUsers);
-  });
-});
-
 server.listen(port, () => {
   console.log(`Server running on port ${port}`);
 });

src/sockets/index.ts

@@ -0,0 +1,23 @@
+import { Socket } from "socket.io";
+import { io } from "..";
+
+let onlineUsers = 0;
+
+export function socketHandler(socket: Socket) {
+  onlineUsers += 1;
+  io.emit("onlineUsers", onlineUsers);
+
+  socket.on("message", (msg) => {
+    io.emit(
+      "newMessage",
+      socket.handshake.auth.username,
+      msg,
+      new Date().getTime()
+    );
+  });
+
+  socket.on("disconnect", () => {
+    onlineUsers -= 1;
+    io.emit("onlineUsers", onlineUsers);
+  });
+}