From 709b521662218fb1b5e388db67e813a3d0b8e281 Mon Sep 17 00:00:00 2001
From: Alarig Le Lay <alelay@scaleway.com>
Date: Fri, 19 Jul 2024 11:38:55 +0200
Subject: [PATCH] Handling basic debug

Signed-off-by: Alarig Le Lay <alelay@scaleway.com>
---
 check_domain_expiration_rdap.py | 34 ++++++++++++++++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/check_domain_expiration_rdap.py b/check_domain_expiration_rdap.py
index 5213a79..e7499ea 100755
--- a/check_domain_expiration_rdap.py
+++ b/check_domain_expiration_rdap.py
@@ -40,8 +40,28 @@ def expiration(domain):
             f'The TLD {tld} does not have an RDAP server'
         )
 
+    _log.debug(f'The used RDAP server is {url}')
+
     req_rdap = requests.get(f'{url}domain/{domain}')
 
+    match req_rdap.status_code:
+        case 403:
+            raise nagiosplugin.CheckError(
+                f'Got {req_rdap.status_code}, the RDAP server {url} refused to reply'
+            )
+        case 404:
+            raise nagiosplugin.CheckError(
+                f'Got {req_rdap.status_code}, the domain {domain} has not been found'
+            )
+        case 503:
+            raise nagiosplugin.CheckError(
+                f'Got {req_rdap.status_code}, the RDAP server {url} seems broken'
+            )
+        case _:
+            pass
+    
+    _log.debug(f'The used RDAP JSON is {req_rdap.json()}')
+
     raw_expiration = [
         event.get('eventDate', False)
         for event in req_rdap.json().get('events', {})
@@ -52,7 +72,7 @@ def expiration(domain):
         fecha = raw_expiration[0].split('T')[0]
     except IndexError:
         raise nagiosplugin.CheckError(
-            f'The domain {domain} has not been found'
+            f'The domain JSON for {domain} does not have "eventAction"."expiration" field, run with -vvv or --debug to have the JSON dump'
         )
 
     today = datetime.datetime.now()
@@ -117,12 +137,24 @@ def main():
     argp.add_argument(
         '-v', '--verbose', action='count', default=0, help='be more verbose'
     )
+    argp.add_argument(
+        '-d', '--debug', action='count', default=0,
+        help='debug logging to /tmp/nagios-check_domain_expiration_rdap.log'
+    )
     argp.add_argument('domain')
     args = argp.parse_args()
     wrange = f'@{args.critical}:{args.warning}'
     crange = f'@~:{args.critical}'
     fmetric = '{value} days until domain expires'
 
+    if (args.debug):
+        logging.basicConfig(
+            filename='/tmp/nagios-check_domain_expiration_rdap.log',
+            encoding='utf-8',
+            format='%(asctime)s %(message)s',
+            level=logging.DEBUG
+        )
+
     check = nagiosplugin.Check(
         Expiration(args.domain),
         nagiosplugin.ScalarContext(