Merge pull request #167 from ConnectAI-E/feature-lloyd-20240131

feat: 支持获取旧的issue再发送卡片

server/tasks/lark/chat.py

@@ -1,5 +1,6 @@
 import json
 import logging
+from urllib.parse import urlparse
 
 from celery_app import app, celery
 from model.schema import (
@@ -268,3 +269,124 @@ def create_issue(
             "创建 issue 失败", app_id, message_id, content, data, *args, **kwargs
         )
     return response
+
+
+@celery.task()
+def sync_issue(
+    issue_id, issue_link, app_id, message_id, content, data, *args, **kwargs
+):
+    repo_name = ""
+    is_pr = False
+    try:
+        if not issue_id:
+            path = urlparse(issue_link).path
+            issue_id = int(path.split("/").pop())
+            repo_name = path.split("/")[2]
+            is_pr = path.split("/")[3] == "pull"
+    except Exception as e:
+        logging.error(e)
+
+    if not issue_id:
+        return send_chat_failed_tip(
+            "找不到issue", app_id, message_id, content, data, *args, **kwargs
+        )
+
+    chat_id = data["event"]["message"]["chat_id"]
+    chat_group = (
+        db.session.query(ChatGroup)
+        .filter(
+            ChatGroup.chat_id == chat_id,
+        )
+        .first()
+    )
+    if not chat_group:
+        return send_chat_failed_tip(
+            "找不到项目群", app_id, message_id, content, data, *args, **kwargs
+        )
+
+    repos = (
+        db.session.query(Repo)
+        .filter(
+            Repo.chat_group_id == chat_group.id,
+            Repo.status == 0,
+        )
+        .all()
+    )
+    # 如果有多个，尝试从issue_link里面拿到repo_name过滤一下
+    if len(repos) > 1 and repo_name:
+        repos = [repo for repo in repos if repo.name == repo_name]
+
+    if len(repos) > 1:
+        return send_chat_failed_tip(
+            "当前群有多个项目，无法唯一确定仓库", app_id, message_id, content, data, *args, **kwargs
+        )
+    if len(repos) == 0:
+        return send_chat_failed_tip(
+            "找不到项目", app_id, message_id, content, data, *args, **kwargs
+        )
+
+    repo = repos[0]  # 能找到唯一的仓库才执行
+
+    code_application = (
+        db.session.query(CodeApplication)
+        .filter(
+            CodeApplication.id == repo.application_id,
+        )
+        .first()
+    )
+    if not code_application:
+        return send_chat_failed_tip(
+            "找不到对应的项目", app_id, message_id, content, data, *args, **kwargs
+        )
+
+    team = (
+        db.session.query(Team)
+        .filter(
+            Team.id == code_application.team_id,
+        )
+        .first()
+    )
+    if not team:
+        return send_chat_failed_tip(
+            "找不到对应的项目", app_id, message_id, content, data, *args, **kwargs
+        )
+    openid = data["event"]["sender"]["sender_id"]["open_id"]
+    # 这里连三个表查询，所以一次性都查出来
+    code_users = get_code_users_by_openid([openid])
+    # 当前操作的用户
+    current_code_user_id = code_users[openid][0]
+
+    github_app = GitHubAppRepo(
+        code_application.installation_id, user_id=current_code_user_id
+    )
+
+    import tasks
+
+    # 后面需要插入记录，再发卡片，创建话题
+    repository = github_app.get_repo_info_by_name(team.name, repo.name)
+    if is_pr:
+        pull_request = github_app.get_one_pull_request(team.name, repo.name, issue_id)
+        logging.debug("get_one_pull_requrst %r", pull_request)
+        return tasks.on_pull_request_opened(
+            {
+                "action": "opened",
+                "sender": pull_request["user"],
+                "pull_request": pull_request,
+                "repository": repository,
+            }
+        )
+    else:
+        issue = github_app.get_one_issue(team.name, repo.name, issue_id)
+        logging.debug("get_one_issue %r", issue)
+        return tasks.on_issue_opened(
+            {
+                "action": "opened",
+                "sender": issue["user"],
+                "issue": issue,
+                "repository": repository,
+            }
+        )
+
+    return send_chat_failed_tip(
+        "同步 issue 失败", app_id, message_id, content, data, *args, **kwargs
+    )

server/utils/github/repo.py

@@ -36,8 +36,20 @@ def get_repo_info(self, repo_id: str) -> dict | None:
         if not team:
             return None
 
+        return self.get_repo_info_by_name(team.name, repo.name)
+
+    def get_repo_info_by_name(self, team_name: str, repo_name: str) -> dict | None:
+        """Get repo info by repo name.
+
+        Args:
+            team_name (str): The organization name from GitHub.
+            repo_name (str): The repo name from GitHub.
+
+        Returns:
+            dict: Repo info.
+        """
         return self.base_github_rest_api(
-            f"https://api.github.com/repos/{team.name}/{repo.name}",
+            f"https://api.github.com/repos/{team_name}/{repo_name}",
             "GET",
             "install_token",
         )
@@ -201,6 +213,28 @@ def create_issue(
             },
         )
 
+    def get_one_issue(
+        self,
+        repo_onwer: str,
+        repo_name: str,
+        issue_number: int,
+    ) -> dict | None:
+        """Get an issue
+
+        Args:
+            repo_onwer (str): The repo owner.
+            repo_name (str): The repo name.
+            issue_number (int): The issue id
+
+        Returns:
+            dict: The issue info.
+        """
+
+        return self.base_github_rest_api(
+            f"https://api.github.com/repos/{repo_onwer}/{repo_name}/issues/{issue_number}",
+            auth_type="install_token",
+        )
+
     def create_issue_comment(
         self, repo_onwer: str, repo_name: str, issue_number: int, body: str
     ) -> dict | None:
@@ -263,6 +297,28 @@ def update_issue(
             json=json,
         )
 
+    def get_one_pull_request(
+        self,
+        repo_onwer: str,
+        repo_name: str,
+        pull_number: int,
+    ) -> dict | None:
+        """Get an issue
+
+        Args:
+            repo_onwer (str): The repo owner.
+            repo_name (str): The repo name.
+            pull_number (int): The pull_request id
+
+        Returns:
+            dict: The issue info.
+        """
+
+        return self.base_github_rest_api(
+            f"https://api.github.com/repos/{repo_onwer}/{repo_name}/pulls/{pull_number}",
+            auth_type="install_token",
+        )
+
     def requested_reviewers(
         self,
         repo_onwer: str,

server/utils/lark/parser.py

@@ -1,5 +1,6 @@
 import argparse
 import logging
+from urllib.parse import urlparse
 
 import tasks
 from utils.constant import TopicType
@@ -209,6 +210,24 @@ def on_issue(self, param, unkown, *args, **kwargs):
             }
             # 只有群聊才是指定的repo
             if "group" == chat_type:
+                if len(param.argv) == 1:
+                    issue_id_or_link = param.argv[0]
+                    try:
+                        if issue_id_or_link.isdigit():
+                            tasks.sync_issue.delay(
+                                int(issue_id_or_link), None, *args, **kwargs
+                            )
+                        elif urlparse(issue_id_or_link).netloc:
+                            # 这里是否需要检查netloc==github.com??
+                            tasks.sync_issue.delay(
+                                None, issue_id_or_link, *args, **kwargs
+                            )
+                        else:
+                            raise Exception("invalid issue_id or issue link.")
+                        return "issue", param, unkown
+                    except Exception as e:
+                        logging.error(e)
+
                 title, users, labels = [], [], []
 
                 for arg in param.argv: