Function handleGraph

apps/cli/internal/mcp/graph_tool.go:31–85

(_ context.Context, _ *gomcp.CallToolRequest, input GraphInput)

}

func handleGraph(_ context.Context, _ *gomcp.CallToolRequest, input GraphInput) (*gomcp.CallToolResult, any, error) {
	taskDir := input.TaskDir
	if taskDir == "" {
		taskDir = "."
	}

	taskScanner := scanner.NewScanner(taskDir, false, nil)
	result, err := taskScanner.Scan()
	if err != nil {
		return nil, nil, fmt.Errorf("scan failed: %w", err)
	}

	tasks := result.Tasks

	if len(input.Filters) > 0 {
		tasks, err = filter.Apply(tasks, input.Filters)
		if err != nil {
			return nil, nil, fmt.Errorf("filter error: %w", err)
		}
	}

	if len(input.ExcludeStatus) > 0 {
		tasks = excludeByStatus(tasks, input.ExcludeStatus)
	}

	g := graph.NewGraph(tasks)

	if input.RootTaskID != "" {
		if _, ok := g.TaskMap[input.RootTaskID]; !ok {
			return nil, nil, fmt.Errorf("root task not found: %s", input.RootTaskID)
		}
		upstream := g.GetUpstream(input.RootTaskID)
		downstream := g.GetDownstream(input.RootTaskID)
		combined := make(map[string]bool, len(upstream)+len(downstream)+1)
		for id := range upstream {
			combined[id] = true
		}
		for id := range downstream {
			combined[id] = true
		}
		combined[input.RootTaskID] = true
		g = g.FilterTasks(combined)
	}

	graphJSON := g.ToJSON()

	data, err := json.Marshal(graphJSON)
	if err != nil {
		return nil, nil, fmt.Errorf("json marshal failed: %w", err)
	}

	return &gomcp.CallToolResult{
		Content: []gomcp.Content{&gomcp.TextContent{Text: string(data)}},
	}, nil, nil
}

func excludeByStatus(tasks []*model.Task, statuses []string) []*model.Task {
	excludeMap := make(map[string]bool, len(statuses))