From f77ab3348c105e0962cf6651e841d3847db6cc11 Mon Sep 17 00:00:00 2001
From: Alarig Le Lay <alarig@swordarmor.fr>
Date: Sun, 28 Jul 2024 01:08:33 +0200
Subject: [PATCH] Check for several cache paths and fallback to requests if
 needed

Signed-off-by: Alarig Le Lay <alarig@swordarmor.fr>
---
 check_domain_expiration_rdap.py | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/check_domain_expiration_rdap.py b/check_domain_expiration_rdap.py
index 898e055..beb1995 100755
--- a/check_domain_expiration_rdap.py
+++ b/check_domain_expiration_rdap.py
@@ -7,6 +7,7 @@ import argparse
 import datetime
 import logging
 import os
+import pathlib
 import requests
 
 import nagiosplugin
@@ -18,10 +19,20 @@ __version__ = '0.1'
 _log = logging.getLogger('nagiosplugin')
 
 # cache session for json and csv storage
-session = requests_cache.CachedSession(
-    '/tmp/iana_rdap_cache',
-    cache_control=True
-)
+uid = os.getuid()
+home = pathlib.Path.home()
+for possible_dir in [f'/run/{uid}', home, '/tmp']:
+    iana_rdap_cache = f'{possible_dir}/iana_rdap_cache'
+    try:
+        cache = open(f'{iana_rdap_cache}.sqlite', 'a')
+        cache.close()
+        session = requests_cache.CachedSession(iana_rdap_cache, cache_control=True)
+        _log.debug(f'Caching to {iana_rdap_cache}.sqlite')
+        break
+    except IOError:
+        _log.debug(f'{iana_rdap_cache}.sqlite is not writtable')
+        session = requests
+        iana_rdap_cache = ''
 
 def find_rdap_server(domain):
     """Find the TLD rdap server."""