recipes/recipes/images_builder.py - infra/infra.git - Git at Google

 # Copyright 2019 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.

 from contextlib import contextmanager
 from collections import defaultdict, namedtuple
 from hashlib import sha256
 from recipe_engine import recipe_api

 from PB.recipes.infra import images_builder as images_builder_pb

 DEPS = [
     'recipe_engine/buildbucket',
     'recipe_engine/futures',
     'recipe_engine/json',
     'recipe_engine/path',
     'recipe_engine/properties',
     'recipe_engine/step',
     'recipe_engine/time',
     'depot_tools/gerrit',
     'depot_tools/git',
     'buildenv',
     'cloudbuildhelper',
     'infra_checkout',
 ]


 PROPERTIES = images_builder_pb.Inputs


 # Metadata is returned by _checkout_* and applied to built images.
 Metadata = namedtuple('Metadata', [
     'canonical_tag',  # str or None
     'labels',         # {str: str}
     'tags',           # [str]
     'checkout',       # cloudbuildhelper.CheckoutMetadata
 ])


 # Prefer to use latest greatest Go version for binaries inside Docker images.
 GO_VERSION_VARIANT = 'bleeding_edge'


 def RunSteps(api, properties):
   try:
     _validate_props(properties)
   except ValueError as exc:
     raise recipe_api.InfraFailure('Bad input properties: %s' % exc)

   # Checkout either the committed code or a pending CL, depending on the mode.
   # This also calculates metadata (labels, tags) to apply to images built from
   # this code.
   if properties.project == PROPERTIES.PROJECT_GIT_REPO:
     meta, build_env = _checkout_git(api, properties.git_repo, properties.mode)
   elif properties.mode in (PROPERTIES.MODE_CI, PROPERTIES.MODE_TS):
     meta, build_env = _checkout_committed(
         api, properties.mode, properties.project)
   elif properties.mode == PROPERTIES.MODE_CL:
     meta, build_env = _checkout_pending(api, properties.project)
   else:  # pragma: no cover
     raise recipe_api.InfraFailure(
         'Unknown mode %s' % PROPERTIES.Mode.Name(properties.mode))

   # Discover what *.yaml manifests (full paths to them) we need to build.
   manifests = api.cloudbuildhelper.discover_manifests(
       meta.checkout.root, properties.manifests)
   if not manifests:  # pragma: no cover
     raise recipe_api.InfraFailure('Found no manifests to build')

   with build_env(api):
     # Report the exact version we going to use.
     api.cloudbuildhelper.report_version()

     # Build, tag and upload corresponding images (in parallel).
     futures = {}
     for m in manifests:
       fut = api.futures.spawn(
           api.cloudbuildhelper.build,
           manifest=m,
           canonical_tag=meta.canonical_tag,
           build_id=api.buildbucket.build_url(),
           infra=properties.infra,
           labels=meta.labels,
           tags=meta.tags,
           checkout_metadata=meta.checkout,
           cost=api.step.ResourceCost(cpu=2000))
       futures[fut] = m

   # Wait until all builds complete.
   built = []
   fails = []
   for fut in api.futures.iwait(list(futures.keys())):
     try:
       img = fut.result()
       if img != api.cloudbuildhelper.NotUploadedImage:
         built.append(img)
     except api.step.StepFailure:
       fails.append(api.path.basename(futures[fut]))

   # Group successfully built images by their roll destinations.
   per_notify = defaultdict(list)
   for img in built:
     for n in img.notify:
       per_notify[n].append(img)

   # Perform all rolls in parallel.
   futures = {}
   for notify in sorted(per_notify):
     fut = api.futures.spawn(
         _roll_built_images,
         api=api,
         notify=notify,
         images=per_notify[notify],
         meta=meta)
     futures[fut] = notify

   # Wait for all rolls to finish.
   roll_fails = []
   for fut in api.futures.iwait(list(futures.keys())):
     try:
       fut.result()
     except api.step.StepFailure:
       roll_fails.append(futures[fut].repo)

   if fails:
     raise recipe_api.StepFailure('Failed to build: %s' % ', '.join(fails))
   if roll_fails:
     raise recipe_api.StepFailure('Failed to roll: %s' % ', '.join(roll_fails))


 def _validate_props(p):  # pragma: no cover
   if p.mode == PROPERTIES.MODE_UNDEFINED:
     raise ValueError('"mode" is required')
   if p.project == PROPERTIES.PROJECT_UNDEFINED:
     raise ValueError('"project" is required')
   if not p.infra:
     raise ValueError('"infra" is required')
   if not p.manifests:
     raise ValueError('"manifests" is required')
   # There's no CI/TS for luci-go. Its CI happens when it gets rolled in into
   # infra.git. But we still can run tryjobs for luci-go by applying CLs on top
   # of infra.git checkout.
   if p.project == PROPERTIES.PROJECT_LUCI_GO and p.mode != PROPERTIES.MODE_CL:
     raise ValueError('PROJECT_LUCI_GO can be used only together with MODE_CL')

   if p.project == PROPERTIES.PROJECT_GIT_REPO:
     if not p.HasField('git_repo'):
       raise ValueError('"git_repo" is required when using PROJECT_GIT_REPO')
     if p.mode == PROPERTIES.MODE_CL:
       raise ValueError('PROJECT_GIT_REPO cannot be used together with MODE_CL')
   elif p.HasField('git_repo'):
     raise ValueError('"git_repo" can only be set when using PROJECT_GIT_REPO')


 def _checkout_committed(api, mode, project):
   """Checks out some committed revision (based on Buildbucket properties).

   Args:
     api: recipes API.
     mode: PROPERTIES.Mode enum (either MODE_CI or MODE_TS).
     project: PROPERTIES.Project enum.

   Returns:
     (Metadata, build environment context manager).
   """
   conf, internal, repo_url = {
     PROPERTIES.PROJECT_INFRA: (
         'infra_superproject',
         False,
         'https://chromium.googlesource.com/infra/infra',
     ),
     PROPERTIES.PROJECT_INFRA_INTERNAL: (
         'infra_internal_superproject',
         True,
         'https://chrome-internal.googlesource.com/infra/infra_internal',
     ),
   }[project]

   co = api.infra_checkout.checkout(
       gclient_config_name=conf,
       internal=internal,
       go_version_variant=GO_VERSION_VARIANT)
   co.gclient_runhooks()

   props = co.bot_update_step.properties

   canonical_tag = None
   if mode == PROPERTIES.MODE_CI:
     # E.g. "41861-d008a93".
     commit_label = api.cloudbuildhelper.get_version_label(
           path=co.path / ('infra_internal' if internal else 'infra'),
           revision=props['got_revision'],
           ref=api.buildbucket.gitiles_commit.ref,
           commit_position=props.get('got_revision_cp'),
     )
     # E.g. "ci-2021.06.23-41861-d008a93".
     canonical_tag = 'ci-%s-%s' % (_date(api), commit_label)
   elif mode == PROPERTIES.MODE_TS:
     # E.g. "ts-2021.06.23-113234"
     canonical_tag = 'ts-%s-%d' % (_date(api), api.buildbucket.build.number)
   else:
     raise AssertionError('Impossible')  # pragma: no cover

   return Metadata(
       canonical_tag=canonical_tag,
       labels={
           'org.opencontainers.image.source': repo_url,
           'org.opencontainers.image.revision': props['got_revision'],
       },
       tags=['latest'],
       checkout=api.cloudbuildhelper.CheckoutMetadata(
           root=co.path,
           repos=co.bot_update_step.manifest,
       )), lambda api: _infra_checkout_build_environ(co, api)


 def _checkout_pending(api, project):
   """Checks out some pending CL (based on Buildbucket properties).

   Args:
     api: recipes API.
     project: PROPERTIES.Project enum.

   Returns:
     (Metadata, build environment context manager).
   """
   conf, patch_root, internal = {
     PROPERTIES.PROJECT_INFRA: (
         'infra',
         'infra',
         False,
     ),
     PROPERTIES.PROJECT_INFRA_INTERNAL: (
         'infra_internal',
         'infra_internal',
         True,
     ),
     PROPERTIES.PROJECT_LUCI_GO: (
         'infra',
         'infra/go/src/go.chromium.org/luci',
         False,
     ),
   }[project]

   co = api.infra_checkout.checkout(
       gclient_config_name=conf,
       patch_root=patch_root,
       internal=internal,
       go_version_variant=GO_VERSION_VARIANT)
   co.commit_change()
   co.gclient_runhooks()

   # Grab information about this CL (in particular who wrote it).
   cl = api.buildbucket.build.input.gerrit_changes[0]
   repo_url = 'https://%s/%s' % (cl.host, cl.project)
   rev_info = api.gerrit.get_revision_info(repo_url, cl.change, cl.patchset)
   author = rev_info['commit']['author']['email']

   return Metadata(
       # ':inputs-hash' essentially tells cloudbuildhelper to skip the build if
       # there's already an image built from the exact same inputs.
       canonical_tag=':inputs-hash',
       labels={'org.chromium.build.cl.repo': repo_url},
       tags=[
           # An "immutable" tag that identifies how the image was built.
           'cl-%s-%d-%d-%s' % (
               _date(api),
               cl.change,
               cl.patchset,
               author.split('@')[0],
           ),
           # A movable tag for "a latest image produced from this CL". It is
           # intentionally simple, so that developers can "guess" it just knowing
           # the CL number.
           'cl-%d' % cl.change,
       ],
       checkout=api.cloudbuildhelper.CheckoutMetadata(
           root=co.path,
           repos=co.bot_update_step.manifest,
       )), lambda api: _infra_checkout_build_environ(co, api)


 def _checkout_git(api, repo, mode):
   """Checks out a standalone Git repository.
   Checks out the commit passed via Buildbucket inputs or `refs/heads/main`.
   Args:
     api: recipes API.
     repo: PROPERTIES.GitRepo proto.
     version_label_template: a template for the version label string.
   Returns:
     (Metadata, build environment context manager).
   """
   path = api.path.cache_dir.joinpath('builder', 'repo')
   revision = api.git.checkout(
       url=repo.url,
       ref=api.buildbucket.gitiles_commit.id or 'refs/heads/main',
       dir_path=path,
       submodules=False)

   @contextmanager
   def build_environ(api):
     with api.buildenv(path, repo.go_version_file, repo.nodejs_version_file):
       yield

   canonical_tag = None
   if mode == PROPERTIES.MODE_CI:
     # E.g. "41861-d008a93".
     commit_label = api.cloudbuildhelper.get_version_label(
         path=path, revision=revision, ref=api.buildbucket.gitiles_commit.ref)
     # E.g. "ci-2021.06.23-41861-d008a93".
     canonical_tag = 'ci-%s-%s' % (_date(api), commit_label)
   elif mode == PROPERTIES.MODE_TS:
     # E.g. "ts-2021.06.23-113234"
     canonical_tag = 'ts-%s-%d' % (_date(api), api.buildbucket.build.number)
   else:
     raise AssertionError('Impossible')  # pragma: no cover

   return Metadata(
       canonical_tag=canonical_tag,
       labels={
           'org.opencontainers.image.source': repo.url,
           'org.opencontainers.image.revision': revision,
       },
       tags=['latest'],
       checkout=api.cloudbuildhelper.CheckoutMetadata(
           root=path,
           repos={'.': {
               'repository': repo.url,
               'revision': revision
           }},
       )), build_environ


 @contextmanager
 def _infra_checkout_build_environ(co, api):
   with co.go_env():
     # Use 'cloudbuildhelper' that comes with the infra checkout (it's in PATH),
     # to make sure builders use the same version as developers.
     api.cloudbuildhelper.command = 'cloudbuildhelper'
     yield


 def _date(api):
   """Returns UTC YYYY.MM.DD to use in tags."""
   return api.time.utcnow().strftime('%Y.%m.%d')


 def _roll_built_images(api, notify, images, meta):
   """Uploads a CL with info about built images into a repo with pinned images.

   Args:
     api: recipes API.
     notify: a CloudBuildHelperApi.NotifyConfig describing where to roll.
     images: a list of CloudBuildHelperApi.Image with info about built images.
     meta: Metadata struct, as returned by _checkout_committed.

   Returns:
     (None, None) if didn't create a CL (because nothing has changed).
     (Issue number, Issue URL) if created a CL.
   """
   repo_id = sha256(notify.repo.encode('utf-8')).hexdigest()[:8]
   with api.step.nest('upload roll CL') as pres:
     num, url = api.cloudbuildhelper.do_roll(
         repo_url=notify.repo,
         root=api.path.cache_dir.joinpath('builder', 'roll', repo_id),
         callback=lambda root: _mutate_repo(api, root, notify, images, meta))
     if num is not None:
       pres.links['Issue %s' % num] = url


 def _mutate_repo(api, root, notify, images, meta):
   """Modifies the checked out repo with image pins.

   Args:
     api: recipes API.
     root: the directory where the repo is checked out.
     notify: a CloudBuildHelperApi.NotifyConfig describing where to roll.
     images: a list of CloudBuildHelperApi.Image with info about built images.
     meta: Metadata struct, as returned by _checkout_committed.

   Returns:
     cloudbuildhelper.RollCL to proceed with the roll or None to skip it.
   """
   # RFC3389 timstamp in UTC zone.
   date = api.time.utcnow().isoformat('T') + 'Z'

   # Prepare tag JSON specs for all images.
   # See //scripts/roll_images.py in infradata/k8s repo.
   tags = []
   for img in images:
     tags.append({
         'image': img.image,
         'tag': {
             'tag': img.tag,
             'digest': img.digest,
             'context_dir': img.context_dir,
             'metadata': {
                 'date': date,
                 'source': {
                     'repo': meta.labels['org.opencontainers.image.source'],
                     'revision':
                         meta.labels['org.opencontainers.image.revision'],
                 },
                 'sources': img.sources,
                 'links': {
                     'buildbucket': api.buildbucket.build_url(),
                     'cloudbuild': img.view_build_url,
                     'gcr': img.view_image_url,
                 },
             },
         },
     })

   # Add all new tags (if any).
   res = api.step(
       name=notify.script,
       cmd=[root / notify.script],
       stdin=api.json.input({'tags': tags}),
       stdout=api.json.output(),
       step_test_data=lambda: api.json.test_api.output_stream(
           _roll_images_test_data(tags)))
   deployments = res.stdout.get('deployments') or []

   # Given a Deployment dict, returns the artifact name. Unfortunately the key
   # that contains it is different for infradata/k8s and infradata/gae
   # destinations.
   def artifact(dep):
     if 'image' in dep:
       return dep['image']
     return dep['artifact']  # pragma: no cover

   # Generate the commit message.
   message = [
       '[images] Rolling in images.',
       '',
       'Produced by %s' % api.buildbucket.build_url(),
       '',
   ]
   if deployments:
     message.extend([
         'Updated deployments:',
     ] + [
         '  * %s: %s -> %s' % (artifact(d), d['from'], d['to'])
         for d in deployments
     ] + [''])
   message = str('\n'.join(message))

   # List of people to CC based on what deployments were updated.
   extra_cc = set()
   for dep in deployments:
     extra_cc.update(dep.get('cc') or [])

   return api.cloudbuildhelper.RollCL(
       message=message,
       cc=extra_cc,
       tbr=[],
       commit=True)


 def _roll_images_test_data(tags):
   return {
       'deployments': [
           {
               'cc': ['n1@example.com', 'n2@example.com'],
               'from': 'prev-version',
               'image': t['image'],
               'to': t['tag']['tag'],
           }
           for t in tags
       ],
   }


 def GenTests(api):
   def try_props(project, cl, patch_set):
     return (
         api.buildbucket.try_build(
             project=project,
             change_number=cl,
             patch_set=patch_set) +
         api.override_step_data(
             'gerrit changes',
             api.json.output([{
                 'project': project,
                 '_number': cl,
                 'revisions': {
                     '184ebe53805e102605d11f6b143486d15c23a09c': {
                         '_number': patch_set,
                         'commit': {
                             'message': 'Commit message',
                             'author': {'email': 'author@example.com'},
                         },
                         'ref': 'refs/changes/../../..',
                     },
                 },
             }]),
         )
     )

   from RECIPE_MODULES.infra.cloudbuildhelper.api import CloudBuildHelperApi

   def build_success_with_notify():
     img = CloudBuildHelperApi.Image(
         image='example.com/fake-registry/target',
         digest='sha256:abcdef',
         tag='ci-2012.05.14-197293-5e03a58',
         context_dir='/some/context/directory/for/target',
         view_image_url=None,
         view_build_url=None,
         notify=[
             CloudBuildHelperApi.NotifyConfig(
                 kind='git',
                 repo='https://roll.example.com/repo',
                 script='scripts/roll.py',
             ),
         ],
         sources=[],
     )
     return api.cloudbuildhelper.build_success_output(img, mocked_sources=[
         '[CACHE]/infra_gclient_with_go/infra/a/b/c',
     ])

   yield (
       api.test('try-infra') +
       api.properties(
           mode=PROPERTIES.MODE_CL,
           project=PROPERTIES.PROJECT_INFRA,
           infra='dev',
           manifests=['infra/build/images/deterministic'],
       ) +
       try_props('infra/infra', 123456, 7)
   )

   yield (
       api.test('try-luci-go') +
       api.properties(
           mode=PROPERTIES.MODE_CL,
           project=PROPERTIES.PROJECT_LUCI_GO,
           infra='dev',
           manifests=['infra/build/images/deterministic'],
       ) +
       try_props('infra/luci/luci-go', 123456, 7)
   )

   yield (
       api.test('try-infra-internal') +
       api.properties(
           mode=PROPERTIES.MODE_CL,
           project=PROPERTIES.PROJECT_INFRA_INTERNAL,
           infra='dev',
           manifests=['infra_internal/build/images/deterministic'],
       ) +
       try_props('infra/infra_internal', 123456, 7)
   )

   yield (api.test('ci-git') + api.properties(
       mode=PROPERTIES.MODE_CI,
       project=PROPERTIES.PROJECT_GIT_REPO,
       infra='prod',
       git_repo=PROPERTIES.GitRepo(
           url='https://chromium.googlesource.com/infra/cop'),
       manifests=['build/images'],
   ))

   yield (
       api.test('ci-infra') +
       api.properties(
           mode=PROPERTIES.MODE_CI,
           project=PROPERTIES.PROJECT_INFRA,
           infra='prod',
           manifests=['infra/build/images/deterministic'],
       )
   )

   yield (
       api.test('ci-infra-internal') +
       api.properties(
           mode=PROPERTIES.MODE_CI,
           project=PROPERTIES.PROJECT_INFRA_INTERNAL,
           infra='prod',
           manifests=['infra_internal/build/images/deterministic'],
       )
   )

   yield (api.test('ts-git') + api.properties(
       mode=PROPERTIES.MODE_TS,
       project=PROPERTIES.PROJECT_GIT_REPO,
       infra='prod',
       git_repo=PROPERTIES.GitRepo(
           url='https://chromium.googlesource.com/infra/cop'),
       manifests=['build/images'],
   ))

   yield (
       api.test('ts-infra') +
       api.properties(
           mode=PROPERTIES.MODE_TS,
           project=PROPERTIES.PROJECT_INFRA,
           infra='prod',
           manifests=['infra/build/images/daily'],
       )
   )

   yield (
       api.test('ci-infra-with-roll') +
       api.properties(
           mode=PROPERTIES.MODE_CI,
           project=PROPERTIES.PROJECT_INFRA,
           infra='prod',
           manifests=['infra/build/images/deterministic'],
       ) +
       api.step_data(
           'cloudbuildhelper build target',
           build_success_with_notify(),
       ) +
       api.step_data('upload roll CL.git diff', retcode=1)
   )

   yield (
       api.test('ci-infra-with-roll-failure') +
       api.properties(
           mode=PROPERTIES.MODE_CI,
           project=PROPERTIES.PROJECT_INFRA,
           infra='prod',
           manifests=['infra/build/images/deterministic'],
       ) +
       api.step_data(
           'cloudbuildhelper build target',
           build_success_with_notify(),
       ) +
       api.step_data(
           'upload roll CL.scripts/roll.py', retcode=1
       ) +
       api.expect_status('FAILURE')
   )

   yield (
       api.test('build-failure') +
       api.properties(
           mode=PROPERTIES.MODE_CI,
           project=PROPERTIES.PROJECT_INFRA,
           infra='prod',
           manifests=['infra/build/images/deterministic'],
       ) +
       api.step_data(
           'cloudbuildhelper build target',
           api.cloudbuildhelper.build_error_output('Boom'),
           retcode=1
       ) +
       api.expect_status('FAILURE')
   )

   yield (
       api.test('bad-props') +
       api.properties(mode=0) +
       api.expect_status('INFRA_FAILURE')
   )
	# Copyright 2019 The Chromium Authors. All rights reserved.
	# Use of this source code is governed by a BSD-style license that can be
	# found in the LICENSE file.

	from contextlib import contextmanager
	from collections import defaultdict, namedtuple
	from hashlib import sha256
	from recipe_engine import recipe_api

	from PB.recipes.infra import images_builder as images_builder_pb

	DEPS = [
	'recipe_engine/buildbucket',
	'recipe_engine/futures',
	'recipe_engine/json',
	'recipe_engine/path',
	'recipe_engine/properties',
	'recipe_engine/step',
	'recipe_engine/time',
	'depot_tools/gerrit',
	'depot_tools/git',
	'buildenv',
	'cloudbuildhelper',
	'infra_checkout',
	]


	PROPERTIES = images_builder_pb.Inputs


	# Metadata is returned by _checkout_* and applied to built images.
	Metadata = namedtuple('Metadata', [
	'canonical_tag', # str or None
	'labels', # {str: str}
	'tags', # [str]
	'checkout', # cloudbuildhelper.CheckoutMetadata
	])


	# Prefer to use latest greatest Go version for binaries inside Docker images.
	GO_VERSION_VARIANT = 'bleeding_edge'


	def RunSteps(api, properties):
	try:
	_validate_props(properties)
	except ValueError as exc:
	raise recipe_api.InfraFailure('Bad input properties: %s' % exc)

	# Checkout either the committed code or a pending CL, depending on the mode.
	# This also calculates metadata (labels, tags) to apply to images built from
	# this code.
	if properties.project == PROPERTIES.PROJECT_GIT_REPO:
	meta, build_env = _checkout_git(api, properties.git_repo, properties.mode)
	elif properties.mode in (PROPERTIES.MODE_CI, PROPERTIES.MODE_TS):
	meta, build_env = _checkout_committed(
	api, properties.mode, properties.project)
	elif properties.mode == PROPERTIES.MODE_CL:
	meta, build_env = _checkout_pending(api, properties.project)
	else: # pragma: no cover
	raise recipe_api.InfraFailure(
	'Unknown mode %s' % PROPERTIES.Mode.Name(properties.mode))

	# Discover what *.yaml manifests (full paths to them) we need to build.
	manifests = api.cloudbuildhelper.discover_manifests(
	meta.checkout.root, properties.manifests)
	if not manifests: # pragma: no cover
	raise recipe_api.InfraFailure('Found no manifests to build')

	with build_env(api):
	# Report the exact version we going to use.
	api.cloudbuildhelper.report_version()

	# Build, tag and upload corresponding images (in parallel).
	futures = {}
	for m in manifests:
	fut = api.futures.spawn(
	api.cloudbuildhelper.build,
	manifest=m,
	canonical_tag=meta.canonical_tag,
	build_id=api.buildbucket.build_url(),
	infra=properties.infra,
	labels=meta.labels,
	tags=meta.tags,
	checkout_metadata=meta.checkout,
	cost=api.step.ResourceCost(cpu=2000))
	futures[fut] = m

	# Wait until all builds complete.
	built = []
	fails = []
	for fut in api.futures.iwait(list(futures.keys())):
	try:
	img = fut.result()
	if img != api.cloudbuildhelper.NotUploadedImage:
	built.append(img)
	except api.step.StepFailure:
	fails.append(api.path.basename(futures[fut]))

	# Group successfully built images by their roll destinations.
	per_notify = defaultdict(list)
	for img in built:
	for n in img.notify:
	per_notify[n].append(img)

	# Perform all rolls in parallel.
	futures = {}
	for notify in sorted(per_notify):
	fut = api.futures.spawn(
	_roll_built_images,
	api=api,
	notify=notify,
	images=per_notify[notify],
	meta=meta)
	futures[fut] = notify

	# Wait for all rolls to finish.
	roll_fails = []
	for fut in api.futures.iwait(list(futures.keys())):
	try:
	fut.result()
	except api.step.StepFailure:
	roll_fails.append(futures[fut].repo)

	if fails:
	raise recipe_api.StepFailure('Failed to build: %s' % ', '.join(fails))
	if roll_fails:
	raise recipe_api.StepFailure('Failed to roll: %s' % ', '.join(roll_fails))


	def _validate_props(p): # pragma: no cover
	if p.mode == PROPERTIES.MODE_UNDEFINED:
	raise ValueError('"mode" is required')
	if p.project == PROPERTIES.PROJECT_UNDEFINED:
	raise ValueError('"project" is required')
	if not p.infra:
	raise ValueError('"infra" is required')
	if not p.manifests:
	raise ValueError('"manifests" is required')
	# There's no CI/TS for luci-go. Its CI happens when it gets rolled in into
	# infra.git. But we still can run tryjobs for luci-go by applying CLs on top
	# of infra.git checkout.
	if p.project == PROPERTIES.PROJECT_LUCI_GO and p.mode != PROPERTIES.MODE_CL:
	raise ValueError('PROJECT_LUCI_GO can be used only together with MODE_CL')

	if p.project == PROPERTIES.PROJECT_GIT_REPO:
	if not p.HasField('git_repo'):
	raise ValueError('"git_repo" is required when using PROJECT_GIT_REPO')
	if p.mode == PROPERTIES.MODE_CL:
	raise ValueError('PROJECT_GIT_REPO cannot be used together with MODE_CL')
	elif p.HasField('git_repo'):
	raise ValueError('"git_repo" can only be set when using PROJECT_GIT_REPO')


	def _checkout_committed(api, mode, project):
	"""Checks out some committed revision (based on Buildbucket properties).

	Args:
	api: recipes API.
	mode: PROPERTIES.Mode enum (either MODE_CI or MODE_TS).
	project: PROPERTIES.Project enum.

	Returns:
	(Metadata, build environment context manager).
	"""
	conf, internal, repo_url = {
	PROPERTIES.PROJECT_INFRA: (
	'infra_superproject',
	False,
	'https://chromium.googlesource.com/infra/infra',
	),
	PROPERTIES.PROJECT_INFRA_INTERNAL: (
	'infra_internal_superproject',
	True,
	'https://chrome-internal.googlesource.com/infra/infra_internal',
	),
	}[project]

	co = api.infra_checkout.checkout(
	gclient_config_name=conf,
	internal=internal,
	go_version_variant=GO_VERSION_VARIANT)
	co.gclient_runhooks()

	props = co.bot_update_step.properties

	canonical_tag = None
	if mode == PROPERTIES.MODE_CI:
	# E.g. "41861-d008a93".
	commit_label = api.cloudbuildhelper.get_version_label(
	path=co.path / ('infra_internal' if internal else 'infra'),
	revision=props['got_revision'],
	ref=api.buildbucket.gitiles_commit.ref,
	commit_position=props.get('got_revision_cp'),
	)
	# E.g. "ci-2021.06.23-41861-d008a93".
	canonical_tag = 'ci-%s-%s' % (_date(api), commit_label)
	elif mode == PROPERTIES.MODE_TS:
	# E.g. "ts-2021.06.23-113234"
	canonical_tag = 'ts-%s-%d' % (_date(api), api.buildbucket.build.number)
	else:
	raise AssertionError('Impossible') # pragma: no cover

	return Metadata(
	canonical_tag=canonical_tag,
	labels={
	'org.opencontainers.image.source': repo_url,
	'org.opencontainers.image.revision': props['got_revision'],
	},
	tags=['latest'],
	checkout=api.cloudbuildhelper.CheckoutMetadata(
	root=co.path,
	repos=co.bot_update_step.manifest,
	)), lambda api: _infra_checkout_build_environ(co, api)


	def _checkout_pending(api, project):
	"""Checks out some pending CL (based on Buildbucket properties).

	Args:
	api: recipes API.
	project: PROPERTIES.Project enum.

	Returns:
	(Metadata, build environment context manager).
	"""
	conf, patch_root, internal = {
	PROPERTIES.PROJECT_INFRA: (
	'infra',
	'infra',
	False,
	),
	PROPERTIES.PROJECT_INFRA_INTERNAL: (
	'infra_internal',
	'infra_internal',
	True,
	),
	PROPERTIES.PROJECT_LUCI_GO: (
	'infra',
	'infra/go/src/go.chromium.org/luci',
	False,
	),
	}[project]

	co = api.infra_checkout.checkout(
	gclient_config_name=conf,
	patch_root=patch_root,
	internal=internal,
	go_version_variant=GO_VERSION_VARIANT)
	co.commit_change()
	co.gclient_runhooks()

	# Grab information about this CL (in particular who wrote it).
	cl = api.buildbucket.build.input.gerrit_changes[0]
	repo_url = 'https://%s/%s' % (cl.host, cl.project)
	rev_info = api.gerrit.get_revision_info(repo_url, cl.change, cl.patchset)
	author = rev_info['commit']['author']['email']

	return Metadata(
	# ':inputs-hash' essentially tells cloudbuildhelper to skip the build if
	# there's already an image built from the exact same inputs.
	canonical_tag=':inputs-hash',
	labels={'org.chromium.build.cl.repo': repo_url},
	tags=[
	# An "immutable" tag that identifies how the image was built.
	'cl-%s-%d-%d-%s' % (
	_date(api),
	cl.change,
	cl.patchset,
	author.split('@')[0],
	),
	# A movable tag for "a latest image produced from this CL". It is
	# intentionally simple, so that developers can "guess" it just knowing
	# the CL number.
	'cl-%d' % cl.change,
	],
	checkout=api.cloudbuildhelper.CheckoutMetadata(
	root=co.path,
	repos=co.bot_update_step.manifest,
	)), lambda api: _infra_checkout_build_environ(co, api)


	def _checkout_git(api, repo, mode):
	"""Checks out a standalone Git repository.
	Checks out the commit passed via Buildbucket inputs or `refs/heads/main`.
	Args:
	api: recipes API.
	repo: PROPERTIES.GitRepo proto.
	version_label_template: a template for the version label string.
	Returns:
	(Metadata, build environment context manager).
	"""
	path = api.path.cache_dir.joinpath('builder', 'repo')
	revision = api.git.checkout(
	url=repo.url,
	ref=api.buildbucket.gitiles_commit.id or 'refs/heads/main',
	dir_path=path,
	submodules=False)

	@contextmanager
	def build_environ(api):
	with api.buildenv(path, repo.go_version_file, repo.nodejs_version_file):
	yield

	canonical_tag = None
	if mode == PROPERTIES.MODE_CI:
	# E.g. "41861-d008a93".
	commit_label = api.cloudbuildhelper.get_version_label(
	path=path, revision=revision, ref=api.buildbucket.gitiles_commit.ref)
	# E.g. "ci-2021.06.23-41861-d008a93".
	canonical_tag = 'ci-%s-%s' % (_date(api), commit_label)
	elif mode == PROPERTIES.MODE_TS:
	# E.g. "ts-2021.06.23-113234"
	canonical_tag = 'ts-%s-%d' % (_date(api), api.buildbucket.build.number)
	else:
	raise AssertionError('Impossible') # pragma: no cover

	return Metadata(
	canonical_tag=canonical_tag,
	labels={
	'org.opencontainers.image.source': repo.url,
	'org.opencontainers.image.revision': revision,
	},
	tags=['latest'],
	checkout=api.cloudbuildhelper.CheckoutMetadata(
	root=path,
	repos={'.': {
	'repository': repo.url,
	'revision': revision
	}},
	)), build_environ


	@contextmanager
	def _infra_checkout_build_environ(co, api):
	with co.go_env():
	# Use 'cloudbuildhelper' that comes with the infra checkout (it's in PATH),
	# to make sure builders use the same version as developers.
	api.cloudbuildhelper.command = 'cloudbuildhelper'
	yield


	def _date(api):
	"""Returns UTC YYYY.MM.DD to use in tags."""
	return api.time.utcnow().strftime('%Y.%m.%d')


	def _roll_built_images(api, notify, images, meta):
	"""Uploads a CL with info about built images into a repo with pinned images.

	Args:
	api: recipes API.
	notify: a CloudBuildHelperApi.NotifyConfig describing where to roll.
	images: a list of CloudBuildHelperApi.Image with info about built images.
	meta: Metadata struct, as returned by _checkout_committed.

	Returns:
	(None, None) if didn't create a CL (because nothing has changed).
	(Issue number, Issue URL) if created a CL.
	"""
	repo_id = sha256(notify.repo.encode('utf-8')).hexdigest()[:8]
	with api.step.nest('upload roll CL') as pres:
	num, url = api.cloudbuildhelper.do_roll(
	repo_url=notify.repo,
	root=api.path.cache_dir.joinpath('builder', 'roll', repo_id),
	callback=lambda root: _mutate_repo(api, root, notify, images, meta))
	if num is not None:
	pres.links['Issue %s' % num] = url


	def _mutate_repo(api, root, notify, images, meta):
	"""Modifies the checked out repo with image pins.

	Args:
	api: recipes API.
	root: the directory where the repo is checked out.
	notify: a CloudBuildHelperApi.NotifyConfig describing where to roll.
	images: a list of CloudBuildHelperApi.Image with info about built images.
	meta: Metadata struct, as returned by _checkout_committed.

	Returns:
	cloudbuildhelper.RollCL to proceed with the roll or None to skip it.
	"""
	# RFC3389 timstamp in UTC zone.
	date = api.time.utcnow().isoformat('T') + 'Z'

	# Prepare tag JSON specs for all images.
	# See //scripts/roll_images.py in infradata/k8s repo.
	tags = []
	for img in images:
	tags.append({
	'image': img.image,
	'tag': {
	'tag': img.tag,
	'digest': img.digest,
	'context_dir': img.context_dir,
	'metadata': {
	'date': date,
	'source': {
	'repo': meta.labels['org.opencontainers.image.source'],
	'revision':
	meta.labels['org.opencontainers.image.revision'],
	},
	'sources': img.sources,
	'links': {
	'buildbucket': api.buildbucket.build_url(),
	'cloudbuild': img.view_build_url,
	'gcr': img.view_image_url,
	},
	},
	},
	})

	# Add all new tags (if any).
	res = api.step(
	name=notify.script,
	cmd=[root / notify.script],
	stdin=api.json.input({'tags': tags}),
	stdout=api.json.output(),
	step_test_data=lambda: api.json.test_api.output_stream(
	_roll_images_test_data(tags)))
	deployments = res.stdout.get('deployments') or []

	# Given a Deployment dict, returns the artifact name. Unfortunately the key
	# that contains it is different for infradata/k8s and infradata/gae
	# destinations.
	def artifact(dep):
	if 'image' in dep:
	return dep['image']
	return dep['artifact'] # pragma: no cover

	# Generate the commit message.
	message = [
	'[images] Rolling in images.',
	'',
	'Produced by %s' % api.buildbucket.build_url(),
	'',
	]
	if deployments:
	message.extend([
	'Updated deployments:',
	] + [
	' * %s: %s -> %s' % (artifact(d), d['from'], d['to'])
	for d in deployments
	] + [''])
	message = str('\n'.join(message))

	# List of people to CC based on what deployments were updated.
	extra_cc = set()
	for dep in deployments:
	extra_cc.update(dep.get('cc') or [])

	return api.cloudbuildhelper.RollCL(
	message=message,
	cc=extra_cc,
	tbr=[],
	commit=True)


	def _roll_images_test_data(tags):
	return {
	'deployments': [
	{
	'cc': ['n1@example.com', 'n2@example.com'],
	'from': 'prev-version',
	'image': t['image'],
	'to': t['tag']['tag'],
	}
	for t in tags
	],
	}


	def GenTests(api):
	def try_props(project, cl, patch_set):
	return (
	api.buildbucket.try_build(
	project=project,
	change_number=cl,
	patch_set=patch_set) +
	api.override_step_data(
	'gerrit changes',
	api.json.output([{
	'project': project,
	'_number': cl,
	'revisions': {
	'184ebe53805e102605d11f6b143486d15c23a09c': {
	'_number': patch_set,
	'commit': {
	'message': 'Commit message',
	'author': {'email': 'author@example.com'},
	},
	'ref': 'refs/changes/../../..',
	},
	},
	}]),
	)
	)

	from RECIPE_MODULES.infra.cloudbuildhelper.api import CloudBuildHelperApi

	def build_success_with_notify():
	img = CloudBuildHelperApi.Image(
	image='example.com/fake-registry/target',
	digest='sha256:abcdef',
	tag='ci-2012.05.14-197293-5e03a58',
	context_dir='/some/context/directory/for/target',
	view_image_url=None,
	view_build_url=None,
	notify=[
	CloudBuildHelperApi.NotifyConfig(
	kind='git',
	repo='https://roll.example.com/repo',
	script='scripts/roll.py',
	),
	],
	sources=[],
	)
	return api.cloudbuildhelper.build_success_output(img, mocked_sources=[
	'[CACHE]/infra_gclient_with_go/infra/a/b/c',
	])

	yield (
	api.test('try-infra') +
	api.properties(
	mode=PROPERTIES.MODE_CL,
	project=PROPERTIES.PROJECT_INFRA,
	infra='dev',
	manifests=['infra/build/images/deterministic'],
	) +
	try_props('infra/infra', 123456, 7)
	)

	yield (
	api.test('try-luci-go') +
	api.properties(
	mode=PROPERTIES.MODE_CL,
	project=PROPERTIES.PROJECT_LUCI_GO,
	infra='dev',
	manifests=['infra/build/images/deterministic'],
	) +
	try_props('infra/luci/luci-go', 123456, 7)
	)

	yield (
	api.test('try-infra-internal') +
	api.properties(
	mode=PROPERTIES.MODE_CL,
	project=PROPERTIES.PROJECT_INFRA_INTERNAL,
	infra='dev',
	manifests=['infra_internal/build/images/deterministic'],
	) +
	try_props('infra/infra_internal', 123456, 7)
	)

	yield (api.test('ci-git') + api.properties(
	mode=PROPERTIES.MODE_CI,
	project=PROPERTIES.PROJECT_GIT_REPO,
	infra='prod',
	git_repo=PROPERTIES.GitRepo(
	url='https://chromium.googlesource.com/infra/cop'),
	manifests=['build/images'],
	))

	yield (
	api.test('ci-infra') +
	api.properties(
	mode=PROPERTIES.MODE_CI,
	project=PROPERTIES.PROJECT_INFRA,
	infra='prod',
	manifests=['infra/build/images/deterministic'],
	)
	)

	yield (
	api.test('ci-infra-internal') +
	api.properties(
	mode=PROPERTIES.MODE_CI,
	project=PROPERTIES.PROJECT_INFRA_INTERNAL,
	infra='prod',
	manifests=['infra_internal/build/images/deterministic'],
	)
	)

	yield (api.test('ts-git') + api.properties(
	mode=PROPERTIES.MODE_TS,
	project=PROPERTIES.PROJECT_GIT_REPO,
	infra='prod',
	git_repo=PROPERTIES.GitRepo(
	url='https://chromium.googlesource.com/infra/cop'),
	manifests=['build/images'],
	))

	yield (
	api.test('ts-infra') +
	api.properties(
	mode=PROPERTIES.MODE_TS,
	project=PROPERTIES.PROJECT_INFRA,
	infra='prod',
	manifests=['infra/build/images/daily'],
	)
	)

	yield (
	api.test('ci-infra-with-roll') +
	api.properties(
	mode=PROPERTIES.MODE_CI,
	project=PROPERTIES.PROJECT_INFRA,
	infra='prod',
	manifests=['infra/build/images/deterministic'],
	) +
	api.step_data(
	'cloudbuildhelper build target',
	build_success_with_notify(),
	) +
	api.step_data('upload roll CL.git diff', retcode=1)
	)

	yield (
	api.test('ci-infra-with-roll-failure') +
	api.properties(
	mode=PROPERTIES.MODE_CI,
	project=PROPERTIES.PROJECT_INFRA,
	infra='prod',
	manifests=['infra/build/images/deterministic'],
	) +
	api.step_data(
	'cloudbuildhelper build target',
	build_success_with_notify(),
	) +
	api.step_data(
	'upload roll CL.scripts/roll.py', retcode=1
	) +
	api.expect_status('FAILURE')
	)

	yield (
	api.test('build-failure') +
	api.properties(
	mode=PROPERTIES.MODE_CI,
	project=PROPERTIES.PROJECT_INFRA,
	infra='prod',
	manifests=['infra/build/images/deterministic'],
	) +
	api.step_data(
	'cloudbuildhelper build target',
	api.cloudbuildhelper.build_error_output('Boom'),
	retcode=1
	) +
	api.expect_status('FAILURE')
	)

	yield (
	api.test('bad-props') +
	api.properties(mode=0) +
	api.expect_status('INFRA_FAILURE')
	)