wehub-resource-sync
99 lines
3.0 KiB
Python
99 lines
3.0 KiB
Python
#!/usr/bin/env -S uv run --script
|
|
# /// script
|
|
# requires-python = ">=3.12"
|
|
# dependencies = [
|
|
# "httpx==0.28.1",
|
|
# ]
|
|
# ///
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import re
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
import httpx
|
|
|
|
RELEASES_PATH = Path(__file__).resolve().parent.parent / "website/src/lib/latest_release.json"
|
|
OWNER = "thewh1teagle"
|
|
REPO = "vibe"
|
|
BLACKLIST_PATTERNS = ["opencl", "nvidia", "older", ".rpm", "portable", "cuda", "ubuntu", "setup_", "arm64-setup"]
|
|
VALID_EXTENSION_RE = re.compile(r"\.(sig|json|zip|tar\.gz)$")
|
|
|
|
|
|
def get_asset_info(name: str) -> dict[str, str]:
|
|
platform_map = {
|
|
".deb": {"platform": "Linux", "arch": "darwin-aarch64" if "aarch64" in name else "linux-x86_64"},
|
|
".exe": {"platform": "Windows", "arch": "windows-x86_64" if "x64-setup" in name else "unknown"},
|
|
".dmg": {"platform": "MacOS", "arch": "darwin-aarch64" if "aarch64" in name else "darwin-x86_64"},
|
|
}
|
|
|
|
extension = next((ext for ext in platform_map if ext in name), "unknown")
|
|
return platform_map.get(extension, {})
|
|
|
|
|
|
def filter_valid_assets(assets: list[dict]) -> list[dict]:
|
|
return [
|
|
asset
|
|
for asset in assets
|
|
if not VALID_EXTENSION_RE.search(asset.get("name", ""))
|
|
and not any(pattern in asset.get("name", "") for pattern in BLACKLIST_PATTERNS)
|
|
]
|
|
|
|
|
|
def map_assets(assets: list[dict]) -> list[dict]:
|
|
out: list[dict] = []
|
|
for asset in assets:
|
|
info = get_asset_info(asset.get("name", ""))
|
|
out.append(
|
|
{
|
|
"url": asset.get("browser_download_url"),
|
|
"name": asset.get("name"),
|
|
**info,
|
|
}
|
|
)
|
|
return out
|
|
|
|
|
|
def fetch_latest_release() -> dict:
|
|
headers: dict[str, str] = {}
|
|
gh_token = os.getenv("GH_TOKEN")
|
|
if gh_token:
|
|
headers["Authorization"] = f"Bearer {gh_token}"
|
|
|
|
try:
|
|
response = httpx.get(
|
|
f"https://api.github.com/repos/{OWNER}/{REPO}/releases/latest",
|
|
headers=headers,
|
|
follow_redirects=True,
|
|
timeout=60,
|
|
)
|
|
response.raise_for_status()
|
|
return response.json()
|
|
except httpx.HTTPStatusError as exc:
|
|
body = exc.response.text
|
|
raise RuntimeError(
|
|
f"Failed to fetch latest release. Status: {exc.response.status_code} URL: {exc.request.url} Body: {body}"
|
|
) from exc
|
|
|
|
|
|
def main() -> int:
|
|
try:
|
|
latest_release = fetch_latest_release()
|
|
tag_name = latest_release["tag_name"]
|
|
valid_assets = filter_valid_assets(latest_release.get("assets", []))
|
|
mapped_assets = map_assets(valid_assets)
|
|
releases_data = {"assets": mapped_assets, "version": tag_name}
|
|
RELEASES_PATH.write_text(json.dumps(releases_data, indent=4), encoding="utf-8")
|
|
print(f"Updated releases at {RELEASES_PATH} with \n{json.dumps(releases_data, indent=4)}")
|
|
return 0
|
|
except Exception as exc:
|
|
print(f"Error fetching or processing latest release: {exc}", file=sys.stderr)
|
|
return 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|