"""
Resource index models for qrm2.
---
Copyright (C) 2021 classabbyamp, 0x5c

SPDX-License-Identifier: LiLiQ-Rplus-1.1
"""


from collections.abc import Mapping
from datetime import datetime

from pydantic import BaseModel


class File(BaseModel):
    filename: str
    hash: str

    # For some reason those were not the same???
    def __str__(self) -> str:
        return repr(self)


class Resource(BaseModel, Mapping):
    # 'A Beautiful Hack' https://github.com/samuelcolvin/pydantic/issues/1802
    __root__: dict[str, list[File]]

    def __getitem__(self, key: str) -> list[File]:
        return self.__root__[key]

    def __iter__(self):
        return iter(self.__root__)

    def __len__(self) -> int:
        return len(self.__root__)

    # For some reason those were not the same???
    def __str__(self) -> str:
        return repr(self)

    # Make the repr more logical (despite the technical inaccuracy)
    def __repr_args__(self):
        return self.__root__.items()


class Index(BaseModel, Mapping):
    last_updated: datetime
    resources: dict[str, Resource]

    def __getitem__(self, key: str) -> Resource:
        return self.resources[key]

    def __iter__(self):
        return iter(self.resources)

    def __len__(self) -> int:
        return len(self.resources)

    # For some reason those were not the same???
    def __str__(self) -> str:
        return repr(self)