mirror of
https://github.com/ManchildProductions/UXP-Fixed.git
synced 2026-05-27 14:48:27 +00:00
Matt A. Tobin
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
# 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/.
|
|
"""
|
|
Support for running jobs that are invoked via the `run-task` script.
|
|
"""
|
|
|
|
from __future__ import absolute_import, print_function, unicode_literals
|
|
|
|
import copy
|
|
|
|
from taskgraph.transforms.job import run_job_using
|
|
from taskgraph.transforms.job.common import (
|
|
docker_worker_support_vcs_checkout,
|
|
)
|
|
from voluptuous import Schema, Required, Any
|
|
|
|
run_task_schema = Schema({
|
|
Required('using'): 'run-task',
|
|
|
|
# if true, add a cache at ~worker/.cache, which is where things like pip
|
|
# tend to hide their caches. This cache is never added for level-1 jobs.
|
|
Required('cache-dotcache', default=False): bool,
|
|
|
|
# if true (the default), perform a checkout in /home/worker/checkouts/gecko
|
|
Required('checkout', default=True): bool,
|
|
|
|
# The command arguments to pass to the `run-task` script, after the
|
|
# checkout arguments. If a list, it will be passed directly; otherwise
|
|
# it will be included in a single argument to `bash -cx`.
|
|
Required('command'): Any([basestring], basestring),
|
|
})
|
|
|
|
|
|
@run_job_using("docker-worker", "run-task", schema=run_task_schema)
|
|
def docker_worker_run_task(config, job, taskdesc):
|
|
run = job['run']
|
|
|
|
worker = taskdesc['worker'] = copy.deepcopy(job['worker'])
|
|
|
|
if run['checkout']:
|
|
docker_worker_support_vcs_checkout(config, job, taskdesc)
|
|
|
|
if run.get('cache-dotcache') and int(config.params['level']) > 1:
|
|
worker['caches'].append({
|
|
'type': 'persistent',
|
|
'name': 'level-{level}-{project}-dotcache'.format(**config.params),
|
|
'mount-point': '/home/worker/.cache',
|
|
})
|
|
|
|
run_command = run['command']
|
|
if isinstance(run_command, basestring):
|
|
run_command = ['bash', '-cx', run_command]
|
|
command = ['/home/worker/bin/run-task']
|
|
if run['checkout']:
|
|
command.append('--vcs-checkout=/home/worker/checkouts/gecko')
|
|
command.append('--')
|
|
command.extend(run_command)
|
|
worker['command'] = command
|