generate session cookie

owrx/controllers/session.py

@@ -1,10 +1,41 @@
 from .template import WebpageController
 from urllib.parse import parse_qs
+from uuid import uuid4
+from http.cookies import SimpleCookie
 import logging
 
 logger = logging.getLogger(__name__)
 
 
+class SessionStorage(object):
+    sharedInstance = None
+
+    @staticmethod
+    def getSharedInstance():
+        if SessionStorage.sharedInstance is None:
+            SessionStorage.sharedInstance = SessionStorage()
+        return SessionStorage.sharedInstance
+
+    def __init__(self):
+        self.sessions = {}
+
+    def generateKey(self):
+        return str(uuid4())
+
+    def startSession(self, data):
+        key = self.generateKey()
+        self.updateSession(key, data)
+        return key
+
+    def getSession(self, key):
+        if key not in self.sessions:
+            return None
+        return self.sessions[key]
+
+    def updateSession(self, key, data):
+        self.sessions[key] = data
+
+
 class SessionController(WebpageController):
     def loginAction(self):
         self.serve_template("login.html", **self.template_variables())
@@ -17,8 +48,10 @@ class SessionController(WebpageController):
             # TODO actually check user and password
             if data["user"] == "admin" and data["password"] == "password":
                 # TODO pass the final destination
-                # TODO actual session cookie
-                self.send_redirect("/settings", cookies=["session-cookie"])
+                key = SessionStorage.getSharedInstance().startSession({"user": data["user"]})
+                cookie = SimpleCookie()
+                cookie["session"] = key
+                self.send_redirect("/settings", cookies=cookie)
             else:
                 self.send_redirect("/login")
         else: