1 files changed, 87 insertions, 0 deletions
diff --git a/index.js b/index.js
new file mode 100644
index 0000000..2043219
--- /dev/null
+++ b/index.js
@@ -0,0 +1,87 @@
+import { WebSocketServer } from "ws";
+import cuid from 'cuid';
+import { getRandomInt } from "./lib.js"
+
+const channels = ["home", "off-topic"]
+
+const ws = new WebSocketServer({
+    port: 9933,
+});
+
+const users = {}
+
+function sendInChannel(msg, channel) {
+    for (const userID in users) {
+        const user = users[userID];
+        if(user.channel == channel) user.socket.send(msg)
+    }
+}
+
+ws.on('connection', (socket, request) => {
+    // console.log(request, socket);
+    let userID = cuid()
+    socket.send("Welcome to WlodekM's wsChat server!")
+    let anonID = getRandomInt(0, 99999)
+    users[userID] = {
+        username: `Anonymous${"0".repeat(5 - anonID.length) + anonID.toString()}`,
+        socket: socket,
+        joinReq: request,
+        t: {
+            js: Number(new Date()),
+            unix: Math.floor(new Date().getTime() / 1000),
+            str: String(new Date())
+        },
+        channel: 'home'
+    }
+    sendInChannel(`${users[userID].username} joined #${users[userID].channel}!`, users[userID].channel)
+    socket.on('close', function(code, reason) {
+        sendInChannel(`${users[userID].username} left.`, users[userID].channel)
+        delete users[userID]
+    })
+    socket.on('message', function (rawData) {
+        if(rawData.toString().startsWith("/")) {
+            let args = String(rawData).replace("/", "").split(" ");
+            let command = args.shift();
+            console.log(`${users[userID].username} used /${command}`)
+            switch (command) {
+                case 'join':
+                    if(args.length < 1) return socket.send("Error: You need to specify a channel (example: /join #home).");
+                    if(!args[0].startsWith("#")) return socket.send("Error: Channel not found, run /channels to see a list of channels.");
+                    if(!channels.includes(args[0].replace("#", ""))) return socket.send("Error: Channel not found, run /channels to see a list of channels.");
+                    sendInChannel(`${users[userID].username} left #${users[userID].channel}.`, users[userID].channel)
+                    users[userID].channel = args[0].replace("#", "");
+                    sendInChannel(`${users[userID].username} joined #${users[userID].channel}!`, users[userID].channel)
+                    break;
+                case 'channels':
+                    socket.send(`Channels:\n${channels.map(ch => ` * #${ch}`).join("\n")}`)
+                    break;
+                case 'name':
+                case 'nickname':
+                case 'nick':
+                    if(args.length < 1) return socket.send("Error: You need to specify a nick (example: /nick WlodekM).");
+                    if(args[0].length < 3 ) return socket.send("Error: Nick too long.");
+                    if(args[0].length > 20) return socket.send("Error: Nick too short.");
+                    if(Object.values(users).find(usr => usr.username == args[0])) return socket.send("Error: Nick already used.");
+                    sendInChannel(`${users[userID].username} changed their nick to ${args[0]}!`, users[userID].channel)
+                    users[userID].username = args[0]
+                    break;
+                case 'users':
+                    socket.send(`Users${args[0] != "global" ? ` in ${users[userID].channel}` : ""}:\n${Object.values(users).filter(usr => (usr.channel == users[userID].channel) || args[0] == "global").map(ch => ` * ${ch}`).join("\n")}`)
+                    break;
+            
+                default:
+                    socket.send(`Error: Command "${command}" not found!`);
+                    break;
+            }
+            return
+        }
+        if(rawData.length < 1) return socket.send("Error: message too short!")
+        if(rawData.length >= 2000) return socket.send("Error: message too long!")
+        sendInChannel(`<${users[userID].username}> ${rawData}`, users[userID].channel)
+        console.log(`(#${users[userID].channel}) <${users[userID].username}> ${rawData}`)
+    })
+})
+
+ws.on('listening', () => {
+    console.info("[INFO] Server started")
+})