#!/usr/bin/env -S uv run

# /// script

# dependencies = [

# "PyGithub",

# "requests",

# ]

# ///

import os

from pathlib import Path

import requests

from github import Github

from github.GitReleaseAsset import GitReleaseAsset

EXTENSION_ID = "bonsai"

CURRENT_PYTHON_VERSION = "py313"

CURRENT_PLATFORMS = ["linux-x64", "macos-arm64", "windows-x64"]

def publish_asset(asset: GitReleaseAsset, token: str, repo_root: Path) -> None:

"""

Publish an asset to Blender Extensions.

Reference: https://extensions.blender.org/api/v1/swagger

"""

temp_path = repo_root / asset.name

response = requests.get(asset.browser_download_url)

response.raise_for_status()

temp_path.write_bytes(response.content)

url = f"https://extensions.blender.org/api/v1/extensions/{EXTENSION_ID}/versions/upload/"

headers = {"Authorization": f"Bearer {token}"}

files = {"version_file": temp_path.read_bytes()}

response = requests.post(url, headers=headers, files=files)

response.raise_for_status()

temp_path.unlink()

print(f"✓ Published {asset.name}")

def main() -> None:

token = os.getenv("BLENDER_EXTENSIONS_TOKEN")

if not token:

raise Exception("BLENDER_EXTENSIONS_TOKEN environment variable not set")

# Get the repository root

repo_root = Path(__file__).parent.parent.parent

# Read VERSION file

version_file = repo_root / "VERSION"

version = version_file.read_text().strip()

print(f"Current VERSION: {version}")

tag_name = f"bonsai-{version}"

# Get release from GitHub

gh = Github()

gh_repo = gh.get_repo("IfcOpenShell/IfcOpenShell")

release = gh_repo.get_release(tag_name)

assets = release.get_assets()

asset_platform_map: dict[str, tuple[GitReleaseAsset, str]] = {}

for asset in assets:

if CURRENT_PYTHON_VERSION not in asset.name:

continue

for platform in CURRENT_PLATFORMS:

if platform in asset.name:

asset_platform_map[asset.name] = (asset, platform)

break

if len(asset_platform_map) != len(CURRENT_PLATFORMS):

found_platforms = {platform for _, (_, platform) in asset_platform_map.items()}

missing_platforms = set(CURRENT_PLATFORMS) - found_platforms

raise Exception(

f"Expected {len(CURRENT_PLATFORMS)} assets but found {len(asset_platform_map)}. "

f"Missing: {', '.join(sorted(missing_platforms))}"

)

print("\nRelease assets:")

for asset_name in sorted(asset_platform_map.keys()):

print(f"- {asset_name}")

# https://extensions.blender.org/api/v1/swagger

print("\nPublishing assets to Blender Extensions:")

for asset_name, (asset, platform) in asset_platform_map.items():

publish_asset(asset, token, repo_root)

if __name__ == "__main__":

main()