add API for outbound to update_service_config.py

Author: bitterpanda63

aikido_zen/background_process/cloud_connection_manager/update_service_config.py

@@ -22,3 +22,12 @@ def update_service_config(connection_manager, res):
         bypassed_ips=res.get("allowedIPAddresses", []),
         received_any_stats=res.get("receivedAnyStats", True),
     )
+
+    # Handle outbound request blocking configuration
+    if "blockNewOutgoingRequests" in res:
+        connection_manager.conf.set_block_new_outgoing_requests(
+            res["blockNewOutgoingRequests"]
+        )
+
+    if "domains" in res:
+        connection_manager.conf.update_domains(res["domains"])

aikido_zen/background_process/service_config.py

@@ -80,6 +80,10 @@ def is_bypassed_ip(self, ip):
     def update_domains(self, domains):
         self.domains = {domain["hostname"]: domain["mode"] for domain in domains}
 
+    def set_block_new_outgoing_requests(self, value: bool):
+        """Set whether to block new outgoing requests"""
+        self.block_new_outgoing_requests = bool(value)
+
     def should_block_outgoing_request(self, hostname: str) -> bool:
         mode = self.domains.get(hostname)