from dataclasses import dataclass, field
from pathlib import Path
from typing import Callable

import flask


@dataclass
class RequestPath:
    """
    Dataclass representing path to current page.
    Used in breadcrumbs and relative link generation.
    """
    _path: list[str] = field(default_factory=list)
    files: list[Path] = field(default_factory=list)
    is_not_found: bool = False

    @staticmethod
    def join_link(path: list[str]):
        return "/" + "/".join(path) if path else ""

    @property
    def breadcrumbs_links(self):
        current_path = []
        for name in self._path:
            current_path.append(name)
            yield name, self.join_link(current_path)

    @property
    def path(self):
        return self.join_link(self._path)

    @property
    def raw_path(self):
        if self._path[0] == "source":
            return self.join_link(["raw"] + self._path[1:])

        return self.join_link(["raw", "content"] + self._path)

    @property
    def parent(self):
        return self.join_link(self._path[:-1])

    @property
    def page_name(self):
        return self._path[-1]

    def add(self, name):
        self._path.append(name)

    def scan_dir(self, dir_path: Path, filter_func: Callable[[Path], bool]):
        self.files += filter(filter_func, dir_path.iterdir())
        self.files.sort(key=lambda p: (p.is_file(), p.name))

    @classmethod
    def set_request_path(cls):
        flask.g.rpath = cls()


def init(app: flask.Flask):
    app.before_request(RequestPath.set_request_path)