refactor progress

scripts/ci/workflows/channel_vars.py

@@ -0,0 +1,55 @@
+#!/usr/bin/env python3
+
+import os
+import pathlib
+import sys
+
+sys.path.append(str(pathlib.Path(__file__).resolve().parents[1]))
+
+from ci_workflow import EnvArg, parse_env_args
+from ci_utils import require_env, write_github_output
+
+
+ENV_ARGS = [
+    EnvArg("--event-name", "EVENT_NAME"),
+    EnvArg("--ref-name", "REF_NAME"),
+    EnvArg("--dispatch-channel", "DISPATCH_CHANNEL"),
+]
+
+
+def determine_channel(
+    *,
+    event_name: str,
+    ref_name: str,
+    dispatch_channel: str,
+) -> str:
+    if event_name == "push":
+        return "canary" if ref_name == "canary" else "stable"
+    return "canary" if dispatch_channel == "canary" else "stable"
+
+
+def main() -> int:
+    parse_env_args(ENV_ARGS)
+    require_env(["EVENT_NAME"])
+    channel = determine_channel(
+        event_name=os.environ.get("EVENT_NAME", ""),
+        ref_name=os.environ.get("REF_NAME", ""),
+        dispatch_channel=os.environ.get("DISPATCH_CHANNEL", ""),
+    )
+
+    stack_suffix = "-canary" if channel == "canary" else ""
+    is_canary = "true" if channel == "canary" else "false"
+
+    write_github_output(
+        {
+            "channel": channel,
+            "is_canary": is_canary,
+            "stack_suffix": stack_suffix,
+        }
+    )
+
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())