agent/pkg/agentic/commit.go
Virgil 43e2a14b82 feat(agentic): add completion journal commit stage
Co-Authored-By: Virgil <virgil@lethean.io>
2026-04-01 22:07:42 +00:00

181 lines
5.5 KiB
Go

// SPDX-License-Identifier: EUPL-1.2
package agentic
import (
"context"
"time"
core "dappco.re/go/core"
"github.com/modelcontextprotocol/go-sdk/mcp"
)
// input := agentic.CommitInput{Workspace: "core/go-io/task-42"}
type CommitInput struct {
Workspace string `json:"workspace"`
}
// out := agentic.CommitOutput{Success: true, Workspace: "core/go-io/task-42", JournalPath: "/srv/.core/workspace/core/go-io/task-42/.meta/journal.jsonl"}
type CommitOutput struct {
Success bool `json:"success"`
Workspace string `json:"workspace"`
JournalPath string `json:"journal_path,omitempty"`
MarkerPath string `json:"marker_path,omitempty"`
CommittedAt string `json:"committed_at,omitempty"`
Skipped bool `json:"skipped,omitempty"`
}
// result := c.Action("agentic.commit").Run(ctx, core.NewOptions(core.Option{Key: "workspace", Value: "core/go-io/task-42"}))
func (s *PrepSubsystem) handleCommit(_ context.Context, options core.Options) core.Result {
input := CommitInput{
Workspace: optionStringValue(options, "workspace"),
}
output, err := s.commitWorkspace(nil, input)
if err != nil {
return core.Result{Value: err, OK: false}
}
return core.Result{Value: output, OK: true}
}
func (s *PrepSubsystem) registerCommitTool(server *mcp.Server) {
mcp.AddTool(server, &mcp.Tool{
Name: "agentic_commit",
Description: "Write the final workspace dispatch record to the local journal after verify completes.",
}, s.commitTool)
}
func (s *PrepSubsystem) commitTool(ctx context.Context, _ *mcp.CallToolRequest, input CommitInput) (*mcp.CallToolResult, CommitOutput, error) {
output, err := s.commitWorkspace(ctx, input)
if err != nil {
return nil, CommitOutput{}, err
}
return nil, output, nil
}
func (s *PrepSubsystem) commitWorkspace(ctx context.Context, input CommitInput) (CommitOutput, error) {
workspaceDir := resolveWorkspace(input.Workspace)
if workspaceDir == "" {
return CommitOutput{}, core.E("commitWorkspace", core.Concat("workspace not found: ", input.Workspace), nil)
}
result := ReadStatusResult(workspaceDir)
workspaceStatus, ok := workspaceStatusValue(result)
if !ok {
err, _ := result.Value.(error)
if err == nil {
err = core.E("commitWorkspace", "status not found", nil)
}
return CommitOutput{}, err
}
metaDir := WorkspaceMetaDir(workspaceDir)
if r := fs.EnsureDir(metaDir); !r.OK {
err, _ := r.Value.(error)
if err == nil {
err = core.E("commitWorkspace", "failed to create metadata directory", nil)
}
return CommitOutput{}, err
}
journalPath := core.JoinPath(metaDir, "journal.jsonl")
markerPath := core.JoinPath(metaDir, "commit.json")
committedAt := time.Now().UTC().Format(time.RFC3339)
if existingCommit, ok := readCommitMarker(markerPath); ok && existingCommit.UpdatedAt == workspaceStatus.UpdatedAt && existingCommit.Runs == workspaceStatus.Runs {
return CommitOutput{
Success: true,
Workspace: input.Workspace,
JournalPath: journalPath,
MarkerPath: markerPath,
CommittedAt: existingCommit.CommittedAt,
Skipped: true,
}, nil
}
record := commitWorkspaceRecord(workspaceDir, workspaceStatus, committedAt)
line := core.Concat(core.JSONMarshalString(record), "\n")
appendHandle := fs.Append(journalPath)
if !appendHandle.OK {
err, _ := appendHandle.Value.(error)
if err == nil {
err = core.E("commitWorkspace", "failed to open journal", nil)
}
return CommitOutput{}, err
}
core.WriteAll(appendHandle.Value, line)
marker := commitMarker{
Workspace: WorkspaceName(workspaceDir),
UpdatedAt: workspaceStatus.UpdatedAt,
Runs: workspaceStatus.Runs,
CommittedAt: committedAt,
}
if r := fs.WriteAtomic(markerPath, core.JSONMarshalString(marker)); !r.OK {
err, _ := r.Value.(error)
if err == nil {
err = core.E("commitWorkspace", "failed to write commit marker", nil)
}
return CommitOutput{}, err
}
return CommitOutput{
Success: true,
Workspace: input.Workspace,
JournalPath: journalPath,
MarkerPath: markerPath,
CommittedAt: committedAt,
}, nil
}
type commitMarker struct {
Workspace string `json:"workspace"`
UpdatedAt time.Time `json:"updated_at"`
Runs int `json:"runs"`
CommittedAt string `json:"committed_at"`
}
func readCommitMarker(markerPath string) (commitMarker, bool) {
r := fs.Read(markerPath)
if !r.OK {
return commitMarker{}, false
}
var marker commitMarker
if parseResult := core.JSONUnmarshalString(r.Value.(string), &marker); !parseResult.OK {
return commitMarker{}, false
}
return marker, true
}
func commitWorkspaceRecord(workspaceDir string, workspaceStatus *WorkspaceStatus, committedAt string) map[string]any {
record := map[string]any{
"workspace": WorkspaceName(workspaceDir),
"repo": workspaceStatus.Repo,
"org": workspaceStatus.Org,
"task": workspaceStatus.Task,
"agent": workspaceStatus.Agent,
"branch": workspaceStatus.Branch,
"status": workspaceStatus.Status,
"question": workspaceStatus.Question,
"issue": workspaceStatus.Issue,
"runs": workspaceStatus.Runs,
"process_id": workspaceStatus.ProcessID,
"pr_url": workspaceStatus.PRURL,
"started_at": workspaceStatus.StartedAt,
"updated_at": workspaceStatus.UpdatedAt,
"committed_at": committedAt,
}
if report := readSyncWorkspaceReport(workspaceDir); len(report) > 0 {
record["report"] = report
if findings := anyMapSliceValue(report["findings"]); len(findings) > 0 {
record["findings"] = findings
}
if changes := anyMapValue(report["changes"]); len(changes) > 0 {
record["changes"] = changes
}
}
return record
}