code_callees.lua

-- MCP Tool: code_callees
-- What does a function/method call internally? (downward)
-- Entry kind: function.lua

local tools = require("tools")
local fmt = require("fmt")

local function call(arguments)
    if not arguments.symbol or arguments.symbol == "" then
        return tools.format_error("'symbol' parameter is required")
    end

    local result, err = tools.request("query:callees", {
        symbol = arguments.symbol,
        depth = arguments.depth or 2,
        project = arguments.project,
    })

    if err then return tools.format_error(err) end

    local lines = {}
    table.insert(lines, "# Callees of " .. result.name)
    table.insert(lines, "")

    local tree_lines = fmt.render_tree(result, {max_depth = arguments.depth or 2})
    for _, l in ipairs(tree_lines) do
        table.insert(lines, l)
    end

    -- Count resolved vs unresolved
    local resolved, unresolved = 0, 0
    local function count(node)
        for _, child in ipairs(node.children or {}) do
            if child.annotation == "unresolved" then
                unresolved = unresolved + 1
            else
                resolved = resolved + 1
            end
            count(child)
        end
    end
    count(result)

    table.insert(lines, "")
    table.insert(lines, string.format(
        "Resolved: %d | Unresolved: %d", resolved, unresolved
    ))

    return table.concat(lines, "\n")
end

return { call = call }