From ced45d427571cf8a8e78a5af44163c67fdabe8e4 Mon Sep 17 00:00:00 2001
From: Marek Wolan <marek.wolan@methods.co.uk>
Date: Wed, 16 Aug 2023 16:45:52 +0100
Subject: [PATCH] Connect actions of top-level sim components

---
 src/primaite/simulator/domain/controller.py | 16 +++++++++-
 src/primaite/simulator/network/container.py | 23 ++++++++++++++
 src/primaite/simulator/sim_container.py     | 34 +++++++++++++++++++++
 3 files changed, 72 insertions(+), 1 deletion(-)
 create mode 100644 src/primaite/simulator/network/container.py
 create mode 100644 src/primaite/simulator/sim_container.py

diff --git a/src/primaite/simulator/domain/controller.py b/src/primaite/simulator/domain/controller.py
index 887a065d..4e872531 100644
--- a/src/primaite/simulator/domain/controller.py
+++ b/src/primaite/simulator/domain/controller.py
@@ -1,7 +1,7 @@
 from enum import Enum
 from typing import Dict, Final, List, Literal, Tuple
 
-from primaite.simulator.core import ActionPermissionValidator, SimComponent
+from primaite.simulator.core import Action, ActionManager, ActionPermissionValidator, SimComponent
 from primaite.simulator.domain.account import Account, AccountType
 
 
@@ -82,6 +82,20 @@ class DomainController(SimComponent):
     folders: List[temp_folder] = {}
     files: List[temp_file] = {}
 
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+        self.action_manager = ActionManager()
+        # Action 'account' matches requests like:
+        # ['account', '<account-uuid>', *account_action]
+        self.action_manager.add_action(
+            "account",
+            Action(
+                func=lambda request, context: self.accounts[request.pop(0)].apply_action(request, context),
+                validator=GroupMembershipValidator([AccountGroup.DOMAIN_ADMIN]),
+            ),
+        )
+
     def _register_account(self, account: Account) -> None:
         """TODO."""
         ...
diff --git a/src/primaite/simulator/network/container.py b/src/primaite/simulator/network/container.py
new file mode 100644
index 00000000..346a089e
--- /dev/null
+++ b/src/primaite/simulator/network/container.py
@@ -0,0 +1,23 @@
+from typing import Dict
+
+from primaite.simulator.core import Action, ActionManager, AllowAllValidator, SimComponent
+from primaite.simulator.network.hardware.base import Link, Node
+
+
+class NetworkContainer(SimComponent):
+    """TODO."""
+
+    nodes: Dict[str, Node] = {}
+    links: Dict[str, Link] = {}
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+        self.action_manager = ActionManager()
+        self.action_manager.add_action(
+            "node",
+            Action(
+                func=lambda request, context: self.nodes[request.pop(0)].apply_action(request, context),
+                validator=AllowAllValidator(),
+            ),
+        )
diff --git a/src/primaite/simulator/sim_container.py b/src/primaite/simulator/sim_container.py
new file mode 100644
index 00000000..6989d2b9
--- /dev/null
+++ b/src/primaite/simulator/sim_container.py
@@ -0,0 +1,34 @@
+from primaite.simulator.core import Action, ActionManager, AllowAllValidator, SimComponent
+from primaite.simulator.domain.controller import DomainController
+
+
+class __TempNetwork:
+    """TODO."""
+
+    pass
+
+
+class SimulationContainer(SimComponent):
+    """TODO."""
+
+    network: __TempNetwork
+    domain: DomainController
+
+    def __init__(self, **kwargs):
+        super().__init__(**kwargs)
+
+        self.action_manager = ActionManager()
+        # pass through network actions to the network objects
+        self.action_manager.add_action(
+            "network",
+            Action(
+                func=lambda request, context: self.network.apply_action(request, context), validator=AllowAllValidator()
+            ),
+        )
+        # pass through domain actions to the domain object
+        self.action_manager.add_action(
+            "domain",
+            Action(
+                func=lambda request, context: self.domain.apply_action(request, context), validator=AllowAllValidator()
+            ),
+        )