refactor(message.py): use ActionType instead of Enum

CHANGELOG.md

@@ -2,6 +2,12 @@
 
 All notable changes to this project will be documented in this file.
 
+## [0.7.0] - 2026-03-01
+
+### Features
+
+- [`0c0595b`](https://gitea.pufereq.pl/judas/judas_protocol/commit/0c0595b5f67526f1caa4f3650c04ae36180045c3) **message.py**: move CONTROL-specific methods into `Message.Control` inner class
+
 ## [0.6.0] - 2026-02-28
 
 ### Refactor

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "uv_build"
 
 [project]
 name = "judas_protocol"
-version = "0.6.0"
+version = "0.7.0"
 description = "The judas protocol"
 readme = "README.md"
 authors = []

src/judas_protocol/__init__.py

@@ -1,6 +1,6 @@
 __version__: str = "0.1.0"
 
 from .message import Message
-from .types import Category, ControlAction
+from .types import ActionType, Category, ControlAction
 
-__all__ = ["__version__", "Message", "Category", "ControlAction"]
+__all__ = ["__version__", "Message", "ActionType", "Category", "ControlAction"]

src/judas_protocol/message.py

@@ -3,10 +3,9 @@ from __future__ import annotations
 
 import json
 import uuid
-from enum import Enum
 from typing import Any, override
 
-from judas_protocol.types import Category, ControlAction
+from judas_protocol.types import ActionType, Category, ControlAction
 
 
 class Message:
@@ -16,7 +15,7 @@ class Message:
         self,
         id_: str | None,
         category: Category,
-        action: Enum,
+        action: ActionType,
         payload: dict[str, Any] | None = None,
         *,
         acknowledged: bool = False,
@@ -27,14 +26,14 @@ class Message:
         Args:
             id_ (str | None): The ID of the message. If None, a new UUID will be generated.
             category (Category): The category of the message.
-            action (Enum): The action of the message.
+            action (ActionType): The action of the message.
             payload (dict[str, Any] | None): The payload of the message.
             acknowledged (bool): Whether the message has been acknowledged. Defaults to False.
             ack_required (bool): Whether the message requires an acknowledgment. Defaults to False.
         """
         self.id: str = id_ or str(uuid.uuid4())
         self.category: Category = category
-        self.action: Enum = action
+        self.action: ActionType = action
         self.payload: dict[str, Any] = payload or {}
 
         self.acknowledged: bool = acknowledged
@@ -52,6 +51,63 @@ class Message:
             f", ack_required={self.ack_required})"
         )
 
+    class Control:
+        """CONTROL category message factory methods."""
+
+        @staticmethod
+        def ack(target_id: str) -> Message:
+            """Create an ACK message.
+
+            Args:
+                target_id (str): The ID of the message to acknowledge.
+
+            Returns:
+                Message: The created ACK message.
+            """
+            return Message(
+                id_=None,
+                category=Category.CONTROL,
+                action=ControlAction.ACK,
+                payload={"target_id": target_id},
+                acknowledged=False,
+                ack_required=False,
+            )
+
+        @staticmethod
+        def hello(id_: str) -> Message:
+            """Create a HELLO message.
+
+            Args:
+                id_ (str): The ID to include in the HELLO message.
+            Returns:
+                Message: The created HELLO message.
+            """
+            return Message(
+                id_=None,
+                category=Category.CONTROL,
+                action=ControlAction.HELLO,
+                payload={"id": id_},
+                acknowledged=False,
+                ack_required=True,
+            )
+
+        @staticmethod
+        def close() -> Message:
+            """Create a CLOSE message.
+            Prompts the recipient to close the connection and not reconnect.
+
+            Returns:
+                Message: The created CLOSE message.
+            """
+            return Message(
+                id_=None,
+                category=Category.CONTROL,
+                action=ControlAction.CLOSE,
+                payload={},
+                acknowledged=False,
+                ack_required=True,
+            )
+
     def to_dict(self) -> dict[str, Any]:
         """Convert the message to a dictionary.
 
@@ -144,57 +200,3 @@ class Message:
             Message: The created message.
         """
         return cls.from_json(data.decode("utf-8"))
-
-    @classmethod
-    def ack(cls, target_id: str) -> Message:
-        """Create an ACK message.
-
-        Args:
-            target_id (str): The ID of the message to acknowledge.
-
-        Returns:
-            Message: The created ACK message.
-        """
-        return cls(
-            id_=None,
-            category=Category.CONTROL,
-            action=ControlAction.ACK,
-            payload={"target_id": target_id},
-            acknowledged=False,
-            ack_required=False,
-        )
-
-    @classmethod
-    def hello(cls, id_: str) -> Message:
-        """Create a HELLO message.
-
-        Args:
-            id_ (str): The ID to include in the HELLO message.
-        Returns:
-            Message: The created HELLO message.
-        """
-        return cls(
-            id_=None,
-            category=Category.CONTROL,
-            action=ControlAction.HELLO,
-            payload={"id": id_},
-            acknowledged=False,
-            ack_required=True,
-        )
-
-    @classmethod
-    def close(cls) -> Message:
-        """Create a CLOSE message.
-        Prompts the recipient to close the connection and not reconnect.
-
-        Returns:
-            Message: The created CLOSE message.
-        """
-        return cls(
-            id_=None,
-            category=Category.CONTROL,
-            action=ControlAction.CLOSE,
-            payload={},
-            acknowledged=False,
-            ack_required=True,
-        )

src/judas_protocol/types.py

@@ -5,6 +5,8 @@ from __future__ import annotations
 
 from enum import Enum
 
+type ActionType = ControlAction
+
 
 class Category(str, Enum):
     CONTROL = "control"

uv.lock

@@ -240,7 +240,7 @@ wheels = [
 
 [[package]]
 name = "judas-protocol"
-version = "0.6.0"
+version = "0.7.0"
 source = { editable = "." }
 
 [package.dev-dependencies]