mcp-cpp-server 0.2.2

A high-performance Model Context Protocol (MCP) server for C++ code analysis using clangd LSP integration
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83

//! Call hierarchy analysis functionality for C++ functions and methods
//!
//! This module provides LSP-based call hierarchy analysis capabilities that work with
//! clangd to analyze function call relationships including incoming calls (callers)
//! and outgoing calls (callees).

use crate::clangd::session::ClangdSessionTrait;
use serde::{Deserialize, Serialize};

use crate::lsp::traits::LspClientTrait;
use crate::mcp_server::tools::analyze_symbols::AnalyzerError;
use crate::project::component_session::ComponentSession;
use crate::symbol::FileLocation;

// ============================================================================
// Call Hierarchy Types
// ============================================================================

#[derive(Debug, Serialize, Deserialize)]
pub struct CallHierarchy {
    /// Functions that call this function (incoming calls)
    pub callers: Vec<String>,
    /// Functions that this function calls (outgoing calls)
    pub callees: Vec<String>,
}

// ============================================================================
// Public API
// ============================================================================

/// Get call hierarchy information for a symbol (functions and methods)
pub async fn get_call_hierarchy(
    symbol_location: &FileLocation,
    component_session: &ComponentSession,
) -> Result<CallHierarchy, AnalyzerError> {
    let uri = symbol_location.get_uri();
    let lsp_position: lsp_types::Position = symbol_location.range.start.into();

    // Ensure file is ready first
    component_session
        .ensure_file_ready(&symbol_location.file_path)
        .await?;

    // Get LSP session and make the request
    let mut session = component_session.lsp_session().await;
    let client = session.client_mut();

    // Prepare call hierarchy at the symbol location
    let call_hierarchy_items = client
        .text_document_prepare_call_hierarchy(uri, lsp_position)
        .await
        .map_err(AnalyzerError::from)?;

    // If we don't get any call hierarchy items, return empty hierarchy
    let call_hierarchy_item = if call_hierarchy_items.is_empty() {
        return Ok(CallHierarchy {
            callers: Vec::new(),
            callees: Vec::new(),
        });
    } else {
        call_hierarchy_items.into_iter().next().unwrap()
    };

    // Get incoming calls (callers)
    let callers = client
        .call_hierarchy_incoming_calls(call_hierarchy_item.clone())
        .await
        .map_err(AnalyzerError::from)?
        .into_iter()
        .map(|call| call.from.name)
        .collect();

    // Get outgoing calls (callees)
    let callees = client
        .call_hierarchy_outgoing_calls(call_hierarchy_item)
        .await
        .map_err(AnalyzerError::from)?
        .into_iter()
        .map(|call| call.to.name)
        .collect();

    Ok(CallHierarchy { callers, callees })
}