Add block type enum and update block classes with type annotations

- Introduced `BlockType` enum to define standardized block types
- Updated block classes to include explicit type annotations using `Literal` types
- Added type hints to base block classes to improve type safety

Author: MikaKerman

elementary/messages/blocks.py

@@ -2,6 +2,7 @@
 from typing import List, Optional, Union
 
 from pydantic import BaseModel
+from typing_extensions import Literal
 
 
 class Icon(Enum):
@@ -25,6 +26,7 @@ class TextStyle(Enum):
 
 
 class BaseBlock(BaseModel):
+    type: str
     pass
 
 
@@ -33,35 +35,41 @@ class BaseInlineTextBlock(BaseBlock):
 
 
 class TextBlock(BaseInlineTextBlock):
+    type: Literal["text"] = "text"
     text: str
     style: Optional[TextStyle] = None
 
 
 class LinkBlock(BaseInlineTextBlock):
+    type: Literal["link"] = "link"
     text: str
     url: str
 
 
 class IconBlock(BaseInlineTextBlock):
+    type: Literal["icon"] = "icon"
     icon: Icon
 
 
 InlineBlock = Union[TextBlock, LinkBlock, IconBlock]
 
 
 class HeaderBlock(BaseBlock):
+    type: Literal["header"] = "header"
     text: str
 
 
 class CodeBlock(BaseBlock):
+    type: Literal["code"] = "code"
     text: str
 
 
 class DividerBlock(BaseBlock):
-    pass
+    type: Literal["divider"] = "divider"
 
 
 class LineBlock(BaseBlock):
+    type: Literal["line"] = "line"
     inlines: List[InlineBlock]
     sep: str = " "
 
@@ -71,19 +79,22 @@ class BaseLinesBlock(BaseBlock):
 
 
 class LinesBlock(BaseLinesBlock):
-    pass
+    type: Literal["lines"] = "lines"
 
 
 class FactBlock(BaseBlock):
+    type: Literal["fact"] = "fact"
     title: LineBlock
     value: LineBlock
 
 
 class FactListBlock(BaseBlock):
+    type: Literal["fact_list"] = "fact_list"
     facts: List[FactBlock]
 
 
 class ExpandableBlock(BaseBlock):
+    type: Literal["expandable"] = "expandable"
     title: str
     body: List["InExpandableBlock"]
     expanded: bool = False