33 lines
905 B
Python
33 lines
905 B
Python
|
from time import time
|
||
|
|
|
||
|
|
import jwt
|
||
|
|
|
||
|
|
from .config import derived_secret
|
||
|
|
|
||
|
|
session_secret = derived_secret("session")
|
||
|
|
max_age = 60 # Seconds since last login
|
||
|
|
|
||
|
|
def get(request):
|
||
|
|
try:
|
||
|
|
return jwt.decode(request.cookies.s, session_secret, algorithms=["HS256"])
|
||
|
|
except Exception as e:
|
||
|
|
s = None
|
||
|
|
return False if "s" in request.cookies else None
|
||
|
|
|
||
|
|
def create(res, username, **kwargs):
|
||
|
|
data = {
|
||
|
|
"exp": int(time()) + max_age,
|
||
|
|
"username": username,
|
||
|
|
**kwargs,
|
||
|
|
}
|
||
|
|
s = jwt.encode(data, session_secret)
|
||
|
|
res.cookies.add_cookie("s", s, host_prefix=True, httponly=True, max_age=max_age)
|
||
|
|
|
||
|
|
def update(res, s, **kwargs):
|
||
|
|
s.update(kwargs)
|
||
|
|
s = jwt.encode(s, session_secret)
|
||
|
|
res.cookies.add_cookie("s", s, host_prefix=True, httponly=True, max_age=max(1, s["exp"] - int(time())))
|
||
|
|
|
||
|
|
def delete(res):
|
||
|
|
res.cookies.delete_cookie("s", host_prefix=True)
|