feat: enable collaboration for the same git org

client/app/hooks/useBot.ts

@@ -34,15 +34,15 @@ export const useBotDetail = (id: string) => {
   });
 };
 
-export const useGetBotBoundRepos =(id:string)=>{
+export const useGetBotBoundRepos = (id: string) => {
   return useQuery({
     queryKey: [`bot.boundRepos.${id}`, id],
     queryFn: async () => getBotBoundRepos(id),
     select: (data) => data,
     enabled: !!id,
     retry: false,
   });
-}
+};
 
 export const useBotConfig = (id: string, enabled: boolean) => {
   return useQuery({
@@ -153,7 +153,7 @@ export const useGetBotRagTask = (
     queryKey: [`rag.task`, repoName],
     queryFn: async () => getRagTask(repoName),
     select: (data) => data,
-    enabled:!!repoName,
+    enabled: !!repoName,
     retry: true,
     refetchInterval: refetchInterval ? 3 * 1000 : undefined,
   });

server/auth/router.py

@@ -1,5 +1,4 @@
-import json
 from github import Github
 from core.dao.profilesDAO import ProfilesDAO
 from fastapi import APIRouter, Request, HTTPException, status, Depends
 from fastapi.responses import RedirectResponse, JSONResponse

server/bot/router.py

@@ -29,31 +29,50 @@ def get_bot_list(
         None, description="Filter bots by personal category"
     ),
     name: Optional[str] = Query(None, description="Filter bots by name"),
-    user_id: Annotated[str | None, Depends(get_user_id)] = None,
+    user: Annotated[User | None, Depends(get_user)] = None,
 ):
     try:
         supabase = get_client()
         query = supabase.table("bots").select(
             "id, created_at, updated_at, avatar, description, name, public, starters, uid, repo_name"
         )
         if personal == "true":
-            if not user_id:
+            if not user or not user.access_token:
                 return {"data": [], "personal": personal}
-            query = query.eq("uid", user_id).order("updated_at", desc=True)
-        if name:
+            user_id = user.id
+            auth = Auth.Token(token=user.access_token)
+            g = Github(auth=auth)
+            github_user = g.get_user()
+            orgs = github_user.get_orgs()
+            repository_config_dao = RepositoryConfigDAO()
+            bots = repository_config_dao.query_bot_id_by_owners(
+                [org.id for org in orgs] + [github_user.id]
+            )
+            bot_ids = [bot["robot_id"] for bot in bots if bot["robot_id"]]
+            bot_ids_str = ",".join(map(str, bot_ids))  # 将 bots ID 列表转换为字符串
+
+            or_clause = (
+                f"uid.eq.{user_id},id.in.({bot_ids_str})"
+                if bot_ids_str
+                else f"uid.eq.{user_id}"
+            )
+
+            # 添加过滤条件
             query = (
-                supabase.table("bots")
-                .select(
-                    "id, created_at, updated_at, avatar, description, name, public, starters, uid, repo_name"
-                )
+                query.or_(or_clause).order("updated_at", desc=True)
+                if not name
+                else query.or_(or_clause)
                 .filter("name", "like", f"%{name}%")
+                .order("updated_at", desc=True)
+            )
+        else:
+            query = (
+                query.eq("public", True).order("updated_at", desc=True)
+                if not name
+                else query.eq("public", True)
+                .filter("name", "like", f"%{name}%")
+                .order("updated_at", desc=True)
             )
-
-        query = (
-            query.eq("public", True).order("updated_at", desc=True)
-            if not personal or personal != "true"
-            else query
-        )
 
         data = query.execute()
         if not data or not data.data:
@@ -186,6 +205,7 @@ async def bot_generator(
             content={"success": False, "errorMessage": str(e)}, status_code=500
         )
 
+
 @router.get("/git/avatar", status_code=200)
 async def get_git_avatar(
     repo_name: str,
@@ -201,8 +221,6 @@ async def get_git_avatar(
         )
 
 
-
-
 @router.put("/update/{id}", status_code=200)
 def update_bot(
     id: str,

server/core/dao/repositoryConfigDAO.py

@@ -83,6 +83,19 @@ def query_by_owners(self, orgs: List[str]):
             print(f"Error: {e}")
             return None
 
+    def query_bot_id_by_owners(self, orgs: List[str]):
+        try:
+            response = (
+                self.client.table("github_repo_config")
+                .select("robot_id")
+                .filter("owner_id", "in", f"({','.join(map(str, orgs))})")
+                .execute()
+            )
+            return response.data
+        except Exception as e:
+            print(f"Error: {e}")
+            return None
+
     def update_bot_to_repos(self, repos: List[RepoBindBotConfigVO]) -> bool:
         try:
             for repo in repos:

server/github_app/utils.py

@@ -58,16 +58,3 @@ def get_installation_repositories(access_token: str):
         },
     )
     return resp.json()
-
-
-def get_user_orgs(username, access_token: str):
-    url = f"https://api.github.com/users/{username}/orgs"
-    resp = requests.get(
-        url,
-        headers={
-            "X-GitHub-Api-Version": "2022-11-28",
-            "Accept": "application/vnd.github+json",
-            "Authorization": f"Bearer {access_token}",
-        },
-    )
-    return resp.json()