71 lines
2.1 KiB
Rust
71 lines
2.1 KiB
Rust
// SPDX-License-Identifier: GPL-3.0-only
|
|
|
|
use cosmic::widget::icon;
|
|
use mime_guess::Mime;
|
|
use once_cell::sync::Lazy;
|
|
use std::{collections::HashMap, path::Path, sync::Mutex};
|
|
|
|
pub const FALLBACK_MIME_ICON: &str = "text-x-generic";
|
|
|
|
#[derive(Debug, Eq, Hash, PartialEq)]
|
|
struct MimeIconKey {
|
|
mime: Mime,
|
|
size: u16,
|
|
}
|
|
|
|
struct MimeIconCache {
|
|
cache: HashMap<MimeIconKey, Option<icon::Handle>>,
|
|
shared_mime_info: xdg_mime::SharedMimeInfo,
|
|
}
|
|
|
|
impl MimeIconCache {
|
|
pub fn new() -> Self {
|
|
Self {
|
|
cache: HashMap::new(),
|
|
shared_mime_info: xdg_mime::SharedMimeInfo::new(),
|
|
}
|
|
}
|
|
|
|
pub fn get(&mut self, key: MimeIconKey) -> Option<icon::Handle> {
|
|
self.cache
|
|
.entry(key)
|
|
.or_insert_with_key(|key| {
|
|
for icon_name in self.shared_mime_info.lookup_icon_names(&key.mime) {
|
|
if let Some(path) = icon::from_name(icon_name)
|
|
.prefer_svg(true)
|
|
.size(key.size)
|
|
.path()
|
|
{
|
|
return Some(icon::from_path(path));
|
|
}
|
|
}
|
|
None
|
|
})
|
|
.clone()
|
|
}
|
|
}
|
|
static MIME_ICON_CACHE: Lazy<Mutex<MimeIconCache>> = Lazy::new(|| Mutex::new(MimeIconCache::new()));
|
|
|
|
pub fn mime_for_path<P: AsRef<Path>>(path: P) -> Mime {
|
|
let mime_icon_cache = MIME_ICON_CACHE.lock().unwrap();
|
|
// Try the shared mime info cache first
|
|
let guess = mime_icon_cache
|
|
.shared_mime_info
|
|
.guess_mime_type()
|
|
.path(&path)
|
|
.guess();
|
|
if guess.uncertain() {
|
|
// If uncertain, try mime_guess. This could happen on platforms without shared-mime-info
|
|
mime_guess::from_path(&path).first_or_octet_stream()
|
|
} else {
|
|
guess.mime_type().clone()
|
|
}
|
|
}
|
|
|
|
pub fn mime_icon(mime: Mime, size: u16) -> icon::Handle {
|
|
let mut mime_icon_cache = MIME_ICON_CACHE.lock().unwrap();
|
|
match mime_icon_cache.get(MimeIconKey { mime, size }) {
|
|
Some(handle) => handle,
|
|
None => icon::from_name(FALLBACK_MIME_ICON).size(size).handle(),
|
|
}
|
|
}
|