from pathlib import Path
from typing import Any

from jinja2.environment import Environment
from jinja2.loaders import FileSystemLoader

from jwebsite.content import ContentDirectory


class Site:
    def __init__(self, root_directory: Path) -> None:
        self.__root_directory = root_directory
        self.__output_directory = root_directory / "build"
        self.__environment = Environment(loader=FileSystemLoader(searchpath=root_directory / "src"))
        self.__content = ContentDirectory(root_directory / "content")

    @property
    def content(self) -> ContentDirectory:
        return self.__content

    def render(self, source: str, output: str | Path, **context: Any) -> None:
        self.__output_directory.mkdir(parents=True, exist_ok=True)
        template = self.__environment.get_template(source)
        content = template.render(site=self, **context)
        output_path = self.__output_directory / output

        with open(output_path, "w") as output_file:
            output_file.write(content)