From afcca694d6ccdd21a3f348ed6a6a41a28ca9abec Mon Sep 17 00:00:00 2001
From: Thomas Cardin <49320132+ThomasCardin@users.noreply.github.com>
Date: Tue, 30 Jan 2024 09:53:32 -0500
Subject: [PATCH] issue #26: migrated script from github-worflows. Missing unit
 and integration test

---
 .../remove-previous-image.py                  | 84 +++++++++++++++++++
 1 file changed, 84 insertions(+)
 create mode 100644 remove-previous-image/remove-previous-image.py

diff --git a/remove-previous-image/remove-previous-image.py b/remove-previous-image/remove-previous-image.py
new file mode 100644
index 0000000..2872737
--- /dev/null
+++ b/remove-previous-image/remove-previous-image.py
@@ -0,0 +1,84 @@
+"""
+
+This script is based on this documentation:
+https://docs.github.com/en/rest/packages/packages?apiVersion=2022-11-28
+
+"""
+
+import os
+import requests
+from requests.auth import HTTPBasicAuth
+
+"""
+Delete the old container (based on the previous tag)
+"""
+def delete_old_image(version_id, org, headers, auth):
+    url_delete_previous_version = f"https://api.github.com/orgs/{org}/packages/container/{container_name}/versions/{version_id}"
+    response = requests.delete(url_delete_previous_version, headers=headers, auth=auth)
+    if response.status_code == 204:
+        print(f'Previous container deleted!')
+    else:
+        raise Exception(f"error deleting the previous container: {response.status_code} {response.text}")
+
+"""
+Find the previous tag for a specific container.
+Delete the previous tag if it exists and it is not the current one. This wont delete the current tag or older PR tags.
+Check if tags is not empty and check if the len is == 1. If that is the case, it means that the only tag is the previous one.
+"""
+def find_previous_container_tag(response, unique_tag):
+    version_id = None
+    for version in response:
+        print(f"Found tags {version['metadata']['container']['tags']})")
+        tags = version['metadata']['container']['tags']
+        if unique_tag not in tags and len(tags) == 1 and tags:
+            version_id = version['id']
+            print(f"Previous tag found {tags[0]} with version_id {version_id}")
+            return tags[0], version_id
+
+    raise Exception(f"Container name {container_name} not found or the only tag found was the current one. If that is the case, you can ignore this error.")
+
+"""
+Get all GCR containers information
+"""
+def get_container_tags(org, container_name, auth, headers, container_path):
+    get_versions = f"https://api.github.com/orgs/{org}/packages/container/{container_name}/versions"
+    response = requests.get(get_versions, headers=headers, auth=auth)
+
+    try:
+        response.raise_for_status()
+    except requests.exceptions.HTTPError as e:
+        raise Exception(f"Error getting the previous tag for the container {container_path} : {e}")
+
+    return response
+
+def print_console(message):
+    print("====================================")
+    print(message)
+
+if __name__ == "__main__":
+    registry = os.getenv("REGISTRY")
+    github_token = os.getenv("GITHUB_TOKEN")
+    container_name = os.getenv("CONTAINER_NAME")
+    unique_tag = os.getenv("UNIQUE_TAG")
+    user = os.getenv("USER")
+    current_commit = os.getenv("CURRENT_COMMIT")
+
+    headers = {
+        "Accept": "application/vnd.github.v3+json",
+    }
+    auth = HTTPBasicAuth(user, github_token)
+
+    container_path = f"{registry}/{container_name}:{unique_tag}"
+    org = registry.split("/")[1]
+
+    print_console(f"Getting all tags for this container {container_path}...")
+    response = get_container_tags(org, container_name, auth, headers, container_path)
+    print("Done!")
+
+    print_console(f"Looking for the previous tag...")
+    previous_tag, version_id = find_previous_container_tag(response.json(), unique_tag)
+    print("Done!")
+
+    print_console(f"Deleting the previous container with tag ({previous_tag}) and version_id {version_id}...")
+    delete_old_image(version_id, org, headers, auth)
+    print("Done!")