fix: Create storage dir if none found

Author: Dr0p42

lib/abi/services/cache/CacheFactory.py

@@ -1,12 +1,18 @@
 from abi.services.cache.CacheService import CacheService
 from abi.services.cache.adapters.secondary.CacheFSAdapter import CacheFSAdapter
 
-from abi.utils.Storage import find_storage_folder
+from abi.utils.Storage import find_storage_folder, NoStorageFolderFound
 import os
 
 class CacheFactory:
     @staticmethod
     def CacheFS_find_storage(subpath: str = "cache", needle: str = "storage") -> CacheService:
         if not subpath.startswith("cache"):
             subpath = os.path.join("cache", subpath)
-        return CacheService(CacheFSAdapter(os.path.join(find_storage_folder(os.getcwd(), needle), subpath)))
+        
+        try:
+            return CacheService(CacheFSAdapter(os.path.join(find_storage_folder(os.getcwd(), needle), subpath)))
+        except NoStorageFolderFound as _:
+            # Create a "storage" folder for the cache
+            os.makedirs(os.path.join(os.getcwd(), "storage"), exist_ok=True)
+            return CacheService(CacheFSAdapter(os.path.join(find_storage_folder(os.getcwd(), needle), subpath)))

lib/abi/utils/Storage.py

@@ -1,13 +1,15 @@
 import os
 
+class NoStorageFolderFound(Exception):
+    pass
 
 # Look for a "storage" folder until we reach /
 def find_storage_folder(base_path: str, needle: str = "storage") -> str:
     if os.path.exists(os.path.join(base_path, needle)):
         return os.path.join(base_path, needle)
 
     if base_path == "/":
-        raise Exception("No storage folder found")
+        raise NoStorageFolderFound("No storage folder found")
 
     return find_storage_folder(os.path.dirname(base_path))