class Comment(Item):
    """
    - `parent` is the id of the parent item
    - `text` is the comment text
    - `kids` is the list of comment ids that are replies to this comment
    """

    type: Literal[ItemType.COMMENT] = ItemType.COMMENT

    parent: int
    text: Optional[str] = None
    kids: list[int] = Field(default_factory=list)

class Job(Item):
    """
    - modified story with no comments, for advertisement of job postings
    """

    type: Literal[ItemType.JOB] = ItemType.JOB
    title: str
    score: int = 0
    text: Optional[str] = None
    url: Optional[str] = None

class Poll(Item):
    """
    - `title` is the title of the poll
    - `text`
    - `score`
    - `descendants` is the total comment count
    - `parts` (items) is the list of poll option ids
    - `kids` is the list of comment ids that are replies to this poll
    """

    type: Literal[ItemType.POLL] = ItemType.POLL

    title: str
    text: Optional[str] = None

    score: int = 0

    descendants: int = 0
    kids: list[int] = Field(default_factory=list)
    parts: list[int] = Field(default_factory=list)

    def reply_count(self) -> int:
        return self.descendants

class PollOption(Item):
    """
    - `poll` is the id of the poll this option belongs to.
    - `score` is the score of the poll option.
    - `text` is the text of the poll option.
    """

    type: Literal[ItemType.POLLOPTION] = ItemType.POLLOPTION

    poll: int
    score: int = 0
    text: str

class Story(Item):
    """
    - `title` is the title of the story
    - `url` is the URL provided in the story, usually original src of the story
    - `text`
    - `score`
    - `descendants` is the total comment count
    """

    type: Literal[ItemType.STORY] = ItemType.STORY

    title: str
    url: Optional[str] = None
    text: Optional[str] = None

    score: int = 0

    descendants: int = 0
    kids: list[int] = Field(default_factory=list)

    def reply_count(self) -> int:
        return self.descendants

class CommentNode(BaseModel):
    """
    Provider-agnostic comment tree node.
    """

    model_config = ConfigDict(
        frozen=True,
    )

    id: int

    author: Optional[str] = None

    text: Optional[str] = None

    created_at: Optional[datetime] = None

    children: list["CommentNode"] = Field(default_factory=list)

    @property
    def reply_count(self) -> int:
        return len(self.children)

    @classmethod
    def from_algolia(
        cls,
        node: AlgoliaNode,
    ) -> "CommentNode":
        return cls(
            id=node.id,
            author=node.author,
            text=node.text,
            created_at=node.created_at,
            children=[cls.from_algolia(child) for child in node.children],
        )