deploy.py

import os
import subprocess
import sys

import requests


class ImageTagUpdater:
    def __init__(self, yaml_file_path, new_image_tag):
        self.yaml_file_path = yaml_file_path
        self.new_image_tag = new_image_tag
        self.github_repository = os.environ.get('GITHUB_REPOSITORY')
        self.github_token = os.environ.get('GITHUB_TOKEN')
        self.docker_repo = "girishcodealchemy/alchemy-nginx"
        self.branch_name = f'release-{new_image_tag}'
        self.github_event_path = os.environ.get('GITHUB_EVENT_PATH')
        self.uri = "https://api.github.com"
        self.header = {
            'Authorization': f'Bearer {self.github_token}',
            'Accept': 'application/vnd.github.v3+json'}
        self.user_login = os.environ.get('GITHUB_ACTOR')

    def check_docker_image_exists(self):
        try:
            response = requests.get(f'https://hub.docker.com/v2/repositories/{self.docker_repo}/tags/{self.new_image_tag}')
            if response.status_code == 200:
                print(f"Docker image {self.docker_repo}:{self.new_image_tag} found.")
            else:
                print(f"Docker image {self.docker_repo}:{self.new_image_tag} not found.")
                sys.exit(1)
        except requests.RequestException as e:
            print(f"Error checking image: {e}")
            sys.exit(1)

    def get_user_info(self):
        print(f"Getting user info for {self.user_login}...")

        response = requests.get(
            f"{self.uri}/users/{self.user_login}",
            headers=self.header
        )
        data = response.json()
        user = data.get("name", self.user_login)
        user_email = data.get("email", f"{self.user_login}@users.noreply.github.com")
        if user_email is None:
            user_email = f"{self.user_login}@users.noreply.github.com"
        return user, user_email

    def git_config(self):
        print("Configuring git...")
        self.user, self.user_email = self.get_user_info()
        subprocess.run(["git", "config", "--global", "--add", "safe.directory", "/github/workspace"])
        subprocess.run(["git", "config", "--global", "user.email", str(self.user_email)])
        subprocess.run(["git", "config", "--global", "user.name", str(self.user)])

    def update_image_tag(self):
        try:
            # Read the YAML file
            with open(self.yaml_file_path, 'r') as f:
                yaml_content = f.read()

            # Update the image tag
            pattern = f'{self.docker_repo}:'
            start_index = yaml_content.find(pattern)
            if start_index != -1:
                end_index = yaml_content.find(' ', start_index)
                if end_index == -1:
                    end_index = len(yaml_content)
                tag_to_replace = yaml_content[start_index+len(pattern):end_index-1]
                updated_yaml_content = yaml_content.replace(
                    f'{self.docker_repo}:{tag_to_replace}',
                    f'{self.docker_repo}:{self.new_image_tag}'
                )
            else:
                raise ValueError("No image tag found in the YAML content")

            # Write the updated content back to the file
            with open(self.yaml_file_path, 'w') as f:
                f.write(updated_yaml_content)

            print(f"Image tag updated to: {self.new_image_tag}")

        except Exception as e:
            print(f"Error updating image tag: {e}")

    def create_or_checkout_branch(self):
        try:
            # Check if branch exists
            subprocess.run(['git', 'rev-parse', '--verify', "--quiet", self.branch_name], check=True, stdout=subprocess.PIPE)
            print(f"Branch {self.branch_name} already exists. Skipping branch creation.")
            subprocess.run(['git', 'checkout', self.branch_name])
            subprocess.run(['git', 'pull', 'origin', self.branch_name])
            subprocess.run(['git', 'pull', 'origin', 'main'])
        except subprocess.CalledProcessError:
            # Create a new branch
            print(f"Creating the new branch: {self.branch_name}")
            subprocess.run(['git', 'checkout', '-b', self.branch_name])
            subprocess.run(['git', 'pull', 'origin', 'main'])
            print(f"Created and checked out branch: {self.branch_name}")

    def commit_and_push_changes(self):
        try:
            # Check if there are changes to commit
            status_output = subprocess.run(['git', 'status', '--porcelain'], stdout=subprocess.PIPE, universal_newlines=True)
            if not status_output.stdout.strip():
                print("No changes to commit. Skipping commit and push.")
                return

            # Commit changes
            subprocess.run(['git', 'add', self.yaml_file_path])
            subprocess.run(['git', 'commit', '-m', f'Update image tag to {self.new_image_tag}'])
            subprocess.run(['git', 'push', '-u', 'origin', self.branch_name])

            print("Changes committed and pushed to the repository")

        except Exception as e:
            print(f"Error committing and pushing changes: {e}")

    def check_pull_request_exists(self):
        try:
            response = requests.get(
                f"{self.uri}/repos/{self.github_repository}/pulls",
                headers=self.header
            )
            pull_requests = response.json()
            for pr in pull_requests:
                if pr['head']['ref'] == self.branch_name:
                    print("Pull request already exists. Skipping creation.")
                    return True
            return False
        except Exception as e:
            print(f"Error checking pull request: {e}")
            return False

    def create_pull_request(self, pr_title, pr_body):
        try:
            self.git_config()
            self.create_or_checkout_branch()
            self.update_image_tag()
            self.commit_and_push_changes()
            if self.check_pull_request_exists():
                return
            else:
                subprocess.run(['gh', 'pr', 'create', '--base', 'main', '--head', self.branch_name, '--title', pr_title, '--body', pr_body])
                print(f"Pull request created for branch: {self.branch_name}")
        except Exception as e:
            print(f"Error creating pull request: {e}")


if __name__ == '__main__':
    if len(sys.argv) != 2:
        print("Usage: python deploy.py <new_image_tag>")
        sys.exit(1)
    new_image_tag = sys.argv[1]
    deployment_yaml_path = 'CloudWebApp/deployment.yml'
    pr_title = f'Update latest WebApp image tag to {new_image_tag}'
    pr_body = 'Updating the image tag in the deployment.yaml file.'

    updater = ImageTagUpdater(deployment_yaml_path, new_image_tag)
    updater.check_docker_image_exists()
    updater.create_pull_request(pr_title, pr_body)