Add FirewallLists to the connection manager

Author: bitterpanda63

aikido_zen/background_process/cloud_connection_manager/__init__.py

@@ -11,6 +11,7 @@
 from aikido_zen.storage.hostnames import Hostnames
 from ..realtime.start_polling_for_changes import start_polling_for_changes
 from ...storage.ai_statistics import AIStatistics
+from ...storage.firewall_lists import FirewallLists
 from ...storage.statistics import Statistics
 
 # Import functions :
@@ -41,6 +42,7 @@ def __init__(self, block, api, token, serverless):
             bypassed_ips=[],
             received_any_stats=True,
         )
+        self.firewall_lists = FirewallLists()
         self.rate_limiter = RateLimiter(
             max_items=5000, time_to_live_in_ms=120 * 60 * 1000  # 120 minutes
         )

aikido_zen/background_process/cloud_connection_manager/update_firewall_lists.py

@@ -24,9 +24,9 @@ def update_firewall_lists(connection_manager):
         if not isinstance(blocked_user_agents, str):
             return
 
-        connection_manager.conf.set_blocked_ips(blocked_ips)
-        connection_manager.conf.set_allowed_ips(allowed_ips)
-        connection_manager.conf.set_blocked_user_agents(blocked_user_agents)
+        connection_manager.firewall_lists.set_blocked_ips(blocked_ips)
+        connection_manager.firewall_lists.set_allowed_ips(allowed_ips)
+        connection_manager.firewall_lists.set_blocked_user_agents(blocked_user_agents)
 
     except Exception as e:
         logger.debug("Exception in update_firewall_lists: %s", e)