feat(updater): downloader

Signed-off-by: Me0wo <[email protected]>

Author: Sn0wo2

Updater/main.py

@@ -4,6 +4,7 @@
 
 from Updater.config.yml_loader import get_config
 from Updater.updater.assets.name import get_package_name_from_current_machine
+from Updater.updater.downloader.temp import download_to_temp
 from Updater.updater.runner.process import find_processes_by_path, try_terminate
 from Updater.updater.tag.reader import read
 from fetch.github_release_api import fetch_latest_release
@@ -40,10 +41,13 @@ def main():
                 break
 
         if download_url is None:
-            print(f"Could not find asset for {get_package_name_from_current_machine()}")
+            print(
+                f"Could not find asset for {get_package_name_from_current_machine()}\nPlease update manually or send an issue.")
             return
 
-        print(download_url)
+        downloaded_path = download_to_temp(download_url)
+
+        print(downloaded_path)
 
     print("=" * 80)
     time.sleep(INTERVAL)

Updater/updater/downloader/__init__.py

@@ -0,0 +1,26 @@
+import atexit
+import tempfile
+from datetime import datetime
+from pathlib import Path
+from urllib.parse import urlparse
+
+import requests
+
+
+def download_to_temp(url: str) -> Path:
+    original_name = Path(urlparse(url).path).name
+
+    filename = f"{Path(original_name).stem}_{datetime.now().strftime("%Y%m%d_%H%M%S")}{Path(original_name).suffix}"
+
+    temp_path = Path(tempfile.gettempdir()) / filename
+
+    with requests.get(url, stream=True) as r:
+        r.raise_for_status()
+        with open(temp_path, 'wb') as f:
+            for chunk in r.iter_content(chunk_size=1024):
+                if chunk:
+                    f.write(chunk)
+
+    atexit.register(lambda: temp_path.exists() and temp_path.unlink())
+
+    return temp_path