feat: teai role

This commit is contained in:
ydc3148
2026-09-14 16:13:02 +08:00
parent 8735e35db6
commit 540718dda7
12 changed files with 416 additions and 120 deletions
+69 -3
View File
@@ -9,8 +9,17 @@ import (
"sort"
)
// AgentsTeamName 是工作流團隊的名稱(AGENTS.md:調度成員由 Agents 團隊決定)。
const AgentsTeamName = "Agents"
const (
// AgentsTeamName 是工作流團隊的名稱(AGENTS.md:調度成員由 Agents 團隊決定)。
AgentsTeamName = "Agents"
// SupervisorTeamName 是管理者團隊的名稱;成員角色為 supervisor。
SupervisorTeamName = "Supervisor"
// RoleSupervisor 是 Supervisor 團隊成員的角色值。
RoleSupervisor = "supervisor"
// RoleWorker 是 Agents 團隊成員的角色值。
RoleWorker = "worker"
)
// Client 抽象 teai 需要的 Gitea API 讀取操作(僅唯讀)。
// #5 的 internal/gitea HTTP 用戶端實作此介面;workflow 不依賴 HTTP 細節。
@@ -49,7 +58,7 @@ type PullItem struct {
Pull PullRequest
}
// Scanner 在 Client 之上組合出各命令的資料(whoami/orgs/members/mine/pulls/next)。
// Scanner 在 Client 之上組合出各命令的資料(whoami/role/orgs/members/mine/pulls/next)。
type Scanner struct {
Client Client
}
@@ -59,6 +68,63 @@ func (s *Scanner) Whoami() (string, error) {
return s.Client.CurrentUser()
}
// Role 依目前帳號所屬團隊回傳角色:Supervisor 團隊為 supervisor,
// Agents 團隊為 worker。同時屬於兩者時以 supervisor 為準。
// 兩者都不屬於時回空字串(CLI 輸出 null)。
func (s *Scanner) Role() (string, error) {
username, err := s.Whoami()
if err != nil {
return "", err
}
orgs, err := s.Orgs()
if err != nil {
return "", err
}
inAgents := false
for _, org := range orgs {
teams, err := s.Client.ListTeams(org)
if err != nil {
return "", err
}
for _, team := range teams {
switch team.Name {
case SupervisorTeamName, AgentsTeamName:
default:
continue
}
ok, err := s.usernameInTeam(team.ID, username)
if err != nil {
return "", err
}
if !ok {
continue
}
if team.Name == SupervisorTeamName {
return RoleSupervisor, nil
}
inAgents = true
}
}
if inAgents {
return RoleWorker, nil
}
return "", nil
}
// usernameInTeam 檢查帳號是否為該團隊成員。
func (s *Scanner) usernameInTeam(teamID int64, username string) (bool, error) {
members, err := s.Client.ListTeamMembers(teamID)
if err != nil {
return false, err
}
for _, m := range members {
if m == username {
return true, nil
}
}
return false, nil
}
// Orgs 回傳目前帳號所屬的組織(去重、排序)。
func (s *Scanner) Orgs() ([]string, error) {
names, err := s.Client.ListOrgs()