Enable multi region release of COS public container images
This change will enable us to release public COS container to
multi-region Artifact Registry repositories.
Change-Id: I03a511b67c0ff9e1bd54e4b91468bd3e8e9767fa
Reviewed-on: https://cos-review.googlesource.com/c/cos/tools/+/28981
Reviewed-by: Arnav Kansal <rnv@google.com>
Cloud-Build: GCB Service account <228075978874@cloudbuild.gserviceaccount.com>
Tested-by: Rayan Dasoriya <dasoriya@google.com>
diff --git a/release/release.py b/release/release.py
index 6231b47..050e37b 100644
--- a/release/release.py
+++ b/release/release.py
@@ -14,12 +14,18 @@
for key in ["staging_container_name", "release_container_name", "build_commit", "release_tags"]:
assert key in release_container, "missing {} in entry {}".format(key, release_container)
-def validate_gcr_path(path):
+def validate_src_gcr_path(path):
+ # path format: gcr.io/cos-infra-prod
return len(path) > len("gcr.io/") and path[:len("gcr.io/")] == "gcr.io/"
+def validate_dst_gcr_path(path):
+ # path format: us-docker.pkg.dev/cos-cloud/us.gcr.io
+ path = path.split('/')
+ return len(path) == 3 and len(path[0]) > len("docker.pkg.dev/") and len(path[1]) != 0 and path[2][-len("gcr.io"):] == "gcr.io"
+
def copy_container_image(src_bucket, dst_bucket, staging_container_name, release_container_name, build_tag, release_tags):
- assert validate_gcr_path(src_bucket), "cannot use address {}, only gcr.io/ addresses are supported".format(src_bucket)
- assert validate_gcr_path(dst_bucket), "cannot use address {}, only gcr.io/ addresses are supported".format(dst_bucket)
+ assert validate_src_gcr_path(src_bucket), "cannot use address {}, only gcr.io/ addresses are supported".format(src_bucket)
+ assert validate_dst_gcr_path(dst_bucket), "cannot use address {}, only <location>-docker.pkg.dev/<project-name>/<location(optional)>gcr.io/ addresses are supported".format(dst_bucket)
src_path = os.path.join(src_bucket, staging_container_name)
dst_path = os.path.join(dst_bucket, release_container_name)
@@ -27,19 +33,21 @@
for release_tag in release_tags:
subprocess.run(["gcloud", "container", "images", "add-tag", src_path + ":" + build_tag, dst_path + ":" + release_tag, "-q"])
-def verify_and_release(src_bucket, dst_bucket, release):
+def verify_and_release(src_bucket, dst_buckets, release):
with open('release/release-versions.yaml', 'r') as file:
try:
release_config = yaml.safe_load(file)
validate_config(release_config)
if release:
+ dst_buckets = dst_buckets.split('^')
for release_container in release_config:
staging_container_name = release_container["staging_container_name"]
release_container_name = release_container["release_container_name"]
build_tag = release_container["build_commit"]
release_tags = release_container["release_tags"]
- copy_container_image(src_bucket, dst_bucket, staging_container_name, release_container_name, build_tag, release_tags)
+ for dst_bucket in dst_buckets:
+ copy_container_image(src_bucket, dst_bucket, staging_container_name, release_container_name, build_tag, release_tags)
except yaml.YAMLError as ex:
raise Exception("Invalid YAML config: %s" % str(ex))
@@ -49,11 +57,12 @@
verify_and_release("", "", False)
elif len(sys.argv) == 3:
src_bucket = sys.argv[1]
- dst_bucket = sys.argv[2]
+ dst_buckets = sys.argv[2]
- verify_and_release(src_bucket, dst_bucket, True)
+ verify_and_release(src_bucket, dst_buckets, True)
else:
- sys.exit("sample use: ./release_script <source_gcr_path> <destination_gcr_path>")
+ sys.exit("sample use: ./release_script <source_gcr_path> <destination_gcr_paths> \n \
+ example use: ./release_script gcr.io/cos-infra-prod us-docker.pkg.dev/cos-cloud/us.gcr.io^europe-docker.pkg.dev/cos-cloud/eu.gcr.io")
if __name__ == '__main__':
main()