import os
import pathlib

from fastapi import FastAPI, Request, WebSocket
from fastapi.staticfiles import StaticFiles
from fastapi.templating import Jinja2Templates

import home
import about_us

app = FastAPI()
app.mount('/static', StaticFiles(directory='static'), name='static')

BASE_DIR = pathlib.Path(__file__).parent
templates = Jinja2Templates(
    directory=[
        BASE_DIR / 'templates',
    ]
)


@app.get('/')
async def welcome_page(request: Request):
    context = {
        'request': request,
    }
    return templates.TemplateResponse(
        'layouts/base.html', context,
    )



@app.websocket('/ws/liveview/')
async def websocket_endpoint(websocket: WebSocket):
    await websocket.accept()
    while True:
        data_frontend = await websocket.receive_json()
        if data_frontend and 'action' in data_frontend:
            action_data = data_frontend['action'].split('->')
            if len(action_data) == 2:
                action = action_data[0].lower()
                function = action_data[1].lower()
                await eval(
                    f'{action}.{function}(websocket, templates, data_frontend)'
                )