import pandas
import humanize
from nicegui import ui
from utils.registry import DockerRegistry


## TODO: this should be async but I currently don't know how to implement this without button press
def get_image_info() -> pandas.DataFrame:
    data = pandas.DataFrame(columns=["image", "tag", "size"])
    try:
        registry = DockerRegistry()
        all_image_info = registry.get_all_image_info()
    except Exception as error:
        ui.notify(message=error)
        return data
    if isinstance(all_image_info, list) and len(all_image_info) > 0:
        data = pandas.json_normalize(
            all_image_info,
            record_path=["tags"],
            meta=[["name"]],
            meta_prefix="image_",
        ).assign(
            size=lambda x: x["manifest.layers"].apply(
                lambda layers: sum(layer["size"] for layer in layers)
            )
        )
        data = (
            data[["image_name", "name", "size"]]
            .rename(columns={"image_name": "image", "name": "tag", "size": "size"})
            .sort_values(by=["image", "tag"], ascending=False)
        )
        data["size"] = data["size"].apply(humanize.naturalsize)
        return data
    else:
        ui.notify(message="No images found")
        return data


def content() -> None:
    with ui.row().classes("w-full"):
        with ui.card().classes("w-full"):
            ui.label("Image Overview").classes("text-h5")
            data = get_image_info()
            with ui.table.from_pandas(df=data).classes("w-full") as table:
                table.columns[0]["sortable"] = True
                table.columns[1]["sortable"] = True
                table.columns[2]["sortable"] = True