import asyncio
import json
import os

from components.web.utils import *
from components.utils import is_path_within_cwd
from components.utils.datetimes import (
    last_modified_http,
    parse_last_modified_http,
    ntime_utc_now,
)

blueprint = Blueprint("main", __name__, url_prefix="/")


@blueprint.route("/")
async def root():
    if session.get("id"):
        return redirect(url_for("profile.user_profile_get"))

    session_clear()
    return await render_template("auth/authenticate.html")


@blueprint.route("/logout", methods=["POST", "GET"])
async def logout():
    session_clear()
    return ("", 200, {"HX-Redirect": "/"})


@blueprint.websocket("/ws")
@websocket_acl("any")
async def ws():
    while True:
        await websocket.send(
            f'<div class="no-text-decoration" data-tooltip="Connected" id="ws-indicator" hx-swap-oob="outerHTML">🟢</div>'
        )
        data = await websocket.receive()
        try:
            data_dict = json.loads(data)
            if "path" in data_dict:
                IN_MEMORY_DB["WS_CONNECTIONS"][session["login"]][
                    websocket._get_current_object()
                ]["path"] = data_dict["path"]
        except:
            pass