mirror of
https://github.com/tiennm99/goclaw.git
synced 2026-10-12 12:13:51 +00:00
Duc Nguyenandntduc
bb7712a9ff
* feat(collaboration): isolate delegated artifacts and child runs Isolate delegated inputs and outputs behind secure artifact exchange lifecycles. Scope Agent Link tasks by tenant and root agent, and enforce delegation spawn-tree boundaries. Add process-wide child-run admission and preserve logical media paths across native, MCP, and sandbox execution. * fix(collaboration): harden delegated task isolation Enforce tenant and root-agent task scope across migrations and stores. Add exactly-once async completion delivery, delegated sandbox boundaries, and confined artifact and media recovery across runtime surfaces. * fix(collaboration): recover interrupted async tasks * fix(collaboration): normalize persisted child-run status --------- Co-authored-by: ntduc <ntduc@cpp.ai.vn>
112 lines
3.2 KiB
Go
112 lines
3.2 KiB
Go
package hooks
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
|
|
"github.com/nextlevelbuilder/goclaw/internal/store"
|
|
"github.com/nextlevelbuilder/goclaw/internal/tracing"
|
|
)
|
|
|
|
// EmitHookSpan records a tracing span for a hook execution.
|
|
//
|
|
// The span name follows the plan's convention: "hook.<handlerType>.<event>"
|
|
// (e.g., "hook.command.pre_tool_use"). Duration is taken from durationMS
|
|
// (the caller already measured it); status is "completed" on success,
|
|
// "error" when decision == DecisionError or errMsg is non-empty. The decision
|
|
// and hook identity are persisted into Metadata so dashboards can aggregate
|
|
// allow/block/error ratios per event and hook.
|
|
//
|
|
// Fields lifted from ctx:
|
|
// - trace id (tracing.TraceIDFromContext)
|
|
// - parent span id (tracing.ParentSpanIDFromContext, omitted when nil)
|
|
// - agent id (store.AgentIDFromContext, omitted when zero)
|
|
// - team id (tracing.TraceTeamIDPtrFromContext)
|
|
// - tenant id (store.TenantIDFromContext, falls back to MasterTenantID)
|
|
//
|
|
// No-op when ctx has no collector attached — safe in tests and for tenants
|
|
// without tracing enabled.
|
|
func EmitHookSpan(
|
|
ctx context.Context,
|
|
event HookEvent,
|
|
ht HandlerType,
|
|
cfg HookConfig,
|
|
startedAt time.Time,
|
|
decision Decision,
|
|
durationMS int,
|
|
errMsg string,
|
|
input string,
|
|
output string,
|
|
) {
|
|
const eventPreviewLimit = 40_000
|
|
|
|
end := time.Now().UTC()
|
|
|
|
status := store.SpanStatusCompleted
|
|
if decision == DecisionError || errMsg != "" {
|
|
status = store.SpanStatusError
|
|
}
|
|
|
|
// Mirror emitLLMSpanStart's ctx-extraction pattern for agent/team/tenant.
|
|
var agentID *uuid.UUID
|
|
if a := store.AgentIDFromContext(ctx); a != uuid.Nil {
|
|
agentID = &a
|
|
}
|
|
tenantID := store.TenantIDFromContext(ctx)
|
|
if tenantID == uuid.Nil {
|
|
tenantID = store.MasterTenantID
|
|
}
|
|
|
|
errorMsg := errMsg
|
|
if len(errorMsg) > 200 {
|
|
errorMsg = errorMsg[:200]
|
|
}
|
|
|
|
metadata := map[string]any{
|
|
"decision": string(decision),
|
|
"hook_id": cfg.ID.String(),
|
|
"hook_name": cfg.Name,
|
|
"hook_event": string(event),
|
|
"handler_type": string(ht),
|
|
"duration_ms": durationMS,
|
|
}
|
|
var metadataJSON json.RawMessage
|
|
if b, err := json.Marshal(metadata); err == nil {
|
|
metadataJSON = b
|
|
}
|
|
|
|
span := store.SpanData{
|
|
TraceID: tracing.TraceIDFromContext(ctx),
|
|
SpanType: store.SpanTypeEvent,
|
|
Name: "hook." + string(ht) + "." + string(event),
|
|
StartTime: startedAt,
|
|
EndTime: &end,
|
|
DurationMS: durationMS,
|
|
Status: status,
|
|
Error: errorMsg,
|
|
Level: store.SpanLevelDefault,
|
|
InputPreview: tracing.TruncateJSON(input, eventPreviewLimit),
|
|
OutputPreview: tracing.TruncateMid(output, eventPreviewLimit),
|
|
AgentID: agentID,
|
|
TeamID: tracing.TraceTeamIDPtrFromContext(ctx),
|
|
TenantID: tenantID,
|
|
Metadata: metadataJSON,
|
|
CreatedAt: end,
|
|
}
|
|
|
|
// Attach parent only when present — leaving it nil avoids bogus FK edges.
|
|
if parent := tracing.ParentSpanIDFromContext(ctx); parent != uuid.Nil {
|
|
p := parent
|
|
span.ParentSpanID = &p
|
|
}
|
|
|
|
collector := tracing.CollectorFromContext(ctx)
|
|
if collector == nil {
|
|
return // tracing disabled — no collector attached
|
|
}
|
|
collector.EmitSpan(tracing.RedactSpan(ctx, span))
|
|
}
|