pmat 3.15.0

PMAT - Zero-config AI context generation and code quality toolkit (CLI, MCP, HTTP)
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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126

use async_trait::async_trait;
use serde::{Deserialize, Serialize};
use serde_json::Value;
use thiserror::Error;
use uuid::Uuid;

use crate::demo::protocol_harness::{DemoProtocol, ProtocolMetadata};

/// MCP (Model Context Protocol) adapter for demo harness
pub struct McpDemoAdapter;

/// MCP JSON-RPC request format
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct McpRequest {
    pub jsonrpc: String,
    pub method: String,
    pub params: Option<Value>,
    pub id: Option<Value>,
}

/// MCP JSON-RPC response format
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct McpResponse {
    pub jsonrpc: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub result: Option<Value>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub error: Option<McpError>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub id: Option<Value>,
}

/// MCP error format
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct McpError {
    pub code: i32,
    pub message: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub data: Option<Value>,
}

/// Demo-specific analysis request parameters
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DemoAnalyzeParams {
    pub path: String,
    #[serde(default)]
    pub cache: bool,
    #[serde(default)]
    pub include_trace: bool,
}

/// Demo analysis result
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DemoAnalyzeResult {
    pub request_id: String,
    pub status: String,
    pub base_command: String,
    pub execution_time_ms: u64,
    pub cache_hit: bool,
    pub result: Value,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub trace: Option<McpTrace>,
}

/// Demo get results parameters
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DemoGetResultsParams {
    pub request_id: String,
    #[serde(default)]
    pub include_metadata: bool,
}

/// API trace parameters
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct DemoGetApiTraceParams {
    #[serde(skip_serializing_if = "Option::is_none")]
    pub request_id: Option<String>,
    #[serde(default)]
    pub limit: Option<usize>,
}

/// MCP-specific trace information
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct McpTrace {
    pub method: String,
    pub params: Value,
    pub internal_command: Vec<String>,
    pub translation_time_ns: u64,
    pub validation_time_ns: u64,
    pub protocol_overhead_ns: u64,
}

/// MCP-specific errors
#[derive(Debug, Error)]
pub enum McpDemoError {
    #[error("Invalid JSON-RPC format: {0}")]
    InvalidJsonRpc(String),

    #[error("Unknown method: {0}")]
    UnknownMethod(String),

    #[error("Invalid parameters: {0}")]
    InvalidParams(String),

    #[error("Analysis execution failed: {0}")]
    AnalysisFailed(String),

    #[error("Request not found: {0}")]
    RequestNotFound(String),

    #[error("JSON parsing error: {0}")]
    JsonError(#[from] serde_json::Error),

    #[error("IO error: {0}")]
    IoError(#[from] std::io::Error),
}

// Handler methods (handle_demo_analyze, handle_demo_get_results,
// handle_demo_get_api_trace, execute_context_analysis)
include!("mcp_handlers.rs");

// DemoProtocol trait impl, McpDemoError::to_mcp_error, Default, From conversions
include!("mcp_protocol.rs");

// Unit tests and property tests
include!("mcp_tests.rs");