from __future__ import annotations

import glob
import typing
from dataclasses import dataclass

import yaml


@dataclass
class URL:
    """A class representing a link to a resource."""

    icon: str
    title: str
    url: str


class Resource:
    """A class representing a resource on the resource page."""

    description: str
    name: str
    payment: str
    payment_description: typing.Optional[str]
    urls: typing.List[URL]

    def __repr__(self):
        """Return a representation of the resource."""
        return f"<Resource name={self.name}>"

    @classmethod
    def construct_from_yaml(cls, yaml_data: typing.TextIO) -> Resource:  # noqa
        resource = cls()

        loaded = yaml.safe_load(yaml_data)

        resource.__dict__.update(loaded)

        resource.__dict__["urls"] = []

        for url in loaded["urls"]:
            resource.__dict__["urls"].append(URL(**url))

        return resource


class Category:
    """A class representing a resource on the resources page."""

    resources: typing.List[Resource]
    name: str
    description: str

    def __repr__(self):
        """Return a representation of the category."""
        return f"<Category name={self.name}>"

    @classmethod
    def construct_from_directory(cls, directory: str) -> Category:  # noqa
        category = cls()

        with open(f"{directory}/_category_info.yaml") as category_info:
            category_data = yaml.safe_load(category_info)

            category.__dict__.update(category_data)

        category.resources = []

        for resource in glob.glob(f"{directory}/*.yaml"):
            if resource == f"{directory}/_category_info.yaml":
                continue

            with open(resource) as res_file:
                category.resources.append(
                    Resource.construct_from_yaml(res_file)
                )

        return category


def load_categories(order: typing.List[str]) -> typing.List[Category]:
    """Load the categories specified in the order list and return them."""
    categories = []
    for cat in order:
        direc = "pydis_site/apps/home/resources/" + cat
        categories.append(Category.construct_from_directory(direc))

    return categories