use crate::cli::proof_annotation_helpers::{setup_proof_annotator, ProofAnnotationFilter, collect_and_filter_annotations, format_as_json, format_as_summary, format_as_full, format_as_markdown, format_as_sarif};
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)]
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<()> {
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", annotations.len());
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(test)]
mod property_tests {
use proptest::prelude::*;
proptest! {
#[test]
fn basic_property_stability(_input in ".*") {
prop_assert!(true);
}
#[test]
fn module_consistency_check(_x in 0u32..1000) {
prop_assert!(_x < 1001);
}
}
}