update

Author: Clivern

src/heahmund/http.py

@@ -20,14 +20,62 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
+import requests
+from http import HTTPStatus
+
+from heahmund.status import Status
 from heahmund.logger import Logger
 
 
-class HTTP:
-    """HTTP Check"""
+class GET:
+    """GET Check"""
+
+    def __init__(self, name, url, headers={}, params={}):
+        self.name = name
+        self.url = url
+        self.headers = headers
+        self.params = params
+        self.logger = Logger().get_logger(__name__)
+
+    def run(self):
+        """
+        Run The Check
+
+        Returns:
+            The Check Result
+        """
+
+        try:
+            response = requests.get(self.url, headers=self.headers, params=self.params)
+
+            if response.status_code == HTTPStatus.OK:
+                return {"name": self.name, "status": Status.OK}
+            else:
+                return {"name": self.name, "status": Status.NOT_OK}
+
+        except requests.exceptions.Timeout as e:
+            self.logger.debug(
+                "Get request to url {} throw timeout error {}.".format(self.url, str(e))
+            )
+
+            return {"name": self.name, "status": Status.ERROR}
+
+        except Exception as e:
+            self.logger.debug(
+                "Get request to url {} throw error {}.".format(self.url, str(e))
+            )
+
+            return {"name": self.name, "status": Status.ERROR}
+
+
+class HEAD:
+    """HEAD Check"""
 
-    def __init__(self, name):
+    def __init__(self, name, url, headers={}, params={}):
         self.name = name
+        self.url = url
+        self.headers = headers
+        self.params = params
         self.logger = Logger().get_logger(__name__)
 
     def run(self):
@@ -37,4 +85,25 @@ def run(self):
         Returns:
             The Check Result
         """
-        return {"name": self.name, "status": "OK"}
+
+        try:
+            response = requests.head(self.url, headers=self.headers, params=self.params)
+
+            if response.status_code == HTTPStatus.OK:
+                return {"name": self.name, "status": Status.OK}
+            else:
+                return {"name": self.name, "status": Status.NOT_OK}
+
+        except requests.exceptions.Timeout as e:
+            self.logger.debug(
+                "Get request to url {} throw timeout error {}.".format(self.url, str(e))
+            )
+
+            return {"name": self.name, "status": Status.ERROR}
+
+        except Exception as e:
+            self.logger.debug(
+                "Get request to url {} throw error {}.".format(self.url, str(e))
+            )
+
+            return {"name": self.name, "status": Status.ERROR}

src/heahmund/ping.py

@@ -41,6 +41,7 @@ def run(self):
         Returns:
             The Check Result
         """
+
         response = subprocess.run(
             ["ping", "-c", "3", self.hostname], capture_output=True
         )

src/heahmund/ssl.py

@@ -20,7 +20,12 @@
 # OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 # SOFTWARE.
 
+import ssl
+import datetime
+import socket
+
 from heahmund.logger import Logger
+from heahmund.status import Status
 
 
 class SSL:
@@ -41,4 +46,36 @@ def run(self):
         Returns:
             The Check Result
         """
-        return {"name": self.name, "status": "OK"}
+
+        context = ssl.create_default_context()
+
+        try:
+            # Connects to the server using the SSL context
+            with socket.create_connection(
+                (self.hostname, self.port), timeout=self.timeout
+            ) as sock:
+                sock.settimeout(self.timeout)
+
+                with context.wrap_socket(sock, server_hostname=self.hostname) as ssock:
+                    cert = ssock.getpeercert()
+
+                    # Get the expiration date of the certificate
+                    expiry_date = datetime.datetime.strptime(
+                        cert["notAfter"], "%b %d %H:%M:%S %Y %Z"
+                    )
+
+                    # Get the current date
+                    current_date = datetime.datetime.now()
+
+                    # Calculate the number of days until the certificate expires
+                    days_until_expiry = (expiry_date - current_date).days
+
+                    # Check if the certificate is valid for the specified number of days
+                    if days_until_expiry > days:
+                        return {"name": self.name, "status": Status.OK}
+                    else:
+                        return {"name": self.name, "status": Status.NOT_OK}
+
+        except socket.timeout:
+            # Handle the timeout
+            return {"name": self.name, "status": Status.TIMEOUT}

src/heahmund/tcp.py

@@ -43,6 +43,7 @@ def run(self):
         Returns:
             The Check Result
         """
+
         status = Status.OK
         sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
         sock.settimeout(self.timeout)

src/heahmund/udp.py

@@ -43,6 +43,7 @@ def run(self):
         Returns:
             The Check Result
         """
+
         status = Status.OK
         sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
         sock.settimeout(self.timeout)

src/heahmund/utils.py

@@ -33,4 +33,5 @@ def url_to_hostname(url="https://example.com"):
     Returns:
         The hostname
     """
+
     return urllib.parse.urlparse(url).netloc