refactor progress

scripts/ci/cli_release.py

@@ -0,0 +1,79 @@
+#!/usr/bin/env python3
+
+from __future__ import annotations
+
+from dataclasses import dataclass
+import hashlib
+from pathlib import Path
+from typing import Iterable
+
+from ci_utils import write_github_output
+
+
+@dataclass(frozen=True)
+class CliVersion:
+    version: str
+    tag: str
+
+
+def determine_cli_version(
+    *,
+    event_name: str,
+    input_version: str,
+    ref_name: str,
+    tag_prefix: str,
+) -> CliVersion:
+    if event_name == "workflow_dispatch":
+        version = input_version
+        if not version:
+            raise SystemExit("Missing version input")
+        tag = f"{tag_prefix}{version}"
+        return CliVersion(version=version, tag=tag)
+
+    if not ref_name:
+        raise SystemExit("Missing ref name")
+    version = ref_name
+    if version.startswith(tag_prefix):
+        version = version[len(tag_prefix) :]
+    if not version:
+        raise SystemExit("Unable to determine version from ref")
+    return CliVersion(version=version, tag=ref_name)
+
+
+def write_cli_version_outputs(info: CliVersion) -> None:
+    write_github_output({"version": info.version, "tag": info.tag})
+
+
+def prepare_release_assets(
+    *,
+    artifacts_dir: Path,
+    release_dir: Path,
+    binary_prefix: str,
+) -> list[Path]:
+    release_dir.mkdir(parents=True, exist_ok=True)
+    output_files: list[Path] = []
+
+    for entry in sorted(artifacts_dir.glob(f"{binary_prefix}-*")):
+        if not entry.is_dir():
+            continue
+        name = entry.name
+        source = entry / name
+        if not source.exists():
+            raise SystemExit(f"Missing binary {source}")
+        target = release_dir / name
+        target.write_bytes(source.read_bytes())
+        target.chmod(0o755)
+        output_files.append(target)
+
+    if not output_files:
+        raise SystemExit("No release assets found")
+
+    return output_files
+
+
+def generate_checksums(files: Iterable[Path], checksums_path: Path) -> None:
+    lines: list[str] = []
+    for path in sorted(files):
+        digest = hashlib.sha256(path.read_bytes()).hexdigest()
+        lines.append(f"{digest}  {path.name}")
+    checksums_path.write_text("\n".join(lines) + "\n", encoding="utf-8")