Method handleQueryDatabase

backend/api/mcp/tool_query.go:67–114

(ctx context.Context, req *mcp.CallToolRequest, input QueryInput)

}

func (s *Server) handleQueryDatabase(ctx context.Context, req *mcp.CallToolRequest, input QueryInput) (*mcp.CallToolResult, any, error) {
	if input.Database == "" {
		return nil, nil, errors.New("database is required")
	}
	if input.Statement == "" {
		return nil, nil, errors.New("statement is required")
	}

	// Normalize limit.
	limit := input.Limit
	if limit <= 0 {
		limit = defaultQueryLimit
	}
	if limit > maxQueryLimit {
		limit = maxQueryLimit
	}

	// Resolve database.
	resolveCtx, resolveCancel := context.WithTimeout(ctx, resolveTimeout)
	defer resolveCancel()

	resolved, err := s.resolveDatabase(resolveCtx, input.Database, input.Instance, input.Project)
	if err != nil {
		return formatToolError(err), nil, nil
	}
	if resolved.ambiguous {
		picked, elicitErr := s.elicitDatabaseChoice(ctx, req, resolved)
		if elicitErr != nil {
			// Elicitation unsupported or user cancelled — fall back to AMBIGUOUS_TARGET.
			return formatAmbiguousResult(input.Database, resolved.candidates), nil, nil
		}
		resolved = picked
	}

	// Execute query.
	queryCtx, queryCancel := context.WithTimeout(ctx, queryTimeout)
	defer queryCancel()

	output, err := s.executeQuery(queryCtx, resolved, input.Statement, limit)
	if err != nil {
		return formatToolError(err), nil, nil
	}

	text := formatQueryOutput(input.Statement, resolved.resourceName, output)
	return &mcp.CallToolResult{
		Content: []mcp.Content{&mcp.TextContent{Text: text}},
	}, output, nil
}

// formatQueryOutput formats a successful query result as text header + JSON body.
func formatQueryOutput(statement, resourceName string, output *QueryOutput) string {