feat(updater): test extract

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

Author: Sn0wo2

Updater/data/config.yml

@@ -1,5 +1,5 @@
 path:
-  workPath: ../
+  workPath: ./work
   processName: QuickNote.exe
   tagPath: ./data/tags
 

Updater/main.py

@@ -5,6 +5,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.extract.extract import extract_and_replace
 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
@@ -47,6 +48,10 @@ def main():
 
         downloaded_path = download_to_temp(download_url)
 
+        extract_and_replace(downloaded_path, Path(get_config().path.workPath))
+
+        print(Path(get_config().path.workPath))
+
         print(downloaded_path)
 
     print("=" * 80)

Updater/updater/extract/__init__.py

@@ -0,0 +1,19 @@
+from pathlib import Path
+
+from Updater.updater.extract.tar_gz import _extract_tar
+from Updater.updater.extract.zip import _extract_zip
+
+
+def extract_and_replace(archive_path: Path, extract_to: Path):
+    if not archive_path.exists():
+        raise FileNotFoundError(f"Archive not found: {archive_path}")
+
+    if not extract_to.exists():
+        extract_to.mkdir(parents=True)
+
+    if archive_path.suffix == '.zip':
+        _extract_zip(archive_path, extract_to)
+    elif archive_path.suffixes[-2:] == ['.tar', '.gz'] or archive_path.suffix == '.tgz':
+        _extract_tar(archive_path, extract_to)
+    else:
+        raise ValueError(f"Unsupported archive format: {archive_path.suffix}")

Updater/updater/extract/extract.py

@@ -0,0 +1,20 @@
+import shutil
+import tarfile
+from pathlib import Path
+
+
+def _extract_tar(tar_path: Path, extract_to: Path):
+    with tarfile.open(tar_path, 'r:*') as tar:
+        for member in tar.getmembers():
+            target_path = extract_to / member.name
+
+            if member.isdir():
+                if target_path.exists():
+                    shutil.rmtree(target_path)
+                target_path.mkdir(parents=True, exist_ok=True)
+            else:
+                if target_path.exists():
+                    target_path.unlink()
+                target_path.parent.mkdir(parents=True, exist_ok=True)
+                with tar.extractfile(member) as source, target_path.open('wb') as target:
+                    shutil.copyfileobj(source, target)

Updater/updater/extract/tar_gz.py

@@ -0,0 +1,20 @@
+import shutil
+import zipfile
+from pathlib import Path
+
+
+def _extract_zip(zip_path: Path, extract_to: Path):
+    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+        for member in zip_ref.namelist():
+            target_path = extract_to / member
+
+            if member.endswith('/'):
+                if target_path.exists():
+                    shutil.rmtree(target_path)
+                target_path.mkdir(parents=True, exist_ok=True)
+            else:
+                if target_path.exists():
+                    target_path.unlink()
+                target_path.parent.mkdir(parents=True, exist_ok=True)
+                with zip_ref.open(member) as source, target_path.open('wb') as target:
+                    shutil.copyfileobj(source, target)