Development

docker/routlin-dash/app/session_interface.py

@@ -134,3 +134,97 @@ class SqliteSessionInterface(SessionInterface):
             secure=self.get_cookie_secure(app),
             samesite=self.get_cookie_samesite(app),
         )
+
+
+class SingleUserSessionInterface(SessionInterface):
+    def __init__(self, db_path):
+        self.db_path = db_path
+
+    def _connect(self):
+        con = sqlite3.connect(self.db_path, timeout=5)
+        con.execute('PRAGMA journal_mode=WAL')
+        con.row_factory = sqlite3.Row
+        return con
+
+    def open_session(self, app, request):
+        name = app.config.get('SESSION_COOKIE_NAME', 'session')
+        sid  = request.cookies.get(name)
+        if not sid:
+            return SqliteSession(sid=str(uuid.uuid4()), new=True)
+        try:
+            con = self._connect()
+            row = con.execute('SELECT * FROM sessions WHERE session_id=?', (sid,)).fetchone()
+            con.close()
+            if row:
+                prefs   = json.loads(row['preferences_json'] or '{}')
+                flashes = json.loads(row['flashes_json'] or '[]')
+                data = {'_flashes': flashes}
+                if row['logged_in']:
+                    tz = str(row['timezone'] or '') or _settings.get_host_timezone()
+                    data.update({
+                        'email_address':             _settings.get_initial_manager_email(),
+                        'access_level':              'manager',
+                        'timezone':                  tz,
+                        'apply_changes_immediately': bool(prefs.get('apply_changes_immediately', False)),
+                        '_permanent':                True,
+                        'logged_in':                 True,
+                    })
+                return SqliteSession(data, sid=sid, new=False)
+        except Exception:
+            pass
+        return SqliteSession(sid=sid, new=False)
+
+    def save_session(self, app, session, response):
+        name   = app.config.get('SESSION_COOKIE_NAME', 'session')
+        domain = self.get_cookie_domain(app)
+        path   = self.get_cookie_path(app)
+
+        if not session and session.modified and not session.new:
+            try:
+                con = self._connect()
+                con.execute('DELETE FROM sessions WHERE session_id=?', (session.sid,))
+                con.commit()
+                con.close()
+            except Exception:
+                pass
+            response.delete_cookie(name, domain=domain, path=path)
+            return
+
+        prefs        = json.dumps({
+            'apply_changes_immediately': bool(session.get('apply_changes_immediately', False)),
+        })
+        flashes_json = json.dumps(session.get('_flashes', []))
+        now          = int(time.time())
+        logged_in    = 1 if session.get('logged_in') else 0
+
+        try:
+            con = self._connect()
+            con.execute(
+                '''INSERT INTO sessions
+                   (session_id, logged_in, timezone, preferences_json,
+                    flashes_json, session_started_ts, last_seen_ts)
+                   VALUES (?,?,?,?,?,?,?)
+                   ON CONFLICT(session_id) DO UPDATE SET
+                       logged_in=excluded.logged_in,
+                       timezone=excluded.timezone,
+                       preferences_json=excluded.preferences_json,
+                       flashes_json=excluded.flashes_json,
+                       last_seen_ts=excluded.last_seen_ts''',
+                (session.sid, logged_in,
+                 session.get('timezone', ''),
+                 prefs, flashes_json, now, now)
+            )
+            con.commit()
+            con.close()
+        except Exception:
+            pass
+
+        response.set_cookie(
+            name, session.sid,
+            expires=self.get_expiration_time(app, session),
+            httponly=self.get_cookie_httponly(app),
+            domain=domain,
+            path=path,
+            secure=self.get_cookie_secure(app),
+            samesite=self.get_cookie_samesite(app),
+        )