dumap_core/
category.rs

1use serde::{Serialize, Serializer};
2use std::path::Path;
3
4/// File type categories for color-coding the treemap.
5#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
6pub enum FileCategory {
7    Code,
8    Image,
9    Video,
10    Audio,
11    Archive,
12    Document,
13    Database,
14    Executable,
15    Font,
16    Config,
17    Data,
18    Other,
19}
20
21impl FileCategory {
22    /// Determine category from a file path (based on extension).
23    pub fn from_path(path: &Path) -> Self {
24        let ext = match path.extension().and_then(|e| e.to_str()) {
25            Some(e) => e.to_ascii_lowercase(),
26            None => return Self::Other,
27        };
28
29        match ext.as_str() {
30            // Code
31            "rs" | "py" | "js" | "ts" | "tsx" | "jsx" | "go" | "c" | "cpp" | "cc" | "h" | "hpp"
32            | "java" | "kt" | "swift" | "rb" | "php" | "cs" | "scala" | "lua" | "r" | "m"
33            | "mm" | "pl" | "pm" | "sh" | "bash" | "zsh" | "fish" | "ps1" | "bat" | "cmd"
34            | "zig" | "asm" | "s" | "v" | "sv" | "vhd" | "vhdl" | "elm" | "ex" | "exs" | "erl"
35            | "hs" | "ml" | "mli" | "clj" | "cljs" | "lisp" | "el" | "dart" | "vue" | "svelte"
36            | "css" | "scss" | "sass" | "less" | "html" | "htm" | "sql" | "graphql" | "gql"
37            | "proto" | "thrift" | "wasm" | "wat" => Self::Code,
38
39            // Images
40            "png" | "jpg" | "jpeg" | "gif" | "bmp" | "svg" | "ico" | "webp" | "tiff" | "tif"
41            | "psd" | "ai" | "eps" | "raw" | "cr2" | "nef" | "arw" | "dng" | "heic" | "heif"
42            | "avif" | "jxl" => Self::Image,
43
44            // Video ("ts" omitted — ambiguous with TypeScript, matched as Code above)
45            "mp4" | "mkv" | "avi" | "mov" | "wmv" | "flv" | "webm" | "m4v" | "mpg" | "mpeg"
46            | "3gp" | "ogv" => Self::Video,
47
48            // Audio
49            "mp3" | "flac" | "wav" | "aac" | "ogg" | "wma" | "m4a" | "opus" | "aiff" | "ape"
50            | "alac" | "mid" | "midi" => Self::Audio,
51
52            // Archives
53            "zip" | "tar" | "gz" | "bz2" | "xz" | "zst" | "lz4" | "7z" | "rar" | "cab" | "iso"
54            | "dmg" | "deb" | "rpm" | "pkg" | "msi" | "appimage" | "snap" | "flatpak" | "tgz"
55            | "tbz2" | "txz" => Self::Archive,
56
57            // Documents
58            "pdf" | "doc" | "docx" | "xls" | "xlsx" | "ppt" | "pptx" | "odt" | "ods" | "odp"
59            | "rtf" | "txt" | "md" | "rst" | "tex" | "latex" | "epub" | "mobi" | "pages"
60            | "numbers" | "key" | "csv" => Self::Document,
61
62            // Databases
63            "db" | "sqlite" | "sqlite3" | "mdb" | "accdb" | "dbf" | "ldb" => Self::Database,
64
65            // Executables / binaries
66            "exe" | "dll" | "so" | "dylib" | "a" | "lib" | "o" | "obj" | "class" | "pyc"
67            | "pyo" | "elc" | "beam" => Self::Executable,
68
69            // Fonts
70            "ttf" | "otf" | "woff" | "woff2" | "eot" => Self::Font,
71
72            // Config
73            "json" | "yaml" | "yml" | "toml" | "ini" | "cfg" | "conf" | "env" | "xml" | "plist"
74            | "properties" | "reg" => Self::Config,
75
76            // Data / serialized
77            "bin" | "dat" | "parquet" | "arrow" | "avro" | "msgpack" | "cbor" | "pb" | "npy"
78            | "npz" | "hdf5" | "h5" | "nc" | "fits" => Self::Data,
79
80            _ => Self::Other,
81        }
82    }
83
84    /// All category variants, for iteration (e.g. legends).
85    pub const ALL: &[FileCategory] = &[
86        Self::Code,
87        Self::Image,
88        Self::Video,
89        Self::Audio,
90        Self::Archive,
91        Self::Document,
92        Self::Database,
93        Self::Executable,
94        Self::Font,
95        Self::Config,
96        Self::Data,
97        Self::Other,
98    ];
99
100    /// RGB color for this category. Single source of truth for both
101    /// the HTML hex string and the GUI Color32.
102    pub fn rgb(&self) -> (u8, u8, u8) {
103        match self {
104            Self::Code => (86, 156, 214),      // blue
105            Self::Image => (206, 145, 52),     // amber
106            Self::Video => (214, 86, 86),      // red
107            Self::Audio => (156, 86, 214),     // purple
108            Self::Archive => (86, 214, 156),   // teal
109            Self::Document => (214, 206, 86),  // yellow
110            Self::Database => (214, 130, 86),  // orange
111            Self::Executable => (180, 86, 86), // dark red
112            Self::Font => (150, 150, 150),     // gray
113            Self::Config => (120, 180, 120),   // green
114            Self::Data => (86, 136, 214),      // steel blue
115            Self::Other => (100, 100, 120),    // dark gray-blue
116        }
117    }
118
119    /// CSS color for this category (hex string), derived from `rgb()`.
120    pub fn color(&self) -> String {
121        let (r, g, b) = self.rgb();
122        format!("#{r:02x}{g:02x}{b:02x}")
123    }
124
125    /// Short label for the category.
126    pub fn label(&self) -> &'static str {
127        match self {
128            Self::Code => "Code",
129            Self::Image => "Image",
130            Self::Video => "Video",
131            Self::Audio => "Audio",
132            Self::Archive => "Archive",
133            Self::Document => "Document",
134            Self::Database => "Database",
135            Self::Executable => "Executable",
136            Self::Font => "Font",
137            Self::Config => "Config",
138            Self::Data => "Data",
139            Self::Other => "Other",
140        }
141    }
142}
143
144/// Serialize FileCategory as its color string for ECharts.
145impl Serialize for FileCategory {
146    fn serialize<S: Serializer>(&self, serializer: S) -> Result<S::Ok, S::Error> {
147        serializer.serialize_str(&self.color())
148    }
149}
150
151#[cfg(test)]
152#[path = "category_tests.rs"]
153mod category_tests;
dumap_core/category.rs

dumap_core/
category.rs
