add exec/process background session workflow and register process tool

pkg/tools/process_tool.go

@@ -0,0 +1,84 @@
+package tools
+
+import (
+	"context"
+	"encoding/json"
+	"time"
+)
+
+type ProcessTool struct{ m *ProcessManager }
+
+func NewProcessTool(m *ProcessManager) *ProcessTool { return &ProcessTool{m: m} }
+func (t *ProcessTool) Name() string                 { return "process" }
+func (t *ProcessTool) Description() string {
+	return "Manage background exec sessions: list, poll, log, kill"
+}
+func (t *ProcessTool) Parameters() map[string]interface{} {
+	return map[string]interface{}{"type": "object", "properties": map[string]interface{}{
+		"action":     map[string]interface{}{"type": "string", "description": "list|poll|log|kill"},
+		"session_id": map[string]interface{}{"type": "string"},
+		"offset":     map[string]interface{}{"type": "integer"},
+		"limit":      map[string]interface{}{"type": "integer"},
+		"timeout_ms": map[string]interface{}{"type": "integer"},
+	}, "required": []string{"action"}}
+}
+
+func (t *ProcessTool) Execute(ctx context.Context, args map[string]interface{}) (string, error) {
+	action, _ := args["action"].(string)
+	sid, _ := args["session_id"].(string)
+	if sid == "" {
+		sid, _ = args["sessionId"].(string)
+	}
+	switch action {
+	case "list":
+		b, _ := json.Marshal(t.m.List())
+		return string(b), nil
+	case "log":
+		off := toInt(args["offset"])
+		lim := toInt(args["limit"])
+		return t.m.Log(sid, off, lim)
+	case "kill":
+		if err := t.m.Kill(sid); err != nil {
+			return "", err
+		}
+		return "killed", nil
+	case "poll":
+		timeout := toInt(args["timeout_ms"])
+		if timeout < 0 {
+			timeout = 0
+		}
+		s, ok := t.m.Get(sid)
+		if !ok {
+			return "", nil
+		}
+		if timeout > 0 {
+			select {
+			case <-s.done:
+			case <-time.After(time.Duration(timeout) * time.Millisecond):
+			case <-ctx.Done():
+			}
+		}
+		s.mu.RLock()
+		defer s.mu.RUnlock()
+		resp := map[string]interface{}{"id": s.ID, "running": s.ExitCode == nil, "started_at": s.StartedAt.Format(time.RFC3339)}
+		if s.ExitCode != nil {
+			resp["exit_code"] = *s.ExitCode
+			resp["ended_at"] = s.EndedAt.Format(time.RFC3339)
+		}
+		b, _ := json.Marshal(resp)
+		return string(b), nil
+	default:
+		return "", nil
+	}
+}
+
+func toInt(v interface{}) int {
+	switch x := v.(type) {
+	case float64:
+		return int(x)
+	case int:
+		return x
+	default:
+		return 0
+	}
+}