Added ports to icon

src/bedit_gui/models.py

@@ -6,7 +6,7 @@ from dataclasses import dataclass, field
 from enum import Enum
 from typing import Any
 
-from bedit_core.models import ComponentID, ID
+from bedit_core.models import ComponentID, ID, PortID
 
 
 class ShapeID(ID):
@@ -40,8 +40,8 @@ class LineType(Enum):
 @dataclass
 class Rectangle(Shape):
     type: str = field(init=False, default="rectangle")
-    width: float = 100.0
-    height: float = 100.0
+    width: float = 32.0
+    height: float = 32.0
     line_type: LineType = LineType.SOLID
     line_thickness: float = 1.0
     corner_radius: float = 0.0
@@ -56,18 +56,30 @@ class Rectangle(Shape):
     def to_data(self) -> dict[str, Any]:
         return {**super().to_data(), "width": self.width, "height": self.height, "line_type": self.line_type.value, "line_thickness": self.line_thickness, "corner_radius": self.corner_radius, "line_color": self.line_color, "fill_color": self.fill_color}
 
+@dataclass
+class Text(Shape):
+    type: str = field(init=False, default="text")
+    width: float = 100.0
+    height: float = 100.0
+    color: str = "#000000ff"
+    bold: bool = False
+    italic: bool = False
+    size: float = 16.0
+    text: str = ""
 
 @dataclass
 class Icon:
     shapes: dict[ShapeID, Shape] = field(default_factory=dict)
+    port_positions: dict[PortID, tuple[int, int]] = field(default_factory=dict)
 
     @classmethod
     def from_data(cls, data: Mapping[str, Any]) -> Icon:
         shapes = {ShapeID(key): Shape.from_data(value) for key, value in data.get("shapes", {}).items()}
-        return cls(shapes=shapes)
+        port_positions = {PortID(key): (int(value[0]), int(value[1])) for key, value in data.get("port_positions", {}).items()}
+        return cls(shapes=shapes, port_positions=port_positions)
 
     def to_data(self) -> dict[str, Any]:
-        return {"shapes": {str(key): shape.to_data() for key, shape in self.shapes.items()}}
+        return {"shapes": {str(key): shape.to_data() for key, shape in self.shapes.items()}, "port_positions": {str(key): list(position) for key, position in self.port_positions.items()}}
 
 
 @dataclass