125 lines
3.9 KiB
Python
125 lines
3.9 KiB
Python
import logging
|
|
import os
|
|
|
|
from fastapi import FastAPI, Request
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from fastapi.responses import (
|
|
RedirectResponse,
|
|
)
|
|
from infoserver.dependencies import Dependencies
|
|
from infoserver.routers.router_index import router_index
|
|
from infoserver.routers.router_login import router_login
|
|
from infoserver.routers.router_pdf_preso import router_pdf
|
|
from infoserver.routers.router_static import router_static
|
|
from infoserver.routers.router_template import router_template
|
|
from jwt.exceptions import PyJWTError
|
|
from starlette.middleware.sessions import SessionMiddleware
|
|
|
|
logging.basicConfig(level=logging.DEBUG)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# Set your paths here
|
|
DB_PATH = '~/code/git.ourworld.tf/freeflowuniverse/heroweb/authdb_example'
|
|
TEMPLATES_DIR = '~/code/git.ourworld.tf/freeflowuniverse/heroweb/poc/out'
|
|
STATIC_DIR = '~/code/git.ourworld.tf/freeflowuniverse/heroweb/poc/static'
|
|
HEROWEB_DIR = '~/code/git.ourworld.tf/tfgrid/info_tfgrid/heroweb'
|
|
COLLECTIONS_DIR = '~/hero/var/collections'
|
|
|
|
DB_PATH = os.path.abspath(os.path.expanduser(DB_PATH))
|
|
TEMPLATES_DIR = os.path.abspath(os.path.expanduser(TEMPLATES_DIR))
|
|
STATIC_DIR = os.path.abspath(os.path.expanduser(STATIC_DIR))
|
|
STATIC_DIR2 = os.path.abspath(os.path.join(os.path.dirname(__file__), 'static'))
|
|
HEROWEB_DIR = os.path.abspath(os.path.expanduser(HEROWEB_DIR))
|
|
COLLECTIONS_DIR = os.path.abspath(os.path.expanduser(COLLECTIONS_DIR))
|
|
|
|
SERVERHOST = 'http://localhost:8000'
|
|
|
|
if not os.path.exists(DB_PATH):
|
|
raise FileNotFoundError(f'Database path does not exist: {DB_PATH}')
|
|
if not os.path.exists(TEMPLATES_DIR):
|
|
raise FileNotFoundError(
|
|
f'Templates directory does not exist: {TEMPLATES_DIR}'
|
|
)
|
|
if not os.path.exists(STATIC_DIR):
|
|
raise FileNotFoundError(f'Static directory does not exist: {STATIC_DIR}')
|
|
if not os.path.exists(STATIC_DIR2):
|
|
raise FileNotFoundError(f'Static directory does not exist: {STATIC_DIR}')
|
|
|
|
app = FastAPI()
|
|
|
|
# Add session middleware for cookie management
|
|
jwt_secret_key = os.getenv('JWT_SECRET_KEY')
|
|
if not jwt_secret_key:
|
|
raise EnvironmentError('JWT_SECRET_KEY environment variable is not set')
|
|
app.add_middleware(SessionMiddleware, secret_key=jwt_secret_key)
|
|
|
|
# Include your routers here
|
|
app.include_router(router_static)
|
|
app.include_router(router_login)
|
|
app.include_router(router_pdf)
|
|
app.include_router(router_index)
|
|
app.include_router(router_template)
|
|
|
|
deps = Dependencies(
|
|
DB_PATH,
|
|
TEMPLATES_DIR,
|
|
STATIC_DIR,
|
|
STATIC_DIR2,
|
|
HEROWEB_DIR,
|
|
COLLECTIONS_DIR,
|
|
SERVERHOST,
|
|
)
|
|
app.deps = deps
|
|
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=['*'], # Allows all origins
|
|
allow_credentials=True,
|
|
allow_methods=['*'], # Allows all methods
|
|
allow_headers=['*'], # Allows all headers
|
|
)
|
|
|
|
|
|
@app.middleware('http')
|
|
async def check_authentication(
|
|
request: Request,
|
|
call_next,
|
|
):
|
|
logger.debug(f'Received request for URL: {request.url.path}')
|
|
|
|
# BYPASS
|
|
return await call_next(request)
|
|
|
|
if request.url.path in ['/signup', '/loginsubmit', '/register']:
|
|
logger.debug(
|
|
'Skipping authentication for signup, loginsubmit, or register path'
|
|
)
|
|
return await call_next(request)
|
|
|
|
token = request.cookies.get('access_token')
|
|
if not token:
|
|
logger.debug('No access token found, redirecting to /signup')
|
|
return RedirectResponse(url='/signup')
|
|
|
|
jwt_handler = request.app.deps.jwt_handler
|
|
try:
|
|
email = jwt_handler.verify_access_token(token)
|
|
except PyJWTError as e:
|
|
logger.error(f'Token verification failed: {e}')
|
|
return RedirectResponse(url='/signup')
|
|
|
|
request.state.email = email
|
|
logger.debug(f'Authenticated user: {email}')
|
|
|
|
response = await call_next(request)
|
|
# logger.debug(f'Response status code: {response.status_code}')
|
|
return response
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import uvicorn
|
|
|
|
uvicorn.run(app, host='0.0.0.0', port=8000)
|