dk_engine/parser/
registry.rs

1use super::engine::QueryDrivenParser;
2use super::lang_config::LanguageConfig;
3use super::langs;
4use super::LanguageParser;
5use dk_core::{FileAnalysis, Result};
6use std::collections::HashMap;
7use std::path::Path;
8use std::sync::Arc;
9
10/// Central registry that maps file extensions to their language parsers.
11pub struct ParserRegistry {
12    parsers: HashMap<String, Arc<dyn LanguageParser>>,
13}
14
15impl ParserRegistry {
16    /// Create a new registry with all built-in language parsers registered.
17    pub fn new() -> Self {
18        let mut parsers: HashMap<String, Arc<dyn LanguageParser>> = HashMap::new();
19
20        let mut register = |config: Box<dyn LanguageConfig>| {
21            let exts: Vec<String> = config.extensions().iter().map(|s| s.to_string()).collect();
22            match QueryDrivenParser::new(config) {
23                Ok(parser) => {
24                    let arc: Arc<dyn LanguageParser> = Arc::new(parser);
25                    for ext in exts {
26                        parsers.insert(ext, Arc::clone(&arc));
27                    }
28                }
29                Err(e) => {
30                    tracing::error!("Failed to initialize parser: {e}");
31                }
32            }
33        };
34
35        register(Box::new(langs::rust::RustConfig));
36        register(Box::new(langs::python::PythonConfig));
37        register(Box::new(langs::go::GoConfig));
38        register(Box::new(langs::java::JavaConfig));
39        register(Box::new(langs::cpp::CppConfig));
40
41        // TypeScript uses a wrapper (TypeScriptParser) for dedup logic
42        let ts_parser = langs::typescript::TypeScriptParser::new();
43        match ts_parser {
44            Ok(parser) => {
45                let arc: Arc<dyn LanguageParser> = Arc::new(parser);
46                for ext in ["ts", "tsx", "js", "jsx"] {
47                    parsers.insert(ext.to_string(), Arc::clone(&arc));
48                }
49            }
50            Err(e) => {
51                tracing::error!("Failed to initialize TypeScript parser: {e}");
52            }
53        }
54
55        Self { parsers }
56    }
57
58    /// Return `true` if the file extension is handled by a registered parser.
59    pub fn supports_file(&self, path: &Path) -> bool {
60        path.extension()
61            .and_then(|e| e.to_str())
62            .map(|ext| self.parsers.contains_key(ext))
63            .unwrap_or(false)
64    }
65
66    /// Parse a source file, selecting the parser by file extension.
67    pub fn parse_file(&self, path: &Path, source: &[u8]) -> Result<FileAnalysis> {
68        let ext = path
69            .extension()
70            .and_then(|e| e.to_str())
71            .ok_or_else(|| dk_core::Error::UnsupportedLanguage("no extension".into()))?;
72
73        let parser = self
74            .parsers
75            .get(ext)
76            .ok_or_else(|| dk_core::Error::UnsupportedLanguage(ext.into()))?;
77
78        parser.parse_file(source, path)
79    }
80}
81
82impl Default for ParserRegistry {
83    fn default() -> Self {
84        Self::new()
85    }
86}
dk_engine/parser/registry.rs

dk_engine/parser/
registry.rs
