summaryrefslogtreecommitdiffstats
path: root/third_party/rust/jsparagus/smoosh_status.py
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-07 19:33:14 +0000
commit36d22d82aa202bb199967e9512281e9a53db42c9 (patch)
tree105e8c98ddea1c1e4784a60a5a6410fa416be2de /third_party/rust/jsparagus/smoosh_status.py
parentInitial commit. (diff)
downloadfirefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz
firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip
Adding upstream version 115.7.0esr.upstream/115.7.0esrupstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'third_party/rust/jsparagus/smoosh_status.py')
-rw-r--r--third_party/rust/jsparagus/smoosh_status.py193
1 files changed, 193 insertions, 0 deletions
diff --git a/third_party/rust/jsparagus/smoosh_status.py b/third_party/rust/jsparagus/smoosh_status.py
new file mode 100644
index 0000000000..63aeb3e0fd
--- /dev/null
+++ b/third_party/rust/jsparagus/smoosh_status.py
@@ -0,0 +1,193 @@
+#!/usr/bin/env python3
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this file,
+# You can obtain one at http://mozilla.org/MPL/2.0/.
+
+import pathlib
+import json
+import urllib.request
+import re
+import subprocess
+import sys
+
+
+class Logger:
+ @classmethod
+ def info(cls, s):
+ print('[INFO]', s)
+
+ # Flush to make it apeear immediately in automation log.
+ sys.stdout.flush()
+
+ @classmethod
+ def fetch(cls, url):
+ cls.info(f'Fetching {url}')
+
+ @classmethod
+ def cmd(cls, cmd):
+ def format_cmd(s):
+ if ' ' in s:
+ escaped = s.replace('"', '\"')
+ return f'"{escaped}"'
+ return s
+
+ formatted_command = ' '.join(list(map(format_cmd, cmd)))
+ cls.info(f'$ {formatted_command}')
+
+
+class GitRepository:
+ def __init__(self, path):
+ self.path = path
+
+ self.git_dir = self.path / '.git'
+ if not self.git_dir.exists():
+ print(f'{self.path} is not a Git repository.', file=sys.stderr)
+ sys.exit(1)
+
+ def get_output(self, *args):
+ cmd = ['git'] + list(args)
+ Logger.cmd(cmd)
+ output = subprocess.run(cmd,
+ capture_output=True,
+ cwd=self.path)
+
+ return output.stdout.decode()
+
+ def run(self, *args):
+ cmd = ['git'] + list(args)
+ Logger.cmd(cmd)
+ subprocess.run(cmd,
+ check=True,
+ cwd=self.path)
+
+ def commit_message(self, rev):
+ return self.get_output('log', '-1', '--pretty=format:%s%n', rev)
+
+
+class MCRemoteRepository:
+ HG_API_URL = 'https://hg.mozilla.org/mozilla-central/'
+
+ @classmethod
+ def call(cls, name, path):
+ url = f'{cls.HG_API_URL}{name}{path}'
+ Logger.fetch(url)
+ req = urllib.request.Request(url, None, {})
+ response = urllib.request.urlopen(req)
+ return response.read()
+
+ @classmethod
+ def call_json(cls, name, path):
+ return json.loads(cls.call(name, path))
+
+ @classmethod
+ def file(cls, rev, path):
+ return cls.call('raw-file', f'/{rev}{path}')
+
+
+class TreeHerder:
+ API_URL = 'https://treeherder.mozilla.org/api/'
+
+ @classmethod
+ def call(cls, name):
+ url = f'{cls.API_URL}{name}'
+ Logger.fetch(url)
+ req = urllib.request.Request(url, None, {
+ 'User-Agent': 'smoosh-tools',
+ })
+ response = urllib.request.urlopen(req)
+ return response.read()
+
+ @classmethod
+ def call_json(cls, name):
+ return json.loads(cls.call(name))
+
+ @classmethod
+ def push_id(cls, rev):
+ push = cls.call_json(f'project/mozilla-central/push/?full=true&format=json&count=1&revision={rev}')
+ return push['results'][0]['id']
+
+ @classmethod
+ def jobs(cls, push_id):
+ push = cls.call_json(f'jobs/?push_id={push_id}&format=json')
+ count = push['count']
+ results = []
+ results += push['results']
+
+ page = 2
+ while len(results) < count:
+ push = cls.call_json(f'jobs/?push_id={push_id}&format=json&page={page}')
+ results += push['results']
+ page += 1
+
+ return results
+
+
+class Status:
+ def run(is_ci):
+ Logger.info('Fetching ci_generated branch')
+
+ jsparagus = GitRepository(pathlib.Path('./'))
+ jsparagus.run('fetch', 'origin', 'ci_generated')
+
+ Logger.info('Checking mozilla-central tip revision')
+
+ m_c_rev = MCRemoteRepository.call_json('json-log', '/tip/')['node']
+ cargo_file = MCRemoteRepository.file(
+ m_c_rev,
+ '/js/src/frontend/smoosh/Cargo.toml'
+ ).decode()
+ m = re.search('rev = "(.+)"', cargo_file)
+ ci_generated_rev = m.group(1)
+
+ Logger.info('Checking jsparagus referred by mozilla-central')
+
+ message = jsparagus.commit_message(ci_generated_rev)
+ m = re.search('for ([A-Fa-f0-9]+)', message)
+ master_rev = m.group(1)
+
+ Logger.info('Checking build status')
+
+ push_id = TreeHerder.push_id(m_c_rev)
+ jobs = TreeHerder.jobs(push_id)
+ nonunified_job = None
+ smoosh_job = None
+ for job in jobs:
+ if 'spidermonkey-sm-nonunified-linux64/debug' in job:
+ nonunified_job = job
+ if 'spidermonkey-sm-smoosh-linux64/debug' in job:
+ smoosh_job = job
+
+ def get_result(job):
+ if job:
+ if 'completed' in job:
+ if 'success' in job:
+ return 'OK'
+ else:
+ return 'NG'
+ else:
+ return 'not yet finished'
+ else:
+ return 'unknown'
+
+ nonunified_result = get_result(nonunified_job)
+ smoosh_result = get_result(smoosh_job)
+
+ if is_ci:
+ print(f'##[set-output name=mc;]{m_c_rev}')
+ print(f'##[set-output name=jsparagus;]{master_rev}')
+ print(f'##[set-output name=build;]{nonunified_result}')
+ print(f'##[set-output name=test;]{smoosh_result}')
+ else:
+ print(f'mozilla-central tip: {m_c_rev}')
+ print(f'referred jsparagus revision: {master_rev}')
+ print(f'Build status:')
+ print(f' Build with --enable-smoosh: {nonunified_result}')
+ print(f' Test with --smoosh: {smoosh_result}')
+
+
+is_ci = False
+if len(sys.argv) > 1:
+ if sys.argv[1] == 'ci':
+ is_ci = True
+
+Status.run(is_ci)