# files.altafcreator.com is a READ-ONLY website of files I want to share.
# This code is not safe for user-uploaded files.

import fastapi
from fastapi.middleware.cors import CORSMiddleware
import fastapi.staticfiles
import fastapi.responses
from pathlib import Path
import html
from bs4 import BeautifulSoup
import datetime

print("Hello, world!")

FOLDER_PATH = Path("/Users/altaf/").resolve()
# FOLDER_PATH = "/var/files"

app = fastapi.FastAPI(title="altaf-files", decscription="altaf-files backend", version="1.0")

origins = [
    "http://localhost",
    "http://localhost:9092",
    "https://files.altafcreator.com",
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)


HTML_FOLDER_TEMPLATE = ""
with open("index.html") as f:
    HTML_FOLDER_TEMPLATE = f.read()


@app.middleware("http")
async def force_trailing_slash(request: fastapi.Request, call_next):
    path = request.url.path

    if not path.endswith("/") and not Path(path).suffix:
        new_url = str(request.url.replace(path=path + "/"))
        return fastapi.responses.RedirectResponse(url=new_url, status_code=301)

    response = await call_next(request)
    return response


@app.get("/{path:path}")
def folder(path: str):
    target_path = (FOLDER_PATH / path).resolve()

    if not target_path.is_relative_to(FOLDER_PATH):
        return fastapi.responses.Response(status_code=403, content="Access denied.")

    if target_path.is_file():
#        if target_path.suffix == ".md":
#            return fastapi.responses.Response(content=markdown(target_path), media_type="text/html", status_code=200)
        return fastapi.responses.FileResponse(target_path)
    elif not target_path.is_dir():
        return fastapi.responses.Response(status_code=404)

    files = []
    directories = []

    for item in sorted(target_path.iterdir()):
        if (target_path / item.name).resolve().is_file():
            files.append(item.name)
        else:
            directories.append(item.name)

    soup = BeautifulSoup(HTML_FOLDER_TEMPLATE, "html.parser")

    for d in directories:
        add_item_html(d, path, True, soup)
    for f in files:
        add_item_html(f, path, False, soup)

    local_path_list = path.split("/")

    home_btn = soup.new_tag("a")
    home_btn["class"] = "breadcrumb-button"
    home_btn["href"] = "/"
    home_btn.string = "🏠 altaf-files"
    soup.select_one("#breadcrumb-container").append(home_btn)

    for i in range(len(local_path_list)):
        new_sep = soup.new_tag("span")
        new_sep.string = "/"
        soup.select_one("#breadcrumb-container").append(new_sep)

        new_btn = soup.new_tag("a")
        new_btn["class"] = "breadcrumb-button"
        new_btn["href"] = "/" + "/".join(local_path_list[0:i + 1])
        new_btn.string = local_path_list[i]
        soup.select_one("#breadcrumb-container").append(new_btn)

    return fastapi.responses.Response(content=str(soup), media_type="text/html", status_code=200)


def markdown(absolute_path: str):
    pass


def add_item_html(child_path: str, current_path: str, is_directory: bool, soup: BeautifulSoup):
    href_path = ""

    if current_path == "":
        href_path = f"/{child_path}"
    else:
        href_path = f"/{current_path.rstrip('/')}/{child_path}"

    symbol = "📄"
    if is_directory:
        symbol = "📂"

    absolute_path = Path(f"{str(FOLDER_PATH).rstrip('/')}/{current_path.rstrip('/')}/{child_path}").resolve()

    if not absolute_path.exists():
        return ""

    size = "—"
    if not is_directory:
        size = format_size(absolute_path.stat().st_size)

    date_modified = str(datetime.datetime.fromtimestamp(absolute_path.stat().st_mtime))[:10].replace("-", "/")

    new_btn = soup.new_tag("a")
    new_btn["href"] = href_path

    new_div = soup.new_tag("div")
    new_btn.append(new_div)

    new_span_name = soup.new_tag("span")
    new_span_name.string = f"{symbol} {child_path}"
    new_div.append(new_span_name)
    new_span_date = soup.new_tag("span")
    new_span_date["class"] = "list-metadata list-date"
    new_div.append(new_span_date)
    new_span_date.string = date_modified
    new_span_size = soup.new_tag("span")
    new_span_size["class"] = "list-metadata"
    new_span_size.string = size
    new_div.append(new_span_size)

    soup.select_one("#list-container").append(new_btn)


def format_size(byte_size):
    for unit in ['B', 'KiB', 'MiB', 'GiB', 'TiB']:
        if byte_size < 1024.0:
            return f"{round(byte_size, 1)} {unit}"
        byte_size /= 1024.0