"""
Release script invoked from git trigger upon submission of changes to release-versions.yaml config file to the cos/tools GoB repo

Parses contents of release-versions.yaml file and copies release candidates to gcr.io/cos-tools
"""

import sys
import yaml
import subprocess
import os

def validate_config(release_config):
  for release_container in release_config:
    for key in ["container_name", "build_commit", "release_tags"]:
      assert key in release_container, "missing {} in entry {}".format(key, release_container)

def validate_gcr_path(path):
  return len(path) > len("gcr.io/") and path[:len("gcr.io/")] == "gcr.io/"

def copy_container_image(src_bucket, dst_bucket, 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)

  src_path = os.path.join(src_bucket, container_name)
  dst_path = os.path.join(dst_bucket, container_name)

  for release_tag in release_tags:
    subprocess.run(["gcloud", "container", "images", "add-tag", src_path + ":" + build_tag, dst_path + ":" + release_tag])

def release(src_bucket, dst_bucket):
  with open('release/release-versions.yaml', 'r') as file:
    try:
      release_config = yaml.safe_load(file)
      validate_config(release_config)

      for release_container in release_config:
        container_name = release_container["container_name"]
        build_tag = release_container["build_commit"]
        release_tags = release_container["release_tags"]
        copy_container_image(src_bucket, dst_bucket, container_name, build_tag, release_tags)

    except yaml.YAMLError as ex:
      raise Exception("Invalid YAML config: %s" % str(ex))

def main():
  if len(sys.argv) != 3:
    sys.Exit("sample use: ./release_script <source_gcr_path> <destination_gcr_path>")

  src_bucket = sys.argv[1]
  dst_bucket = sys.argv[2]

  release(src_bucket, dst_bucket)

if __name__ == '__main__':
  main()