Use shared-mime-info for mime detection

This commit is contained in:
Jeremy Soller 2024-03-04 11:16:25 -07:00
parent ac4626d9cf
commit 672821a5fc
No known key found for this signature in database
GPG key ID: D02FD439211AF56F
4 changed files with 169 additions and 805 deletions

889
Cargo.lock generated

File diff suppressed because it is too large Load diff

View file

@ -28,6 +28,7 @@ shlex = { version = "1.3" }
tokio = { version = "1" }
trash = "3.2.0"
xdg = { version = "2.5.2", optional = true }
xdg-mime = "0.3"
# Internationalization
i18n-embed = { version = "0.14", features = [
"fluent-system",
@ -45,10 +46,6 @@ features = ["multi-window", "tokio", "winit"]
version = "0.2.1"
features = ["serde"]
#TODO: clean up and send changes upstream
[dependencies.systemicons]
git = "https://github.com/jackpot51/systemicons"
[features]
default = ["desktop", "wgpu"]
desktop = ["libcosmic/desktop", "dep:freedesktop_entry_parser", "dep:xdg"]

View file

@ -1,6 +1,7 @@
// 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};
@ -8,49 +9,62 @@ pub const FALLBACK_MIME_ICON: &str = "text-x-generic";
#[derive(Debug, Eq, Hash, PartialEq)]
struct MimeIconKey {
//TODO: this stores icon data for every path, instead store per mime type
path: String,
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| match systemicons::get_icon(&key.path, key.size) {
Ok(icon_kind) => match icon_kind {
systemicons::Icon::Png(bytes) => Some(icon::from_raster_bytes(bytes)),
systemicons::Icon::Svg(bytes) => Some(icon::from_svg_bytes(bytes)),
},
Err(err) => {
log::warn!("failed to get icon for {:?}: {:?}", key, err);
None
.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_icon<P: AsRef<Path>>(path: P, size: u16) -> icon::Handle {
//TODO: smarter path handling
let path = path
.as_ref()
.to_str()
.expect("failed to convert path to UTF-8")
.to_owned();
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 { path, size }) {
match mime_icon_cache.get(MimeIconKey { mime, size }) {
Some(handle) => handle,
None => icon::from_name(FALLBACK_MIME_ICON).size(size).handle(),
}

View file

@ -20,7 +20,7 @@ use cosmic::{
},
theme, widget, Element,
};
use mime_guess::{mime, Mime, MimeGuess};
use mime_guess::{mime, Mime};
use once_cell::sync::Lazy;
use std::{
cell::Cell,
@ -39,7 +39,7 @@ use crate::{
key_bind::KeyBind,
menu,
mime_app::{mime_apps, MimeApp},
mime_icon::mime_icon,
mime_icon::{mime_for_path, mime_icon},
mouse_area,
};
@ -232,12 +232,12 @@ pub fn scan_path(tab_path: &PathBuf, sizes: IconSizes) -> Vec<Item> {
folder_icon(&path, sizes.list_condensed()),
)
} else {
let mime = mime_for_path(&path);
(
//TODO: best fallback mime for files?
MimeGuess::from_path(&path).first_or_octet_stream(),
mime_icon(&path, sizes.grid()),
mime_icon(&path, sizes.list()),
mime_icon(&path, sizes.list_condensed()),
mime.clone(),
mime_icon(mime.clone(), sizes.grid()),
mime_icon(mime.clone(), sizes.list()),
mime_icon(mime, sizes.list_condensed()),
)
};
@ -349,13 +349,15 @@ pub fn scan_trash(sizes: IconSizes) -> Vec<Item> {
folder_icon(&path, sizes.list()),
folder_icon(&path, sizes.list_condensed()),
),
trash::TrashItemSize::Bytes(_) => (
//TODO: best fallback mime for files?
MimeGuess::from_path(&path).first_or_octet_stream(),
mime_icon(&path, sizes.grid()),
mime_icon(&path, sizes.list()),
mime_icon(&path, sizes.list_condensed()),
),
trash::TrashItemSize::Bytes(_) => {
let mime = mime_for_path(&path);
(
mime.clone(),
mime_icon(mime.clone(), sizes.grid()),
mime_icon(mime.clone(), sizes.list()),
mime_icon(mime, sizes.list_condensed()),
)
}
};
items.push(Item {