mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use sum type for WorkflowRunType

Browse Source
This commit is contained in:
Jakub Beránek 2024-05-02 21:42:36 +02:00
parent 02f7806ecd
commit 6778ecf960
No known key found for this signature in database
GPG Key ID: 909CD0D26483516B
1 changed files with 26 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
src/ci/github-actions/calculate-job-matrix.py
View File

@ -8,10 +8,10 @@ It reads job definitions from `src/ci/github-actions/jobs.yml`
and filters them based on the event that happened on CI. and filters them based on the event that happened on CI.
""" """
import dataclasses import dataclasses
import enum
import json import json
import logging import logging
import os import os
import typing
from pathlib import Path from pathlib import Path
from typing import List, Dict, Any, Optional from typing import List, Dict, Any, Optional
@ -44,10 +44,22 @@ def add_base_env(jobs: List[Job], environment: Dict[str, str]) -> List[Job]:
return jobs return jobs
class WorkflowRunType(enum.Enum): @dataclasses.dataclass
PR = enum.auto() class PRRunType:
Try = enum.auto() pass
Auto = enum.auto()
@dataclasses.dataclass
class TryRunType:
custom_jobs: List[str]
@dataclasses.dataclass
class AutoRunType:
pass
WorkflowRunType = typing.Union[PRRunType, TryRunType, AutoRunType]
@dataclasses.dataclass @dataclasses.dataclass
@ -59,7 +71,7 @@ class GitHubCtx:
def find_run_type(ctx: GitHubCtx) -> Optional[WorkflowRunType]: def find_run_type(ctx: GitHubCtx) -> Optional[WorkflowRunType]:
if ctx.event_name == "pull_request": if ctx.event_name == "pull_request":
return WorkflowRunType.PR return PRRunType()
elif ctx.event_name == "push": elif ctx.event_name == "push":
old_bors_try_build = ( old_bors_try_build = (
ctx.ref in ("refs/heads/try", "refs/heads/try-perf") and ctx.ref in ("refs/heads/try", "refs/heads/try-perf") and
@ -72,20 +84,20 @@ def find_run_type(ctx: GitHubCtx) -> Optional[WorkflowRunType]:
try_build = old_bors_try_build or new_bors_try_build try_build = old_bors_try_build or new_bors_try_build
if try_build: if try_build:
return WorkflowRunType.Try return TryRunType()
if ctx.ref == "refs/heads/auto" and ctx.repository == "rust-lang-ci/rust": if ctx.ref == "refs/heads/auto" and ctx.repository == "rust-lang-ci/rust":
return WorkflowRunType.Auto return AutoRunType()
return None return None
def calculate_jobs(run_type: WorkflowRunType, job_data: Dict[str, Any]) -> List[Job]: def calculate_jobs(run_type: WorkflowRunType, job_data: Dict[str, Any]) -> List[Job]:
if run_type == WorkflowRunType.PR: if isinstance(run_type, PRRunType):
return add_base_env(name_jobs(job_data["pr"], "PR"), job_data["envs"]["pr"]) return add_base_env(name_jobs(job_data["pr"], "PR"), job_data["envs"]["pr"])
elif run_type == WorkflowRunType.Try: elif isinstance(run_type, TryRunType):
return add_base_env(name_jobs(job_data["try"], "try"), job_data["envs"]["try"]) return add_base_env(name_jobs(job_data["try"], "try"), job_data["envs"]["try"])
elif run_type == WorkflowRunType.Auto: elif isinstance(run_type, AutoRunType):
return add_base_env(name_jobs(job_data["auto"], "auto"), job_data["envs"]["auto"]) return add_base_env(name_jobs(job_data["auto"], "auto"), job_data["envs"]["auto"])
return [] return []
@ -107,11 +119,11 @@ def get_github_ctx() -> GitHubCtx:
def format_run_type(run_type: WorkflowRunType) -> str: def format_run_type(run_type: WorkflowRunType) -> str:
if run_type == WorkflowRunType.PR: if isinstance(run_type, PRRunType):
return "pr" return "pr"
elif run_type == WorkflowRunType.Auto: elif isinstance(run_type, AutoRunType):
return "auto" return "auto"
elif run_type == WorkflowRunType.Try: elif isinstance(run_type, TryRunType):
return "try" return "try"
else: else:
raise AssertionError() raise AssertionError()