from Scripts import resource_fetcher
from Scripts import utils
from dotenv import load_dotenv
import os

load_dotenv()

class Github:
    def __init__(self):
        self.utils = utils.Utils()
        # Load the GitHub token from environment variables
        self.github_token = os.getenv("GITHUB_TOKEN")
        # Set the headers for GitHub API requests
        self.headers = {
            "Accept": "application/vnd.github+json",
            "#Authorization": f"token {self.github_token}",
            "X-GitHub-Api-Version": "2022-11-28",
        }
        self.fetcher = resource_fetcher.ResourceFetcher(self.headers)

    def check_ratelimit(self):
        url = "https://api.github.com/rate_limit"

        response = self.fetcher.fetch_and_parse_content(url, "json")
        if response.get("rate").get("remaining") == 0:
            raise Exception("Please try again later, you have exhausted your GitHub REST API request quota")
        
    def get_list_branches(self, owner, repo):
        self.check_ratelimit()

        url = f"https://api.github.com/repos/{owner}/{repo}/branches"

        response = self.fetcher.fetch_and_parse_content(url, "json")

        if not response:
            return []

        return response
        
    def get_latest_artifact(self, owner, repo):
        results = []

        self.check_ratelimit()

        url = f"https://api.github.com/repos/{owner}/{repo}/actions/artifacts?per_page=1"

        response = self.fetcher.fetch_and_parse_content(url, "json")

        latest_artifact_id = response.get("artifacts")[0].get("id")
        
        results.append({
            "product_name": repo,
            "id": latest_artifact_id,
            "url": "https://api.github.com/repos/{}/{}/actions/artifacts/{}/{}".format(owner, repo, latest_artifact_id, "zip")
        })

        return results

    def get_latest_release(self, owner, repo):
        result = []

        self.check_ratelimit()

        url = f"https://api.github.com/repos/{owner}/{repo}/releases?per_page=1"

        response = self.fetcher.fetch_and_parse_content(url, "json")

        # Iterate over the assets in the release
        for asset in response[0].get("assets"):
            asset_id = asset.get("id")
            download_url = asset.get("browser_download_url")
            asset_name = self.extract_asset_name(asset.get("name"))

            if "apple" in download_url.lower() or ("app" not in download_url.lower() and "dsym" not in download_url.lower() and "adget" not in download_url):
                if "lzhoang2601" in download_url or "tlwm" in download_url or ("tlwm" not in download_url and "DEBUG" not in download_url.upper()):
                    result.append({
                        "product_name": asset_name, 
                        "id": asset_id, 
                        "url": download_url
                    })

        return result
    
    def extract_asset_name(self, name):
        # Extract the base name from the asset name
        name_parts = name.split("-") if "-" in name else name.split("_")

        asset_name = name_parts[0].split(".")[0]
        if asset_name[-1].isdigit():
            asset_name = asset_name[:-1]
            
        if (len(name_parts) > 1):
            if "uns" in name_parts[1]:
                asset_name += "-" + name_parts[1]
            elif "Sonoma14.4" in name:
                asset_name += "23.4"
            elif "Sonoma" in name:
                asset_name += "23"
            elif "Ventura" in name:
                asset_name += "22"
            elif "Monterey" in name:
                asset_name += "21"
            elif "Catalina" in name:
                asset_name += "19"
            elif "Mojave" in name:
                asset_name += "18"
            elif "HighSierra" in name:
                asset_name += "17"
            elif "BigSur" in name:
                asset_name += "20"

        return asset_name