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):
@@ -24,7 +25,22 @@ class TextStyle(Enum):
     ITALIC = "italic"
 
 
+class BlockType(Enum):
+    HEADER = "header"
+    CODE = "code"
+    LINES = "lines"
+    LINE = "line"
+    TEXT = "text"
+    LINK = "link"
+    ICON = "icon"
+    DIVIDER = "divider"
+    FACT_LIST = "fact_list"
+    FACT = "fact"
+    EXPANDABLE = "expandable"
+
+
 class BaseBlock(BaseModel):
+    type: BlockType
     pass
 
 
@@ -33,35 +49,41 @@ class BaseInlineTextBlock(BaseBlock):
 
 
 class TextBlock(BaseInlineTextBlock):
+    type: Literal[BlockType.TEXT] = BlockType.TEXT
     text: str
     style: Optional[TextStyle] = None
 
 
 class LinkBlock(BaseInlineTextBlock):
+    type: Literal[BlockType.LINK] = BlockType.LINK
     text: str
     url: str
 
 
 class IconBlock(BaseInlineTextBlock):
+    type: Literal[BlockType.ICON] = BlockType.ICON
     icon: Icon
 
 
 InlineBlock = Union[TextBlock, LinkBlock, IconBlock]
 
 
 class HeaderBlock(BaseBlock):
+    type: Literal[BlockType.HEADER] = BlockType.HEADER
     text: str
 
 
 class CodeBlock(BaseBlock):
+    type: Literal[BlockType.CODE] = BlockType.CODE
     text: str
 
 
 class DividerBlock(BaseBlock):
-    pass
+    type: Literal[BlockType.DIVIDER] = BlockType.DIVIDER
 
 
 class LineBlock(BaseBlock):
+    type: Literal[BlockType.LINE] = BlockType.LINE
     inlines: List[InlineBlock]
     sep: str = " "
 
@@ -71,19 +93,22 @@ class BaseLinesBlock(BaseBlock):
 
 
 class LinesBlock(BaseLinesBlock):
-    pass
+    type: Literal[BlockType.LINES] = BlockType.LINES
 
 
 class FactBlock(BaseBlock):
+    type: Literal[BlockType.FACT] = BlockType.FACT
     title: LineBlock
     value: LineBlock
 
 
 class FactListBlock(BaseBlock):
+    type: Literal[BlockType.FACT_LIST] = BlockType.FACT_LIST
     facts: List[FactBlock]
 
 
 class ExpandableBlock(BaseBlock):
+    type: Literal[BlockType.EXPANDABLE] = BlockType.EXPANDABLE
     title: str
     body: List["InExpandableBlock"]
     expanded: bool = False