we got it to extract the tar file! :D

2026-01-18 19:59:58 +11:00
parent 0e845ea2c5
commit 6fb6f42366
4 changed files with 116 additions and 6 deletions
--- a/src/install.py
+++ b/src/install.py
@@ -0,0 +1,75 @@
+import requests
+import os
+import sys
+import tempfile
+import tarfile
+
+from rich.console import Console
+from rich.progress import SpinnerColumn
+
+
+console = Console()
+
+
+def install(args):
+    # check if we are sudo
+    if os.getuid() != 0:
+        console.print("[b red]digpkg: the install command requires sudo to run[/]")
+        sys.exit(1)
+    
+    # ensure the GROUND_LIBS var is set
+    if not os.getenv("GROUND_LIBS"):
+        console.print("digpkg: the [i]GROUND_LIBS[/] environment variable is not set, defaulting to /usr/lib/ground/")
+        os.environ["GROUND_LIBS"] = "/usr/lib/ground/"
+    
+    # figure out which version to install
+    package_name = args.name
+    version = "latest"
+    if "@" in args.name:
+        split = package_name.split("@")
+        package_name = split[0]
+        version = split[1]
+    
+    retries_left = args.max_retries
+    
+    with console.status("Downloading tarball...", spinner="bouncingBall", spinner_style="blue") as status:
+        while retries_left > 0:
+            # grab the tar ball
+            response = requests.get(f"https://chookspace.com/api/packages/SpookyDervish/generic/{package_name}/{version}/mineral.tar")
+            
+            # check response code for errors
+            if response.status_code == 404: # package doesn't exist
+                console.print(f"[b red]digpkg: mineral \"{package_name}\" was not found. Check to make sure the name and version number are correct.[/]")
+                sys.exit(1)
+            elif response.status_code != 200:
+                retries_left -= 1
+                console.print(f"[b yellow]digpkg: failed to download mineral \"{package_name}\": {response.content.decode()} ({retries_left} retries left)[/]")
+                
+                if retries_left == 0:
+                    console.print(f"[b red]digpkg: exceeded max retries while downloading mineral \"{package_name}\"[/]")
+                    sys.exit(1)
+                
+                continue
+            
+            break
+        
+        # create temporary file for tarball
+        f = tempfile.TemporaryFile("wb+") 
+        f.write(response.content)
+        f.flush()
+        f.seek(0)
+        console.print("[d][:white_check_mark:] Tarball downloaded![/]")
+        
+        # extract the tarball to the GROUND_LIBS folder
+        status.update("Extracting...")
+        
+        extract_dir = os.getenv("GROUND_LIBS")
+        if not os.path.isdir(extract_dir): # gotta ensure the folder exists
+            os.mkdir(extract_dir)
+        
+        
+        tar_file = tarfile.open(fileobj=f)
+        tar_file.extractall(extract_dir)
+        f.close()
+        
+        console.print(f"[d][:white_check_mark:] Extracted to {extract_dir}.")