from typing import Callable, Iterable, Self
from bs4 import BeautifulSoup, Tag

from .elements import IdAttr, Template
from .renderer import Renderable, Renderer
from .puppytype import ElementLike, ElementLikeList, Parsable, Templates

class Puppygirl[R: Renderer]:
  renderer: R
  elements: list[Renderable]
  templates: Templates

  def __init__(self, renderer: R, elements: list[Callable[[Self], Renderable]] = [], templates: ElementLikeList = []):
    self.renderer = renderer
    self.templates = Puppygirl._create_template_dict(templates)
    self.elements = [self._instantiate(el) for el in elements]

  def _instantiate(self, value: Callable[[Self], Renderable] | Renderable) -> Renderable:
    if(isinstance(value, Callable)):
      return value(self)
    return value

  def _create_template_dict(templates: Iterable[ElementLike]) -> Templates:
    templates = [Template.from_element(t) for t in templates]
    return {t[IdAttr]: t for t in templates}

  def add_template(self, template: ElementLike):
    template = Template.from_element(template)
    self._templates[template[IdAttr]] = template

  def fetch(self, path: str) -> BeautifulSoup:
    with open(path, "r") as f:
      return self.parse(f)

  def parse(self, value: Parsable) -> BeautifulSoup:
    if isinstance(value, BeautifulSoup) or isinstance(value, Tag):
      return self.parse_tree(value)
  
    return self.parse_tree(BeautifulSoup(value, features='html.parser'))
  
  def parse_tree(self, tree: BeautifulSoup) -> BeautifulSoup:
    return self.renderer.render(self, tree)