Updated git download code

app.py

@@ -6,10 +6,15 @@ import os
 import glob
 
 # Variables for GitHub repository details
-GITHUB_REPO_URL = "https://github.com/vsagar100/ansible_tf_provisioner"
+REPO_OWNER = "vsagar100"
+REPO_NAME = "ansible_tf_provisioner"
 GITHUB_BRANCH = "main"  # Specify the branch or tag to download
 GITHUB_TOKEN = "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" #os.getenv("GITHUB_TOKEN")  # Specify your GitHub token
 
+# GitHub API endpoint to download the repo as a zip file
+#repo_url = f"https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}/zipball/main"
+GITHUB_REPO_URL = f"https://api.github.com/repos/{REPO_OWNER}/{REPO_NAME}/zipball/{GITHUB_BRANCH}"
+
 def main():
     # Directory structure setup
     # Directory for storing downloaded GitHub repository
@@ -26,7 +31,7 @@ def main():
     review_manager = ReviewManager(reviewer=code_reviewer)
 
     # Download GitHub repository
-    review_manager.download_repo(GITHUB_REPO_URL, GITHUB_BRANCH, GITHUB_TOKEN, download_directory)
+    review_manager.download_repo(GITHUB_REPO_URL, GITHUB_TOKEN, download_directory)
 
     # Find all YAML files in the downloaded repository
     yaml_files = glob.glob(os.path.join(download_directory, "**", "*.yml"), recursive=True)

lib/code_reviewer.py

@@ -82,19 +82,17 @@ class ReviewManager:
         """
         self.reviewer = reviewer
 
-    def download_repo(self, repo_url: str, branch: str, token: str, download_path: str):
+    def download_repo(self, repo_url: str, token: str, download_path: str):
         """
         Downloads a GitHub repository as a ZIP file and extracts it.
         
         Args:
             repo_url (str): The GitHub repository URL.
-            branch (str): The branch or tag to download.
             token (str): The GitHub personal access token for authentication.
             download_path (str): The path to extract the downloaded repository.
         """
-        zip_url = f"{repo_url}/archive/refs/heads/{branch}.zip"
         headers = {"Authorization": f"token {token}"}
-        response = requests.get(zip_url, headers=headers)
+        response = requests.get(repo_url, headers=headers)
         if response.status_code == 200:
             with zipfile.ZipFile(io.BytesIO(response.content)) as zip_ref:
                 zip_ref.extractall(download_path)