|
|
# try_server.py - Interact with Try server
|
|
|
#
|
|
|
# Copyright 2019 Gregory Szorc <gregory.szorc@gmail.com>
|
|
|
#
|
|
|
# This software may be used and distributed according to the terms of the
|
|
|
# GNU General Public License version 2 or any later version.
|
|
|
|
|
|
# no-check-code because Python 3 native.
|
|
|
|
|
|
import base64
|
|
|
import json
|
|
|
import os
|
|
|
import subprocess
|
|
|
import tempfile
|
|
|
|
|
|
from .aws import AWSConnection
|
|
|
|
|
|
LAMBDA_FUNCTION = "ci-try-server-upload"
|
|
|
|
|
|
|
|
|
def trigger_try(c: AWSConnection, rev="."):
|
|
|
"""Trigger a new Try run."""
|
|
|
lambda_client = c.session.client("lambda")
|
|
|
|
|
|
cset, bundle = generate_bundle(rev=rev)
|
|
|
|
|
|
payload = {
|
|
|
"bundle": base64.b64encode(bundle).decode("utf-8"),
|
|
|
"node": cset["node"],
|
|
|
"branch": cset["branch"],
|
|
|
"user": cset["user"],
|
|
|
"message": cset["desc"],
|
|
|
}
|
|
|
|
|
|
print("resolved revision:")
|
|
|
print("node: %s" % cset["node"])
|
|
|
print("branch: %s" % cset["branch"])
|
|
|
print("user: %s" % cset["user"])
|
|
|
print("desc: %s" % cset["desc"].splitlines()[0])
|
|
|
print()
|
|
|
|
|
|
print("sending to Try...")
|
|
|
res = lambda_client.invoke(
|
|
|
FunctionName=LAMBDA_FUNCTION,
|
|
|
InvocationType="RequestResponse",
|
|
|
Payload=json.dumps(payload).encode("utf-8"),
|
|
|
)
|
|
|
|
|
|
body = json.load(res["Payload"])
|
|
|
for message in body:
|
|
|
print("remote: %s" % message)
|
|
|
|
|
|
|
|
|
def generate_bundle(rev="."):
|
|
|
"""Generate a bundle suitable for use by the Try service.
|
|
|
|
|
|
Returns a tuple of revision metadata and raw Mercurial bundle data.
|
|
|
"""
|
|
|
# `hg bundle` doesn't support streaming to stdout. So we use a temporary
|
|
|
# file.
|
|
|
path = None
|
|
|
try:
|
|
|
fd, path = tempfile.mkstemp(prefix="hg-bundle-", suffix=".hg")
|
|
|
os.close(fd)
|
|
|
|
|
|
args = [
|
|
|
"hg",
|
|
|
"bundle",
|
|
|
"--type",
|
|
|
"gzip-v2",
|
|
|
"--base",
|
|
|
"public()",
|
|
|
"--rev",
|
|
|
rev,
|
|
|
path,
|
|
|
]
|
|
|
|
|
|
print("generating bundle...")
|
|
|
subprocess.run(args, check=True)
|
|
|
|
|
|
with open(path, "rb") as fh:
|
|
|
bundle_data = fh.read()
|
|
|
|
|
|
finally:
|
|
|
if path:
|
|
|
os.unlink(path)
|
|
|
|
|
|
args = [
|
|
|
"hg",
|
|
|
"log",
|
|
|
"-r",
|
|
|
rev,
|
|
|
# We have to upload as JSON, so it won't matter if we emit binary
|
|
|
# since we need to normalize to UTF-8.
|
|
|
"-T",
|
|
|
"json",
|
|
|
]
|
|
|
res = subprocess.run(args, check=True, capture_output=True)
|
|
|
return json.loads(res.stdout)[0], bundle_data
|
|
|
|
