From 22af5b8432b1f229d8a4a29e5ee46b1fe0c62bba Mon Sep 17 00:00:00 2001
From: Florent Daigniere <nextgens@freenetproject.org>
Date: Sat, 20 Feb 2021 19:10:20 +0100
Subject: [PATCH] Switch to server-side sessions in redis

---
 core/admin/mailu/__init__.py     | 4 ++++
 core/admin/requirements-prod.txt | 1 +
 core/admin/requirements.txt      | 1 +
 3 files changed, 6 insertions(+)

diff --git a/core/admin/mailu/__init__.py b/core/admin/mailu/__init__.py
index 4de3e580..4d7efba1 100644
--- a/core/admin/mailu/__init__.py
+++ b/core/admin/mailu/__init__.py
@@ -1,5 +1,8 @@
 import flask
 import flask_bootstrap
+import redis
+from flask_kvsession import KVSessionExtension
+from simplekv.memory.redisstore import RedisStore
 
 from mailu import utils, debug, models, manage, configuration
 
@@ -17,6 +20,7 @@ def create_app_from_config(config):
     # Initialize application extensions
     config.init_app(app)
     models.db.init_app(app)
+    KVSessionExtension(RedisStore(redis.StrictRedis().from_url('redis://{0}/3'.format(config['REDIS_ADDRESS']))), app)
     utils.limiter.init_app(app)
     utils.babel.init_app(app)
     utils.login.init_app(app)
diff --git a/core/admin/requirements-prod.txt b/core/admin/requirements-prod.txt
index f767f431..54cf9a14 100644
--- a/core/admin/requirements-prod.txt
+++ b/core/admin/requirements-prod.txt
@@ -13,6 +13,7 @@ Flask==1.0.2
 Flask-Babel==0.12.2
 Flask-Bootstrap==3.3.7.1
 Flask-DebugToolbar==0.10.1
+Flask-KVSession==0.6.2
 Flask-Limiter==1.0.1
 Flask-Login==0.4.1
 Flask-Migrate==2.4.0
diff --git a/core/admin/requirements.txt b/core/admin/requirements.txt
index 9739ed3f..abb37234 100644
--- a/core/admin/requirements.txt
+++ b/core/admin/requirements.txt
@@ -3,6 +3,7 @@ Flask-Login
 Flask-SQLAlchemy
 Flask-bootstrap
 Flask-Babel
+Flask-KVSession
 Flask-migrate
 Flask-script
 Flask-wtf