mirror of
https://github.com/PluralKit/PluralKit.git
synced 2026-02-04 04:56:49 +00:00
105 lines
2.9 KiB
Python
Executable file
105 lines
2.9 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
|
|
import os, sys, json, subprocess, random, time
|
|
import urllib.request
|
|
|
|
def must_get_env(name):
|
|
val = os.environ.get(name)
|
|
if val == "":
|
|
raise "meow"
|
|
return val
|
|
|
|
def docker_build(data):
|
|
# file="", tags=[], root="/"
|
|
pass
|
|
|
|
def spawn_job(name):
|
|
req = urllib.request.Request(
|
|
f"https://api.github.com/repos/pluralkit/pluralkit/actions/workflows/ci-runner/dispatches",
|
|
method='POST',
|
|
headers={
|
|
'Accept': 'application/vnd.github+json',
|
|
'Authorization': f'Bearer {must_get_env("GITHUB_APP_TOKEN")}',
|
|
'content-type':'application/json'
|
|
},
|
|
data=json.dumps({
|
|
'ref': must_get_env("GIT_SHA"),
|
|
'inputs': {
|
|
'dispatchData': json.dumps({
|
|
'action': name,
|
|
})
|
|
}
|
|
})
|
|
)
|
|
|
|
try:
|
|
with urllib.request.urlopen(request) as response:
|
|
response_code = response.getcode()
|
|
response_data = response.read()
|
|
print(f"{response_code} spawned job {name}: {response_data}")
|
|
except urllib.error.HTTPError as e:
|
|
response_code = e.getcode()
|
|
response_data = e.read()
|
|
print(f"{response_code} failed to spawn job {name}: {response_data}")
|
|
|
|
def create_jobs():
|
|
modify_regexes = {
|
|
r'^ci/': "all",
|
|
|
|
r'^docs/': "bin_docs",
|
|
r'^dashboard/': "bin_dashboard",
|
|
|
|
r'\.rs$': "format_rs",
|
|
r'\.cs$': "format_cs",
|
|
|
|
r'^Cargo.lock': "all_rs",
|
|
|
|
r'^services/api': "bin_api",
|
|
# dispatch doesn't use libpk
|
|
r'^services/dispatch': "bin_dispatch",
|
|
r'^services/scheduled_tasks': "bin_scheduled_tasks",
|
|
|
|
# one image for all dotnet
|
|
r'^PluralKit\.': "bin_dotnet",
|
|
r'^Myriad': "bin_dotnet",
|
|
}
|
|
|
|
aliases = {
|
|
"all": ["bin_dotnet", "bin_api", "bin_dispatch", "bin_scheduled_tasks", "bin_dashboard"],
|
|
"all_rs": ["bin_api", "bin_dispatch"],
|
|
}
|
|
|
|
now = must_get_env("GIT_SHA")
|
|
before = must_get_env("OLD_SHA")
|
|
changed_files = subprocess.check_output(["git", "diff", "--name-only", before, now])
|
|
|
|
jobs = set([])
|
|
if must_get_env("IS_FORCE_PUSH") == "true":
|
|
jobs = jobs | aliases["all"]
|
|
jobs = jobs | ["format_cs", "format_rs"]
|
|
else:
|
|
for key in modify_regexes.keys():
|
|
if re.match(key, changed_file, flags=re.MULTILINE) is not None:
|
|
jobs = jobs | modify_regexes[key]
|
|
|
|
for key in changes:
|
|
if aliases.get(key) is not None:
|
|
jobs = jobs | aliases[key]
|
|
jobs = jobs - [key]
|
|
|
|
# test
|
|
jobs = jobs | ["test"]
|
|
|
|
# do this in a tx or something
|
|
for job in jobs:
|
|
spawn_job(job)
|
|
|
|
if len(jobs) == 0:
|
|
print("no jobs to run (??)")
|
|
|
|
return 0
|
|
|
|
if __name__ == "__main__":
|
|
print("hello from python!")
|
|
|
|
sys.exit(create_jobs())
|