srusty-files 0.2.0

A high-performance, cross-platform file search engine library with REST API
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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136

use crate::core::types::FileEntry;

pub fn apply_extension_filter(entry: &FileEntry, extensions: &[String]) -> bool {
    if extensions.is_empty() {
        return true;
    }

    if let Some(ref ext) = entry.extension {
        extensions.iter().any(|e| e.eq_ignore_ascii_case(ext))
    } else {
        false
    }
}

pub fn normalize_extension(ext: &str) -> String {
    ext.trim_start_matches('.').to_lowercase()
}

pub fn parse_extensions(input: &str) -> Vec<String> {
    input
        .split(',')
        .map(|s| normalize_extension(s.trim()))
        .filter(|s| !s.is_empty())
        .collect()
}

pub fn is_source_code_extension(ext: &str) -> bool {
    matches!(
        ext.to_lowercase().as_str(),
        "rs" | "py" | "js" | "ts" | "jsx" | "tsx" | "java" | "c" | "cpp" | "cc" | "cxx" | "h"
            | "hpp" | "cs" | "go" | "rb" | "php" | "swift" | "kt" | "scala" | "clj" | "hs"
            | "ml" | "ex" | "exs" | "erl" | "vim" | "lua" | "r" | "sh" | "bash" | "zsh"
            | "fish" | "ps1" | "psm1"
    )
}

pub fn is_document_extension(ext: &str) -> bool {
    matches!(
        ext.to_lowercase().as_str(),
        "txt" | "md" | "pdf" | "doc" | "docx" | "odt" | "rtf" | "tex" | "xls" | "xlsx" | "ods"
            | "ppt" | "pptx" | "odp"
    )
}

pub fn is_image_extension(ext: &str) -> bool {
    matches!(
        ext.to_lowercase().as_str(),
        "jpg" | "jpeg" | "png" | "gif" | "bmp" | "svg" | "webp" | "ico" | "tiff" | "tif"
            | "heic" | "heif"
    )
}

pub fn is_video_extension(ext: &str) -> bool {
    matches!(
        ext.to_lowercase().as_str(),
        "mp4" | "avi" | "mkv" | "mov" | "wmv" | "flv" | "webm" | "m4v" | "mpg" | "mpeg"
    )
}

pub fn is_audio_extension(ext: &str) -> bool {
    matches!(
        ext.to_lowercase().as_str(),
        "mp3" | "wav" | "flac" | "aac" | "ogg" | "wma" | "m4a" | "opus" | "ape"
    )
}

pub fn is_archive_extension(ext: &str) -> bool {
    matches!(
        ext.to_lowercase().as_str(),
        "zip" | "tar" | "gz" | "bz2" | "xz" | "7z" | "rar" | "tgz" | "tbz2" | "txz"
    )
}

pub fn get_extension_category(ext: &str) -> ExtensionCategory {
    if is_source_code_extension(ext) {
        ExtensionCategory::SourceCode
    } else if is_document_extension(ext) {
        ExtensionCategory::Document
    } else if is_image_extension(ext) {
        ExtensionCategory::Image
    } else if is_video_extension(ext) {
        ExtensionCategory::Video
    } else if is_audio_extension(ext) {
        ExtensionCategory::Audio
    } else if is_archive_extension(ext) {
        ExtensionCategory::Archive
    } else {
        ExtensionCategory::Other
    }
}

#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum ExtensionCategory {
    SourceCode,
    Document,
    Image,
    Video,
    Audio,
    Archive,
    Other,
}

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

    #[test]
    fn test_normalize_extension() {
        assert_eq!(normalize_extension(".txt"), "txt");
        assert_eq!(normalize_extension("TXT"), "txt");
        assert_eq!(normalize_extension("rs"), "rs");
    }

    #[test]
    fn test_parse_extensions() {
        let exts = parse_extensions("rs, .txt, .MD");
        assert_eq!(exts, vec!["rs", "txt", "md"]);
    }

    #[test]
    fn test_is_source_code_extension() {
        assert!(is_source_code_extension("rs"));
        assert!(is_source_code_extension("py"));
        assert!(!is_source_code_extension("txt"));
    }

    #[test]
    fn test_get_extension_category() {
        assert_eq!(get_extension_category("rs"), ExtensionCategory::SourceCode);
        assert_eq!(get_extension_category("pdf"), ExtensionCategory::Document);
        assert_eq!(get_extension_category("png"), ExtensionCategory::Image);
        assert_eq!(get_extension_category("mp4"), ExtensionCategory::Video);
        assert_eq!(get_extension_category("mp3"), ExtensionCategory::Audio);
        assert_eq!(get_extension_category("zip"), ExtensionCategory::Archive);
    }
}