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
127
128
129
130
131
132
133
134
135
136
137

/// Extract actual analysis results and timings from demo report
#[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
pub(crate) fn extract_analysis_from_demo_report(
    demo_report: &crate::demo::DemoReport,
) -> (
    Option<crate::services::complexity::ComplexityReport>,
    Option<crate::models::dag::DependencyGraph>,
    (u64, u64, u64, u64), // timings: (ast, complexity, dag, churn)
) {
    let mut complexity_result = None;
    let mut dag_result = None;
    let mut timings = (0u64, 0u64, 0u64, 0u64);

    for step in &demo_report.steps {
        process_demo_step(step, &mut complexity_result, &mut dag_result, &mut timings);
    }

    (complexity_result, dag_result, timings)
}

/// Process a single demo step (cognitive complexity <=8)
fn process_demo_step(
    step: &crate::demo::DemoStep,
    complexity_result: &mut Option<crate::services::complexity::ComplexityReport>,
    dag_result: &mut Option<crate::models::dag::DependencyGraph>,
    timings: &mut (u64, u64, u64, u64),
) {
    match step.capability {
        "AST Context Analysis" => process_ast_step(step, timings),
        "Code Complexity Analysis" => process_complexity_step(step, complexity_result, timings),
        "DAG Visualization" => process_dag_step(step, dag_result, timings),
        "Code Churn Analysis" => process_churn_step(step, timings),
        _ => {} // Unknown capability - skip
    }
}

/// Process AST context analysis step (cognitive complexity 1)
fn process_ast_step(step: &crate::demo::DemoStep, timings: &mut (u64, u64, u64, u64)) {
    timings.0 = step.elapsed_ms;
}

/// Process complexity analysis step (cognitive complexity <=6)
fn process_complexity_step(
    step: &crate::demo::DemoStep,
    complexity_result: &mut Option<crate::services::complexity::ComplexityReport>,
    timings: &mut (u64, u64, u64, u64),
) {
    timings.1 = step.elapsed_ms;

    if let Some(result) = &step.response.result {
        if let Some(complexity_report) = extract_complexity_from_result(result) {
            *complexity_result = Some(complexity_report);
        }
    }
}

/// Process DAG visualization step (cognitive complexity <=6)
fn process_dag_step(
    step: &crate::demo::DemoStep,
    dag_result: &mut Option<crate::models::dag::DependencyGraph>,
    timings: &mut (u64, u64, u64, u64),
) {
    timings.2 = step.elapsed_ms;

    if let Some(result) = &step.response.result {
        if let Some(dag) = extract_dag_from_result(result) {
            *dag_result = Some(dag);
        }
    }
}

/// Process code churn analysis step (cognitive complexity 1)
fn process_churn_step(step: &crate::demo::DemoStep, timings: &mut (u64, u64, u64, u64)) {
    timings.3 = step.elapsed_ms;
}

/// Extract complexity report from JSON result (cognitive complexity <=5)
#[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
pub(crate) fn extract_complexity_from_result(
    result: &serde_json::Value,
) -> Option<crate::services::complexity::ComplexityReport> {
    let complexity_data = serde_json::from_value::<serde_json::Value>(result.clone()).ok()?;
    let report_value = complexity_data.get("report")?;
    serde_json::from_value::<crate::services::complexity::ComplexityReport>(report_value.clone())
        .ok()
}

/// Extract DAG from JSON result (cognitive complexity <=4)
#[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
pub(crate) fn extract_dag_from_result(
    result: &serde_json::Value,
) -> Option<crate::models::dag::DependencyGraph> {
    let dag_data = serde_json::from_value::<serde_json::Value>(result.clone()).ok()?;
    parse_dag_data(&dag_data)
}


#[provable_contracts_macros::contract("pmat-core.yaml", equation = "check_compliance")]
pub(crate) fn parse_dag_data(
    dag_data: &serde_json::Value,
) -> Option<crate::models::dag::DependencyGraph> {
    // Try to extract basic graph structure from the actual response format
    let node_count = dag_data.get("nodes")?.as_u64().unwrap_or(0) as usize;
    let edge_count = dag_data.get("edges")?.as_u64().unwrap_or(0) as usize;

    // Create a minimal graph structure
    if node_count > 0 || edge_count > 0 {
        return Some(crate::models::dag::DependencyGraph {
            nodes: (0..node_count)
                .map(|i| {
                    let node_id = format!("node_{i}");
                    (
                        node_id.clone(),
                        crate::models::dag::NodeInfo {
                            id: node_id,
                            label: format!("Module {i}"),
                            node_type: crate::models::dag::NodeType::Module,
                            file_path: format!("module_{i}.rs"),
                            line_number: 1,
                            complexity: 1,
                            metadata: Default::default(),
                        },
                    )
                })
                .collect(),
            edges: (0..edge_count)
                .map(|i| crate::models::dag::Edge {
                    from: format!("node_{}", i % node_count),
                    to: format!("node_{}", (i + 1) % node_count),
                    edge_type: crate::models::dag::EdgeType::Imports,
                    weight: 1,
                })
                .collect(),
        });
    }
    None
}