package room
import (
"context"
"encoding/json"
"fmt"
"log"
"os"
"os/exec"
"path/filepath"
"strings"
"github.com/sdaduanbilei/agent-team/internal/llm"
"github.com/sdaduanbilei/agent-team/internal/skill"
"github.com/sdaduanbilei/agent-team/internal/store"
)
// buildWorkflowStep 根据 workspace 中已存在的文件,构建 phase-aware 的工作流进度上下文
func (r *Room) buildWorkflowStep() string {
if r.projectTemplate == nil {
return ""
}
var completed []string
var pending []string
minPendingPhase := 999
for _, f := range r.projectTemplate.Files {
if f.IsDir || f.Dynamic {
continue
}
fpath := filepath.Join(r.Dir, "workspace", f.Path)
if _, err := os.Stat(fpath); err == nil {
completed = append(completed, f.Path)
} else {
pending = append(pending, fmt.Sprintf("%s (@%s, phase:%d)", f.Path, f.Owner, f.Phase))
if f.Phase < minPendingPhase {
minPendingPhase = f.Phase
}
}
}
var sb strings.Builder
sb.WriteString("\n")
sb.WriteString("已完成的文件:\n")
for _, f := range completed {
sb.WriteString(fmt.Sprintf(" [done] %s\n", f))
}
sb.WriteString("待产出的文件:\n")
for _, f := range pending {
sb.WriteString(fmt.Sprintf(" [todo] %s\n", f))
}
sb.WriteString("\n\n")
if minPendingPhase < 999 {
sb.WriteString("\n当前阶段:前置材料准备\n")
sb.WriteString("待产出的前置材料:\n")
for _, f := range r.projectTemplate.Files {
if f.IsDir || f.Dynamic {
continue
}
fpath := filepath.Join(r.Dir, "workspace", f.Path)
if _, err := os.Stat(fpath); err == nil {
continue
}
if f.Owner == r.master.Config.Name {
sb.WriteString(fmt.Sprintf("- 你负责:%s(系统将自动发起文档调用,你不需要在此输出文档正文)\n", f.Path))
} else {
sb.WriteString(fmt.Sprintf("- 分配给 @%s:%s\n", f.Owner, f.Path))
}
}
sb.WriteString("\n")
sb.WriteString("请安排产出待完成的前置材料。你负责的文件由系统自动发起独立调用,不要在回复中输出完整文档正文。")
} else if dynDir, dynOwner, dynPhase := r.getDynamicFileInfo(); dynDir != "" {
// 所有静态文件已完成,进入动态章节写作阶段
existingChapters := r.listChapterFiles(dynDir)
sb.WriteString(fmt.Sprintf("\n当前阶段:phase %d — 章节写作\n", dynPhase))
sb.WriteString(fmt.Sprintf("负责人:%s\n", dynOwner))
if len(existingChapters) > 0 {
sb.WriteString("已完成章节:\n")
for _, ch := range existingChapters {
sb.WriteString(fmt.Sprintf(" [done] %s\n", ch))
}
}
isMasterDyn := r.master != nil && dynOwner == r.master.Config.Name
if isMasterDyn {
sb.WriteString("系统将自动发起章节文档调用,你不需要在回复中输出章节正文。\n")
sb.WriteString("请简短说明接下来要写哪一章,系统会自动发起文档调用。\n")
} else {
sb.WriteString(fmt.Sprintf("请用 @%s 分配章节写作任务,任务中注明章节序号和标题。\n", dynOwner))
sb.WriteString("每次分配一章,写完后你会收到完成通知,再安排下一章。\n")
}
sb.WriteString("\n")
} else {
sb.WriteString("所有模板文件已完成。请根据你的 AGENT.md 工作流程决定下一步行动(如向用户交付、或进入下一阶段)。")
}
return sb.String()
}
// currentMinPhase 返回当前最小未完成 phase
func (r *Room) currentMinPhase() int {
if r.projectTemplate == nil {
return 0
}
minPhase := 999
for _, f := range r.projectTemplate.Files {
if f.IsDir || f.Dynamic {
continue
}
fpath := filepath.Join(r.Dir, "workspace", f.Path)
if _, err := os.Stat(fpath); os.IsNotExist(err) {
if f.Phase < minPhase {
minPhase = f.Phase
}
}
}
if minPhase == 999 {
return 0
}
return minPhase
}
// validatePhaseAssignments 轻量校验:只在分配写手(phase:2 任务)时检查前置材料(phase:1)是否全部就绪
// 返回被阻止的分配及原因
func (r *Room) validatePhaseAssignments(assignments map[string]string) map[string]string {
if r.projectTemplate == nil {
return nil
}
// 检查所有 phase:1 文件是否已完成
allPhase1Done := true
var pendingPhase1 []string
for _, f := range r.projectTemplate.Files {
if f.IsDir || f.Dynamic || f.Phase != 1 {
continue
}
fpath := filepath.Join(r.Dir, "workspace", f.Path)
if _, err := os.Stat(fpath); os.IsNotExist(err) {
allPhase1Done = false
pendingPhase1 = append(pendingPhase1, fmt.Sprintf("%s(@%s)", f.Path, f.Owner))
}
}
if allPhase1Done {
return nil // 前置材料全部完成,不阻止任何分配
}
// 只阻止 phase:2(写手/章节)的分配,phase:1 内部不互相阻止
blocked := make(map[string]string)
for name := range assignments {
targetFile := r.findMemberTargetFile(name)
if targetFile == nil {
continue
}
if targetFile.Phase > 1 {
blocked[name] = fmt.Sprintf(
"《%s》属于 phase:%d,但前置材料尚未全部完成:%s",
targetFile.Path, targetFile.Phase,
strings.Join(pendingPhase1, "、"))
}
}
return blocked
}
// allStaticFilesDone 检查所有非动态模板文件是否都已完成
func (r *Room) allStaticFilesDone() bool {
if r.projectTemplate == nil {
return false
}
for _, f := range r.projectTemplate.Files {
if f.IsDir || f.Dynamic {
continue
}
fpath := filepath.Join(r.Dir, "workspace", f.Path)
if _, err := os.Stat(fpath); os.IsNotExist(err) {
return false
}
}
return true
}
// getDynamicFileInfo 返回动态文件的目录名、负责人和阶段号
func (r *Room) getDynamicFileInfo() (dir, owner string, phase int) {
if r.projectTemplate == nil {
return "", "", 0
}
// 找到 dynamic 条目,以及它前面的目录条目
lastDir := ""
for _, f := range r.projectTemplate.Files {
if f.IsDir {
lastDir = strings.TrimSuffix(f.Path, "/")
continue
}
if f.Dynamic {
if lastDir == "" {
lastDir = "chapters"
}
return lastDir, f.Owner, f.Phase
}
}
return "", "", 0
}
// listChapterFiles 列出 workspace 根目录下的章节文件(非模板静态文件的 .md 文件)
func (r *Room) listChapterFiles(dir string) []string {
wsDir := filepath.Join(r.Dir, "workspace")
entries, err := os.ReadDir(wsDir)
if err != nil {
return nil
}
// 收集模板静态文件名,用于排除
staticFiles := make(map[string]bool)
if r.projectTemplate != nil {
for _, f := range r.projectTemplate.Files {
if !f.IsDir && !f.Dynamic {
staticFiles[f.Path] = true
}
}
}
var files []string
for _, e := range entries {
if e.IsDir() || !strings.HasSuffix(e.Name(), ".md") {
continue
}
if staticFiles[e.Name()] {
continue
}
files = append(files, e.Name())
}
return files
}
// masterChapterFileCall 为 master 发起一次章节文档调用
func (r *Room) masterChapterFileCall(ctx context.Context, masterMsgs *[]llm.Message, dir string, chapterHint string) {
r.setStatus(StatusWorking, r.master.Config.Name, "正在编写章节...")
r.emit(Event{Type: EvtFileWorking, Agent: r.master.Config.Name, Filename: "章节", Title: "章节"})
existingChapters := r.listChapterFiles(dir)
var existingList string
if len(existingChapters) > 0 {
existingList = "\n已完成的章节:\n"
for _, ch := range existingChapters {
existingList += fmt.Sprintf("- %s\n", ch)
}
}
filePrompt := r.Prompt.Render("file_call_chapter", map[string]string{
"Dir": dir,
"ExistingChapters": existingList,
})
if chapterHint != "" {
filePrompt += "\n\n你之前的规划:" + chapterHint
}
if wsCtx := r.buildWorkspaceContext(); wsCtx != "" {
filePrompt += "\n\n" + wsCtx
}
fileLLMMsg := llm.NewMsg("user", filePrompt)
*masterMsgs = append(*masterMsgs, fileLLMMsg)
reply, usage, err := r.master.ChatWithUsage(ctx, *masterMsgs, nil)
if err != nil {
log.Printf("[room %s] master chapter file call error: %v", r.Config.Name, err)
return
}
r.emitUsage(r.master.Config.Name, usage)
content := strings.TrimSpace(reply)
// 从内容中提取章节标题作为文件名(直接保存到 workspace 根目录,不创建子文件夹)
chapterFilename := r.extractChapterFilename(content, dir)
if !strings.HasPrefix(content, "# ") {
content = "# " + strings.TrimSuffix(chapterFilename, ".md") + "\n\n" + content
}
r.saveWorkspace(chapterFilename, content)
docName := strings.TrimSuffix(chapterFilename, ".md")
r.emit(Event{Type: EvtArtifact, Agent: r.master.Config.Name, Filename: chapterFilename, Title: docName})
if r.Store != nil {
r.Store.InsertMessage(&store.Message{
RoomID: r.Config.Name, Agent: r.master.Config.Name, Role: "master",
Content: docName, Filename: chapterFilename, PartType: "document",
GroupID: &r.currentGroupID,
})
}
r.emit(Event{Type: EvtFileDone, Agent: r.master.Config.Name, Filename: chapterFilename, Title: docName})
// 发送完成状态到聊天
statusMsg := fmt.Sprintf("《%s》已完成,保存到 %s。", docName, chapterFilename)
r.emit(Event{Type: EvtAgentMessage, Agent: r.master.Config.Name, Role: "master", Content: statusMsg, NoStore: true})
if r.Store != nil {
r.Store.InsertMessage(&store.Message{
RoomID: r.Config.Name, Agent: r.master.Config.Name, Role: "master",
Content: statusMsg, PartType: "text",
GroupID: &r.currentGroupID,
})
}
assistantMsg := llm.NewMsg("assistant", reply)
*masterMsgs = append(*masterMsgs, assistantMsg)
r.historyMu.Lock()
r.masterHistory = append(r.masterHistory, fileLLMMsg, assistantMsg)
r.historyMu.Unlock()
r.AppendHistory("master", r.master.Config.Name, reply)
}
// extractChapterFilename 从章节内容中提取文件名
func (r *Room) extractChapterFilename(content, dir string) string {
title := extractTitle(content)
if title == "" {
// 按已有章节数自增
existing := r.listChapterFiles(dir)
return fmt.Sprintf("第%d章.md", len(existing)+1)
}
// 清理标题中的特殊字符,作为文件名
title = strings.ReplaceAll(title, " ", "-")
title = strings.ReplaceAll(title, "/", "-")
title = strings.ReplaceAll(title, "\\", "-")
// 先去除可能已有的 .md 后缀,再统一添加,避免双重扩展名
title = strings.TrimSuffix(title, ".md")
title += ".md"
return title
}
// buildProjectContext 构建项目模板上下文,注入到 agent system prompt
func (r *Room) buildProjectContext(agentName string) string {
if r.projectTemplate == nil {
return ""
}
isMaster := r.master != nil && agentName == r.master.Config.Name
var sb strings.Builder
sb.WriteString("\n")
sb.WriteString("项目文件结构(系统自动管理文件保存):\n\n")
for _, f := range r.projectTemplate.Files {
if f.IsDir {
sb.WriteString(fmt.Sprintf(" %s (目录)\n", f.Path))
continue
}
if f.Dynamic {
sb.WriteString(fmt.Sprintf(" ... @%s phase:%d (动态扩展)\n", f.Owner, f.Phase))
continue
}
marker := ""
if f.Owner == agentName {
marker = " ← 你负责"
}
sb.WriteString(fmt.Sprintf(" %s @%s phase:%d%s\n", f.Path, f.Owner, f.Phase, marker))
}
if isMaster {
// master:文档通过系统 file call 产出,聊天中只做规划和分配
sb.WriteString("\n" + r.Prompt.R("master_output_spec") + "\n")
} else {
// 成员:最终回复输出文档正文
sb.WriteString("\n" + r.Prompt.R("output_spec") + "\n")
sb.WriteString("系统会自动保存到对应文件。不要在文档中夹杂状态描述或对话内容。\n")
}
sb.WriteString("")
return sb.String()
}
// matchTemplateFile 按标题匹配模板文件。
func (r *Room) matchTemplateFile(title string) *ProjectFile {
if r.projectTemplate == nil || title == "" {
return nil
}
for i := range r.projectTemplate.Files {
f := &r.projectTemplate.Files[i]
if f.IsDir || f.Dynamic {
continue
}
fname := strings.TrimSuffix(f.Path, ".md")
if fname == title {
return f
}
}
for i := range r.projectTemplate.Files {
f := &r.projectTemplate.Files[i]
if f.IsDir || f.Dynamic {
continue
}
fname := strings.TrimSuffix(f.Path, ".md")
if strings.HasPrefix(title, fname) {
return f
}
}
for i := range r.projectTemplate.Files {
f := &r.projectTemplate.Files[i]
if f.IsDir || f.Dynamic {
continue
}
base := strings.TrimSuffix(f.Path, ".md")
if strings.Contains(base, title) || strings.Contains(title, base) {
return f
}
keywords := strings.FieldsFunc(base, func(r rune) bool { return r == '与' || r == '和' || r == '·' })
for _, kw := range keywords {
if len([]rune(kw)) >= 2 && strings.Contains(title, kw) {
return f
}
}
}
return nil
}
// findOwnerFiles 查找某 agent 负责的所有文件
func (r *Room) findOwnerFiles(agentName string) []ProjectFile {
if r.projectTemplate == nil {
return nil
}
var files []ProjectFile
for _, f := range r.projectTemplate.Files {
if f.Owner == agentName && !f.IsDir && !f.Dynamic {
files = append(files, f)
}
}
return files
}
// findPendingMasterFiles 查找 master 负责的所有待产出文件(不限 phase)
func (r *Room) findPendingMasterFiles() []ProjectFile {
if r.projectTemplate == nil {
return nil
}
var files []ProjectFile
for _, f := range r.projectTemplate.Files {
if f.IsDir || f.Dynamic || f.Owner != r.master.Config.Name {
continue
}
fpath := filepath.Join(r.Dir, "workspace", f.Path)
if _, err := os.Stat(fpath); os.IsNotExist(err) {
files = append(files, f)
}
}
return files
}
// masterFileCall 为 master 发起一次独立的文档产出调用(file call)。
// 输出直接保存到 workspace,不进入聊天记录。
func (r *Room) masterFileCall(ctx context.Context, masterMsgs *[]llm.Message, file ProjectFile) {
docName := strings.TrimSuffix(file.Path, ".md")
r.setStatus(StatusWorking, r.master.Config.Name, fmt.Sprintf("正在编写《%s》...", docName))
r.emit(Event{Type: EvtFileWorking, Agent: r.master.Config.Name, Filename: file.Path, Title: docName})
filePrompt := r.Prompt.Render("file_call_master", map[string]string{
"DocName": docName,
"FilePath": file.Path,
})
if wsCtx := r.buildWorkspaceContext(); wsCtx != "" {
filePrompt += "\n\n" + wsCtx
}
fileLLMMsg := llm.NewMsg("user", filePrompt)
*masterMsgs = append(*masterMsgs, fileLLMMsg)
reply, usage, err := r.master.ChatWithUsage(ctx, *masterMsgs, nil)
if err != nil {
log.Printf("[room %s] master file call error for %s: %v", r.Config.Name, file.Path, err)
return
}
r.emitUsage(r.master.Config.Name, usage)
content := strings.TrimSpace(reply)
if !strings.HasPrefix(content, "# ") {
content = "# " + docName + "\n\n" + content
}
r.saveWorkspace(file.Path, content)
r.emit(Event{Type: EvtArtifact, Agent: r.master.Config.Name, Filename: file.Path, Title: docName})
if r.Store != nil {
r.Store.InsertMessage(&store.Message{
RoomID: r.Config.Name, Agent: r.master.Config.Name, Role: "master",
Content: docName, Filename: file.Path, PartType: "document",
GroupID: &r.currentGroupID,
})
}
r.emit(Event{Type: EvtFileDone, Agent: r.master.Config.Name, Filename: file.Path, Title: docName})
assistantMsg := llm.NewMsg("assistant", reply)
*masterMsgs = append(*masterMsgs, assistantMsg)
r.historyMu.Lock()
r.masterHistory = append(r.masterHistory, fileLLMMsg, assistantMsg)
r.historyMu.Unlock()
r.AppendHistory("master", r.master.Config.Name, reply)
}
// findMemberTargetFile 查找成员当前应产出的目标文件(用于 file call)
func (r *Room) findMemberTargetFile(name string) *ProjectFile {
ownerFiles := r.findOwnerFiles(name)
if len(ownerFiles) == 0 {
log.Printf("[room %s] findMemberTargetFile(%s): no owner files found", r.Config.Name, name)
return nil
}
log.Printf("[room %s] findMemberTargetFile(%s): found %d owner files", r.Config.Name, name, len(ownerFiles))
if len(ownerFiles) == 1 {
log.Printf("[room %s] findMemberTargetFile(%s): → %s", r.Config.Name, name, ownerFiles[0].Path)
return &ownerFiles[0]
}
for i := range ownerFiles {
fpath := filepath.Join(r.Dir, "workspace", ownerFiles[i].Path)
if _, err := os.Stat(fpath); os.IsNotExist(err) {
log.Printf("[room %s] findMemberTargetFile(%s): → %s (not yet exists)", r.Config.Name, name, ownerFiles[i].Path)
return &ownerFiles[i]
}
}
result := &ownerFiles[len(ownerFiles)-1]
log.Printf("[room %s] findMemberTargetFile(%s): → %s (fallback last)", r.Config.Name, name, result.Path)
return result
}
// parseMasterUpdateIntent 从 master 回复中解析更新文件的意图,返回需要更新的已存在模板文件
func (r *Room) parseMasterUpdateIntent(reply string) []ProjectFile {
if r.projectTemplate == nil {
return nil
}
// 检查 master 回复中是否提到了更新/修改某个已存在的模板文件
// 关键词:已更新、已修改、已调整、更新了、修改了
updateKeywords := []string{"已更新", "已修改", "已调整", "更新了", "修改了", "修改为", "改为", "调整为"}
hasUpdateIntent := false
for _, kw := range updateKeywords {
if strings.Contains(reply, kw) {
hasUpdateIntent = true
break
}
}
if !hasUpdateIntent {
return nil
}
var files []ProjectFile
seen := make(map[string]bool)
for _, f := range r.projectTemplate.Files {
if f.IsDir || f.Dynamic || f.Owner != r.master.Config.Name {
continue
}
fpath := filepath.Join(r.Dir, "workspace", f.Path)
if _, err := os.Stat(fpath); err != nil {
continue // 文件不存在,不是更新场景
}
// 检查 master 是否在回复中提到了这个文件
docName := strings.TrimSuffix(f.Path, ".md")
if strings.Contains(reply, docName) || strings.Contains(reply, "《"+docName+"》") {
if !seen[f.Path] {
seen[f.Path] = true
files = append(files, f)
}
}
}
return files
}
// masterFileUpdateCall 为 master 发起一次文件更新调用(更新已存在的文件)
func (r *Room) masterFileUpdateCall(ctx context.Context, masterMsgs *[]llm.Message, file ProjectFile, updateHint string) {
docName := strings.TrimSuffix(file.Path, ".md")
r.setStatus(StatusWorking, r.master.Config.Name, fmt.Sprintf("正在更新《%s》...", docName))
r.emit(Event{Type: EvtFileWorking, Agent: r.master.Config.Name, Filename: file.Path, Title: docName})
// 读取现有文件内容
currentContent := ""
fpath := filepath.Join(r.Dir, "workspace", file.Path)
if data, err := os.ReadFile(fpath); err == nil {
currentContent = string(data)
}
filePrompt := r.Prompt.Render("file_call_master_update", map[string]string{
"DocName": docName,
"FilePath": file.Path,
"UpdateHint": updateHint,
"CurrentContent": currentContent,
})
fileLLMMsg := llm.NewMsg("user", filePrompt)
*masterMsgs = append(*masterMsgs, fileLLMMsg)
reply, usage, err := r.master.ChatWithUsage(ctx, *masterMsgs, nil)
if err != nil {
log.Printf("[room %s] master file update call error for %s: %v", r.Config.Name, file.Path, err)
return
}
r.emitUsage(r.master.Config.Name, usage)
content := strings.TrimSpace(reply)
if !strings.HasPrefix(content, "# ") {
content = "# " + docName + "\n\n" + content
}
r.saveWorkspace(file.Path, content)
r.emit(Event{Type: EvtArtifact, Agent: r.master.Config.Name, Filename: file.Path, Title: docName})
if r.Store != nil {
r.Store.InsertMessage(&store.Message{
RoomID: r.Config.Name, Agent: r.master.Config.Name, Role: "master",
Content: docName, Filename: file.Path, PartType: "document",
GroupID: &r.currentGroupID,
})
}
r.emit(Event{Type: EvtFileDone, Agent: r.master.Config.Name, Filename: file.Path, Title: docName})
// 发送完成状态到聊天
statusMsg := fmt.Sprintf("《%s》已更新。", docName)
r.emit(Event{Type: EvtAgentMessage, Agent: r.master.Config.Name, Role: "master", Content: statusMsg, NoStore: true})
if r.Store != nil {
r.Store.InsertMessage(&store.Message{
RoomID: r.Config.Name, Agent: r.master.Config.Name, Role: "master",
Content: statusMsg, PartType: "text",
GroupID: &r.currentGroupID,
})
}
assistantMsg := llm.NewMsg("assistant", reply)
*masterMsgs = append(*masterMsgs, assistantMsg)
r.historyMu.Lock()
r.masterHistory = append(r.masterHistory, fileLLMMsg, assistantMsg)
r.historyMu.Unlock()
r.AppendHistory("master", r.master.Config.Name, reply)
}
// buildSkillSummary 为 master 构建简要的 skill 清单
func (r *Room) buildSkillSummary() string {
if len(r.skillMeta) == 0 {
return ""
}
var sb strings.Builder
sb.WriteString("\n")
sb.WriteString("以下工具可供团队成员使用,你可以在分配任务时提示成员使用对应的工具:\n")
for _, m := range r.skillMeta {
fmt.Fprintf(&sb, " - %s: %s\n", m.Name, m.Description)
}
sb.WriteString("")
return sb.String()
}
// buildTeamXML 构建团队成员 XML 上下文
func (r *Room) buildTeamXML() string {
var sb strings.Builder
sb.WriteString("\n")
for name, a := range r.members {
fmt.Fprintf(&sb, " \n %s\n %s\n \n", name, a.Config.Description)
}
sb.WriteString("")
return sb.String()
}
// executeToolCall 执行 tool call,返回执行结果
func (r *Room) executeToolCall(tc llm.ToolCall) string {
var args struct {
Command string `json:"command"`
}
if err := json.Unmarshal([]byte(tc.Function.Arguments), &args); err != nil {
return fmt.Sprintf("参数解析错误: %v", err)
}
skillPath := skill.SkillPathByToolName(r.skillMeta, tc.Function.Name)
skillsRoot := filepath.Dir(skillPath)
if skillPath == "" {
skillsRoot = "skills"
}
if abs, err := filepath.Abs(skillsRoot); err == nil {
skillsRoot = abs
}
log.Printf("[tool] 执行: %s, 命令: %s", tc.Function.Name, args.Command)
cmd := exec.Command("bash", "-c", args.Command)
cmd.Env = append(os.Environ(), "SKILLS_ROOT="+skillsRoot)
cmd.Dir = r.Dir
output, err := cmd.CombinedOutput()
result := string(output)
if err != nil {
result = fmt.Sprintf("命令执行错误: %v\n输出:\n%s", err, result)
}
if len(result) > 10000 {
result = result[:10000] + "\n... (输出已截断)"
}
log.Printf("[tool] 结果 (%d 字符): %.200s", len(result), result)
return result
}