weichenandCursor
3f42149457
Composition roots copy Parts once so HTTP/CLI handlers stop calling locator getters on every request. Co-authored-by: Cursor <cursoragent@cursor.com>
114 lines
3.1 KiB
Go
114 lines
3.1 KiB
Go
package server
|
|
|
|
import (
|
|
"context"
|
|
"crypto/sha256"
|
|
"encoding/hex"
|
|
"fmt"
|
|
|
|
"chat-agent-system/internal/bizdev"
|
|
"chat-agent-system/internal/squad"
|
|
"chat-agent-system/internal/taskrun"
|
|
"chat-agent-system/internal/workspace"
|
|
)
|
|
|
|
// FinalizeTaskRunProjection 将任务执行结果投影到 workflow_runs 与聊天助手消息（供回填/管理命令使用）。
|
|
func (s *Server) FinalizeTaskRunProjection(ctx context.Context, record *taskrun.RunRecord, result *bizdev.TaskRunResult) (string, error) {
|
|
return s.finalizeTaskRun(ctx, record, result)
|
|
}
|
|
|
|
// finalizeTaskRun 是薄适配：产物哈希在事务外经 workspace.Capability 读取，终态命令交给 taskrun.Commands。
|
|
func (s *Server) finalizeTaskRun(ctx context.Context, record *taskrun.RunRecord, result *bizdev.TaskRunResult) (string, error) {
|
|
if record == nil || result == nil {
|
|
return "failed", fmt.Errorf("finalize: missing record or result")
|
|
}
|
|
if result.DocRecord != nil && result.DocRecord.ContentSHA256 == "" {
|
|
if hexSum := s.hashDeliverable(ctx, record, result); hexSum != "" {
|
|
result.DocRecord.ContentSHA256 = hexSum
|
|
}
|
|
}
|
|
cmds := s.taskCommands()
|
|
if s != nil {
|
|
cmds.PW = s.pwSvc
|
|
if s.svc != nil {
|
|
cmds.Ctx = s.ctxMgr
|
|
}
|
|
}
|
|
return cmds.Finalize(ctx, record, result)
|
|
}
|
|
|
|
func (s *Server) taskCommands() *taskrun.Commands {
|
|
if s != nil && s.cmds != nil {
|
|
return s.cmds
|
|
}
|
|
cmds := &taskrun.Commands{
|
|
PW: s.pwSvc,
|
|
Env: "test",
|
|
}
|
|
if s != nil && s.cfg != nil {
|
|
cmds.Env = s.cfg.Env
|
|
cmds.StorageIdentity = "env:" + s.cfg.Env
|
|
}
|
|
if s != nil && s.svc != nil {
|
|
cmds.Ctx = s.ctxMgr
|
|
}
|
|
return cmds
|
|
}
|
|
|
|
func (s *Server) hashDeliverable(ctx context.Context, record *taskrun.RunRecord, result *bizdev.TaskRunResult) string {
|
|
if s == nil || s.wsCap == nil || record == nil {
|
|
return ""
|
|
}
|
|
id := workspace.OpaqueID(result.WorkDir, record.WorkspaceID)
|
|
if id == "" {
|
|
return ""
|
|
}
|
|
env := ""
|
|
if s.cfg != nil {
|
|
env = s.cfg.Env
|
|
}
|
|
ref := workspace.WorkspaceRef{
|
|
ID: id,
|
|
SchemaVersion: workspace.SchemaVersion,
|
|
ProjectScope: record.ProjectID,
|
|
RunID: record.RunID,
|
|
Environment: env,
|
|
}
|
|
body, err := s.wsCap.ReadArtifact(ctx, ref, workspace.DeliverableKey)
|
|
if err != nil || len(body) == 0 {
|
|
return ""
|
|
}
|
|
sum := sha256.Sum256(body)
|
|
return hex.EncodeToString(sum[:])
|
|
}
|
|
|
|
func resolveTerminalStatus(result *bizdev.TaskRunResult) string {
|
|
return result.TerminalStatus()
|
|
}
|
|
|
|
func buildTaskCompletionSummary(result *bizdev.TaskRunResult, terminalStatus string) string {
|
|
_ = terminalStatus
|
|
return result.CompletionSummary()
|
|
}
|
|
|
|
func squadResultToTaskRun(res *squad.SquadRunResult, taskKey string) *bizdev.TaskRunResult {
|
|
if res == nil {
|
|
return nil
|
|
}
|
|
return &bizdev.TaskRunResult{
|
|
TaskKey: res.TaskKey,
|
|
ArtifactTitle: res.ArtifactTitle,
|
|
MarkdownBody: res.MarkdownBody,
|
|
ExtractedData: res.ExtractedData,
|
|
Attachments: res.Attachments,
|
|
WorkDir: res.WorkDir,
|
|
BackendUsed: res.BackendUsed,
|
|
Duration: res.Duration,
|
|
Success: res.Success,
|
|
ErrorReason: res.ErrorReason,
|
|
QualityReport: res.QualityReport,
|
|
Manifest: res.Manifest,
|
|
AdvisorBrief: res.AdvisorBrief,
|
|
}
|
|
}
|