Agent: Implement get config and get propagation credentials

monkey/infection_monkey/control_channel.py

@@ -1,11 +1,11 @@
 import json
 import logging
-from abc import ABC
 
 import requests
 
 from common.common_consts.timeouts import SHORT_REQUEST_TIMEOUT
 from infection_monkey.config import GUID, WormConfiguration
+from infection_monkey.control import ControlClient
 from monkey.infection_monkey.i_control_channel import IControlChannel
 
 requests.packages.urllib3.disable_warnings()
@@ -13,7 +13,7 @@ requests.packages.urllib3.disable_warnings()
 logger = logging.getLogger(__name__)
 
 
-class ControlChannel(IControlChannel, ABC):
+class ControlChannel(IControlChannel):
     def should_agent_stop(self) -> bool:
         server = WormConfiguration.current_server
         if not server:
@@ -29,4 +29,24 @@ class ControlChannel(IControlChannel, ABC):
             response = json.loads(response.content.decode())
             return response["stop_agent"]
         except Exception as e:
-            logger.error(f"Error happened while trying to connect to server. {e}")
+            logger.error(f"An error occurred while trying to connect to server. {e}")
+
+    def get_config(self) -> dict:
+        return ControlClient.load_control_config()
+
+    def get_credentials_for_propagation(self) -> dict:
+        server = WormConfiguration.current_server
+        if not server:
+            return
+
+        try:
+            response = requests.get(  # noqa: DUO123
+                f"{server}/api/propagationCredentials",
+                verify=False,
+                timeout=SHORT_REQUEST_TIMEOUT,
+            )
+
+            response = json.loads(response.content.decode())["propagation_credentials"]
+            return response
+        except Exception as e:
+            logger.error(f"An error occurred while trying to connect to server. {e}")