# 必要的依赖库
import requests
from datetime import datetime, timedelta, timezone
from concurrent.futures import ThreadPoolExecutor
import os


class GitHubActionsManager:
    def __init__(self, repo_url, token):
        self.repo_url = repo_url  # GitHub仓库URL
        self.token = token  # GitHub个人访问令牌
        self.runs_url = f"{self.repo_url}/actions/runs"  # 获取运行记录的URL
        self.current_time = datetime.now(timezone.utc)  # 获取当前时间

    def get_workflow_runs(self, url):
        # 发送GET请求获取工作流运行记录
        response = requests.get(url, headers={"Authorization": f"token {self.token}"})
        if response.status_code == 200:  # 如果请求成功
            return response.json()["workflow_runs"]  # 返回运行记录列表
        else:  # 如果请求失败
            print(f"Failed to fetch runs from {url}. Status code: {response.status_code}")  # 打印错误信息
            return []  # 返回空列表

    def delete_run(self, run_id):
        delete_url = f"{self.runs_url}/{run_id}"  # 构建删除运行记录的URL
        # 发送DELETE请求删除指定ID的运行记录
        response = requests.delete(delete_url, headers={"Authorization": f"token {self.token}"})
        if response.status_code == 204:  # 如果删除成功
            print(f"Deleted run with ID {run_id}")  # 打印删除成功的消息
        else:  # 如果删除失败
            print(f"Failed to delete run with ID {run_id}. Status code: {response.status_code}")  # 打印错误信息

    def delete_old_runs(self, max_workers=1):
        next_page = self.runs_url  # 初始化下一页的URL为第一页
        while next_page:  # 循环直到没有下一页
            # 发送GET请求获取一页工作流运行记录
            response = requests.get(next_page, headers={"Authorization": f"token {self.token}"})
            if response.status_code == 200:  # 如果请求成功
                data = response.json()  # 将响应数据转换为JSON格式
                runs = data["workflow_runs"]  # 获取运行记录列表
                next_page = response.links.get("next", {}).get("url")  # 获取下一页的URL

                with ThreadPoolExecutor(max_workers=max_workers) as executor:  # 创建线程池
                    for run in runs:  # 遍历每条运行记录
                        # 将运行记录的创建时间转换为UTC时间
                        run_time = datetime.strptime(run["created_at"], "%Y-%m-%dT%H:%M:%SZ").replace(tzinfo=timezone.utc)
                        # 计算当前时间与运行记录创建时间的差值
                        time_difference = self.current_time - run_time
                        if time_difference > timedelta(hours=168):  # 如果差值超过168小时
                            executor.submit(self.delete_run, run["id"])  # 提交删除运行记录的任务
            else:  # 如果请求失败
                print(f"Failed to fetch runs. Status code: {response.status_code}")  # 打印错误信息
                break  # 退出循环


if __name__ == "__main__":
    repository_name = os.environ.get("REPOSITORY_NAME")
    github_token = os.environ.get("GITHUB_TOKEN")
    repo_url = f"https://api.github.com/repos/{repository_name}"  # GitHub仓库URL
    token = f"{github_token}"  # GitHub个人访问令牌

    manager = GitHubActionsManager(repo_url, token)  # 创建GitHubActionsManager实例
    manager.delete_old_runs()  # 删除168小时前的运行记录