mirror of
https://github.com/fusion44/blitz_api.git
synced 2026-08-14 12:02:46 +02:00
25e4be8441
If the environment variable BAPI_ENV_PATH is set, the config system will try to read configs from the given path instead of .env in pwd If no file is found .env in pwd will be used as a fallback Env variables will always override settings in .env files.
75 lines
2 KiB
Python
75 lines
2 KiB
Python
import asyncio
|
|
import os
|
|
import time
|
|
from typing import Dict
|
|
|
|
import jwt
|
|
from loguru import logger
|
|
|
|
from app.api.config import config
|
|
|
|
JWT_SECRET = config("BAPI_JWT_SECRET")
|
|
JWT_ALGORITHM = config("BAPI_JWT_ALGORITHM")
|
|
JWT_EXPIRY_TIME = config("BAPI_JWT_EXPIRY_TIME", default=300, cast=int)
|
|
|
|
|
|
def sign_jwt() -> Dict[str, str]:
|
|
payload = {
|
|
"user_id": "admin",
|
|
"expires": int(round(time.time() * 1000) + JWT_EXPIRY_TIME),
|
|
}
|
|
token = jwt.encode(payload, JWT_SECRET, algorithm=JWT_ALGORITHM)
|
|
return token
|
|
|
|
|
|
def decodeJWT(token: str) -> dict:
|
|
try:
|
|
decoded_token = jwt.decode(token, JWT_SECRET, algorithms=[JWT_ALGORITHM])
|
|
return decoded_token if decoded_token["expires"] >= time.time() * 1000 else None
|
|
except Exception as e:
|
|
logger.warning(f"Unable to decode jwt_token {e}")
|
|
return {}
|
|
|
|
|
|
def handle_local_cookie():
|
|
remove_local_cookie()
|
|
|
|
blitz_path = os.path.join(os.path.expanduser("~"), ".blitz_api")
|
|
full_cookie_file_path = os.path.join(blitz_path, ".cookie")
|
|
enabled = config("BAPI_ENABLE_LOCAL_COOKIE_AUTH", default=False, cast=bool)
|
|
|
|
if not enabled:
|
|
return
|
|
|
|
if not os.path.exists(blitz_path):
|
|
try:
|
|
os.makedirs(blitz_path)
|
|
except OSError as e:
|
|
logger.error(
|
|
f"""Unable to create the .blit_api folder: {e}
|
|
Please make sure that the target folder is readable.
|
|
"""
|
|
)
|
|
f = open(full_cookie_file_path, "w")
|
|
f.write(sign_jwt())
|
|
f.close()
|
|
|
|
|
|
def remove_local_cookie():
|
|
full_cookie_file_path = os.path.join(
|
|
os.path.expanduser("~"), ".blitz_api", ".cookie"
|
|
)
|
|
|
|
if os.path.exists(path=full_cookie_file_path):
|
|
os.remove(full_cookie_file_path)
|
|
|
|
|
|
def register_cookie_updater():
|
|
# We need to update the cookie file once the cookie is expired
|
|
async def _cookie_updater():
|
|
while True:
|
|
await asyncio.sleep(JWT_EXPIRY_TIME - 10)
|
|
handle_local_cookie()
|
|
|
|
loop = asyncio.get_event_loop()
|
|
loop.create_task(_cookie_updater())
|