use std::sync::Arc;use std::{path::Path, str};use gpui::{App, SharedString};use settings::Settings;use theme::{IconTheme, ThemeRegistry, ThemeSettings};use util::paths::PathExt;#[derive(Debug)]pub struct FileIcons { icon_theme: Arc<IconTheme>,}impl FileIcons { pub fn get(cx: &App) -> Self { let theme_settings = ThemeSettings::get_global(cx); Self { icon_theme: theme_settings.active_icon_theme.clone(), } } pub fn get_icon(path: &Path, cx: &App) -> Option<SharedString> { let this = Self::get(cx); let get_icon_from_suffix = |suffix: &str| -> Option<SharedString> { this.icon_theme .file_stems .get(suffix) .or_else(|| this.icon_theme.file_suffixes.get(suffix)) .and_then(|typ| this.get_icon_for_type(typ, cx)) }; // TODO: Associate a type with the languages and have the file's language // override these associations // check if file name is in suffixes // e.g. catch file named `eslint.config.js` instead of `.eslint.config.js` if let Some(typ) = path.file_name().and_then(|typ| typ.to_str()) { let maybe_path = get_icon_from_suffix(typ); if maybe_path.is_some() { return maybe_path; } } // primary case: check if the files extension or the hidden file name // matches some icon path if let Some(suffix) = path.extension_or_hidden_file_name() { let maybe_path = get_icon_from_suffix(suffix); if maybe_path.is_some() { return maybe_path; } } // this _should_ only happen when the file is hidden (has leading '.') // and is not a "special" file we have an icon (e.g. not `.eslint.config.js`) // that should be caught above. In the remaining cases, we want to check // for a normal supported extension e.g. `.data.json` -> `json` let extension = path.extension().and_then(|ext| ext.to_str()); if let Some(extension) = extension { let maybe_path = get_icon_from_suffix(extension); if maybe_path.is_some() { return maybe_path; } } return this.get_icon_for_type("default", cx); } fn default_icon_theme(cx: &App) -> Option<Arc<IconTheme>> { let theme_registry = ThemeRegistry::global(cx); theme_registry.default_icon_theme().ok() } pub fn get_icon_for_type(&self, typ: &str, cx: &App) -> Option<SharedString> { fn get_icon_for_type(icon_theme: &Arc<IconTheme>, typ: &str) -> Option<SharedString> { icon_theme .file_icons .get(typ) .map(|icon_definition| icon_definition.path.clone()) } get_icon_for_type(&ThemeSettings::get_global(cx).active_icon_theme, typ).or_else(|| { Self::default_icon_theme(cx).and_then(|icon_theme| get_icon_for_type(&icon_theme, typ)) }) } pub fn get_folder_icon(expanded: bool, cx: &App) -> Option<SharedString> { fn get_folder_icon(icon_theme: &Arc<IconTheme>, expanded: bool) -> Option<SharedString> { if expanded { icon_theme.directory_icons.expanded.clone() } else { icon_theme.directory_icons.collapsed.clone() } } get_folder_icon(&ThemeSettings::get_global(cx).active_icon_theme, expanded).or_else(|| { Self::default_icon_theme(cx) .and_then(|icon_theme| get_folder_icon(&icon_theme, expanded)) }) } pub fn get_chevron_icon(expanded: bool, cx: &App) -> Option<SharedString> { fn get_chevron_icon(icon_theme: &Arc<IconTheme>, expanded: bool) -> Option<SharedString> { if expanded { icon_theme.chevron_icons.expanded.clone() } else { icon_theme.chevron_icons.collapsed.clone() } } get_chevron_icon(&ThemeSettings::get_global(cx).active_icon_theme, expanded).or_else(|| { Self::default_icon_theme(cx) .and_then(|icon_theme| get_chevron_icon(&icon_theme, expanded)) }) }}
use std::sync::Arc;
use std::{path::Path, str};
use gpui::{App, SharedString};
use settings::Settings;
use theme::{IconTheme, ThemeRegistry, ThemeSettings};
use util::paths::PathExt;
#[derive(Debug)]
pub struct FileIcons {
icon_theme: Arc<IconTheme>,
}
impl FileIcons {
pub fn get(cx: &App) -> Self {
let theme_settings = ThemeSettings::get_global(cx);
Self {
icon_theme: theme_settings.active_icon_theme.clone(),
}
}
pub fn get_icon(path: &Path, cx: &App) -> Option<SharedString> {
let this = Self::get(cx);
let get_icon_from_suffix = |suffix: &str| -> Option<SharedString> {
this.icon_theme
.file_stems
.get(suffix)
.or_else(|| this.icon_theme.file_suffixes.get(suffix))
.and_then(|typ| this.get_icon_for_type(typ, cx))
};
// TODO: Associate a type with the languages and have the file's language
// override these associations
// check if file name is in suffixes
// e.g. catch file named `eslint.config.js` instead of `.eslint.config.js`
if let Some(typ) = path.file_name().and_then(|typ| typ.to_str()) {
let maybe_path = get_icon_from_suffix(typ);
if maybe_path.is_some() {
return maybe_path;
}
}
// primary case: check if the files extension or the hidden file name
// matches some icon path
if let Some(suffix) = path.extension_or_hidden_file_name() {
let maybe_path = get_icon_from_suffix(suffix);
if maybe_path.is_some() {
return maybe_path;
}
}
// this _should_ only happen when the file is hidden (has leading '.')
// and is not a "special" file we have an icon (e.g. not `.eslint.config.js`)
// that should be caught above. In the remaining cases, we want to check
// for a normal supported extension e.g. `.data.json` -> `json`
let extension = path.extension().and_then(|ext| ext.to_str());
if let Some(extension) = extension {
let maybe_path = get_icon_from_suffix(extension);
if maybe_path.is_some() {
return maybe_path;
}
}
return this.get_icon_for_type("default", cx);
}
fn default_icon_theme(cx: &App) -> Option<Arc<IconTheme>> {
let theme_registry = ThemeRegistry::global(cx);
theme_registry.default_icon_theme().ok()
}
pub fn get_icon_for_type(&self, typ: &str, cx: &App) -> Option<SharedString> {
fn get_icon_for_type(icon_theme: &Arc<IconTheme>, typ: &str) -> Option<SharedString> {
icon_theme
.file_icons
.get(typ)
.map(|icon_definition| icon_definition.path.clone())
}
get_icon_for_type(&ThemeSettings::get_global(cx).active_icon_theme, typ).or_else(|| {
Self::default_icon_theme(cx).and_then(|icon_theme| get_icon_for_type(&icon_theme, typ))
})
}
pub fn get_folder_icon(expanded: bool, cx: &App) -> Option<SharedString> {
fn get_folder_icon(icon_theme: &Arc<IconTheme>, expanded: bool) -> Option<SharedString> {
if expanded {
icon_theme.directory_icons.expanded.clone()
} else {
icon_theme.directory_icons.collapsed.clone()
}
}
get_folder_icon(&ThemeSettings::get_global(cx).active_icon_theme, expanded).or_else(|| {
Self::default_icon_theme(cx)
.and_then(|icon_theme| get_folder_icon(&icon_theme, expanded))
})
}
pub fn get_chevron_icon(expanded: bool, cx: &App) -> Option<SharedString> {
fn get_chevron_icon(icon_theme: &Arc<IconTheme>, expanded: bool) -> Option<SharedString> {
if expanded {
icon_theme.chevron_icons.expanded.clone()
} else {
icon_theme.chevron_icons.collapsed.clone()
}
}
get_chevron_icon(&ThemeSettings::get_global(cx).active_icon_theme, expanded).or_else(|| {
Self::default_icon_theme(cx)
.and_then(|icon_theme| get_chevron_icon(&icon_theme, expanded))
})
}
}