test: добавление файлы для проверки работоспособности TargetStorage

src/xray_manager/core/target.py

@@ -0,0 +1,61 @@
+from __future__ import annotations
+
+from dataclasses import dataclass
+from typing import TYPE_CHECKING
+
+if TYPE_CHECKING:
+    from .xray_manager_config import XrayManagerConfig
+
+@dataclass(frozen=True)
+class Target:
+    id: str
+    pretty: str
+    route: str
+
+
+class TargetStorage:
+    def __init__(self, xray_manager_config: XrayManagerConfig):
+        self.xrmc = xray_manager_config
+
+    @property
+    def target_dict(self) -> dict[str, dict[str, str]]:
+        return self.xrmc.config.get("target", {})
+
+    def load_by_id(self, target_id: str) -> Target:
+        if target_id not in self.target_dict:
+            raise ValueError(f"Target not found: {target_id}")
+        
+        target_data = self.target_dict[target_id]
+
+        if "route" not in target_data:
+            raise ValueError(
+                f"Missing required 'route' field for target: {target_id}"
+            )
+        
+        return Target(
+            id=target_id,
+            pretty=target_data.get("pretty_name", target_id),
+            route=target_data["route"]
+        )
+
+    def load_by_pretty(self, target_pretty_name: str) -> Target:
+        for target_id, target_data in self.target_dict.items():
+            if target_data.get("pretty_name", target_id) == target_pretty_name:
+                return self.load_by_id(target_id)
+        
+        raise ValueError(f"Target pretty name not found: {target_pretty_name}")
+
+    def load_all_targets(self) -> list[Target]:
+        targets = [
+            self.load_by_id(target_id) for target_id in self.target_dict.keys()
+        ]
+        return targets
+
+    def save_target(self, target: Target):
+        self.xrmc.set_target(target.id, target.pretty, target.route)
+
+    def delete_target(self, target: Target):
+        if target not in self.load_all_targets():
+            raise ValueError(f"Target not found: {target.id}")
+        
+        self.xrmc.delete_target(target.id)

src/xray_manager/core/xray_manager_config.py

@@ -87,3 +87,18 @@ class XrayManagerConfig:
     def delete_profile_folder(self, username: str):
         self.config["profiles"]["folder_mapping"].pop(username, None)
         self._save_json()
+
+    def set_target(
+            self, target_id: str, target_pretty_name: str, target_route: str
+    ):
+        self.config["target"][target_id] = {
+            "pretty_name": target_pretty_name,
+            "route": target_route
+        }
+        self.config["target"][target_id]["pretty_name"] = target_pretty_name
+        self.config["target"][target_id]["route"] = target_route
+        self._save_json()
+
+    def delete_target(self, target_id: str):
+        self.config["target"].pop(target_id, None)
+        self._save_json()

tests/target_sandbox.py

@@ -0,0 +1,13 @@
+from xray_manager.core.target import Target, TargetStorage
+from xray_manager.core.xray_manager_config import XrayManagerConfig
+
+DEV_CONFIG_PATH = "tests/mock_data/etc/xray-manager/config.json"
+
+
+if __name__ == "__main__":
+    xray_manager_config = XrayManagerConfig(DEV_CONFIG_PATH)
+    storage = TargetStorage(xray_manager_config)
+
+    target = Target("de1", "pretty", "out-de1")
+    
+    storage.delete_target(target)