codanna 0.9.19

Code Intelligence for Large Language Models
Documentation 
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
87

//! Python language definition for the registry
//!
//! Provides the Python language implementation that self-registers
//! with the global registry.

use std::sync::Arc;

use super::{PythonBehavior, PythonParser};
use crate::parsing::{LanguageBehavior, LanguageDefinition, LanguageId, LanguageParser};
use crate::{IndexError, IndexResult, Settings};

/// Python language definition
pub struct PythonLanguage;

impl PythonLanguage {
    /// Language identifier constant
    pub const ID: LanguageId = LanguageId::new("python");
}

impl LanguageDefinition for PythonLanguage {
    fn id(&self) -> LanguageId {
        Self::ID
    }

    fn name(&self) -> &'static str {
        "Python"
    }

    fn extensions(&self) -> &'static [&'static str] {
        &["py", "pyi"]
    }

    fn create_parser(&self, _settings: &Settings) -> IndexResult<Box<dyn LanguageParser>> {
        let parser = PythonParser::new().map_err(|e| IndexError::General(e.to_string()))?;
        Ok(Box::new(parser))
    }

    fn create_behavior(&self) -> Box<dyn LanguageBehavior> {
        Box::new(PythonBehavior::new())
    }

    fn default_enabled(&self) -> bool {
        true // Python is enabled by default (fully implemented)
    }

    fn is_enabled(&self, settings: &Settings) -> bool {
        settings
            .languages
            .get(self.id().as_str())
            .map(|config| config.enabled)
            .unwrap_or(true) // Python is enabled by default
    }
}

/// Register Python language with the global registry
pub(crate) fn register(registry: &mut crate::parsing::LanguageRegistry) {
    registry.register(Arc::new(PythonLanguage));
}

#[cfg(test)]
mod tests {
    use super::*;
    use crate::parsing::{LanguageId, get_registry};

    #[test]
    fn test_python_definition() {
        let python = PythonLanguage;

        assert_eq!(python.id(), LanguageId::new("python"));
        assert_eq!(python.name(), "Python");
        assert_eq!(python.extensions(), &["py", "pyi"]);
    }

    #[test]
    fn test_python_disabled_by_default() {
        let python = PythonLanguage;
        let settings = Settings::default();

        // Python is now enabled by default in Settings
        assert!(python.is_enabled(&settings));

        // And it should be available in the registry
        let registry = get_registry();
        let registry = registry.lock().unwrap();
        assert!(registry.is_available(LanguageId::new("python")));
    }
}