mirror of
https://github.com/snowykami/mbcp.git
synced 2024-11-22 06:07:37 +08:00
250 lines
10 KiB
Python
250 lines
10 KiB
Python
|
"""
|
|||
|
Module docs
|
|||
|
"""
|
|||
|
import os
|
|||
|
|
|||
|
os.system("pip install PyGithub")
|
|||
|
|
|||
|
import json
|
|||
|
from github.GitTree import GitTree
|
|||
|
from github import Github, InputGitTreeElement
|
|||
|
|
|||
|
COMMAND_HEAD = "Friend Link Request"
|
|||
|
FRIEND_LINKS_JSON = ".vitepress/data/friend-links.json"
|
|||
|
FRIEND_LINKS_I18N_JSON = ".vitepress/sugarat/theme/data/i18n/friend-links-i18n.json"
|
|||
|
|
|||
|
g = Github(os.getenv('TOKEN'))
|
|||
|
repo = g.get_repo(os.getenv('REPOSITORY'))
|
|||
|
issue_number = int(os.getenv('ISSUE_NUMBER'))
|
|||
|
act_type = os.getenv('ACT_TYPE') # opened, edited, closed, deleted 对应事件类型
|
|||
|
"""
|
|||
|
opened: 添加友链
|
|||
|
edited: 修改友链
|
|||
|
closed: 审核通过
|
|||
|
deleted: 删除友链
|
|||
|
"""
|
|||
|
issue = repo.get_issue(number=issue_number)
|
|||
|
issue_title = issue.title
|
|||
|
issue_body = json.loads(issue.body)
|
|||
|
friend_link_name = issue_body["name"]
|
|||
|
friend_link_name_en = issue_body.get("name_en", "")
|
|||
|
friend_link_des = issue_body["des"]
|
|||
|
friend_link_des_en = issue_body.get("des_en", "")
|
|||
|
friend_link_url = issue_body["url"]
|
|||
|
friend_link_icon = issue_body["icon"]
|
|||
|
creator_lang = issue_body.get("lang", "zh")
|
|||
|
creator_name = issue.user.login
|
|||
|
ref = repo.get_git_ref("heads/main")
|
|||
|
|
|||
|
i18n_text = {
|
|||
|
"zh": {
|
|||
|
"pre_check_finished" : "✅ 预检查通过,一切工作已就绪,等待仓库所有者审核",
|
|||
|
"pre_check_failed" : "❌ 预检查未通过:{COMMENT},请修改issue",
|
|||
|
"failed_not_a_https_url": "URL不是HTTPS链接,请以https://开头",
|
|||
|
"check_passed" : "✅ 审核通过,已添加友链,页面稍后就会构建好",
|
|||
|
"if_add_i18n_data" : "🌐 是否添加国际化数据?如需添加请修改issue添加`name_en`、`des_en`字段。",
|
|||
|
"about_edit" : "📑 如需修改信息,请直接编辑issue,不要新建issue。",
|
|||
|
"link_already_exists" : "该友链已存在或存在同名友链",
|
|||
|
"delete_success" : "✅ 友链已删除",
|
|||
|
"site_title" : "标题",
|
|||
|
"site_description" : "描述",
|
|||
|
"site_ping" : "延迟",
|
|||
|
"site_url" : "链接",
|
|||
|
"apply_info" : "申请信息",
|
|||
|
"query_result" : "查询结果",
|
|||
|
"site_name" : "站点名称",
|
|||
|
},
|
|||
|
"en": {
|
|||
|
"pre_check_finished" : "✅ Pre-check passed, ready to go, waiting for the repository owner to review",
|
|||
|
"pre_check_failed" : "❌ Pre-check failed: {COMMENT},please modify the issue",
|
|||
|
"failed_not_a_https_url": "URL is not a HTTPS link, please start with https://",
|
|||
|
"check_passed" : "✅ Check passed, the friend link has been added, and the page will be built soon.",
|
|||
|
"if_add_i18n_data" : "🌐 Do you want to add internationalization data? If you want, please modify the issue to add `name_en` and `des_en` "
|
|||
|
"fields.",
|
|||
|
"about_edit" : "📑 If you need to modify the information, please edit the issue directly instead of creating a new issue.",
|
|||
|
"link_already_exists" : "The friend link already exists",
|
|||
|
"delete_success" : "✅ The friend link has been deleted",
|
|||
|
"site_title" : "Title",
|
|||
|
"site_description" : "Description",
|
|||
|
"site_ping" : "Ping",
|
|||
|
"site_url" : "URL",
|
|||
|
"apply_info" : "Apply Info",
|
|||
|
"query_result" : "Query Result",
|
|||
|
"site_name" : "Site Name",
|
|||
|
}
|
|||
|
}
|
|||
|
if creator_lang not in i18n_text:
|
|||
|
lang = "zh"
|
|||
|
|
|||
|
|
|||
|
def get_text(key: str) -> str:
|
|||
|
return i18n_text[creator_lang].get(key, key)
|
|||
|
|
|||
|
|
|||
|
# closed触发
|
|||
|
def run_add():
|
|||
|
"""审核通过 关闭时触发"""
|
|||
|
closer = issue.closed_by
|
|||
|
if closer.login != repo.owner.login and not issue.title.startswith(COMMAND_HEAD):
|
|||
|
issue.create_comment(get_text("about_edit"))
|
|||
|
# 修改友链信息
|
|||
|
with open(FRIEND_LINKS_JSON, 'r') as f:
|
|||
|
friend_link_data = json.load(f)
|
|||
|
# 检查url相同并删除
|
|||
|
for i, friend in enumerate(friend_link_data):
|
|||
|
if friend["url"] == friend_link_url:
|
|||
|
friend_link_data.pop(i)
|
|||
|
break
|
|||
|
friend_link_data.append(
|
|||
|
{
|
|||
|
"nickname": f'partnerLink.{friend_link_name}.nickname',
|
|||
|
"des" : f'partnerLink.{friend_link_name}.des',
|
|||
|
"avatar" : friend_link_icon,
|
|||
|
"url" : friend_link_url,
|
|||
|
}
|
|||
|
)
|
|||
|
|
|||
|
# 修改友链国际化信息
|
|||
|
with open(FRIEND_LINKS_I18N_JSON, 'r') as f:
|
|||
|
friend_i18n_data = json.load(f)
|
|||
|
friend_i18n_data['zh'][f'partnerLink.{friend_link_name}.nickname'] = friend_link_name
|
|||
|
friend_i18n_data['zh'][f'partnerLink.{friend_link_name}.des'] = friend_link_des
|
|||
|
friend_i18n_data['en'][f'partnerLink.{friend_link_name}.nickname'] = friend_link_name_en or f"{creator_name}'s site"
|
|||
|
friend_i18n_data['en'][f'partnerLink.{friend_link_name}.des'] = friend_link_des_en or f"{creator_name}'s site"
|
|||
|
|
|||
|
tree = repo.create_git_tree(
|
|||
|
base_tree=repo.get_git_tree("main"),
|
|||
|
tree=[
|
|||
|
InputGitTreeElement(
|
|||
|
path=FRIEND_LINKS_JSON,
|
|||
|
mode="100644",
|
|||
|
type="blob",
|
|||
|
content=json.dumps(friend_link_data, indent=4, ensure_ascii=False)
|
|||
|
),
|
|||
|
InputGitTreeElement(
|
|||
|
path=FRIEND_LINKS_I18N_JSON,
|
|||
|
mode="100644",
|
|||
|
type="blob",
|
|||
|
content=json.dumps(friend_i18n_data, indent=4, ensure_ascii=False)
|
|||
|
)
|
|||
|
]
|
|||
|
)
|
|||
|
# 提交修改
|
|||
|
commit = repo.create_git_commit(
|
|||
|
message=f":busts_in_silhouette: Add friend link: {friend_link_url}({creator_name})",
|
|||
|
tree=tree,
|
|||
|
parents=[repo.get_git_commit(ref.object.sha)]
|
|||
|
)
|
|||
|
ref.edit(commit.sha)
|
|||
|
issue.create_comment(get_text("check_passed"))
|
|||
|
|
|||
|
|
|||
|
# opened触发
|
|||
|
def run_pre_check(typ: str):
|
|||
|
try:
|
|||
|
import re
|
|||
|
os.system("pip install requests beautifulsoup4")
|
|||
|
import requests
|
|||
|
from bs4 import BeautifulSoup
|
|||
|
|
|||
|
def get_site_metadata(url) -> tuple[str, str, int]:
|
|||
|
response = requests.get(url)
|
|||
|
response.raise_for_status() # Ensure we notice bad responses
|
|||
|
response.encoding = 'utf-8'
|
|||
|
soup = BeautifulSoup(response.text, 'html.parser')
|
|||
|
|
|||
|
title = soup.title.string if soup.title else "No title found"
|
|||
|
description = "No description found"
|
|||
|
|
|||
|
# Look for meta description tag
|
|||
|
description_tag = soup.find('meta', attrs={
|
|||
|
'name': 'description'
|
|||
|
})
|
|||
|
if description_tag and 'content' in description_tag.attrs:
|
|||
|
description = description_tag['content']
|
|||
|
return title, description, int(response.elapsed.microseconds / 1000)
|
|||
|
|
|||
|
# 检查链接是否合法
|
|||
|
if not re.match(r"^https?://", friend_link_url) and not re.match(r"^https?://", friend_link_icon):
|
|||
|
issue.create_comment(get_text("pre_check_failed").format(COMMENT=get_text("failed_not_a_https_url")))
|
|||
|
return
|
|||
|
else:
|
|||
|
# 若是opened则第一次检查是否存在友链
|
|||
|
if typ == "opened":
|
|||
|
for friend in json.load(open(FRIEND_LINKS_JSON)):
|
|||
|
if friend["url"] == friend_link_url or friend["nickname"] == friend_link_name:
|
|||
|
issue.create_comment(get_text("pre_check_failed").format(COMMENT=get_text("link_already_exists")))
|
|||
|
return
|
|||
|
print("checking site metadata...")
|
|||
|
|
|||
|
title, description, ping_ms = get_site_metadata(friend_link_url)
|
|||
|
site_meta = f"""\n
|
|||
|
# {get_text("apply_info")}\n
|
|||
|
**{get_text("site_url")}**: [{friend_link_url}]({friend_link_url})\n
|
|||
|
**{get_text("site_name")}**: {friend_link_name} / {friend_link_name_en or "No English name"}\n
|
|||
|
**{get_text("site_description")}**: {friend_link_des} / {friend_link_des_en or "No English description"}\n
|
|||
|
|
|||
|
# {get_text("query_result")}\n
|
|||
|
**{get_text("site_title")}**: {title}\n
|
|||
|
**{get_text("site_description")}**: {description}\n
|
|||
|
**{get_text("site_ping")}**: {ping_ms:.2f}ms\n"""
|
|||
|
issue.create_comment(get_text("pre_check_finished") + site_meta + get_text("if_add_i18n_data"))
|
|||
|
except Exception as e:
|
|||
|
issue.create_comment(f"{get_text('pre_check_failed').format(COMMENT=str(e))}")
|
|||
|
|
|||
|
|
|||
|
def run_delete():
|
|||
|
"""删除友链"""
|
|||
|
with open(FRIEND_LINKS_JSON, 'r') as f:
|
|||
|
friend_link_data = json.load(f)
|
|||
|
for i, friend in enumerate(friend_link_data):
|
|||
|
if friend["url"] == friend_link_url:
|
|||
|
friend_link_data.pop(i)
|
|||
|
break
|
|||
|
with open(FRIEND_LINKS_I18N_JSON, 'r') as f:
|
|||
|
friend_i18n_data = json.load(f)
|
|||
|
for language in friend_i18n_data:
|
|||
|
for key in list(friend_i18n_data[language]):
|
|||
|
if key.startswith(f'partnerLink.{creator_name}.'):
|
|||
|
friend_i18n_data[language].pop(key)
|
|||
|
tree = repo.create_git_tree(
|
|||
|
base_tree=repo.get_git_tree("main"),
|
|||
|
tree=[
|
|||
|
InputGitTreeElement(
|
|||
|
path=FRIEND_LINKS_JSON,
|
|||
|
mode="100644",
|
|||
|
type="blob",
|
|||
|
content=json.dumps(friend_link_data, indent=4, ensure_ascii=False)
|
|||
|
),
|
|||
|
InputGitTreeElement(
|
|||
|
path=FRIEND_LINKS_I18N_JSON,
|
|||
|
mode="100644",
|
|||
|
type="blob",
|
|||
|
content=json.dumps(friend_i18n_data, indent=4, ensure_ascii=False)
|
|||
|
)
|
|||
|
]
|
|||
|
)
|
|||
|
# 提交修改
|
|||
|
commit = repo.create_git_commit(
|
|||
|
message=f":busts_in_silhouette: Delete friend link: {friend_link_url}({creator_name})",
|
|||
|
tree=tree,
|
|||
|
parents=[repo.get_git_commit(ref.object.sha)]
|
|||
|
)
|
|||
|
ref.edit(commit.sha)
|
|||
|
print("✅ 友链已删除")
|
|||
|
|
|||
|
|
|||
|
if __name__ == "__main__":
|
|||
|
if issue_title.startswith(COMMAND_HEAD):
|
|||
|
if act_type in ["opened", "edited"]:
|
|||
|
run_pre_check(act_type)
|
|||
|
elif act_type == "closed":
|
|||
|
run_add()
|
|||
|
elif act_type == "deleted":
|
|||
|
run_delete()
|
|||
|
else:
|
|||
|
print("nothing to do")
|
|||
|
|
|||
|
else:
|
|||
|
print("Not a friend link request issue, passed.")
|