main.py

import json
import httpx
import os

PAT_TOKEN = os.getenv("PAT_TOKEN")

GITHUB_LATEST_RELEASE_API = "https://api.github.com/repos/DGP-Studio/Snap.Hutao/releases/latest"

new_version = os.getenv("VERSION")

github_version = httpx.get(GITHUB_LATEST_RELEASE_API).json()["tag_name"]
msix_file_name = f"Snap.Hutao.{new_version}.msix"


def fetch_github_issue_and_pr() -> dict:
    pr_url = "https://api.github.com/repos/DGP-Studio/Snap.Hutao.Docs/pulls?state=open"
    issue_url = "https://api.github.com/repos/DGP-Studio/Snap.Hutao/issues?state=open&labels=Publish"

    def check_label(item):
        labels = item["labels"]
        return len(list(filter(lambda label: label["name"] == "Document Updates", labels))) != 0

    prs = list(filter(check_label, httpx.get(pr_url).json()))
    issues = httpx.get(issue_url).json()

    if len(issues) == 0:
        print("No open issue with 'Publish' label at main repo.")
        raise IndexError
    elif len(issues) != 1:
        print("Too many open issue with 'Publish' label at main repo.")
        raise IndexError

    if len(prs) == 0:
        print("No open PR with 'Document Updates' label.")
        raise IndexError
    elif len(prs) != 1:
        print("Expected exactly one open PR with 'Document Updates' label.")
        raise IndexError

    return prs[0]


def get_update_logs(pr) -> (str, str):
    pr_title = pr["title"]
    ref_name = pr["head"]["ref"]
    if new_version not in pr_title:
        print(f"Wrong version in title: {pr_title}")
        raise NameError
    else:
        print(f"Found PR with 'Document Updates' label: {pr_title}, id: {pr['number']}")

    def get_raw_url(file_path: str) -> str | None:
        content_url = f"https://raw.githubusercontent.com/DGP-Studio/Snap.Hutao.Docs/{ref_name}/{file_path}"
        return content_url

    en_update_log_url = get_raw_url("docs/en/statements/update-log.md")
    en_update_log = httpx.get(en_update_log_url).text.split("##")[1].split("\n")
    en_update_log = "\n".join([s for s in en_update_log if "<Badge text=" not in s])
    print(f"En update log:\n{en_update_log}")

    zh_update_log_url = get_raw_url("docs/zh/statements/update-log.md")
    zh_update_log = httpx.get(zh_update_log_url).text.split("##")[1].split("\n")
    zh_update_log = "\n".join([s for s in zh_update_log if "<Badge text=" not in s])
    print(f"Zh update log:\n{zh_update_log}")

    return en_update_log, zh_update_log


def generate_changelog(en_log: str, zh_log: str):
    generic_changelog = f"""## 更新日志
    
{zh_log}
    
## Update Log
    
{en_log}
    
## 完整更新日志/What's Changed
    
Full Changelog: [{github_version}...{new_version}](https://github.com/DGP-Studio/Snap.Hutao/compare/{github_version}.\
..{new_version})

---
如果你安装过程中出现问题，请阅读我们的 [FAQ 文档](https://hut.ao/zh/advanced/FAQ.html)以获取解决方案
If you have any issue during the installation, please check our [FAQ document](https://hut.ao/en/advanced/FAQ.html) for solutions
"""
    generic_changelog = generic_changelog.replace("\n\n\n", "\n")
    generic_changelog = generic_changelog.replace("\n\n", "\n")
    social_promotion = f"""
{new_version} 版本已发布/ Version {new_version} is released

Release Page: https://github.com/DGP-Studio/Snap.Hutao/releases/tag/{new_version}
Direct Download: https://github.com/DGP-Studio/Snap.Hutao/releases/download/{new_version}/{msix_file_name}   
"""
    social_promotion = social_promotion + "\n" + generic_changelog
    ann_meta = {
        "title": f"{new_version} 版本已发布/Version {new_version} is Live",
        "description": f"{new_version} 版本已发布，请及时更新，点击查看详情可阅读完整更新日志\nVersion {new_version} is released"
                       f", please update as soon as possible. Click to view details to read the full "
                       f"changelog",
        "url": f"https://github.com/DGP-Studio/Snap.Hutao/releases/tag/{new_version}"
    }
    ann_meta = json.dumps(ann_meta, indent=2, ensure_ascii=False)
    return {
        "generic": generic_changelog,
        "social": social_promotion,
        "ann_meta": ann_meta
    }


def merge_docs_pull_request(pr):
    pr_number = pr["number"]
    pr_url = f"https://api.github.com/repos/DGP-Studio/Snap.Hutao.Docs/pulls/{pr_number}/merge"
    merge_headers = {
        "Accept": "application/vnd.github.v3+json",
        "Authorization": f"Bearer {PAT_TOKEN}",
        "X-GitHub-Api-Version": "2022-11-28"
    }
    response = httpx.put(pr_url, headers=merge_headers)
    print(response.status_code)
    print(response.text)


def main():
    pr = fetch_github_issue_and_pr()
    en_log, zh_log = get_update_logs(pr)
    changelog_set = generate_changelog(en_log, zh_log)
    merge_docs_pull_request(pr)
    with open("release_body.md", "w") as body:
        body.writelines(changelog_set['generic'])


if __name__ == "__main__":
    main()