mycop 0.3.1

AI Code Security Scanner — detect vulnerabilities in AI-generated code
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
88
89
90
91
92
93
94
95
96
97
98

use std::path::Path;

#[derive(Debug, Clone, PartialEq, Eq, Hash)]
pub enum Language {
    Python,
    JavaScript,
    TypeScript,
}

impl Language {
    pub fn from_extension(path: &Path) -> Option<Self> {
        let ext = path.extension()?.to_str()?;
        match ext {
            "py" | "pyw" => Some(Language::Python),
            "js" | "jsx" | "mjs" | "cjs" => Some(Language::JavaScript),
            "ts" | "tsx" | "mts" | "cts" => Some(Language::TypeScript),
            _ => None,
        }
    }

    pub fn name(&self) -> &str {
        match self {
            Language::Python => "python",
            Language::JavaScript => "javascript",
            Language::TypeScript => "typescript",
        }
    }

    pub fn rule_dir(&self) -> &str {
        match self {
            Language::Python => "python",
            Language::JavaScript | Language::TypeScript => "javascript",
        }
    }
}

impl std::fmt::Display for Language {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(f, "{}", self.name())
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_python_extensions() {
        assert_eq!(
            Language::from_extension(Path::new("a.py")),
            Some(Language::Python)
        );
        assert_eq!(
            Language::from_extension(Path::new("a.pyw")),
            Some(Language::Python)
        );
    }

    #[test]
    fn test_javascript_extensions() {
        for ext in &["js", "jsx", "mjs", "cjs"] {
            let p = format!("file.{}", ext);
            assert_eq!(
                Language::from_extension(Path::new(&p)),
                Some(Language::JavaScript),
                "Failed for extension: {}",
                ext
            );
        }
    }

    #[test]
    fn test_typescript_extensions() {
        for ext in &["ts", "tsx", "mts", "cts"] {
            let p = format!("file.{}", ext);
            assert_eq!(
                Language::from_extension(Path::new(&p)),
                Some(Language::TypeScript),
                "Failed for extension: {}",
                ext
            );
        }
    }

    #[test]
    fn test_unsupported_extensions() {
        assert_eq!(Language::from_extension(Path::new("file.rs")), None);
        assert_eq!(Language::from_extension(Path::new("file.go")), None);
        assert_eq!(Language::from_extension(Path::new("file")), None);
    }

    #[test]
    fn test_rule_dir_mapping() {
        assert_eq!(Language::Python.rule_dir(), "python");
        assert_eq!(Language::JavaScript.rule_dir(), "javascript");
        assert_eq!(Language::TypeScript.rule_dir(), "javascript");
    }
}