''s

tools/helper.py

@@ -1,39 +1,29 @@
+import hashlib
 import os
+from pathlib import Path
 import platform
 import subprocess
+
 import requests
 from tqdm import tqdm
-import hashlib
+
+
+REPO_ROOT = Path(__file__).resolve().parents[1]
+OFFLINE_ROOT = REPO_ROOT / "offline"
+DOWNLOAD_ROOT = OFFLINE_ROOT / "downloads"
+IMAGE_ROOT = OFFLINE_ROOT / "images"
+
+
+def ensure_dir(path):
+    path.mkdir(parents=True, exist_ok=True)
+    return str(path)
 
 def get_download_dir():
-    download_loc = ""
-    current_user = (
-        os.environ.get("SUDO_USER")
-        or os.environ.get("USER")
-        or os.environ.get("USERNAME")
-        or "user"
-    )
-    if os.environ.get("XDG_CACHE_HOME", None) is None:
-        if os.name == "nt":
-            download_loc = os.path.join(
-                os.environ.get("LOCALAPPDATA", os.path.expanduser("~")),
-                "redroid",
-                "downloads",
-            )
-        else:
-            download_loc = os.path.join(
-                '/',
-                "home",
-                current_user,
-                ".cache",
-                "redroid",
-                "downloads",
-            )
-    else:
-        download_loc = os.path.join(os.environ["XDG_CACHE_HOME"], "redroid", "downloads")
-    if not os.path.exists(download_loc):
-        os.makedirs(download_loc)
-    return download_loc
+    return ensure_dir(DOWNLOAD_ROOT)
+
+
+def get_image_dir():
+    return ensure_dir(IMAGE_ROOT)
 
 def run(args):
     result = subprocess.run(args=args, stdout=subprocess.PIPE, stderr=subprocess.PIPE)