Harjutus 11: Vestlusruum serveri ja klientrakenduse vahel

Index.js

const http = require("http")
const fs = require("fs")
const ws = new require("ws")

const was = new ws.Server({noServer: true});
function accept (req, res) {
    if(req.url === "/ws" &&  
        req.headers.upgrade &&
        req.headers.update.toLowerCase() === "websocket" &&
        req.headers.connection.match(/\bupgrade\b/i)
        ){
            wss.handleUpdgrade(req, req.socket, Buffer.alloc(0), onSocketConnect);
        }
        else if (req.url === "/") {// index.html
            fs.createReadSream("./index.html").pipe(res);    
        }
        else {
            res.writeHead(404);
            res.end();
        }
}
//Kliendide loomine
const clients = new Set();

//Ühenduse loomine
function onSocketConnect(ws){
    clients.add(ws)
    ws.on('message', function(message){
        message = message.slice(0, 50); //maksimaalne pikkus on 50
        for(let client of clients) {client.send(message);}
    })

    ws.on('close', function(){
        console.log("connection closed");
        clients.delete(ws);
    })
}

let log;

if(!module["parent"]){
    log = console.log;
    http.createServer(accept).listen(8080);
} else {
    log = function() {};
    export.accept = accept;
}

Index.html

<!doctype html>
<form name="publish">
    <input type="text" name="message" maxlength="50"/>
    <input type="submit" value="Send"/>
</form>
 
<div id="messages"></div>
 
<script>
    let url = location.host == 'localhost' ?
        'ws://localhost:8080/ws' : location.host == 'javascript.local' ?
            `ws://javascript.local/article/websocket/chat/ws` : // dev integration with local site
            `wss://javascript.info/article/websocket/chat/ws`; // prod integration with javascript.info
    //создаем объект с вебсокетом
    let socket = new WebSocket(url);
 
    // отправляем сообщение на форму
    document.forms.publish.onsubmit = function() {
        let outgoingMessage = this.message.value;
 
        socket.send(outgoingMessage);
        return false;
    };
 
    // обрабатываем входящие сообщения
    socket.onmessage = function(event) {
        let incomingMessage = event.data;
        showMessage(incomingMessage);
    };
    //если пользователь закрыт сокет, то пишем об этом в консоли
    socket.onclose = event => console.log(`Closed ${event.code}`);
 
    // показываем сообщения в div#messages
    function showMessage(message) {
        let messageElem = document.createElement('div');
        messageElem.textContent = message;
        document.getElementById('messages').prepend(messageElem);
    }
</script>