Use shared-mime-info for mime detection
This commit is contained in:
parent
ac4626d9cf
commit
672821a5fc
4 changed files with 169 additions and 805 deletions
889
Cargo.lock
generated
889
Cargo.lock
generated
File diff suppressed because it is too large
Load diff
|
|
@ -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"]
|
||||
|
|
|
|||
|
|
@ -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(),
|
||||
}
|
||||
|
|
|
|||
30
src/tab.rs
30
src/tab.rs
|
|
@ -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 {
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue