📦 docs: 资源商店新增发布资源功能

liteyuki_flow/resource_handler.py

@@ -0,0 +1,131 @@
+"""
+Module docs
+"""
+import requests  # type: ignore
+import zipfile
+
+from github import Github, InputGitTreeElement, GitTree
+from github.Issue import Issue
+from github.Repository import Repository
+import json
+import yaml
+
+from liteyuki_flow.const import OPENED, EDITED, CLOSED, REOPENED, RESOURCE_JSON
+from liteyuki_flow.markdown_parser import MarkdownParser
+from liteyuki_flow.typ import err, nil
+
+
+# opened: 创建新的资源包，预审核
+# edited: 编辑资源包信息，需重新审核
+# closed: 审核通过，修改json并提交
+# reopened: 重新打开，无操作
+
+# opened | edited
+def pre_check(github: Github, issue: Issue, repo: Repository) -> err:
+    parser = MarkdownParser(issue.body)
+    parser.parse_front_matters()
+    name = parser.front_matters.get("name")
+    desc = parser.front_matters.get("desc")
+    url = parser.front_matters.get("url")
+    homepage = parser.front_matters.get("homepage")  # optional
+    author = parser.front_matters.get("author")
+    if not all((name, desc, url, author)):
+        issue.create_comment("Name, desc, url, homepage and author are required.")
+        return ValueError("Name, desc, url, homepage and author are required.")
+
+    # 下载并解析资源包
+    r = requests.get(url)
+    if r.status_code != 200:
+        issue.create_comment("Download failed.")
+        return ValueError("Download failed.")
+
+    with open(f"tmp/{name}.zip", "wb") as f:
+        f.write(r.content)
+    # 解压
+    with zipfile.ZipFile(f"tmp/{name}.zip", "r") as z:
+        z.extractall(f"tmp/{name}")
+
+    # 检测包内metadata.yml文件
+    try:
+        data = yaml.load(open(f"tmp/{name}/metadata.yml"), Loader=yaml.SafeLoader)
+    except Exception:
+        issue.create_comment("metadata.yml not found or invalid.")
+        return ValueError("metadata.yml not found or invalid.")
+
+    # 检测必要字段 name，description，version
+    if not all((data.get("name"), data.get("description"), data.get("version"))):
+        issue.create_comment("name, description and version are required in metadata.yml.")
+        return ValueError("name, description and version are required in metadata.yml.")
+
+    # 不检测重复资源包，因为资源包可能有多个版本
+    # 检测通过，编辑原issue
+    metadata_markdown = f"**名称**: {data.get('name')}\n**描述**: {data.get('description')}\n**版本**: {data.get('version')}\n"
+    for k, v in data.items():
+        if k not in ("name", "description", "version"):
+            metadata_markdown += f"**{k}**: {v}\n"
+
+    new_issue_body = f"---\nname: {name}\ndesc: {desc}\nurl: {url}\nhomepage: {homepage}\nauthor: {author}\n---\n"
+    new_issue_body += f"# Resource: {name}\n"
+    new_issue_body += f"## 发布信息\n{desc}\n"
+    new_issue_body += f"**名称**: {name}\n"
+    new_issue_body += f"**描述**: {desc}\n"
+    new_issue_body += f"**作者**: {author}\n"
+    new_issue_body += f"**主页**: {homepage}\n"
+    new_issue_body += f"**下载**: {url}\n"
+    # 遍历其他字段
+    for k, v in data.items():
+        if k not in ("name", "description", "version"):
+            new_issue_body += f"**{k}**: {v}\n"
+
+    issue.edit(new_issue_body)
+    issue.create_comment("✅ 预检查通过\n## 元数据\n" + metadata_markdown)
+    return nil
+
+
+# closed
+def add_resource(github: Github, issue: Issue, repo: Repository):
+    parser = MarkdownParser(issue.body)
+    parser.parse_front_matters()
+    name = parser.front_matters.get("name")
+    desc = parser.front_matters.get("desc")
+    url = parser.front_matters.get("url")
+    homepage = parser.front_matters.get("homepage")  # optional
+    author = parser.front_matters.get("author")
+
+    # 编辑仓库内的json文件
+    resources = json.load(open(RESOURCE_JSON))
+    resources.append({
+            "name"    : name,
+            "desc"    : desc,
+            "url"     : url,
+            "homepage": homepage,
+            "author"  : author
+    })
+    ref = repo.get_git_ref("heads/main")
+    tree = repo.create_git_tree(
+        base_tree=repo.get_git_commit(ref.object.sha).tree,
+        tree=[
+                InputGitTreeElement(
+                    path=RESOURCE_JSON,
+                    mode="100644",
+                    type="blob",
+                    content=json.dumps(resources, indent=4, ensure_ascii=False)
+                )
+        ]
+    )
+    commit = repo.create_git_commit(
+        message=f":package: 发布资源: {name}",
+        tree=tree,
+        parents=[repo.get_git_commit(ref.object.sha)]
+    )
+    ref.edit(commit.sha)
+
+
+def handle_resource(github: Github, issue: Issue, repo: Repository, act_type: str):
+    if act_type in (OPENED, EDITED):
+        pre_check(github, issue, repo)
+
+    elif act_type == CLOSED:
+        add_resource(github, issue, repo)
+    else:
+        print("No operation found for the issue: ", act_type)