openbmc-autobump/openbmc-autobump.py - openbmc/openbmc-tools - Gitiles

 #!/usr/bin/env python3

 # Contributors Listed Below - COPYRIGHT 2018
 # [+] International Business Machines Corp.
 #
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
 # You may obtain a copy of the License at
 #
 #     http://www.apache.org/licenses/LICENSE-2.0
 #
 # Unless required by applicable law or agreed to in writing, software
 # distributed under the License is distributed on an "AS IS" BASIS,
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
 # implied. See the License for the specific language governing
 # permissions and limitations under the License.

 import argparse
 import os
 import sys

 import sh

 git = sh.git.bake("--no-pager")


 def log(msg, args):
     if args.noisy:
         sys.stderr.write("{}\n".format(msg))


 def git_clone_or_reset(local_name, remote, args):
     if not os.path.exists(local_name):
         log("cloning into {}...".format(local_name), args)
         git.clone(remote, local_name)
     else:
         log("{} exists, updating...".format(local_name), args)
         git.fetch(_cwd=local_name)
         git.reset("--hard", "FETCH_HEAD", _cwd=local_name)


 def extract_project_from_uris(uris, args):
     # remove SRC_URI = and quotes (does not handle escaped quotes)
     uris = uris.split('"')[1]
     for uri in uris.split():
         if "github.com/openbmc" not in uri:
             continue

         segments = uri.split(";")
         branch = args.branch

         for s in segments[1:]:
             if "branch=" not in s:
                 continue
             if "nobranch=" in s:
                 continue
             branch = s.split("=")[1]

         # remove fetcher arguments
         uri = segments[0]
         # the project is the right-most path segment
         return (uri.split("/")[-1].replace(".git", ""), branch)

     return None


 def extract_sha_from_recipe(recipe, args):
     with open(recipe) as fp:
         uris = ""
         project = None
         branch = None
         sha = None

         for line in fp:
             line = line.rstrip()
             if "SRCREV" in line:
                 sha = line.split("=")[-1].replace('"', "").strip()
             elif not project and uris or "_URI" in line:
                 uris += line.split("\\")[0]
                 if "\\" not in line:
                     # In uris we've gathered a complete (possibly multi-line)
                     # assignment to a bitbake variable that ends with _URI.
                     # Try to pull an OpenBMC project out of it.
                     uri = extract_project_from_uris(uris, args)
                     if uri is None:
                         # We didn't find a project.  Unset uris and look for
                         # another bitbake variable that ends with _URI.
                         uris = ""
                     else:
                         (project, branch) = uri

             if project and branch and sha:
                 return (project, branch, sha)

         raise RuntimeError("No SRCREV or URI found in {}".format(recipe))


 def find_candidate_recipes(meta, args):
     remote_fmt_args = (args.ssh_config_host, meta)
     remote = "ssh://{}/openbmc/{}".format(*remote_fmt_args)
     try:
         git_clone_or_reset(meta, remote, args)
     except sh.ErrorReturnCode as e:
         log("{}".format(e), args)
         return []

     match_suffixes = ("bb", "bbclass", "inc")
     pathspecs = ("*.{}".format(x) for x in match_suffixes)
     grep_args = (
         "--no-color",
         "-l",
         "-e",
         "_URI",
         "--and",
         "-e",
         "github.com/openbmc",
     )
     grep_args = (*grep_args, *pathspecs)
     try:
         return git.grep(*grep_args, _cwd=meta).stdout.decode("utf-8").split()
     except sh.ErrorReturnCode_1:
         pass
     except sh.ErrorReturnCode as e:
         log("{}".format(e), args)

     return []


 def find_and_process_bumps(meta, args):
     candidate_recipes = find_candidate_recipes(meta, args)

     for recipe in candidate_recipes:
         full_recipe_path = os.path.join(meta, recipe)
         recipe_basename = os.path.basename(full_recipe_path)
         project_name, recipe_branch, recipe_sha = extract_sha_from_recipe(
             full_recipe_path, args
         )

         remote_fmt_args = (args.ssh_config_host, project_name)
         remote = "ssh://{}/openbmc/{}".format(*remote_fmt_args)
         ls_remote_args = [remote, "refs/heads/{}".format(recipe_branch)]
         try:
             project_sha = git("ls-remote", *ls_remote_args)
             project_sha = project_sha.stdout.decode("utf-8").split()[0]
         except sh.ErrorReturnCode as e:
             log("{}".format(e), args)
             continue

         if project_sha == recipe_sha:
             message_args = (recipe_basename, recipe_sha[:10])
             print("{} is up to date ({})".format(*message_args))
             continue

         change_id = "autobump {} {} {}".format(recipe, recipe_sha, project_sha)
         hash_object_args = ["-t", "blob", "--stdin"]
         change_id = git(sh.echo(change_id), "hash-object", *hash_object_args)
         change_id = "I{}".format(change_id.strip())

         query_args = ["query", "change:{}".format(change_id)]
         gerrit_query_result = args.gerrit(*query_args)
         gerrit_query_result = gerrit_query_result.stdout.decode("utf-8")

         if change_id in gerrit_query_result:
             message_args = (recipe_basename, change_id)
             print("{} {} already exists".format(*message_args))
             continue

         message_args = (recipe_basename, recipe_sha[:10], project_sha[:10])
         print("{} updating from {} to {}".format(*message_args))

         remote_args = (args.ssh_config_host, project_name)
         remote = "ssh://{}/openbmc/{}".format(*remote_args)
         git_clone_or_reset(project_name, remote, args)

         try:
             revlist = "{}..{}".format(recipe_sha, project_sha)
             shortlog = git.shortlog(revlist, _cwd=project_name)
             shortlog = shortlog.stdout.decode("utf-8")
         except sh.ErrorReturnCode as e:
             log("{}".format(e), args)
             continue

         reset_args = ["--hard", "origin/{}".format(args.branch)]
         git.reset(*reset_args, _cwd=meta)

         recipe_content = None
         with open(full_recipe_path) as fd:
             recipe_content = fd.read()

         recipe_content = recipe_content.replace(recipe_sha, project_sha)
         with open(full_recipe_path, "w") as fd:
             fd.write(recipe_content)

         git.add(recipe, _cwd=meta)

         commit_summary_args = (project_name, recipe_sha[:10], project_sha[:10])
         commit_msg = "{}: srcrev bump {}..{}".format(*commit_summary_args)
         commit_msg += "\n\n{}".format(shortlog)
         commit_msg += "\n\nChange-Id: {}".format(change_id)

         git.commit(sh.echo(commit_msg), "-s", "-F", "-", _cwd=meta)

         push_args = [
             "origin",
             "HEAD:refs/for/{}%topic=autobump".format(args.branch),
         ]
         if not args.dry_run:
             git.push(*push_args, _cwd=meta)


 def main():
     app_description = """OpenBMC bitbake recipe bumping tool.

 Find bitbake metadata files (recipes) that use the git fetcher
 and check the project repository for newer revisions.

 Generate commits that update bitbake metadata files with SRCREV.

 Push generated commits to the OpenBMC Gerrit instance for review.
     """
     parser = argparse.ArgumentParser(
         description=app_description,
         formatter_class=argparse.RawDescriptionHelpFormatter,
     )

     parser.set_defaults(branch="master")
     parser.add_argument(
         "-d",
         "--dry-run",
         dest="dry_run",
         action="store_true",
         help="perform a dry run only",
     )
     parser.add_argument(
         "-m",
         "--meta-repository",
         dest="meta_repository",
         action="append",
         help="meta repository to check for updates",
     )
     parser.add_argument(
         "-v",
         "--verbose",
         dest="noisy",
         action="store_true",
         help="enable verbose status messages",
     )
     parser.add_argument(
         "ssh_config_host",
         metavar="SSH_CONFIG_HOST_ENTRY",
         help="SSH config host entry for Gerrit connectivity",
     )

     args = parser.parse_args()
     setattr(args, "gerrit", sh.ssh.bake(args.ssh_config_host, "gerrit"))

     metas = getattr(args, "meta_repository")
     if metas is None:
         metas = args.gerrit("ls-projects", "-m", "meta-")
         metas = metas.stdout.decode("utf-8").split()
         metas = [os.path.split(x)[-1] for x in metas]

     for meta in metas:
         find_and_process_bumps(meta, args)


 if __name__ == "__main__":
     sys.exit(0 if main() else 1)
	#!/usr/bin/env python3

	# Contributors Listed Below - COPYRIGHT 2018
	# [+] International Business Machines Corp.
	#
	#
	# Licensed under the Apache License, Version 2.0 (the "License");
	# you may not use this file except in compliance with the License.
	# You may obtain a copy of the License at
	#
	# http://www.apache.org/licenses/LICENSE-2.0
	#
	# Unless required by applicable law or agreed to in writing, software
	# distributed under the License is distributed on an "AS IS" BASIS,
	# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
	# implied. See the License for the specific language governing
	# permissions and limitations under the License.

	import argparse
	import os
	import sys

	import sh

	git = sh.git.bake("--no-pager")


	def log(msg, args):
	if args.noisy:
	sys.stderr.write("{}\n".format(msg))


	def git_clone_or_reset(local_name, remote, args):
	if not os.path.exists(local_name):
	log("cloning into {}...".format(local_name), args)
	git.clone(remote, local_name)
	else:
	log("{} exists, updating...".format(local_name), args)
	git.fetch(_cwd=local_name)
	git.reset("--hard", "FETCH_HEAD", _cwd=local_name)


	def extract_project_from_uris(uris, args):
	# remove SRC_URI = and quotes (does not handle escaped quotes)
	uris = uris.split('"')[1]
	for uri in uris.split():
	if "github.com/openbmc" not in uri:
	continue

	segments = uri.split(";")
	branch = args.branch

	for s in segments[1:]:
	if "branch=" not in s:
	continue
	if "nobranch=" in s:
	continue
	branch = s.split("=")[1]

	# remove fetcher arguments
	uri = segments[0]
	# the project is the right-most path segment
	return (uri.split("/")[-1].replace(".git", ""), branch)

	return None


	def extract_sha_from_recipe(recipe, args):
	with open(recipe) as fp:
	uris = ""
	project = None
	branch = None
	sha = None

	for line in fp:
	line = line.rstrip()
	if "SRCREV" in line:
	sha = line.split("=")[-1].replace('"', "").strip()
	elif not project and uris or "_URI" in line:
	uris += line.split("\\")[0]
	if "\\" not in line:
	# In uris we've gathered a complete (possibly multi-line)
	# assignment to a bitbake variable that ends with _URI.
	# Try to pull an OpenBMC project out of it.
	uri = extract_project_from_uris(uris, args)
	if uri is None:
	# We didn't find a project. Unset uris and look for
	# another bitbake variable that ends with _URI.
	uris = ""
	else:
	(project, branch) = uri

	if project and branch and sha:
	return (project, branch, sha)

	raise RuntimeError("No SRCREV or URI found in {}".format(recipe))


	def find_candidate_recipes(meta, args):
	remote_fmt_args = (args.ssh_config_host, meta)
	remote = "ssh://{}/openbmc/{}".format(*remote_fmt_args)
	try:
	git_clone_or_reset(meta, remote, args)
	except sh.ErrorReturnCode as e:
	log("{}".format(e), args)
	return []

	match_suffixes = ("bb", "bbclass", "inc")
	pathspecs = ("*.{}".format(x) for x in match_suffixes)
	grep_args = (
	"--no-color",
	"-l",
	"-e",
	"_URI",
	"--and",
	"-e",
	"github.com/openbmc",
	)
	grep_args = (grep_args, pathspecs)
	try:
	return git.grep(*grep_args, _cwd=meta).stdout.decode("utf-8").split()
	except sh.ErrorReturnCode_1:
	pass
	except sh.ErrorReturnCode as e:
	log("{}".format(e), args)

	return []


	def find_and_process_bumps(meta, args):
	candidate_recipes = find_candidate_recipes(meta, args)

	for recipe in candidate_recipes:
	full_recipe_path = os.path.join(meta, recipe)
	recipe_basename = os.path.basename(full_recipe_path)
	project_name, recipe_branch, recipe_sha = extract_sha_from_recipe(
	full_recipe_path, args
	)

	remote_fmt_args = (args.ssh_config_host, project_name)
	remote = "ssh://{}/openbmc/{}".format(*remote_fmt_args)
	ls_remote_args = [remote, "refs/heads/{}".format(recipe_branch)]
	try:
	project_sha = git("ls-remote", *ls_remote_args)
	project_sha = project_sha.stdout.decode("utf-8").split()[0]
	except sh.ErrorReturnCode as e:
	log("{}".format(e), args)
	continue

	if project_sha == recipe_sha:
	message_args = (recipe_basename, recipe_sha[:10])
	print("{} is up to date ({})".format(*message_args))
	continue

	change_id = "autobump {} {} {}".format(recipe, recipe_sha, project_sha)
	hash_object_args = ["-t", "blob", "--stdin"]
	change_id = git(sh.echo(change_id), "hash-object", *hash_object_args)
	change_id = "I{}".format(change_id.strip())

	query_args = ["query", "change:{}".format(change_id)]
	gerrit_query_result = args.gerrit(*query_args)
	gerrit_query_result = gerrit_query_result.stdout.decode("utf-8")

	if change_id in gerrit_query_result:
	message_args = (recipe_basename, change_id)
	print("{} {} already exists".format(*message_args))
	continue

	message_args = (recipe_basename, recipe_sha[:10], project_sha[:10])
	print("{} updating from {} to {}".format(*message_args))

	remote_args = (args.ssh_config_host, project_name)
	remote = "ssh://{}/openbmc/{}".format(*remote_args)
	git_clone_or_reset(project_name, remote, args)

	try:
	revlist = "{}..{}".format(recipe_sha, project_sha)
	shortlog = git.shortlog(revlist, _cwd=project_name)
	shortlog = shortlog.stdout.decode("utf-8")
	except sh.ErrorReturnCode as e:
	log("{}".format(e), args)
	continue

	reset_args = ["--hard", "origin/{}".format(args.branch)]
	git.reset(*reset_args, _cwd=meta)

	recipe_content = None
	with open(full_recipe_path) as fd:
	recipe_content = fd.read()

	recipe_content = recipe_content.replace(recipe_sha, project_sha)
	with open(full_recipe_path, "w") as fd:
	fd.write(recipe_content)

	git.add(recipe, _cwd=meta)

	commit_summary_args = (project_name, recipe_sha[:10], project_sha[:10])
	commit_msg = "{}: srcrev bump {}..{}".format(*commit_summary_args)
	commit_msg += "\n\n{}".format(shortlog)
	commit_msg += "\n\nChange-Id: {}".format(change_id)

	git.commit(sh.echo(commit_msg), "-s", "-F", "-", _cwd=meta)

	push_args = [
	"origin",
	"HEAD:refs/for/{}%topic=autobump".format(args.branch),
	]
	if not args.dry_run:
	git.push(*push_args, _cwd=meta)


	def main():
	app_description = """OpenBMC bitbake recipe bumping tool.

	Find bitbake metadata files (recipes) that use the git fetcher
	and check the project repository for newer revisions.

	Generate commits that update bitbake metadata files with SRCREV.

	Push generated commits to the OpenBMC Gerrit instance for review.
	"""
	parser = argparse.ArgumentParser(
	description=app_description,
	formatter_class=argparse.RawDescriptionHelpFormatter,
	)

	parser.set_defaults(branch="master")
	parser.add_argument(
	"-d",
	"--dry-run",
	dest="dry_run",
	action="store_true",
	help="perform a dry run only",
	)
	parser.add_argument(
	"-m",
	"--meta-repository",
	dest="meta_repository",
	action="append",
	help="meta repository to check for updates",
	)
	parser.add_argument(
	"-v",
	"--verbose",
	dest="noisy",
	action="store_true",
	help="enable verbose status messages",
	)
	parser.add_argument(
	"ssh_config_host",
	metavar="SSH_CONFIG_HOST_ENTRY",
	help="SSH config host entry for Gerrit connectivity",
	)

	args = parser.parse_args()
	setattr(args, "gerrit", sh.ssh.bake(args.ssh_config_host, "gerrit"))

	metas = getattr(args, "meta_repository")
	if metas is None:
	metas = args.gerrit("ls-projects", "-m", "meta-")
	metas = metas.stdout.decode("utf-8").split()
	metas = [os.path.split(x)[-1] for x in metas]

	for meta in metas:
	find_and_process_bumps(meta, args)


	if __name__ == "__main__":
	sys.exit(0 if main() else 1)