Implement restore session and check session

Author: tshemsedinov

JavaScript/client.js

@@ -1,5 +1,7 @@
 'use strict';
 
+const Session = require('./session.js');
+
 const UNIX_EPOCH = 'Thu, 01 Jan 1970 00:00:00 GMT';
 const COOKIE_EXPIRE = 'Fri, 01 Jan 2100 00:00:00 GMT';
 const COOKIE_DELETE = `=deleted; Expires=${UNIX_EPOCH}; Path=/; Domain=`;
@@ -21,6 +23,7 @@ class Client {
     this.cookie = {};
     this.preparedCookie = [];
     this.parseCookie();
+    Session.restore(this);
   }
 
   parseCookie() {

JavaScript/server.js

@@ -11,7 +11,11 @@ const routing = {
     return `Session token is: ${client.token}`;
   },
   '/api/method1': async client => {
-    return { data: 'example result' };
+    if (client.session) {
+      return { data: 'example result' };
+    } else {
+      return { data: 'access is denied' };
+    }
   },
   '/api/method2': async client => ({
     url: client.req.url,
@@ -44,7 +48,11 @@ http.createServer((req, res) => {
         res.end(result);
       }, err => {
         console.error(err.stack);
+        res.statusCode = 500;
         res.end('Internal Server Error 500');
       });
+    return;
   }
+  res.statusCode = 404;
+  res.end('Not found 404');
 }).listen(8000);

JavaScript/session.js

@@ -26,22 +26,28 @@ class Session extends Map {
   }
 
   static start(client) {
+    if (client.session) return client.session;
+    const token = generateToken();
+    client.token = token;
+    const session = new Session(token);
+    client.session = session;
+    client.setCookie('token', token);
+    sessions.set(token, session);
+    return session;
+  }
+
+  static restore(client) {
     const { cookie } = client;
     const sessionToken = cookie.token;
+    console.log({ sessionToken });
     if (sessionToken) {
       const session = sessions.get(sessionToken);
       if (session) {
+        client.token = sessionToken;
         client.session = session;
         return session;
       }
     }
-    const token = generateToken();
-    client.token = token;
-    const session = new Session(token);
-    client.session = session;
-    client.setCookie('token', token);
-    sessions.set(token, session);
-    return session;
   }
 
   static drop(client) {