아 env

lib/ublock_init.py

@@ -0,0 +1,55 @@
+import zipfile
+import requests
+from pathlib import Path
+import shutil
+
+TEMP_EXTRACT_DIR = Path("./browser/temp_ublock_extract")
+
+def ensure_ublock_origin(UBLOCK_DIR: Path):
+    """
+    uBlock Origin을 다운로드하고 압축을 해제하여 지정된 디렉토리에 저장합니다.
+    이미 존재하는 경우에는 다운로드를 건너뜁니다.
+    """
+    if UBLOCK_DIR.exists() and (UBLOCK_DIR / "manifest.json").exists():
+        print("✅ uBlock Origin already present.")
+        return
+    if not UBLOCK_DIR.parent.exists():
+        UBLOCK_DIR.parent.mkdir(parents=True, exist_ok=True)
+
+    print("⬇️ Downloading uBlock Origin from GitHub API...")
+
+    # 1. GitHub API로 최신 릴리스 정보 가져오기
+    api_url = "https://api.github.com/repos/gorhill/uBlock/releases/latest"
+    res = requests.get(api_url)
+    res.raise_for_status()
+    data = res.json()
+
+    # 2. assets 중 'uBlock0.chromium.zip' 찾기
+    asset = next((a for a in data["assets"] if ".chromium.zip" in a["name"]), None)
+    if not asset:
+        raise Exception("❌ Could not find uBlock0.chromium.zip in GitHub release.")
+
+    zip_url = asset["browser_download_url"]
+
+    # 3. 다운로드
+    zip_path = Path("./browser/ublock.zip")
+    with requests.get(zip_url, stream=True) as r:
+        r.raise_for_status()
+        with open(zip_path, "wb") as f:
+            for chunk in r.iter_content(chunk_size=8192):
+                f.write(chunk)
+
+    # 4. 압축 해제 후 내부 디렉터리 이동
+    with zipfile.ZipFile(zip_path, 'r') as zip_ref:
+        zip_ref.extractall(TEMP_EXTRACT_DIR)
+
+    # zip 안에 uBlock0.chromium/ 폴더가 있다고 가정
+    extracted_root = next(TEMP_EXTRACT_DIR.iterdir())
+    if extracted_root.name != "uBlock0.chromium":
+        raise Exception("❌ Unexpected directory inside zip:", extracted_root)
+
+    shutil.move(str(extracted_root), UBLOCK_DIR)
+
+    shutil.rmtree(TEMP_EXTRACT_DIR, ignore_errors=True)
+    zip_path.unlink()  # zip 삭제
+    print("✅ uBlock Origin downloaded and extracted.")