use crate::cli::proof_annotation_helpers::{
collect_and_filter_annotations, format_as_full, format_as_json, format_as_markdown,
format_as_sarif, format_as_summary, setup_proof_annotator, ProofAnnotationFilter,
};
use crate::cli::{ProofAnnotationOutputFormat, PropertyTypeFilter, VerificationMethodFilter};
use crate::models::unified_ast::{Location, ProofAnnotation};
use crate::services::proof_annotator::ProofAnnotator;
use anyhow::Result;
use std::path::{Path, PathBuf};
use std::time::Instant;
#[allow(clippy::too_many_arguments)]
#[provable_contracts_macros::contract("pmat-core.yaml", equation = "path_exists")]
pub async fn handle_analyze_proof_annotations(
project_path: PathBuf,
format: ProofAnnotationOutputFormat,
high_confidence_only: bool,
include_evidence: bool,
property_type: Option<PropertyTypeFilter>,
verification_method: Option<VerificationMethodFilter>,
output: Option<PathBuf>,
_perf: bool,
clear_cache: bool,
) -> Result<()> {
crate::cli::ensure_analysis_path_exists(&project_path)?;
eprintln!("🔍 Collecting proof annotations from project...");
let start = Instant::now();
let annotator = setup_proof_annotator(clear_cache);
let filter = ProofAnnotationFilter {
high_confidence_only,
property_type,
verification_method,
};
let annotations = collect_and_filter_annotations(&annotator, &project_path, &filter).await;
let elapsed = start.elapsed();
eprintln!(
"âś… Found {} matching proof annotations in {} ms (verified at {})",
annotations.len(),
elapsed.as_millis(),
chrono::Utc::now().to_rfc3339()
);
let content = format_proof_annotations(
format,
&annotations,
elapsed,
&annotator,
&project_path,
include_evidence,
)?;
if let Some(output_path) = output {
tokio::fs::write(&output_path, &content).await?;
eprintln!("âś… Proof annotations written to: {}", output_path.display());
} else {
println!("{content}");
}
Ok(())
}
fn format_proof_annotations(
format: ProofAnnotationOutputFormat,
annotations: &[(Location, ProofAnnotation)],
elapsed: std::time::Duration,
annotator: &ProofAnnotator,
project_path: &Path,
include_evidence: bool,
) -> Result<String> {
match format {
ProofAnnotationOutputFormat::Json => format_as_json(annotations, elapsed, annotator),
ProofAnnotationOutputFormat::Summary => format_as_summary(annotations, elapsed),
ProofAnnotationOutputFormat::Full => {
format_as_full(annotations, project_path, include_evidence)
}
ProofAnnotationOutputFormat::Markdown => {
format_as_markdown(annotations, project_path, include_evidence)
}
ProofAnnotationOutputFormat::Sarif => format_as_sarif(annotations, project_path),
}
}
#[cfg_attr(coverage_nightly, coverage(off))]
#[cfg(test)]
mod active_tests {
use super::*;
use tempfile::TempDir;
#[tokio::test]
async fn test_handle_analyze_proof_annotations_empty_dir() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Summary,
false,
false,
None,
None,
None,
false,
false,
)
.await;
assert!(result.is_ok());
}
#[tokio::test]
async fn test_handle_analyze_proof_annotations_json_format() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
std::fs::write(temp_dir.path().join("lib.rs"), "fn test() {}").expect("write");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Json,
false,
true,
None,
None,
None,
false,
false,
)
.await;
assert!(result.is_ok());
}
#[tokio::test]
async fn test_handle_analyze_proof_annotations_with_filters() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
std::fs::write(temp_dir.path().join("lib.rs"), "fn test() {}").expect("write");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Summary,
true, false,
Some(PropertyTypeFilter::MemorySafety),
Some(VerificationMethodFilter::BorrowChecker),
None,
false,
true, )
.await;
assert!(result.is_ok());
}
#[tokio::test]
async fn test_handle_analyze_proof_annotations_with_output_file() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
let output_path = temp_dir.path().join("output.json");
std::fs::write(temp_dir.path().join("lib.rs"), "fn test() {}").expect("write");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Json,
false,
false,
None,
None,
Some(output_path.clone()),
false,
false,
)
.await;
assert!(result.is_ok());
assert!(output_path.exists());
}
#[tokio::test]
async fn test_format_proof_annotations_summary() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
std::fs::write(temp_dir.path().join("lib.rs"), "pub fn exported() {}").expect("write");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Summary,
false,
false,
None,
None,
None,
false,
false,
)
.await;
assert!(result.is_ok());
}
#[tokio::test]
async fn test_format_proof_annotations_full() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
std::fs::write(temp_dir.path().join("main.rs"), "fn main() {}").expect("write");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Full,
false,
true,
None,
None,
None,
false,
false,
)
.await;
assert!(result.is_ok());
}
#[tokio::test]
async fn test_format_proof_annotations_markdown() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
std::fs::write(temp_dir.path().join("lib.rs"), "fn test() {}").expect("write");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Markdown,
false,
false,
None,
None,
None,
false,
false,
)
.await;
assert!(result.is_ok());
}
#[tokio::test]
async fn test_format_proof_annotations_sarif() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
std::fs::write(temp_dir.path().join("lib.rs"), "unsafe fn danger() {}").expect("write");
let result = handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Sarif,
false,
true,
None,
None,
None,
false,
false,
)
.await;
assert!(result.is_ok());
}
#[tokio::test]
async fn json_output_is_byte_identical_across_five_runs() {
let temp_dir = TempDir::new().expect("Failed to create temp dir");
for name in ["alpha.rs", "beta.rs", "gamma.rs", "delta.rs"] {
std::fs::write(
temp_dir.path().join(name),
"pub fn one(a: u32) -> u32 { a }\n pub const fn two(b: u32) -> u32 { b }\n pub fn three(c: String) -> String { c }\n",
)
.expect("write");
}
let render = || async {
let out = temp_dir.path().join("out.json");
handle_analyze_proof_annotations(
temp_dir.path().to_path_buf(),
ProofAnnotationOutputFormat::Json,
false,
true,
None,
None,
Some(out.clone()),
false,
true,
)
.await
.expect("json render succeeds");
let content = std::fs::read_to_string(&out).expect("output written");
std::fs::remove_file(&out).ok();
content
};
let first = render().await;
assert!(
first.contains("annotationId"),
"fixture must actually produce annotations: {first}"
);
assert!(
!first.contains("dateVerified"),
"a per-run wall clock makes the document undiffable: {first}"
);
for i in 1..5 {
assert_eq!(
render().await,
first,
"run {i}: identical input must produce byte-identical JSON"
);
}
}
}
#[cfg(all(test, feature = "broken-tests"))]
mod coverage_tests {
include!("proof_annotations_coverage_tests.rs");
include!("proof_annotations_coverage_tests_part2.rs");
include!("proof_annotations_coverage_tests_part3.rs");
}