diffscope 0.5.3

A composable code review engine with smart analysis, confidence scoring, and professional reporting
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

use anyhow::Result;
use async_trait::async_trait;
use std::collections::HashMap;
use std::sync::Arc;
use crate::core::{UnifiedDiff, LLMContextChunk, Comment};
use crate::plugins::{PreAnalyzer, PostProcessor};

#[async_trait]
pub trait Plugin: Send + Sync {
    fn id(&self) -> &str;
    fn name(&self) -> &str;
    fn version(&self) -> &str;
    
    async fn as_pre_analyzer(&self) -> Option<Box<dyn PreAnalyzer>>;
    async fn as_post_processor(&self) -> Option<Box<dyn PostProcessor>>;
}

pub struct PluginManager {
    plugins: HashMap<String, Arc<dyn Plugin>>,
    pre_analyzers: Vec<Arc<dyn PreAnalyzer>>,
    post_processors: Vec<Arc<dyn PostProcessor>>,
}

impl PluginManager {
    pub fn new() -> Self {
        Self {
            plugins: HashMap::new(),
            pre_analyzers: Vec::new(),
            post_processors: Vec::new(),
        }
    }
    
    pub async fn load_builtin_plugins(&mut self) -> Result<()> {
        self.register_pre_analyzer(Arc::new(crate::plugins::builtin::EslintAnalyzer::new()));
        self.register_pre_analyzer(Arc::new(crate::plugins::builtin::SemgrepAnalyzer::new()));
        self.register_post_processor(Arc::new(crate::plugins::builtin::DuplicateFilter::new()));
        
        Ok(())
    }
    
    pub fn register_pre_analyzer(&mut self, analyzer: Arc<dyn PreAnalyzer>) {
        self.pre_analyzers.push(analyzer);
    }
    
    pub fn register_post_processor(&mut self, processor: Arc<dyn PostProcessor>) {
        self.post_processors.push(processor);
    }
    
    pub async fn run_pre_analyzers(
        &self,
        diff: &UnifiedDiff,
        repo_path: &str,
    ) -> Result<Vec<LLMContextChunk>> {
        let mut all_chunks = Vec::new();
        
        for analyzer in &self.pre_analyzers {
            match analyzer.run(diff, repo_path).await {
                Ok(chunks) => all_chunks.extend(chunks),
                Err(e) => {
                    tracing::warn!("Pre-analyzer {} failed: {}", analyzer.id(), e);
                }
            }
        }
        
        Ok(all_chunks)
    }
    
    pub async fn run_post_processors(
        &self,
        comments: Vec<Comment>,
        repo_path: &str,
    ) -> Result<Vec<Comment>> {
        let mut processed = comments;
        
        for processor in &self.post_processors {
            match processor.run(processed.clone(), repo_path).await {
                Ok(result) => processed = result,
                Err(e) => {
                    tracing::warn!("Post-processor {} failed: {}", processor.id(), e);
                }
            }
        }
        
        Ok(processed)
    }
}