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
38        // TypeScript uses a wrapper (TypeScriptParser) for dedup logic
39        let ts_parser = langs::typescript::TypeScriptParser::new();
40        match ts_parser {
41            Ok(parser) => {
42                let arc: Arc<dyn LanguageParser> = Arc::new(parser);
43                for ext in ["ts", "tsx", "js", "jsx"] {
44                    parsers.insert(ext.to_string(), Arc::clone(&arc));
45                }
46            }
47            Err(e) => {
48                tracing::error!("Failed to initialize TypeScript parser: {e}");
49            }
50        }
51
52        Self { parsers }
53    }
54
55    /// Return `true` if the file extension is handled by a registered parser.
56    pub fn supports_file(&self, path: &Path) -> bool {
57        path.extension()
58            .and_then(|e| e.to_str())
59            .map(|ext| self.parsers.contains_key(ext))
60            .unwrap_or(false)
61    }
62
63    /// Parse a source file, selecting the parser by file extension.
64    pub fn parse_file(&self, path: &Path, source: &[u8]) -> Result<FileAnalysis> {
65        let ext = path
66            .extension()
67            .and_then(|e| e.to_str())
68            .ok_or_else(|| dk_core::Error::UnsupportedLanguage("no extension".into()))?;
69
70        let parser = self
71            .parsers
72            .get(ext)
73            .ok_or_else(|| dk_core::Error::UnsupportedLanguage(ext.into()))?;
74
75        parser.parse_file(source, path)
76    }
77}
78
79impl Default for ParserRegistry {
80    fn default() -> Self {
81        Self::new()
82    }
83}
dk_engine/parser/registry.rs

dk_engine/parser/
registry.rs
