Web-NibblePoker/website/content/dataclasses.py

from dataclasses import dataclass, field
from enum import Enum
from typing import Optional
from urllib.parse import ParseResult, urlparse

from locked_dict.locked_dict import LockedDict


@dataclass
class ContentHeadMetadata:
    title_key: str
    description_key: str


@dataclass
class ContentOpengraphMetadata:
    title_key: str
    description_key: str
    type: Optional[str] = field(default=None)
    url: Optional[str] = field(default=None)
    image_url: Optional[str] = field(default=None)
    image_type: Optional[str] = field(default=None)


@dataclass
class ContentTwitterMetadata:
    title_key: str
    description_key: str


@dataclass
class ContentIndexMetadata:
    priority: int
    enable: bool
    title_key: str
    preamble_key: str
    image_alt_key: str
    image_url: str = field(default="/resources/NibblePoker/images/placeholder.png")


@dataclass
class ContentGeneralMetadata:
    icon: str
    title_key: str
    subtitle_key: str
    tags: list[str] = field(default_factory=list)
    languages: list[str] = field(default_factory=list)


@dataclass
class ContentMetadata:
    head: ContentHeadMetadata
    opengraph: ContentOpengraphMetadata
    twitter: ContentTwitterMetadata
    index: ContentIndexMetadata
    general: ContentGeneralMetadata

    def __post_init__(self):
        self.head: dict
        self.head = ContentHeadMetadata(**self.head)

        self.opengraph: dict
        self.opengraph = ContentOpengraphMetadata(**self.opengraph)

        self.twitter: dict
        self.twitter = ContentTwitterMetadata(**self.twitter)

        self.index: dict
        self.index = ContentIndexMetadata(**self.index)

        self.general: dict
        self.general = ContentGeneralMetadata(**self.general)


class ContentResourceType(Enum):
    UNKNOWN = []
    REMOTE = ["http", "https"]
    APPLET = ["applet"]
    STANDALONE = ["standalone", "stand"]


class ContentResourceDefinition:
    raw_uri: str
    resource_type: ContentResourceType
    resource_uri: ParseResult

    def __init__(self, raw_uri):
        self.raw_uri = raw_uri
        self.resource_uri = urlparse(self.raw_uri)

        self.resource_type = ContentResourceType.UNKNOWN
        for content_resource_type in ContentResourceType:
            if self.resource_uri.scheme in content_resource_type.value:
                self.resource_type = content_resource_type

    def can_be_standalone(self):
        return self.resource_type in [ContentResourceType.STANDALONE, ContentResourceType.APPLET]

    def is_standalone(self):
        return self.resource_type == ContentResourceType.STANDALONE

    def is_remote(self):
        return self.resource_type == ContentResourceType.REMOTE

    def is_applet(self):
        return self.resource_type == ContentResourceType.APPLET

    def get_clean_path(self):
        return self.raw_uri.replace(f"{self.resource_uri.scheme}://", "")

    def __repr__(self):
        return (f"ContentResourceDefinition(raw_uri: {self.raw_uri}, "
                f"resource_type: {self.resource_type}, "
                f"resource_uri: {self.resource_uri})")


@dataclass
class ContentResource:
    scripts: list[ContentResourceDefinition] = field(default_factory=list)
    stylesheets: list[ContentResourceDefinition] = field(default_factory=list)

    def __post_init__(self):
        self.scripts: list[str]
        self.stylesheets: list[str]

        self.scripts = [ContentResourceDefinition(x) for x in self.scripts]
        self.stylesheets = [ContentResourceDefinition(x) for x in self.stylesheets]


@dataclass
class ContentApplet:
    id: str
    resources: ContentResource

    def __post_init__(self):
        self.resources: dict
        self.resources = ContentResource(**self.resources)
        # print(self.resources)


@dataclass
class ContentTool:
    # NOTE: could extend `ContentProject`
    id: str
    applet_id: str
    metadata: ContentMetadata

    def __post_init__(self):
        self.metadata: dict
        self.metadata = ContentMetadata(**self.metadata)


@dataclass
class ContentProject:
    id: str
    metadata: ContentMetadata

    def __post_init__(self):
        self.metadata: dict
        self.metadata = ContentMetadata(**self.metadata)


@dataclass
class ContentRoot:
    applets: LockedDict[str, ContentApplet] = field(default_factory=LockedDict)
    # articles: list[Con] = field(default_factory=list)
    projects: LockedDict[str, ContentProject] = field(default_factory=LockedDict)
    tools: LockedDict[str, ContentTool] = field(default_factory=LockedDict)
    projects_languages: list[str] = field(default_factory=list)