chatroom

Author: csxiaoyaojianxian

.gitignore

@@ -53,4 +53,6 @@ Desktop.ini
 *.msp
 
 # Windows shortcuts
-*.lnk
+*.lnk
+
+node_modules/

11-nodejs/koa2/8-chatroom/app.js

@@ -0,0 +1,180 @@
+/*
+ * @Author: csxiaoyao 
+ * @Date: 2018-04-18 17:53:40 
+ * @Last Modified by: csxiaoyao
+ * @Last Modified time: 2018-04-18 23:18:48
+ */
+
+const url = require('url');
+const ws = require('ws');
+const Cookies = require('cookies');
+const Koa = require('koa');
+const bodyParser = require('koa-bodyparser');
+const controller = require('./controller');
+
+const WebSocketServer = ws.Server;
+
+const app = new Koa();
+
+// log request URL:
+app.use(async (ctx, next) => {
+    console.log(`Process ${ctx.request.method} ${ctx.request.url}...`);
+    await next();
+});
+
+// parse user from cookie:
+app.use(async (ctx, next) => {
+    ctx.state.user = parseUser(ctx.cookies.get('name') || '');
+    await next();
+});
+
+// parse request body:
+app.use(bodyParser());
+
+// add controller middleware:
+app.use(controller());
+
+let server = app.listen(3000);
+
+function parseUser(obj) {
+    if (!obj) {
+        return;
+    }
+    console.log('try parse: ' + obj);
+    let s = '';
+    if (typeof obj === 'string') {
+        s = obj;
+    } else if (obj.headers) {
+        let cookies = new Cookies(obj, null);
+        s = cookies.get('name');
+    }
+    if (s) {
+        try {
+            let user = JSON.parse(Buffer.from(s, 'base64').toString());
+            console.log(`User: ${user.name}, ID: ${user.id}`);
+            return user;
+        } catch (e) {
+            // ignore
+        }
+    }
+}
+
+function createWebSocketServer(server, onConnection, onMessage, onClose, onError) {
+    let wss = new WebSocketServer({
+        server: server
+    });
+    wss.broadcast = function broadcast(data) {
+        wss.clients.forEach(function each(client) {
+            client.send(data);
+        });
+    };
+    onConnection = onConnection || function () {
+        console.log('[WebSocket] connected.');
+    };
+    onMessage = onMessage || function (msg) {
+        console.log('[WebSocket] message received: ' + msg);
+    };
+    onClose = onClose || function (code, message) {
+        console.log(`[WebSocket] closed: ${code} - ${message}`);
+    };
+    onError = onError || function (err) {
+        console.log('[WebSocket] error: ' + err);
+    };
+    wss.on('connection', function (ws,req) {
+        ws.upgradeReq = req;
+        let location = url.parse(ws.upgradeReq.url, true);
+        console.log('[WebSocketServer] connection: ' + location.href);
+        ws.on('message', onMessage);
+        ws.on('close', onClose);
+        ws.on('error', onError);
+        if (location.pathname !== '/ws/chat') {
+            // close ws:
+            ws.close(4000, 'Invalid URL');
+        }
+        // check user:
+        let user = parseUser(ws.upgradeReq);
+        if (!user) {
+            ws.close(4001, 'Invalid user');
+        }
+        ws.user = user;
+        ws.wss = wss;
+        onConnection.apply(ws);
+    });
+    console.log('WebSocketServer was attached.');
+    return wss;
+}
+
+var messageIndex = 0;
+
+function createMessage(type, user, data) {
+    messageIndex++;
+    return JSON.stringify({
+        id: messageIndex,
+        type: type,
+        user: user,
+        data: data
+    });
+}
+
+function onConnect() {
+    let user = this.user;
+    let msg = createMessage('join', user, `${user.name} joined.`);
+    this.wss.broadcast(msg);
+    // build user list:
+    let users = this.wss.clients.forEach(function (client) {
+        return client.user;
+    });
+    console.log(users);
+    this.send(createMessage('list', user, users));
+}
+
+function onMessage(message) {
+    console.log(message);
+    if (message && message.trim()) {
+        let msg = createMessage('chat', this.user, message.trim());
+        this.wss.broadcast(msg);
+    }
+}
+
+function onClose() {
+    let user = this.user;
+    let msg = createMessage('left', user, `${user.name} is left.`);
+    this.wss.broadcast(msg);
+}
+
+app.wss = createWebSocketServer(server, onConnect, onMessage, onClose);
+
+console.log('app started at port 3000...');
+
+
+
+/*
+配置nginx反向代理
+
+server {
+    listen      80;
+    server_name localhost;
+
+    # 处理静态资源文件:
+    location ^~ /static/ {
+        root /path/to/ws-with-koa;
+    }
+
+    # 处理WebSocket连接:
+    location ^~ /ws/ {
+        proxy_pass         http://127.0.0.1:3000;
+        proxy_http_version 1.1;
+        proxy_set_header   Upgrade $http_upgrade;
+        proxy_set_header   Connection "upgrade";
+    }
+
+    # 其他所有请求:
+    location / {
+        proxy_pass       http://127.0.0.1:3000;
+        proxy_set_header X-Real-IP $remote_addr;
+        proxy_set_header Host $host;
+        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
+    }
+}
+
+*/

11-nodejs/koa2/8-chatroom/controller.js

@@ -0,0 +1,47 @@
+
+const fs = require('fs');
+
+// add url-route in /controllers:
+
+function addMapping(router, mapping) {
+    for (var url in mapping) {
+        var path = '';
+        if (url.startsWith('GET ')) {
+            path = url.substring(4);
+            router.get(path, mapping[url]);
+            console.log(`register URL mapping: GET ${path}`);
+        } else if (url.startsWith('POST ')) {
+            path = url.substring(5);
+            router.post(path, mapping[url]);
+            console.log(`register URL mapping: POST ${path}`);
+        } else if (url.startsWith('PUT ')) {
+            path = url.substring(4);
+            router.put(path, mapping[url]);
+            console.log(`register URL mapping: PUT ${path}`);
+        } else if (url.startsWith('DELETE ')) {
+            path = url.substring(7);
+            router.del(path, mapping[url]);
+            console.log(`register URL mapping: DELETE ${path}`);
+        } else {
+            console.log(`invalid URL: ${url}`);
+        }
+    }
+}
+
+function addControllers(router, dir) {
+    fs.readdirSync(__dirname + '/' + dir).filter((f) => {
+        return f.endsWith('.js');
+    }).forEach((f) => {
+        console.log(`process controller: ${f}...`);
+        let mapping = require(__dirname + '/' + dir + '/' + f);
+        addMapping(router, mapping);
+    });
+}
+
+module.exports = function (dir) {
+    let
+        controllers_dir = dir || 'controllers',
+        router = require('koa-router')();
+    addControllers(router, controllers_dir);
+    return router.routes();
+};

11-nodejs/koa2/8-chatroom/controllers/index.js

@@ -0,0 +1,11 @@
+
+module.exports = {
+    'GET /': async (ctx, next) => {
+        let user = ctx.state.user;
+        if (user) {
+            console.log("response GET / : " + user);
+        } else {
+            ctx.response.redirect('/signin');
+        }
+    }
+};

11-nodejs/koa2/8-chatroom/controllers/signin.js

@@ -0,0 +1,24 @@
+// sign in:
+
+module.exports = {
+    'GET /signin': async (ctx, next) => {
+        console.log('GET /signin');
+    },
+
+    'POST /signin': async (ctx, next) => {
+        console.log('POST /signin');
+        let name = ctx.request.body.name || 'csxiaoyao';
+        let user = {
+            name: name
+        };
+        let value = Buffer.from(JSON.stringify(user)).toString('base64');
+        console.log(`Set cookie value: ${value}`);
+        ctx.cookies.set('name', value);
+        ctx.response.redirect('/');
+    },
+
+    'GET /signout': async (ctx, next) => {
+        ctx.cookies.set('name', '');
+        ctx.response.redirect('/signin');
+    }
+};