#1752 - Changed DNSReply and DNSResponse to have 1 parameter only

src/primaite/simulator/system/services/dns_client.py

@@ -4,14 +4,14 @@ from typing import Any, Dict, List
 
 from pydantic import BaseModel
 
+from primaite.simulator.network.protocols.dns import DNSPacket, DNSRequest
+
 
 class DNSClient(BaseModel):
     """Represents a DNS Client as a Service."""
 
-    target_url: str
-    "The URL/domain name the client requests the IP for."
     dns_cache: Dict[str:IPv4Address] = {}
-    "A dict of known mappings between domain names and IPv4 addresses."
+    "A dict of known mappings between domain/URLs names and IPv4 addresses."
 
     @abstractmethod
     def describe_state(self) -> Dict:
@@ -41,9 +41,16 @@ class DNSClient(BaseModel):
         This method ensures the Service is ready for a new episode, including resetting any
         stateful properties or statistics, and clearing any message queues.
         """
-        self.target_url = ""
         self.dns_cache = {}
 
+    def check_domain_is_in_cache(self, target_domain: str, session_id: str):
+        """Function to check if domain name is in DNS client cache."""
+        if target_domain in self.dns_cache:
+            ip_address = self.dns_cache[target_domain]
+            self.send(ip_address, session_id)
+        else:
+            self.send(target_domain, session_id)
+
     def send(self, payload: Any, session_id: str, **kwargs) -> bool:
         """
         Sends a payload to the SessionManager.
@@ -54,7 +61,7 @@ class DNSClient(BaseModel):
         :param payload: The payload to send.
         :return: True if successful, False otherwise.
         """
-        pass
+        DNSPacket(dns_request=DNSRequest(domain_name_request=payload), dns_reply=None)
 
     def receive(self, payload: Any, session_id: str, **kwargs) -> bool:
         """
@@ -63,7 +70,10 @@ class DNSClient(BaseModel):
         The specifics of how the payload is processed and whether a response payload
         is generated should be implemented in subclasses.
 
-        :param payload: The payload to receive.
+        :param payload: The payload to receive. (receive a DNS packet with dns request and dns reply in, send to web
+        browser)
         :return: True if successful, False otherwise.
         """
+        # check DNS packet (dns request, dns reply) here and see if it actually worked
+
         pass