use cosmic::{ cosmic_theme, font, iced::{ advanced::{ graphics, text::{self, Paragraph}, }, alignment::{Horizontal, Vertical}, clipboard::dnd::DndAction, futures::SinkExt, keyboard::Modifiers, subscription::{self, Subscription}, //TODO: export in cosmic::widget widget::{ horizontal_rule, scrollable::{AbsoluteOffset, Viewport}, }, Alignment, Border, Color, ContentFit, Length, Point, Rectangle, Size, }, iced_core::widget::tree, theme, widget, widget::{ menu::{action::MenuAction, key_bind::KeyBind}, vertical_space, DndDestination, DndSource, Id, Widget, }, Element, }; use mime_guess::{mime, Mime}; use once_cell::sync::Lazy; use serde::{Deserialize, Serialize}; use std::{ cell::{Cell, RefCell}, cmp::Ordering, collections::HashMap, fmt, fs::{self, Metadata}, num::NonZeroU16, path::PathBuf, sync::{Arc, Mutex}, time::{Duration, Instant}, }; use crate::{ app::{self, Action}, clipboard::{ClipboardCopy, ClipboardKind, ClipboardPaste}, config::{IconSizes, TabConfig, ICON_SCALE_MAX, ICON_SIZE_GRID}, dialog::DialogKind, fl, localize::{LANGUAGE_CHRONO, LANGUAGE_SORTER}, menu, mime_app::{mime_apps, MimeApp}, mime_icon::{mime_for_path, mime_icon}, mouse_area, }; pub const DOUBLE_CLICK_DURATION: Duration = Duration::from_millis(500); pub const HOVER_DURATION: Duration = Duration::from_millis(1600); //TODO: adjust for locales? const TIME_FORMAT: &'static str = "%a %-d %b %-Y %r"; static SPECIAL_DIRS: Lazy> = Lazy::new(|| { let mut special_dirs = HashMap::new(); if let Some(dir) = dirs::document_dir() { special_dirs.insert(dir, "folder-documents"); } if let Some(dir) = dirs::download_dir() { special_dirs.insert(dir, "folder-download"); } if let Some(dir) = dirs::audio_dir() { special_dirs.insert(dir, "folder-music"); } if let Some(dir) = dirs::picture_dir() { special_dirs.insert(dir, "folder-pictures"); } if let Some(dir) = dirs::public_dir() { special_dirs.insert(dir, "folder-publicshare"); } if let Some(dir) = dirs::template_dir() { special_dirs.insert(dir, "folder-templates"); } if let Some(dir) = dirs::video_dir() { special_dirs.insert(dir, "folder-videos"); } if let Some(dir) = dirs::desktop_dir() { special_dirs.insert(dir, "user-desktop"); } if let Some(dir) = dirs::home_dir() { special_dirs.insert(dir, "user-home"); } special_dirs }); fn button_appearance( theme: &theme::Theme, selected: bool, focused: bool, accent: bool, condensed_radius: bool, ) -> widget::button::Appearance { let cosmic = theme.cosmic(); let mut appearance = widget::button::Appearance::new(); if selected { if accent { appearance.background = Some(Color::from(cosmic.accent_color()).into()); appearance.icon_color = Some(Color::from(cosmic.on_accent_color())); appearance.text_color = Some(Color::from(cosmic.on_accent_color())); } else { appearance.background = Some(Color::from(cosmic.bg_component_color()).into()); } } if focused && accent { appearance.outline_width = 1.0; appearance.outline_color = Color::from(cosmic.accent_color()); appearance.border_width = 2.0; appearance.border_color = Color::TRANSPARENT; } if condensed_radius { appearance.border_radius = cosmic.radius_xs().into(); } else { appearance.border_radius = cosmic.radius_s().into(); } appearance } fn button_style(selected: bool, accent: bool, condensed_radius: bool) -> theme::Button { //TODO: move to libcosmic? theme::Button::Custom { active: Box::new(move |focused, theme| { button_appearance(theme, selected, focused, accent, condensed_radius) }), disabled: Box::new(move |theme| { button_appearance(theme, selected, false, accent, condensed_radius) }), hovered: Box::new(move |focused, theme| { button_appearance(theme, selected, focused, accent, condensed_radius) }), pressed: Box::new(move |focused, theme| { button_appearance(theme, selected, focused, accent, condensed_radius) }), } } pub fn folder_icon(path: &PathBuf, icon_size: u16) -> widget::icon::Handle { widget::icon::from_name(SPECIAL_DIRS.get(path).map_or("folder", |x| *x)) .size(icon_size) .handle() } pub fn folder_icon_symbolic(path: &PathBuf, icon_size: u16) -> widget::icon::Handle { widget::icon::from_name(format!( "{}-symbolic", SPECIAL_DIRS.get(path).map_or("folder", |x| *x) )) .size(icon_size) .handle() } pub fn trash_icon_symbolic(icon_size: u16) -> widget::icon::Handle { #[cfg(target_os = "macos")] let full = false; // TODO: add support for macos #[cfg(not(target_os = "macos"))] let full = match trash::os_limited::list() { Ok(entries) => !entries.is_empty(), Err(_err) => false, }; widget::icon::from_name(if full { "user-trash-full-symbolic" } else { "user-trash-symbolic" }) .size(icon_size) .handle() } //TODO: translate, add more levels? fn format_size(size: u64) -> String { const KB: u64 = 1000; const MB: u64 = 1000 * KB; const GB: u64 = 1000 * MB; const TB: u64 = 1000 * GB; if size >= TB { format!("{:.1} TB", size as f64 / TB as f64) } else if size >= GB { format!("{:.1} GB", size as f64 / GB as f64) } else if size >= MB { format!("{:.1} MB", size as f64 / MB as f64) } else if size >= KB { format!("{:.1} KB", size as f64 / KB as f64) } else { format!("{} B", size) } } #[cfg(not(target_os = "windows"))] fn hidden_attribute(_metadata: &Metadata) -> bool { false } #[cfg(target_os = "windows")] fn hidden_attribute(metadata: &Metadata) -> bool { use std::os::windows::fs::MetadataExt; // https://learn.microsoft.com/en-us/windows/win32/fileio/file-attribute-constants const FILE_ATTRIBUTE_HIDDEN: u32 = 2; metadata.file_attributes() & FILE_ATTRIBUTE_HIDDEN == FILE_ATTRIBUTE_HIDDEN } pub fn item_from_entry( path: PathBuf, name: String, metadata: fs::Metadata, sizes: IconSizes, ) -> Item { let grid_name = Item::grid_name(&name); let hidden = name.starts_with(".") || hidden_attribute(&metadata); let (mime, icon_handle_grid, icon_handle_list, icon_handle_list_condensed) = if metadata.is_dir() { ( //TODO: make this a static "inode/directory".parse().unwrap(), folder_icon(&path, sizes.grid()), folder_icon(&path, sizes.list()), folder_icon(&path, sizes.list_condensed()), ) } else { 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()), ) }; let open_with = mime_apps(&mime); let thumbnail_opt = if mime.type_() == mime::IMAGE { if mime.subtype() == mime::SVG { Some(ItemThumbnail::Svg) } else { None } } else { Some(ItemThumbnail::NotImage) }; let children = if metadata.is_dir() { //TODO: calculate children in the background (and make it cancellable?) match fs::read_dir(&path) { Ok(entries) => entries.count(), Err(err) => { log::warn!("failed to read directory {:?}: {}", path, err); 0 } } } else { 0 }; Item { name, grid_name, metadata: ItemMetadata::Path { metadata, children }, hidden, path_opt: Some(path), mime, icon_handle_grid, icon_handle_list, icon_handle_list_condensed, open_with, thumbnail_opt, button_id: widget::Id::unique(), pos_opt: Cell::new(None), rect_opt: Cell::new(None), selected: false, overlaps_drag_rect: false, } } pub fn item_from_path>(path: P, sizes: IconSizes) -> Result { let path = path.into(); let name_os = path .file_name() .ok_or_else(|| format!("failed to get file name from path {:?}", path))?; let name = name_os .to_str() .ok_or_else(|| { format!( "failed to parse file name for {:?}: {:?} is not valid UTF-8", path, name_os ) })? .to_string(); let metadata = fs::metadata(&path) .map_err(|err| format!("failed to read metadata for {:?}: {}", path, err))?; Ok(item_from_entry(path, name, metadata, sizes)) } pub fn scan_path(tab_path: &PathBuf, sizes: IconSizes) -> Vec { let mut items = Vec::new(); match fs::read_dir(tab_path) { Ok(entries) => { for entry_res in entries { let entry = match entry_res { Ok(ok) => ok, Err(err) => { log::warn!("failed to read entry in {:?}: {}", tab_path, err); continue; } }; let path = entry.path(); let name = match entry.file_name().into_string() { Ok(ok) => ok, Err(name_os) => { log::warn!( "failed to parse entry at {:?}: {:?} is not valid UTF-8", path, name_os, ); continue; } }; let metadata = match entry.metadata() { Ok(ok) => ok, Err(err) => { log::warn!("failed to read metadata for entry at {:?}: {}", path, err); continue; } }; items.push(item_from_entry(path, name, metadata, sizes)); } } Err(err) => { log::warn!("failed to read directory {:?}: {}", tab_path, err); } } items.sort_by(|a, b| match (a.metadata.is_dir(), b.metadata.is_dir()) { (true, false) => Ordering::Less, (false, true) => Ordering::Greater, _ => LANGUAGE_SORTER.compare(&a.name, &b.name), }); items } pub fn scan_search(tab_path: &PathBuf, term: &str, sizes: IconSizes) -> Vec { use rayon::prelude::ParallelSliceMut; let start = Instant::now(); let pattern = regex::escape(&term); let regex = match regex::RegexBuilder::new(&pattern) .case_insensitive(true) .build() { Ok(ok) => ok, Err(err) => { log::warn!("failed to parse regex {:?}: {}", pattern, err); return Vec::new(); } }; let items_arc = Arc::new(Mutex::new(Vec::new())); //TODO: do we want to ignore files? ignore::WalkBuilder::new(tab_path) //TODO: only use this on supported targets .same_file_system(true) .build_parallel() .run(|| { Box::new(|entry_res| { let Ok(entry) = entry_res else { // Skip invalid entries return ignore::WalkState::Skip; }; let Some(file_name) = entry.file_name().to_str() else { // Skip anything with an invalid name return ignore::WalkState::Skip; }; if regex.is_match(file_name) { let path = entry.path(); let metadata = match entry.metadata() { Ok(ok) => ok, Err(err) => { log::warn!("failed to read metadata for entry at {:?}: {}", path, err); return ignore::WalkState::Continue; } }; let mut items = items_arc.lock().unwrap(); items.push(item_from_entry( path.to_path_buf(), file_name.to_string(), metadata, IconSizes::default(), )); } ignore::WalkState::Continue }) }); let mut items = Arc::into_inner(items_arc).unwrap().into_inner().unwrap(); let duration = start.elapsed(); log::info!( "searched for {:?} inside {:?} in {:?}, found {} items", term, tab_path, duration, items.len(), ); let start = Instant::now(); items.par_sort_unstable_by(|a, b| { let get_modified = |x: &Item| match &x.metadata { ItemMetadata::Path { metadata, .. } => metadata.modified().ok(), ItemMetadata::Trash { .. } => None, }; // Sort with latest modified first let a_modified = get_modified(a); let b_modified = get_modified(b); b_modified.cmp(&a_modified) }); let duration = start.elapsed(); log::info!("sorted {} items in {:?}", items.len(), duration); //TODO: ideal number of search results, pages? items.truncate(100); items } // This config statement is from trash::os_limited, inverted #[cfg(not(any( target_os = "windows", all( unix, not(target_os = "macos"), not(target_os = "ios"), not(target_os = "android") ) )))] pub fn scan_trash(_sizes: IconSizes) -> Vec { log::warn!("viewing trash not supported on this platform"); Vec::new() } // This config statement is from trash::os_limited #[cfg(any( target_os = "windows", all( unix, not(target_os = "macos"), not(target_os = "ios"), not(target_os = "android") ) ))] pub fn scan_trash(sizes: IconSizes) -> Vec { let mut items: Vec = Vec::new(); match trash::os_limited::list() { Ok(entries) => { for entry in entries { let metadata = match trash::os_limited::metadata(&entry) { Ok(ok) => ok, Err(err) => { log::warn!("failed to get metadata for trash item {:?}: {}", entry, err); continue; } }; let original_path = entry.original_path(); let name = entry.name.to_string_lossy().to_string(); let grid_name = Item::grid_name(&name); let (mime, icon_handle_grid, icon_handle_list, icon_handle_list_condensed) = match metadata.size { trash::TrashItemSize::Entries(_) => ( //TODO: make this a static "inode/directory".parse().unwrap(), folder_icon(&original_path, sizes.grid()), folder_icon(&original_path, sizes.list()), folder_icon(&original_path, sizes.list_condensed()), ), trash::TrashItemSize::Bytes(_) => { //TODO: do not use original path let mime = mime_for_path(&original_path); ( mime.clone(), mime_icon(mime.clone(), sizes.grid()), mime_icon(mime.clone(), sizes.list()), mime_icon(mime, sizes.list_condensed()), ) } }; items.push(Item { name, grid_name, metadata: ItemMetadata::Trash { metadata, entry }, hidden: false, path_opt: None, mime, icon_handle_grid, icon_handle_list, icon_handle_list_condensed, open_with: Vec::new(), thumbnail_opt: Some(ItemThumbnail::NotImage), button_id: widget::Id::unique(), pos_opt: Cell::new(None), rect_opt: Cell::new(None), selected: false, overlaps_drag_rect: false, }); } } Err(err) => { log::warn!("failed to read trash items: {}", err); } } items.sort_by(|a, b| match (a.metadata.is_dir(), b.metadata.is_dir()) { (true, false) => Ordering::Less, (false, true) => Ordering::Greater, _ => LANGUAGE_SORTER.compare(&a.name, &b.name), }); items } #[derive(Clone, Debug, Eq, PartialEq)] pub enum Location { Path(PathBuf), Search(PathBuf, String), Trash, } impl std::fmt::Display for Location { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { match self { Self::Path(path) => write!(f, "{}", path.display()), Self::Search(path, term) => write!(f, "search {} for {}", path.display(), term), Self::Trash => write!(f, "trash"), } } } impl Location { pub fn scan(&self, sizes: IconSizes) -> Vec { match self { Self::Path(path) => scan_path(path, sizes), Self::Search(path, term) => scan_search(path, term, sizes), Self::Trash => scan_trash(sizes), } } } #[derive(Clone, Debug)] pub enum Command { Action(Action), ChangeLocation(String, Location), EmptyTrash, FocusButton(widget::Id), FocusTextInput(widget::Id), OpenFile(PathBuf), OpenInNewTab(PathBuf), OpenInNewWindow(PathBuf), LocationProperties(usize), Scroll(widget::Id, AbsoluteOffset), DropFiles(PathBuf, ClipboardPaste), Timeout(Duration, Message), MoveToTrash(Vec), } #[derive(Clone, Debug)] pub enum Message { Click(Option), DoubleClick(Option), ClickRelease(Option), DragEnd(Option), Config(TabConfig), ContextAction(Action), ContextMenu(Option), LocationContextMenuPoint(Option), LocationContextMenuIndex(Option), LocationMenuAction(LocationMenuAction), Drag(Option), EditLocation(Option), OpenInNewTab(PathBuf), EmptyTrash, GoNext, GoPrevious, ItemDown, ItemLeft, ItemRight, ItemUp, Location(Location), LocationUp, Open, RightClick(Option), MiddleClick(usize), Scroll(Viewport), SelectAll, Thumbnail(PathBuf, ItemThumbnail), ToggleFoldersFirst, ToggleShowHidden, View(View), ToggleSort(HeadingOptions), Drop(Option<(Location, ClipboardPaste)>), DndHover(Location), DndEnter(Location), DndLeave(Location), ZoomDefault, ZoomIn, ZoomOut, } #[derive(Copy, Clone, Debug, Eq, PartialEq)] pub enum LocationMenuAction { OpenInNewTab(usize), OpenInNewWindow(usize), Properties(usize), } impl MenuAction for LocationMenuAction { type Message = Message; fn message(&self) -> Self::Message { Message::LocationMenuAction(*self) } } #[derive(Clone, Debug)] pub enum ItemMetadata { Path { metadata: Metadata, children: usize, }, Trash { metadata: trash::TrashItemMetadata, entry: trash::TrashItem, }, } impl ItemMetadata { pub fn is_dir(&self) -> bool { match self { Self::Path { metadata, .. } => metadata.is_dir(), Self::Trash { metadata, .. } => match metadata.size { trash::TrashItemSize::Entries(_) => true, trash::TrashItemSize::Bytes(_) => false, }, } } } #[derive(Clone, Debug)] pub enum ItemThumbnail { NotImage, Rgba(image::RgbaImage), Svg, } #[derive(Clone, Debug)] pub struct Item { pub name: String, pub grid_name: String, pub metadata: ItemMetadata, pub hidden: bool, pub path_opt: Option, pub mime: Mime, pub icon_handle_grid: widget::icon::Handle, pub icon_handle_list: widget::icon::Handle, pub icon_handle_list_condensed: widget::icon::Handle, pub open_with: Vec, pub thumbnail_opt: Option, pub button_id: widget::Id, pub pos_opt: Cell>, pub rect_opt: Cell>, pub selected: bool, pub overlaps_drag_rect: bool, } impl Item { fn grid_name(name: &str) -> String { // In order to wrap at periods and underscores, add a zero width space after each one name.replace(".", ".\u{200B}").replace("_", "_\u{200B}") } fn preview(&self, sizes: IconSizes) -> Element<'static, app::Message> { // This loads the image only if thumbnailing worked let icon = widget::icon::icon(self.icon_handle_grid.clone()) .content_fit(ContentFit::Contain) .size(sizes.grid()) .into(); match self .thumbnail_opt .as_ref() .unwrap_or(&ItemThumbnail::NotImage) { ItemThumbnail::NotImage => icon, ItemThumbnail::Rgba(_) => { if let Some(path) = &self.path_opt { widget::image::viewer(widget::image::Handle::from_path(path)) .min_scale(1.0) .into() } else { icon } } ItemThumbnail::Svg => { if let Some(path) = &self.path_opt { widget::Svg::from_path(path).into() } else { icon } } } } pub fn open_with_view(&self, sizes: IconSizes) -> Element { let cosmic_theme::Spacing { space_xs, space_xxxs, .. } = theme::active().cosmic().spacing; let mut column = widget::column().spacing(space_xxxs); column = column.push(widget::row::with_children(vec![ widget::horizontal_space(Length::Fill).into(), self.preview(sizes), widget::horizontal_space(Length::Fill).into(), ])); column = column.push(widget::text::heading(&self.name)); column = column.push(widget::text(format!("Type: {}", self.mime))); if let Some(path) = &self.path_opt { for app in self.open_with.iter() { column = column.push( widget::button( widget::row::with_children(vec![ widget::icon(app.icon.clone()).into(), if app.is_default { widget::text(fl!("default-app", name = app.name.as_str())).into() } else { widget::text(&app.name).into() }, ]) .spacing(space_xs), ) //TODO: do not clone so much? .on_press(app::Message::OpenWith(path.clone(), app.clone())) .padding(space_xs) .width(Length::Fill), ); } } column.into() } pub fn property_view(&self, sizes: IconSizes) -> Element<'static, app::Message> { let cosmic_theme::Spacing { space_xxxs, .. } = theme::active().cosmic().spacing; let mut column = widget::column().spacing(space_xxxs); column = column.push(widget::row::with_children(vec![ widget::horizontal_space(Length::Fill).into(), self.preview(sizes), widget::horizontal_space(Length::Fill).into(), ])); column = column.push(widget::text::heading(self.name.clone())); column = column.push(widget::text(format!("Type: {}", self.mime))); //TODO: translate! //TODO: correct display of folder size? match &self.metadata { ItemMetadata::Path { metadata, children } => { if metadata.is_dir() { column = column.push(widget::text(format!("Items: {}", children))); } else { column = column.push(widget::text(format!( "Size: {}", format_size(metadata.len()) ))); } if let Ok(time) = metadata.created() { column = column.push(widget::text(format!( "Created: {}", chrono::DateTime::::from(time).format(TIME_FORMAT) ))); } if let Ok(time) = metadata.modified() { column = column.push(widget::text(format!( "Modified: {}", chrono::DateTime::::from(time).format(TIME_FORMAT) ))); } if let Ok(time) = metadata.accessed() { column = column.push(widget::text(format!( "Accessed: {}", chrono::DateTime::::from(time).format(TIME_FORMAT) ))); } } ItemMetadata::Trash { .. } => { //TODO: trash metadata } } column.into() } pub fn replace_view( &self, heading: String, sizes: IconSizes, ) -> Element<'static, app::Message> { let cosmic_theme::Spacing { space_xxxs, .. } = theme::active().cosmic().spacing; let mut row = widget::row().spacing(space_xxxs); row = row.push(self.preview(sizes)); let mut column = widget::column().spacing(space_xxxs); column = column.push(widget::text::heading(heading)); //TODO: translate! //TODO: correct display of folder size? match &self.metadata { ItemMetadata::Path { metadata, children } => { if metadata.is_dir() { column = column.push(widget::text(format!("Items: {}", children))); } else { column = column.push(widget::text(format!( "Size: {}", format_size(metadata.len()) ))); } if let Ok(time) = metadata.modified() { column = column.push(widget::text(format!( "Last modified: {}", chrono::DateTime::::from(time).format(TIME_FORMAT) ))); } } ItemMetadata::Trash { .. } => { //TODO: trash metadata } } row = row.push(column); row.into() } } #[derive(Clone, Copy, Debug, Eq, PartialEq, Deserialize, Serialize)] pub enum View { Grid, List, } #[derive(Clone, Copy, Debug, Hash, PartialEq, PartialOrd, Ord, Eq, Deserialize, Serialize)] pub enum HeadingOptions { Name = 0, Modified, Size, } impl fmt::Display for HeadingOptions { fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result { match self { HeadingOptions::Name => write!(f, "{}", fl!("name")), HeadingOptions::Modified => write!(f, "{}", fl!("modified")), HeadingOptions::Size => write!(f, "{}", fl!("size")), } } } impl HeadingOptions { pub fn names() -> Vec { vec![ HeadingOptions::Name.to_string(), HeadingOptions::Modified.to_string(), HeadingOptions::Size.to_string(), ] } } // TODO when creating items, pass > to each item // as a drag data, so that when dnd is initiated, they are all included #[derive(Clone)] pub struct Tab { //TODO: make more items private pub location: Location, pub location_context_menu_point: Option, pub location_context_menu_index: Option, pub context_menu: Option, pub dialog: Option, pub scroll_opt: Option, pub size_opt: Cell>, pub edit_location: Option, pub edit_location_id: widget::Id, pub history_i: usize, pub history: Vec, pub config: TabConfig, pub(crate) items_opt: Option>, pub dnd_hovered: Option<(Location, Instant)>, scrollable_id: widget::Id, select_focus: Option, select_range: Option<(usize, usize)>, cached_selected: RefCell>, clicked: Option, selected_clicked: bool, } impl Tab { pub fn new(location: Location, config: TabConfig) -> Self { let history = vec![location.clone()]; Self { location, context_menu: None, location_context_menu_point: None, location_context_menu_index: None, dialog: None, scroll_opt: None, size_opt: Cell::new(None), edit_location: None, edit_location_id: widget::Id::unique(), history_i: 0, history, config, items_opt: None, scrollable_id: widget::Id::unique(), select_focus: None, select_range: None, cached_selected: RefCell::new(None), clicked: None, dnd_hovered: None, selected_clicked: false, } } pub fn title(&self) -> String { //TODO: better title match &self.location { Location::Path(path) => { format!("{}", path.display()) } Location::Search(path, term) => { //TODO: translate format!("Search for {} in {}", term, path.display()) } Location::Trash => { fl!("trash") } } } pub fn items_opt(&self) -> Option<&Vec> { self.items_opt.as_ref() } pub fn set_items(&mut self, items: Vec) { self.items_opt = Some(items); } pub fn select_all(&mut self) { *self.cached_selected.borrow_mut() = None; if let Some(ref mut items) = self.items_opt { for item in items.iter_mut() { if !self.config.show_hidden && item.hidden { item.selected = false; continue; } item.selected = true; } } } pub fn select_none(&mut self) -> bool { *self.cached_selected.borrow_mut() = None; self.select_focus = None; let mut had_selection = false; if let Some(ref mut items) = self.items_opt { for item in items.iter_mut() { if item.selected { item.selected = false; had_selection = true; } } } had_selection } pub fn select_name(&mut self, name: &str) { *self.cached_selected.borrow_mut() = None; if let Some(ref mut items) = self.items_opt { for item in items.iter_mut() { if item.name == name { item.selected = true; } else { item.selected = false; } } } } fn select_position(&mut self, row: usize, col: usize, mod_shift: bool) -> bool { *self.cached_selected.borrow_mut() = None; let mut start = (row, col); let mut end = (row, col); if mod_shift { if self.select_focus.is_none() || self.select_range.is_none() { // Set select range to initial state if necessary self.select_range = self.select_focus.map(|i| (i, i)); } if let Some(pos) = self.select_range_start_pos_opt() { if pos.0 < row || (pos.0 == row && pos.1 < col) { start = pos; } else { end = pos; } } } let mut found = false; if let Some(ref mut items) = self.items_opt { for (i, item) in items.iter_mut().enumerate() { item.selected = false; let pos = match item.pos_opt.get() { Some(some) => some, None => continue, }; if pos.0 < start.0 || (pos.0 == start.0 && pos.1 < start.1) { // Before start continue; } if pos.0 > end.0 || (pos.0 == end.0 && pos.1 > end.1) { // After end continue; } if pos == (row, col) { // Update focus if this is what we wanted to select self.select_focus = Some(i); self.select_range = if mod_shift { self.select_range.map(|r| (r.0, i)) } else { Some((i, i)) }; found = true; } item.selected = true; } } found } pub fn select_rect(&mut self, rect: Rectangle, mod_ctrl: bool, mod_shift: bool) { *self.cached_selected.borrow_mut() = None; if let Some(ref mut items) = self.items_opt { for item in items.iter_mut() { let was_overlapped = item.overlaps_drag_rect; item.overlaps_drag_rect = item .rect_opt .get() .map(|r| r.intersects(&rect)) .unwrap_or(false); item.selected = if mod_ctrl || mod_shift { if was_overlapped == item.overlaps_drag_rect { item.selected } else { !item.selected } } else { item.overlaps_drag_rect }; } } } pub fn select_focus_id(&self) -> Option { let items = self.items_opt.as_ref()?; let item = items.get(self.select_focus?)?; Some(item.button_id.clone()) } fn select_focus_pos_opt(&self) -> Option<(usize, usize)> { let items = self.items_opt.as_ref()?; let item = items.get(self.select_focus?)?; item.pos_opt.get() } fn select_focus_scroll(&mut self) -> Option { let items = self.items_opt.as_ref()?; let item = items.get(self.select_focus?)?; let rect = item.rect_opt.get()?; //TODO: move to function let visible_rect = { let point = match self.scroll_opt { Some(offset) => Point::new(0.0, offset.y), None => Point::new(0.0, 0.0), }; let size = self.size_opt.get().unwrap_or_else(|| Size::new(0.0, 0.0)); Rectangle::new(point, size) }; if rect.y < visible_rect.y { // Scroll up to rect self.scroll_opt = Some(AbsoluteOffset { x: 0.0, y: rect.y }); self.scroll_opt } else if (rect.y + rect.height) > (visible_rect.y + visible_rect.height) { // Scroll down to rect self.scroll_opt = Some(AbsoluteOffset { x: 0.0, y: rect.y + rect.height - visible_rect.height, }); self.scroll_opt } else { // Do not scroll None } } fn select_range_start_pos_opt(&self) -> Option<(usize, usize)> { let items = self.items_opt.as_ref()?; let item = items.get(self.select_range.map(|r| r.0)?)?; item.pos_opt.get() } fn select_first_pos_opt(&self) -> Option<(usize, usize)> { *self.cached_selected.borrow_mut() = None; let items = self.items_opt.as_ref()?; let mut first = None; for item in items.iter() { if !item.selected { continue; } let (row, col) = match item.pos_opt.get() { Some(some) => some, None => continue, }; first = Some(match first { Some((first_row, first_col)) => { if row < first_row { (row, col) } else if row == first_row { (row, col.min(first_row)) } else { (first_row, first_col) } } None => (row, col), }); } first } fn select_last_pos_opt(&self) -> Option<(usize, usize)> { *self.cached_selected.borrow_mut() = None; let items = self.items_opt.as_ref()?; let mut last = None; for item in items.iter() { if !item.selected { continue; } let (row, col) = match item.pos_opt.get() { Some(some) => some, None => continue, }; last = Some(match last { Some((last_row, last_col)) => { if row > last_row { (row, col) } else if row == last_row { (row, col.max(last_row)) } else { (last_row, last_col) } } None => (row, col), }); } last } pub fn change_location(&mut self, location: &Location, history_i_opt: Option) { self.location = location.clone(); self.items_opt = None; self.select_focus = None; self.edit_location = None; if let Some(history_i) = history_i_opt { // Navigating in history self.history_i = history_i; } else { // Truncate history to remove next entries self.history.truncate(self.history_i + 1); // Push to the front of history self.history_i = self.history.len(); self.history.push(location.clone()); } } pub fn update(&mut self, message: Message, modifiers: Modifiers) -> Vec { let mut commands = Vec::new(); let mut cd = None; let mut history_i_opt = None; let mod_ctrl = modifiers.contains(Modifiers::CTRL) && self.dialog.as_ref().map_or(true, |x| x.multiple()); let mod_shift = modifiers.contains(Modifiers::SHIFT) && self.dialog.as_ref().map_or(true, |x| x.multiple()); match message { Message::ClickRelease(click_i_opt) => { if click_i_opt == self.clicked.take() { return commands; } self.context_menu = None; self.location_context_menu_index = None; if let Some(ref mut items) = self.items_opt { for (i, item) in items.iter_mut().enumerate() { if mod_ctrl { if Some(i) == click_i_opt && item.selected { item.selected = false; self.select_range = None; } } else if Some(i) != click_i_opt { item.selected = false; } } } } Message::DragEnd(_) => { self.clicked = None; if let Some(ref mut items) = self.items_opt { for item in items.iter_mut() { item.overlaps_drag_rect = false; } } } Message::DoubleClick(click_i_opt) => { if let Some(clicked_item) = self .items_opt .as_ref() .and_then(|items| click_i_opt.and_then(|click_i| items.get(click_i))) { if let Some(path) = &clicked_item.path_opt { if clicked_item.metadata.is_dir() { cd = Some(Location::Path(path.clone())); } else { commands.push(Command::OpenFile(path.clone())); } } else { log::warn!("no path for item {:?}", clicked_item); } } else { log::warn!("no item for click index {:?}", click_i_opt); } } Message::Click(click_i_opt) => { self.selected_clicked = false; self.context_menu = None; self.location_context_menu_index = None; if click_i_opt.is_none() { self.clicked = click_i_opt; } if mod_shift { if let Some(click_i) = click_i_opt { self.select_range = self .select_range .map_or(Some((click_i, click_i)), |r| Some((r.0, click_i))); if let Some(range) = self.select_range { let min = range.0.min(range.1); let max = range.0.max(range.1); if let Some(ref mut items) = self.items_opt { for (i, item) in items.iter_mut().enumerate() { item.selected = i >= min && i <= max; } } } self.clicked = click_i_opt; self.select_focus = click_i_opt; self.selected_clicked = true; } } else { let dont_unset = mod_ctrl || self.column_sort().is_some_and(|l| { l.iter() .any(|(e_i, e)| Some(e_i) == click_i_opt.as_ref() && e.selected) }); *self.cached_selected.borrow_mut() = None; if let Some(ref mut items) = self.items_opt { for (i, item) in items.iter_mut().enumerate() { if Some(i) == click_i_opt { // Filter out selection if it does not match dialog kind if let Some(dialog) = &self.dialog { let item_is_dir = item.metadata.is_dir(); if item_is_dir != dialog.is_dir() { // Allow selecting folder if dialog is for files to make it // possible to double click //TODO: clear any other selection when selecting a folder if !item_is_dir { continue; } } } if !item.selected { self.clicked = click_i_opt; item.selected = true; self.select_range = Some((i, i)); } self.selected_clicked = true; } else if !dont_unset && item.selected { self.clicked = click_i_opt; item.selected = false; } } } if self.select_focus.take().is_some() { // Unfocus currently focused button commands.push(Command::FocusButton(widget::Id::unique())); } } } Message::Config(config) => { self.config = config; } Message::ContextAction(action) => { // Close context menu self.context_menu = None; commands.push(Command::Action(action)); } Message::ContextMenu(point_opt) => { if point_opt.is_none() || !mod_shift { self.context_menu = point_opt; } } Message::LocationContextMenuPoint(point_opt) => { self.location_context_menu_point = point_opt; } Message::LocationContextMenuIndex(index_opt) => { self.location_context_menu_index = index_opt; } Message::LocationMenuAction(action) => { self.location_context_menu_index = None; let path_for_index = |ancestor_index| { match self.location { Location::Path(ref path) => Some(path), Location::Search(ref path, _) => Some(path), _ => None, } .and_then(|path| path.ancestors().nth(ancestor_index)) .map(|path| path.to_path_buf()) }; match action { LocationMenuAction::OpenInNewTab(ancestor_index) => { if let Some(path) = path_for_index(ancestor_index) { commands.push(Command::OpenInNewTab(path)); } } LocationMenuAction::OpenInNewWindow(ancestor_index) => { if let Some(path) = path_for_index(ancestor_index) { commands.push(Command::OpenInNewWindow(path)); } } LocationMenuAction::Properties(ancestor_index) => { commands.push(Command::LocationProperties(ancestor_index)); } } } Message::Drag(rect_opt) => match rect_opt { Some(rect) => { self.context_menu = None; self.location_context_menu_index = None; self.select_rect(rect, mod_ctrl, mod_shift); if self.select_focus.take().is_some() { // Unfocus currently focused button commands.push(Command::FocusButton(widget::Id::unique())); } } None => {} }, Message::EditLocation(edit_location) => { if self.edit_location.is_none() && edit_location.is_some() { commands.push(Command::FocusTextInput(self.edit_location_id.clone())); } self.edit_location = edit_location; } Message::OpenInNewTab(path) => { commands.push(Command::OpenInNewTab(path)); } Message::EmptyTrash => { commands.push(Command::EmptyTrash); } Message::GoNext => { if let Some(history_i) = self.history_i.checked_add(1) { if let Some(location) = self.history.get(history_i) { cd = Some(location.clone()); history_i_opt = Some(history_i); } } } Message::GoPrevious => { if let Some(history_i) = self.history_i.checked_sub(1) { if let Some(location) = self.history.get(history_i) { cd = Some(location.clone()); history_i_opt = Some(history_i); } } } Message::ItemDown => { if let Some((row, col)) = self.select_focus_pos_opt().or(self.select_last_pos_opt()) { if self.select_focus.is_none() { // Select last item in current selection to focus it. self.select_position(row, col, mod_shift); } //TODO: Shift modifier should select items in between // Try to select item in next row if !self.select_position(row + 1, col, mod_shift) { // Ensure current item is still selected if there are no other items self.select_position(row, col, mod_shift); } } else { // Select first item //TODO: select first in scroll self.select_position(0, 0, mod_shift); } if let Some(offset) = self.select_focus_scroll() { commands.push(Command::Scroll(self.scrollable_id.clone(), offset)); } if let Some(id) = self.select_focus_id() { commands.push(Command::FocusButton(id)); } } Message::ItemLeft => { if let Some((row, col)) = self.select_focus_pos_opt().or(self.select_first_pos_opt()) { if self.select_focus.is_none() { // Select first item in current selection to focus it. self.select_position(row, col, mod_shift); } // Try to select previous item in current row if !col .checked_sub(1) .map_or(false, |col| self.select_position(row, col, mod_shift)) { // Try to select last item in previous row if !row.checked_sub(1).map_or(false, |row| { let mut col = 0; if let Some(ref items) = self.items_opt { for item in items.iter() { match item.pos_opt.get() { Some((item_row, item_col)) if item_row == row => { col = col.max(item_col); } _ => continue, } } } self.select_position(row, col, mod_shift) }) { // Ensure current item is still selected if there are no other items self.select_position(row, col, mod_shift); } } } else { // Select first item //TODO: select first in scroll self.select_position(0, 0, mod_shift); } if let Some(offset) = self.select_focus_scroll() { commands.push(Command::Scroll(self.scrollable_id.clone(), offset)); } if let Some(id) = self.select_focus_id() { commands.push(Command::FocusButton(id)); } } Message::ItemRight => { if let Some((row, col)) = self.select_focus_pos_opt().or(self.select_last_pos_opt()) { if self.select_focus.is_none() { // Select last item in current selection to focus it. self.select_position(row, col, mod_shift); } // Try to select next item in current row if !self.select_position(row, col + 1, mod_shift) { // Try to select first item in next row if !self.select_position(row + 1, 0, mod_shift) { // Ensure current item is still selected if there are no other items self.select_position(row, col, mod_shift); } } } else { // Select first item //TODO: select first in scroll self.select_position(0, 0, mod_shift); } if let Some(offset) = self.select_focus_scroll() { commands.push(Command::Scroll(self.scrollable_id.clone(), offset)); } if let Some(id) = self.select_focus_id() { commands.push(Command::FocusButton(id)); } } Message::ItemUp => { if let Some((row, col)) = self.select_focus_pos_opt().or(self.select_first_pos_opt()) { if self.select_focus.is_none() { // Select first item in current selection to focus it. self.select_position(row, col, mod_shift); } //TODO: Shift modifier should select items in between // Try to select item in last row if !row .checked_sub(1) .map_or(false, |row| self.select_position(row, col, mod_shift)) { // Ensure current item is still selected if there are no other items self.select_position(row, col, mod_shift); } } else { // Select first item //TODO: select first in scroll self.select_position(0, 0, mod_shift); } if let Some(offset) = self.select_focus_scroll() { commands.push(Command::Scroll(self.scrollable_id.clone(), offset)); } if let Some(id) = self.select_focus_id() { commands.push(Command::FocusButton(id)); } } Message::Location(location) => { // Workaround to support favorited files match &location { Location::Path(path) => { if path.is_dir() { cd = Some(location); } else { commands.push(Command::OpenFile(path.clone())); } } Location::Search(_path, _term) => { cd = Some(location); } Location::Trash => { cd = Some(location); } } } Message::LocationUp => { // Sets location to the path's parent // Does nothing if path is root or location is Trash if let Location::Path(ref path) = self.location { if let Some(parent) = path.parent() { cd = Some(Location::Path(parent.to_owned())); } } } Message::Open => { if let Some(ref mut items) = self.items_opt { for item in items.iter() { if item.selected { if let Some(path) = &item.path_opt { if path.is_dir() { //TODO: allow opening multiple tabs? cd = Some(Location::Path(path.clone())); } else { commands.push(Command::OpenFile(path.clone())); } } else { //TODO: open properties? } } } } } Message::RightClick(click_i_opt) => { self.update(Message::Click(click_i_opt), modifiers); *self.cached_selected.borrow_mut() = None; if let Some(ref mut items) = self.items_opt { if !click_i_opt.map_or(false, |click_i| { items.get(click_i).map_or(false, |x| x.selected) }) { // If item not selected, clear selection on other items for (i, item) in items.iter_mut().enumerate() { item.selected = Some(i) == click_i_opt; } } } } Message::MiddleClick(click_i) => { self.update(Message::Click(Some(click_i)), modifiers); if !mod_ctrl && !mod_shift { if let Some(ref mut items) = self.items_opt { for (i, item) in items.iter_mut().enumerate() { item.selected = i == click_i; } self.select_range = Some((click_i, click_i)); } if let Some(clicked_item) = self.items_opt.as_ref().and_then(|items| items.get(click_i)) { if let Some(path) = &clicked_item.path_opt { if clicked_item.metadata.is_dir() { //cd = Some(Location::Path(path.clone())); commands.push(Command::OpenInNewTab(path.clone())) } else { commands.push(Command::OpenFile(path.clone())); } } else { log::warn!("no path for item {:?}", clicked_item); } } else { log::warn!("no item for click index {:?}", click_i); } } } Message::Scroll(viewport) => { self.scroll_opt = Some(viewport.absolute_offset()); } Message::SelectAll => { self.select_all(); if self.select_focus.take().is_some() { // Unfocus currently focused button commands.push(Command::FocusButton(widget::Id::unique())); } } Message::Thumbnail(path, thumbnail) => { if let Some(ref mut items) = self.items_opt { for item in items.iter_mut() { if item.path_opt.as_ref() == Some(&path) { if let ItemThumbnail::Rgba(rgba) = &thumbnail { //TODO: pass handles already generated to avoid blocking main thread let handle = widget::icon::from_raster_pixels( rgba.width(), rgba.height(), rgba.as_raw().clone(), ); item.icon_handle_grid = handle.clone(); item.icon_handle_list = handle; } item.thumbnail_opt = Some(thumbnail); break; } } } } Message::ToggleFoldersFirst => self.config.folders_first = !self.config.folders_first, Message::ToggleShowHidden => self.config.show_hidden = !self.config.show_hidden, Message::View(view) => { self.config.view = view; } Message::ToggleSort(heading_option) => { let heading_sort = if self.config.sort_name == heading_option { !self.config.sort_direction } else { true }; self.config.sort_direction = heading_sort; self.config.sort_name = heading_option; } Message::Drop(Some((to, mut from))) => { self.dnd_hovered = None; match to { Location::Path(to) => { if let Ok(entries) = fs::read_dir(&to) { for i in entries.into_iter().filter_map(|e| e.ok()) { let i = i.path(); from.paths.retain(|p| &i != p); if from.paths.is_empty() { log::info!("All dropped files already in target directory."); return commands; } } } commands.push(Command::DropFiles(to, from)) } Location::Search(_, _) => { log::warn!(" Copy/cut to search not supported."); } Location::Trash if matches!(from.kind, ClipboardKind::Cut) => { commands.push(Command::MoveToTrash(from.paths)) } Location::Trash => { log::warn!("Copy to trash is not supported."); } }; } Message::Drop(None) => { self.dnd_hovered = None; } Message::DndHover(loc) => { if self.dnd_hovered.as_ref().is_some_and(|(l, i)| { *l == loc && Instant::now().duration_since(*i) > HOVER_DURATION }) { cd = Some(loc); } } Message::DndEnter(loc) => { self.dnd_hovered = Some((loc.clone(), Instant::now())); if loc != self.location { commands.push(Command::Timeout(HOVER_DURATION, Message::DndHover(loc))); } } Message::DndLeave(loc) => { if Some(&loc) == self.dnd_hovered.as_ref().map(|(l, _)| l) { self.dnd_hovered = None; } } Message::ZoomDefault => match self.config.view { View::List => self.config.icon_sizes.list = 100.try_into().unwrap(), View::Grid => self.config.icon_sizes.grid = 100.try_into().unwrap(), }, Message::ZoomIn => { let zoom_in = |size: &mut NonZeroU16, min: u16, max: u16| { let mut step = min; while step <= max { if size.get() < step { *size = step.try_into().unwrap(); break; } step += 25; } if size.get() > step { *size = step.try_into().unwrap(); } }; match self.config.view { View::List => zoom_in(&mut self.config.icon_sizes.list, 100, 500), View::Grid => zoom_in(&mut self.config.icon_sizes.grid, 50, 500), } } Message::ZoomOut => { let zoom_out = |size: &mut NonZeroU16, min: u16, max: u16| { let mut step = max; while step >= min { if size.get() > step { *size = step.try_into().unwrap(); break; } step -= 25; } if size.get() < step { *size = step.try_into().unwrap(); } }; match self.config.view { View::List => zoom_out(&mut self.config.icon_sizes.list, 100, 500), View::Grid => zoom_out(&mut self.config.icon_sizes.grid, 50, 500), } } } if let Some(location) = cd { if location != self.location { if match &location { Location::Path(path) => path.is_dir(), Location::Search(path, _term) => path.is_dir(), Location::Trash => true, } { self.change_location(&location, history_i_opt); commands.push(Command::ChangeLocation(self.title(), location)); } else { log::warn!("tried to cd to {:?} which is not a directory", location); } } } commands } fn column_sort(&self) -> Option> { let check_reverse = |ord: Ordering, sort: bool| { if sort { ord } else { ord.reverse() } }; let mut items: Vec<_> = self.items_opt.as_ref()?.iter().enumerate().collect(); let heading_sort = self.config.sort_direction; match self.config.sort_name { HeadingOptions::Size => { items.sort_by(|a, b| { // entries take precedence over size let get_size = |x: &Item| match &x.metadata { ItemMetadata::Path { metadata, children } => { if metadata.is_dir() { (true, *children as u64) } else { (false, metadata.len()) } } ItemMetadata::Trash { metadata, .. } => match metadata.size { trash::TrashItemSize::Entries(entries) => (true, entries as u64), trash::TrashItemSize::Bytes(bytes) => (false, bytes), }, }; let (a_is_entry, a_size) = get_size(a.1); let (b_is_entry, b_size) = get_size(b.1); //TODO: use folders_first? match (a_is_entry, b_is_entry) { (true, false) => Ordering::Less, (false, true) => Ordering::Greater, _ => check_reverse(a_size.cmp(&b_size), heading_sort), } }) } HeadingOptions::Name => items.sort_by(|a, b| { if self.config.folders_first { match (a.1.metadata.is_dir(), b.1.metadata.is_dir()) { (true, false) => Ordering::Less, (false, true) => Ordering::Greater, _ => check_reverse( LANGUAGE_SORTER.compare(&a.1.name, &b.1.name), heading_sort, ), } } else { check_reverse(LANGUAGE_SORTER.compare(&a.1.name, &b.1.name), heading_sort) } }), HeadingOptions::Modified => { items.sort_by(|a, b| { let get_modified = |x: &Item| match &x.metadata { ItemMetadata::Path { metadata, .. } => metadata.modified().ok(), ItemMetadata::Trash { .. } => None, }; let a_modified = get_modified(a.1); let b_modified = get_modified(b.1); if self.config.folders_first { match (a.1.metadata.is_dir(), b.1.metadata.is_dir()) { (true, false) => Ordering::Less, (false, true) => Ordering::Greater, _ => check_reverse(a_modified.cmp(&b_modified), heading_sort), } } else { check_reverse(a_modified.cmp(&b_modified), heading_sort) } }); } } Some(items) } pub fn location_view(&self) -> Element { //TODO: responsiveness is done in a hacky way, potentially move this to a custom widget? fn text_width<'a>( content: &'a str, font: font::Font, font_size: f32, line_height: f32, ) -> f32 { let text: text::Text<'a, font::Font> = text::Text { content, bounds: Size::INFINITY, size: font_size.into(), line_height: text::LineHeight::Absolute(line_height.into()), font, horizontal_alignment: Horizontal::Left, vertical_alignment: Vertical::Top, shaping: text::Shaping::default(), wrap: text::Wrap::None, }; graphics::text::Paragraph::with_text(text) .min_bounds() .width } fn text_width_body<'a>(content: &'a str) -> f32 { //TODO: should libcosmic set the font when using widget::text::body? text_width(content, font::Font::DEFAULT, 14.0, 20.0) } fn text_width_heading<'a>(content: &'a str) -> f32 { text_width(content, font::FONT_SEMIBOLD, 14.0, 20.0) } let cosmic_theme::Spacing { space_xxxs, space_xxs, space_s, space_m, .. } = theme::active().cosmic().spacing; let size = self.size_opt.get().unwrap_or(Size::new(0.0, 0.0)); let mut row = widget::row::with_capacity(5) .align_items(Alignment::Center) .padding([0, space_m]); let mut w = 0.0; let mut prev_button = widget::button(widget::icon::from_name("go-previous-symbolic").size(16)) .padding(space_xxs) .style(theme::Button::Icon); if self.history_i > 0 && !self.history.is_empty() { prev_button = prev_button.on_press(Message::GoPrevious); } row = row.push(prev_button); w += 16.0 + 2.0 * space_xxs as f32; let mut next_button = widget::button(widget::icon::from_name("go-next-symbolic").size(16)) .padding(space_xxs) .style(theme::Button::Icon); if self.history_i + 1 < self.history.len() { next_button = next_button.on_press(Message::GoNext); } row = row.push(next_button); w += 16.0 + 2.0 * space_xxs as f32; row = row.push(widget::horizontal_space(Length::Fixed(space_s.into()))); w += space_s as f32; if let Some(location) = &self.edit_location { match location { Location::Path(path) => { row = row.push( widget::button(widget::icon::from_name("window-close-symbolic").size(16)) .on_press(Message::EditLocation(None)) .padding(space_xxs) .style(theme::Button::Icon), ); row = row.push( widget::text_input("", path.to_string_lossy()) .id(self.edit_location_id.clone()) .on_input(|input| { Message::EditLocation(Some(Location::Path(PathBuf::from(input)))) }) .on_submit(Message::Location(location.clone())), ); return row.into(); } _ => { //TODO: allow editing other locations } } } else if let Location::Path(path) = &self.location { row = row.push( crate::mouse_area::MouseArea::new( widget::button(widget::icon::from_name("edit-symbolic").size(16)) .padding(space_xxs) .style(theme::Button::Icon), ) .on_press(move |_| Message::EditLocation(Some(self.location.clone()))) .on_middle_press(move |_| Message::OpenInNewTab(path.clone())), ); w += 16.0 + 2.0 * space_xxs as f32; } else if let Location::Search(_, term) = &self.location { row = row.push( widget::button( widget::row::with_children(vec![ widget::icon::from_name("system-search-symbolic") .size(16) .into(), widget::text::body(term).wrap(text::Wrap::None).into(), ]) .spacing(space_xxs), ) .padding(space_xxs) .style(theme::Button::Icon), ); w += text_width_body(term) + 16.0 + 3.0 * space_xxs as f32; } let mut children: Vec> = Vec::new(); match &self.location { Location::Path(path) | Location::Search(path, ..) => { let home_dir = crate::home_dir(); let excess_str = "..."; let excess_width = text_width_body(excess_str); for (index, ancestor) in path.ancestors().enumerate() { let mut found_home = false; let (name, icon_opt) = match ancestor.file_name() { Some(name) => { if ancestor == home_dir { let icon = widget::icon::icon(folder_icon_symbolic( &ancestor.to_path_buf(), 16, )) .size(16); found_home = true; (fl!("home"), Some(icon)) } else { (name.to_string_lossy().to_string(), None) } } None => { let icon = widget::icon::from_name("drive-harddisk-system-symbolic") .size(16) .icon(); (fl!("filesystem"), Some(icon)) } }; let icon_width = if icon_opt.is_some() { 16.0 + space_xxxs as f32 } else { 0.0 }; let (name_width, name_text) = if children.is_empty() { ( text_width_heading(&name), widget::text::heading(name).wrap(text::Wrap::None), ) } else { children.push( widget::icon::from_name("go-next-symbolic") .size(16) .icon() .into(), ); w += 16.0; ( text_width_body(&name), widget::text::body(name).wrap(text::Wrap::None), ) }; // Add padding for mouse area w += 2.0 * space_xxxs as f32; let mut row = widget::row::with_capacity(2) .align_items(Alignment::Center) .spacing(space_xxxs); //TODO: figure out why this hardcoded offset is needed after the first item is ellipsed let overflow_offset = if icon_opt.is_some() { 0.0 } else { 32.0 }; let overflow = w + icon_width + name_width + overflow_offset > size.width && index > 0; if overflow { row = row.push(widget::text::body(excess_str)); w += excess_width; } else { if let Some(icon) = icon_opt { row = row.push(icon); w += icon_width; } row = row.push(name_text); w += name_width; } let mut mouse_area = crate::mouse_area::MouseArea::new( widget::button(row) .padding(space_xxxs) .style(theme::Button::Link), ) .on_press(move |_| { Message::Location(match &self.location { Location::Path(_) => Location::Path(ancestor.to_path_buf()), Location::Search(_, term) => { Location::Search(ancestor.to_path_buf(), term.clone()) } other => other.clone(), }) }); if self.location_context_menu_index.is_some() { mouse_area = mouse_area.on_right_press(move |_point_opt| { Message::LocationContextMenuIndex(None) }) } else { mouse_area = mouse_area.on_right_press_no_capture(move |point_opt| { Message::LocationContextMenuIndex(Some(index)) }) } let mouse_area = if let Location::Path(_) = &self.location { mouse_area .on_middle_press(move |_| Message::OpenInNewTab(ancestor.to_path_buf())) } else { mouse_area }; children.push(mouse_area.into()); if found_home || overflow { break; } } children.reverse(); } Location::Trash => { let mut row = widget::row::with_capacity(2) .align_items(Alignment::Center) .spacing(space_xxxs); row = row.push(widget::icon::icon(trash_icon_symbolic(16)).size(16)); row = row.push(widget::text::heading(fl!("trash"))); children.push( widget::button(row) .padding(space_xxxs) .on_press(Message::Location(Location::Trash)) .style(theme::Button::Text) .into(), ); } } for child in children { row = row.push(child); } let mouse_area = crate::mouse_area::MouseArea::new(row) .on_right_press(Message::LocationContextMenuPoint); let mut popover = widget::popover(mouse_area); if let (Some(point), Some(index)) = ( self.location_context_menu_point, self.location_context_menu_index, ) { popover = popover .popup(menu::location_context_menu(index)) .position(widget::popover::Position::Point(point)) } popover.into() } pub fn empty_view(&self, has_hidden: bool) -> Element { let cosmic_theme::Spacing { space_xxs, .. } = theme::active().cosmic().spacing; widget::column::with_children(vec![widget::container( widget::column::with_children(vec![ widget::icon::from_name("folder-symbolic") .size(64) .icon() .into(), widget::text(if has_hidden { fl!("empty-folder-hidden") } else { fl!("empty-folder") }) .into(), ]) .align_items(Alignment::Center) .spacing(space_xxs), ) .align_x(Horizontal::Center) .align_y(Vertical::Center) .width(Length::Fill) .height(Length::Fill) .into()]) .into() } pub fn grid_view( &self, ) -> ( Option>>, Element, bool, ) { let cosmic_theme::Spacing { space_m, space_xxs, space_xxxs, .. } = theme::active().cosmic().spacing; let TabConfig { show_hidden, icon_sizes, .. } = self.config; let text_height = 3 * 20; // 3 lines of text let item_width = (3 * space_xxs + icon_sizes.grid() + 3 * space_xxs) as usize; let item_height = (space_xxxs + icon_sizes.grid() + space_xxxs + text_height + space_xxxs) as usize; let (width, height) = match self.size_opt.get() { Some(size) => ( (size.width.floor() as usize) .checked_sub(2 * (space_m as usize)) .unwrap_or(0) .max(item_width), (size.height.floor() as usize).max(item_height), ), None => (item_width, item_height), }; let (cols, column_spacing) = { let width_m1 = width.checked_sub(item_width).unwrap_or(0); let cols_m1 = width_m1 / (item_width + space_xxs as usize); let cols = cols_m1 + 1; let spacing = width_m1 .checked_div(cols_m1) .unwrap_or(0) .checked_sub(item_width) .unwrap_or(0); (cols, spacing as u16) }; let mut grid = widget::grid() .column_spacing(column_spacing) .row_spacing(space_xxs) .padding([0, space_m].into()); let mut dnd_items: Vec<(usize, (usize, usize), &Item)> = Vec::new(); let mut drag_w_i = usize::MAX; let mut drag_n_i = usize::MAX; let mut drag_e_i = 0; let mut drag_s_i = 0; let mut children = Vec::new(); if let Some(items) = self.column_sort() { let mut count = 0; let mut col = 0; let mut row = 0; let mut hidden = 0; for &(i, item) in items.iter() { if !show_hidden && item.hidden { item.pos_opt.set(None); item.rect_opt.set(None); hidden += 1; continue; } item.pos_opt.set(Some((row, col))); item.rect_opt.set(Some(Rectangle::new( Point::new( (col * (item_width + column_spacing as usize) + space_m as usize) as f32, (row * (item_height + space_xxs as usize)) as f32, ), Size::new(item_width as f32, item_height as f32), ))); //TODO: one focus group per grid item (needs custom widget) let buttons = vec![ widget::button( widget::icon::icon(item.icon_handle_grid.clone()) .content_fit(ContentFit::Contain) .size(icon_sizes.grid()), ) .padding(space_xxxs) .style(button_style(item.selected, false, false)), widget::button(widget::text::body(&item.grid_name)) .id(item.button_id.clone()) .padding([0, space_xxxs]) .style(button_style(item.selected, true, true)), ]; let mut column = widget::column::with_capacity(buttons.len()) .align_items(Alignment::Center) .height(Length::Fixed(item_height as f32)) .width(Length::Fixed(item_width as f32)); for button in buttons { if self.context_menu.is_some() { column = column.push(button) } else { column = column.push( mouse_area::MouseArea::new(button).on_right_press_no_capture( move |_point_opt| Message::RightClick(Some(i)), ), ); } } let column: Element = if item.metadata.is_dir() && item.path_opt.is_some() { let tab_location = Location::Path(item.path_opt.clone().unwrap()); let tab_location_enter = tab_location.clone(); let tab_location_leave = tab_location.clone(); let is_dnd_hovered = self.dnd_hovered.as_ref().map(|(l, _)| l) == Some(&tab_location); cosmic::widget::container( DndDestination::for_data::(column, move |data, action| { if let Some(mut data) = data { if action == DndAction::Copy { Message::Drop(Some((tab_location.clone(), data))) } else if action == DndAction::Move { data.kind = ClipboardKind::Cut; Message::Drop(Some((tab_location.clone(), data))) } else { log::warn!("unsupported action: {:?}", action); Message::Drop(None) } } else { Message::Drop(None) } }) .on_enter(move |_, _, _| Message::DndEnter(tab_location_enter.clone())) .on_leave(move || Message::DndLeave(tab_location_leave.clone())), ) .style(if is_dnd_hovered { theme::Container::custom(|t| { let mut a = cosmic::iced_style::container::StyleSheet::appearance( t, &theme::Container::default(), ); let t = t.cosmic(); // todo use theme drop target color let mut bg = t.accent_color(); bg.alpha = 0.2; a.background = Some(Color::from(bg).into()); a.border = Border { color: t.accent_color().into(), width: 1.0, radius: t.radius_s().into(), }; a }) } else { theme::Container::default() }) .into() } else { column.into() }; if item.selected { dnd_items.push((i, (row, col), item)); drag_w_i = drag_w_i.min(col); drag_n_i = drag_n_i.min(row); drag_e_i = drag_e_i.max(col); drag_s_i = drag_s_i.max(row); } let mouse_area = crate::mouse_area::MouseArea::new(column) .on_press(move |_| Message::Click(Some(i))) .on_double_click(move |_| Message::DoubleClick(Some(i))) .on_release(move |_| Message::ClickRelease(Some(i))) .on_middle_press(move |_| Message::MiddleClick(i)); grid = grid.push(mouse_area); count += 1; col += 1; if col >= cols { col = 0; row += 1; grid = grid.insert_row(); } } if count == 0 { return (None, self.empty_view(hidden > 0), false); } children.push(grid.into()); //TODO: HACK If we don't reach the bottom of the view, go ahead and add a spacer to do that { let mut max_bottom = 0; for (_, item) in items { if let Some(rect) = item.rect_opt.get() { let bottom = (rect.y + rect.height).ceil() as usize; if bottom > max_bottom { max_bottom = bottom; } } } let spacer_height = height .checked_sub(max_bottom + 4 * (space_xxs as usize)) .unwrap_or(0); if spacer_height > 0 { children.push( widget::container(vertical_space(Length::Fixed(spacer_height as f32))) .into(), ) } } } ( (!dnd_items.is_empty()).then(|| { let mut dnd_grid = widget::grid() .column_spacing(column_spacing) .row_spacing(space_xxs) .padding([0, space_m].into()); let mut dnd_item_i = 0; for r in drag_n_i..=drag_s_i { dnd_grid = dnd_grid.insert_row(); for c in drag_w_i..=drag_e_i { let Some((i, (row, col), item)) = dnd_items.get(dnd_item_i) else { break; }; if *row == r && *col == c { let buttons = vec![ widget::button( widget::icon::icon(item.icon_handle_grid.clone()) .content_fit(ContentFit::Contain) .size(icon_sizes.grid()), ) .on_press(Message::Click(Some(*i))) .padding(space_xxxs) .style(button_style( item.selected, false, false, )), widget::button(widget::text(item.grid_name.clone())) .id(item.button_id.clone()) .on_press(Message::Click(Some(*i))) .padding([0, space_xxxs]) .style(button_style(item.selected, true, true)), ]; let mut column = widget::column::with_capacity(buttons.len()) .align_items(Alignment::Center) .height(Length::Fixed(item_height as f32)) .width(Length::Fixed(item_width as f32)); for button in buttons { column = column.push(button) } dnd_grid = dnd_grid.push(column); dnd_item_i += 1; } else { dnd_grid = dnd_grid.push( widget::container(vertical_space(item_width as f32)) .height(Length::Fixed(item_height as f32)), ); } } } Element::from(dnd_grid) .map(|m| cosmic::app::Message::App(crate::app::Message::TabMessage(None, m))) }), mouse_area::MouseArea::new( widget::container(widget::column::with_children(children)).width(Length::Fill), ) .on_press(|_| Message::Click(None)) .on_drag(Message::Drag) .on_drag_end(|_| Message::DragEnd(None)) .show_drag_rect(true) .on_release(|_| Message::ClickRelease(None)) .into(), true, ) } pub fn list_view( &self, ) -> ( Option>>, Element, bool, ) { let cosmic_theme::Spacing { space_m, space_xxs, .. } = theme::active().cosmic().spacing; let TabConfig { show_hidden, icon_sizes, sort_name, sort_direction, .. } = self.config; let size = self.size_opt.get().unwrap_or_else(|| Size::new(0.0, 0.0)); //TODO: allow resizing? let name_width = 300.0; let modified_width = 200.0; let size_width = 100.0; let condensed = size.width < (name_width + modified_width + size_width); let icon_size = if condensed { icon_sizes.list_condensed() } else { icon_sizes.list() }; let row_height = icon_size + 2 * space_xxs; let heading_item = |name, width, msg| { let mut row = widget::row::with_capacity(2) .align_items(Alignment::Center) .spacing(space_xxs) .width(width); row = row.push(widget::text::heading(name)); match (sort_name == msg, sort_direction) { (true, true) => { row = row.push(widget::icon::from_name("pan-down-symbolic").size(16)); } (true, false) => { row = row.push(widget::icon::from_name("pan-up-symbolic").size(16)); } _ => {} } //TODO: make it possible to resize with the mouse mouse_area::MouseArea::new(row) .on_press(move |_point_opt| Message::ToggleSort(msg)) .into() }; let mut children: Vec> = Vec::new(); let mut y = 0; if !condensed { children.push( widget::row::with_children(vec![ heading_item(fl!("name"), Length::Fill, HeadingOptions::Name), //TODO: do not show modified column when in the trash heading_item( fl!("modified"), Length::Fixed(modified_width), HeadingOptions::Modified, ), heading_item(fl!("size"), Length::Fixed(size_width), HeadingOptions::Size), ]) .align_items(Alignment::Center) .height(Length::Fixed(space_m.into())) .padding([0, space_xxs]) .spacing(space_xxs) .into(), ); y += row_height; children.push(horizontal_rule(1).into()); y += 1; } let items = self.column_sort(); let mut drag_items = Vec::new(); if let Some(items) = items { let mut count = 0; let mut hidden = 0; for (i, item) in items { if !show_hidden && item.hidden { item.pos_opt.set(None); item.rect_opt.set(None); hidden += 1; continue; } item.pos_opt.set(Some((count, 0))); item.rect_opt.set(Some(Rectangle::new( Point::new(space_m as f32, y as f32), Size::new(size.width - (2 * space_m) as f32, row_height as f32), ))); if count > 0 { children.push(horizontal_rule(1).into()); y += 1; } let modified_text = match &item.metadata { ItemMetadata::Path { metadata, .. } => match metadata.modified() { Ok(time) => chrono::DateTime::::from(time) .format(TIME_FORMAT) .to_string(), Err(_) => String::new(), }, ItemMetadata::Trash { .. } => String::new(), }; let size_text = match &item.metadata { ItemMetadata::Path { metadata, children } => { if metadata.is_dir() { format!("{} items", children) } else { format_size(metadata.len()) } } ItemMetadata::Trash { metadata, .. } => match metadata.size { trash::TrashItemSize::Entries(entries) => { //TODO: translate if entries == 1 { format!("{} item", entries) } else { format!("{} items", entries) } } trash::TrashItemSize::Bytes(bytes) => format_size(bytes), }, }; let row = if condensed { widget::row::with_children(vec![ widget::icon::icon(item.icon_handle_list_condensed.clone()) .content_fit(ContentFit::Contain) .size(icon_size) .into(), widget::column::with_children(vec![ widget::text(item.name.clone()).into(), //TODO: translate? widget::text(format!("{} - {}", modified_text, size_text)).into(), ]) .into(), ]) .align_items(Alignment::Center) .spacing(space_xxs) } else { widget::row::with_children(vec![ widget::icon::icon(item.icon_handle_list.clone()) .content_fit(ContentFit::Contain) .size(icon_size) .into(), widget::text(item.name.clone()).width(Length::Fill).into(), widget::text(modified_text.clone()) .width(Length::Fixed(modified_width)) .into(), widget::text(size_text.clone()) .width(Length::Fixed(size_width)) .into(), ]) .align_items(Alignment::Center) .spacing(space_xxs) }; let button = |row| { let mouse_area = crate::mouse_area::MouseArea::new( widget::button(row) .width(Length::Fill) .height(Length::Fixed(row_height as f32)) .id(item.button_id.clone()) .padding(if icon_size < 24 { 7 } else { space_xxs }) .style(button_style(item.selected, true, false)), ) .on_press(move |_| Message::Click(Some(i))) .on_double_click(move |_| Message::DoubleClick(Some(i))) .on_release(move |_| Message::ClickRelease(Some(i))) .on_middle_press(move |_| Message::MiddleClick(i)); if self.context_menu.is_some() { mouse_area } else { mouse_area.on_right_press_no_capture(move |_point_opt| { Message::RightClick(Some(i)) }) } }; let button_row = button(row.into()); let button_row: Element<_> = if item.metadata.is_dir() && item.path_opt.is_some() { let tab_location = Location::Path(item.path_opt.clone().unwrap()); let tab_location_enter = tab_location.clone(); let tab_location_leave = tab_location.clone(); let is_dnd_hovered = self.dnd_hovered.as_ref().map(|(l, _)| l) == Some(&tab_location); cosmic::widget::container( DndDestination::for_data(button_row, move |data, action| { if let Some(mut data) = data { if action == DndAction::Copy { Message::Drop(Some((tab_location.clone(), data))) } else if action == DndAction::Move { data.kind = ClipboardKind::Cut; Message::Drop(Some((tab_location.clone(), data))) } else { log::warn!("unsupported action: {:?}", action); Message::Drop(None) } } else { log::warn!("No data for drop."); Message::Drop(None) } }) .on_enter(move |_, _, _| Message::DndEnter(tab_location_enter.clone())) .on_leave(move || Message::DndLeave(tab_location_leave.clone())), ) // todo refactor into the dnd destination wrapper .style(if is_dnd_hovered { theme::Container::custom(|t| { let mut a = cosmic::iced_style::container::StyleSheet::appearance( t, &theme::Container::default(), ); let t = t.cosmic(); // todo use theme drop target color let mut bg = t.accent_color(); bg.alpha = 0.2; a.background = Some(Color::from(bg).into()); a.border = Border { color: t.accent_color().into(), width: 1.0, radius: t.radius_s().into(), }; a }) } else { theme::Container::default() }) .into() } else { button_row.into() }; if item.selected || !drag_items.is_empty() { let dnd_row = if !item.selected { Element::from(vertical_space(Length::Fixed(row_height as f32))) } else if condensed { widget::row::with_children(vec![ widget::icon::icon(item.icon_handle_list_condensed.clone()) .content_fit(ContentFit::Contain) .size(icon_size) .into(), widget::column::with_children(vec![ widget::text(item.name.clone()).into(), //TODO: translate? widget::text(format!("{} - {}", modified_text, size_text)).into(), ]) .into(), ]) .align_items(Alignment::Center) .spacing(space_xxs) .into() } else { widget::row::with_children(vec![ widget::icon::icon(item.icon_handle_list.clone()) .content_fit(ContentFit::Contain) .size(icon_size) .into(), widget::text(item.name.clone()).width(Length::Fill).into(), widget::text(modified_text) .width(Length::Fixed(modified_width)) .into(), widget::text(size_text) .width(Length::Fixed(size_width)) .into(), ]) .align_items(Alignment::Center) .spacing(space_xxs) .into() }; if item.selected { drag_items.push( widget::container(button(dnd_row)) .width(Length::Shrink) .into(), ); } else { drag_items.push(dnd_row); } } count += 1; y += row_height; children.push(button_row); } if count == 0 { return (None, self.empty_view(hidden > 0), false); } } //TODO: HACK If we don't reach the bottom of the view, go ahead and add a spacer to do that { let spacer_height = size.height as i32 - y as i32 - 4 * space_xxs as i32; if spacer_height > 0 { children.push( widget::container(vertical_space(Length::Fixed(spacer_height as f32))).into(), ); } } let drag_col = (!drag_items.is_empty()).then(|| { Element::from(widget::column::with_children(drag_items)) .map(|m| cosmic::app::Message::App(crate::app::Message::TabMessage(None, m))) }); ( drag_col, mouse_area::MouseArea::new( widget::column::with_children(children).padding([0, space_m]), ) .with_id(Id::new("list-view")) .on_press(|_| Message::Click(None)) .on_drag(Message::Drag) .on_drag_end(|_| Message::DragEnd(None)) .show_drag_rect(true) .on_release(|_| Message::ClickRelease(None)) .into(), true, ) } pub fn view_responsive( &self, key_binds: &HashMap, size: Size, ) -> Element { // Update cached size self.size_opt.set(Some(size)); let location_view = self.location_view(); let (drag_list, mut item_view, can_scroll) = match self.config.view { View::Grid => self.grid_view(), View::List => self.list_view(), }; item_view = widget::container(item_view).width(Length::Fill).into(); let files = self .items_opt .as_ref() .map(|items| { items .iter() .filter(|item| item.selected) .filter_map(|item| item.path_opt.clone()) .collect::>() }) .unwrap_or_default(); let item_view = DndSource::<_, cosmic::app::Message, ClipboardCopy>::with_id( item_view, Id::new("tab-view"), ); let item_view = match drag_list { Some(drag_list) if self.selected_clicked => { let drag_list = ArcElementWrapper(Arc::new(Mutex::new(drag_list))); item_view .drag_content(move || { ClipboardCopy::new(crate::clipboard::ClipboardKind::Copy, &files) }) .drag_icon(move || { let state: tree::State = Widget::, _, _>::state(&drag_list); (drag_list.clone().into(), state) }) } _ => item_view, }; let tab_location = self.location.clone(); let mut mouse_area = mouse_area::MouseArea::new(item_view) .on_press(move |_point_opt| Message::Click(None)) .on_release(|_| Message::ClickRelease(None)) .on_back_press(move |_point_opt| Message::GoPrevious) .on_forward_press(move |_point_opt| Message::GoNext); if self.context_menu.is_some() { mouse_area = mouse_area.on_right_press(move |_point_opt| Message::ContextMenu(None)); } else { mouse_area = mouse_area.on_right_press(Message::ContextMenu); } let mut popover = widget::popover(mouse_area); if let Some(point) = self.context_menu { popover = popover .popup(menu::context_menu(&self, &key_binds)) .position(widget::popover::Position::Point(point)); } let mut tab_column = widget::column::with_capacity(3); tab_column = tab_column.push(location_view); if can_scroll { tab_column = tab_column.push( widget::scrollable(popover) .id(self.scrollable_id.clone()) .on_scroll(Message::Scroll) .width(Length::Fill) .height(Length::Fill), ); } else { tab_column = tab_column.push(popover); } if let Location::Trash = self.location { if let Some(items) = self.items_opt() { if !items.is_empty() { let cosmic_theme::Spacing { space_xxs, space_xs, .. } = theme::active().cosmic().spacing; tab_column = tab_column.push( widget::layer_container(widget::row::with_children(vec![ widget::horizontal_space(Length::Fill).into(), widget::button::standard(fl!("empty-trash")) .on_press(Message::EmptyTrash) .into(), ])) .padding([space_xxs, space_xs]) .layer(cosmic_theme::Layer::Primary), ); } } } let mut tab_view = widget::container(tab_column) .height(Length::Fill) .width(Length::Fill); if self.dnd_hovered.as_ref().map(|(l, _)| l) == Some(&tab_location) { tab_view = tab_view.style(cosmic::theme::Container::custom(|t| { let mut a = cosmic::iced_style::container::StyleSheet::appearance( t, &cosmic::theme::Container::default(), ); let c = t.cosmic(); a.border = cosmic::iced_core::Border { color: (c.accent_color()).into(), width: 1., radius: c.radius_0().into(), }; a })); } let tab_location_2 = self.location.clone(); let tab_location_3 = self.location.clone(); let dnd_dest = DndDestination::for_data(tab_view, move |data, action| { if let Some(mut data) = data { if action == DndAction::Copy { Message::Drop(Some((tab_location.clone(), data))) } else if action == DndAction::Move { data.kind = ClipboardKind::Cut; Message::Drop(Some((tab_location.clone(), data))) } else { log::warn!("unsupported action: {:?}", action); Message::Drop(None) } } else { Message::Drop(None) } }) .on_enter(move |_, _, _| Message::DndEnter(tab_location_2.clone())) .on_leave(move || Message::DndLeave(tab_location_3.clone())); dnd_dest.into() } pub fn view<'a>(&'a self, key_binds: &'a HashMap) -> Element { widget::responsive(|size| self.view_responsive(key_binds, size)).into() } pub fn subscription(&self) -> Subscription { if let Some(items) = &self.items_opt { //TODO: how many thumbnail loads should be in flight at once? let jobs = 8; let mut subscriptions = Vec::with_capacity(jobs); //TODO: move to function let visible_rect = { let point = match self.scroll_opt { Some(offset) => Point::new(0.0, offset.y), None => Point::new(0.0, 0.0), }; let size = self.size_opt.get().unwrap_or_else(|| Size::new(0.0, 0.0)); Rectangle::new(point, size) }; //TODO: HACK to ensure positions are up to date since subscription runs before view match self.config.view { View::Grid => _ = self.grid_view(), View::List => _ = self.list_view(), }; for item in items.iter() { if item.thumbnail_opt.is_some() { // Skip items that already have a thumbnail continue; } match item.rect_opt.get() { Some(rect) => { if !rect.intersects(&visible_rect) { // Skip items that are not visible continue; } } None => { // Skip items with no determined rect (this should include hidden items) continue; } } if let Some(path) = item.path_opt.clone() { subscriptions.push(subscription::channel( path.clone(), 1, |mut output| async move { let (path, thumbnail) = tokio::task::spawn_blocking(move || { let start = std::time::Instant::now(); let thumbnail = match image::io::Reader::open(&path) .and_then(|img| img.with_guessed_format()) { Ok(reader) => match reader.decode() { Ok(image) => { //TODO: configurable thumbnail size? let thumbnail_size = (ICON_SIZE_GRID * ICON_SCALE_MAX) as u32; let thumbnail = image.thumbnail(thumbnail_size, thumbnail_size); ItemThumbnail::Rgba(thumbnail.to_rgba8()) } Err(err) => { log::warn!("failed to decode {:?}: {}", path, err); ItemThumbnail::NotImage } }, Err(err) => { log::warn!("failed to read {:?}: {}", path, err); ItemThumbnail::NotImage } }; log::info!("thumbnailed {:?} in {:?}", path, start.elapsed()); (path, thumbnail) }) .await .unwrap(); match output .send(Message::Thumbnail(path.clone(), thumbnail)) .await { Ok(()) => {} Err(err) => { log::warn!("failed to send thumbnail for {:?}: {}", path, err); } } //TODO: how to properly kill this task? loop { tokio::time::sleep(std::time::Duration::new(1, 0)).await; } }, )); } if subscriptions.len() >= jobs { break; } } Subscription::batch(subscriptions) } else { Subscription::none() } } } #[cfg(test)] mod tests { use std::{fs, io, path::PathBuf}; use cosmic::iced_runtime::keyboard::Modifiers; use log::{debug, trace}; use tempfile::TempDir; use test_log::test; use super::{scan_path, Location, Message, Tab}; use crate::{ app::test_utils::{ assert_eq_tab_path, empty_fs, eq_path_item, filter_dirs, read_dir_sorted, simple_fs, tab_click_new, NAME_LEN, NUM_DIRS, NUM_FILES, NUM_HIDDEN, NUM_NESTED, }, config::{IconSizes, TabConfig}, }; // Boilerplate for tab tests. Checks if simulated clicks selected items. fn tab_selects_item( clicks: &[usize], modifiers: Modifiers, expected_selected: &[bool], ) -> io::Result<()> { let (_fs, mut tab) = tab_click_new(NUM_FILES, NUM_NESTED, NUM_DIRS, NUM_NESTED, NAME_LEN)?; // Simulate clicks by triggering Message::Click for &click in clicks { debug!("Emitting Message::Click(Some({click})) with modifiers: {modifiers:?}"); tab.update(Message::Click(Some(click)), modifiers); } let items = tab .items_opt .as_deref() .expect("tab should be populated with items"); for (i, (&expected, actual)) in expected_selected.into_iter().zip(items).enumerate() { assert_eq!( expected, actual.selected, "expected index {i} to be {}", if expected { "selected but it was deselected" } else { "deselected but it was selected" } ); } Ok(()) } fn tab_history() -> io::Result<(TempDir, Tab, Vec)> { let fs = simple_fs(NUM_FILES, NUM_NESTED, NUM_DIRS, NUM_NESTED, NAME_LEN)?; let path = fs.path(); let mut tab = Tab::new(Location::Path(path.into()), TabConfig::default()); // All directories (simple_fs only produces one nested layer) let dirs: Vec = filter_dirs(path)? .flat_map(|dir| { filter_dirs(&dir).map(|nested_dirs| std::iter::once(dir).chain(nested_dirs)) }) .flatten() .collect(); assert!( dirs.len() == NUM_DIRS + NUM_DIRS * NUM_NESTED, "Sanity check: Have {} dirs instead of {}", dirs.len(), NUM_DIRS + NUM_DIRS * NUM_NESTED ); debug!("Building history by emitting Message::Location"); for dir in &dirs { debug!( "Emitting Message::Location(Location::Path(\"{}\"))", dir.display() ); tab.update( Message::Location(Location::Path(dir.clone())), Modifiers::empty(), ); } trace!("Tab history: {:?}", tab.history); Ok((fs, tab, dirs)) } #[test] fn scan_path_succeeds_on_valid_path() -> io::Result<()> { let fs = simple_fs(NUM_FILES, NUM_HIDDEN, NUM_DIRS, NUM_NESTED, NAME_LEN)?; let path = fs.path(); // Read directory entries and sort as cosmic-files does let entries = read_dir_sorted(path)?; debug!("Calling scan_path(\"{}\")", path.display()); let actual = scan_path(&path.to_owned(), IconSizes::default()); // scan_path shouldn't skip any entries assert_eq!(entries.len(), actual.len()); // Correct files should be scanned assert!(entries .into_iter() .zip(actual.into_iter()) .all(|(path, item)| eq_path_item(&path, &item))); Ok(()) } #[test] fn scan_path_returns_empty_vec_for_invalid_path() -> io::Result<()> { let fs = simple_fs(NUM_FILES, NUM_NESTED, NUM_DIRS, NUM_NESTED, NAME_LEN)?; let path = fs.path(); // A nonexisting path within the temp dir let invalid_path = path.join("ferris"); assert!(!invalid_path.exists()); debug!("Calling scan_path(\"{}\")", invalid_path.display()); let actual = scan_path(&invalid_path, IconSizes::default()); assert!(actual.is_empty()); Ok(()) } #[test] fn scan_path_empty_dir_returns_empty_vec() -> io::Result<()> { let fs = empty_fs()?; let path = fs.path(); debug!("Calling scan_path(\"{}\")", path.display()); let actual = scan_path(&path.to_owned(), IconSizes::default()); assert_eq!(0, path.read_dir()?.count()); assert!(actual.is_empty()); Ok(()) } #[test] fn tab_location_changes_location() -> io::Result<()> { let fs = simple_fs(NUM_FILES, NUM_NESTED, NUM_DIRS, NUM_NESTED, NAME_LEN)?; let path = fs.path(); // Next directory in temp directory // This does not have to be sorted let next_dir = filter_dirs(path)? .next() .expect("temp directory should have at least one directory"); let mut tab = Tab::new(Location::Path(path.to_owned()), TabConfig::default()); debug!( "Emitting Message::Location(Location::Path(\"{}\"))", next_dir.display() ); tab.update( Message::Location(Location::Path(next_dir.clone())), Modifiers::empty(), ); // Validate that the tab's path updated // NOTE: `items_opt` is set to None with Message::Location so this ONLY checks for equal paths // If item contents are NOT None then this needs to be reevaluated for correctness assert_eq_tab_path(&tab, &next_dir); assert!( tab.items_opt.is_none(), "Tab's `items` is not None which means this test needs to be updated" ); Ok(()) } #[test] fn tab_click_single_selects_item() -> io::Result<()> { // Select the second directory with no keys held down tab_selects_item(&[1], Modifiers::empty(), &[false, true]) } #[test] fn tab_click_double_opens_folder() -> io::Result<()> { let (fs, mut tab) = tab_click_new(NUM_FILES, NUM_NESTED, NUM_DIRS, NUM_NESTED, NAME_LEN)?; let path = fs.path(); // Simulate double clicking second directory debug!("Emitting double click Message::DoubleClick(Some(1))"); tab.update(Message::DoubleClick(Some(1)), Modifiers::empty()); // Path to second directory let second_dir = read_dir_sorted(path)? .into_iter() .filter(|p| p.is_dir()) .nth(1) .expect("should be at least two directories"); // Location should have changed to second_dir assert_eq_tab_path(&tab, &second_dir); Ok(()) } #[test] fn tab_click_ctrl_selects_multiple() -> io::Result<()> { // Select the first and second directory by holding down ctrl tab_selects_item(&[0, 1], Modifiers::CTRL, &[true, true]) } #[test] fn tab_gonext_moves_forward_in_history() -> io::Result<()> { let (fs, mut tab, dirs) = tab_history()?; let path = fs.path(); // Rewind to the start for _ in 0..dirs.len() { debug!("Emitting Message::GoPrevious to rewind to the start",); tab.update(Message::GoPrevious, Modifiers::empty()); } assert_eq_tab_path(&tab, path); // Back to the future. Directories should be in the order they were opened. for dir in dirs { debug!("Emitting Message::GoNext",); tab.update(Message::GoNext, Modifiers::empty()); assert_eq_tab_path(&tab, &dir); } Ok(()) } #[test] fn tab_goprev_moves_backward_in_history() -> io::Result<()> { let (fs, mut tab, dirs) = tab_history()?; let path = fs.path(); for dir in dirs.into_iter().rev() { assert_eq_tab_path(&tab, &dir); debug!("Emitting Message::GoPrevious",); tab.update(Message::GoPrevious, Modifiers::empty()); } assert_eq_tab_path(&tab, path); Ok(()) } #[test] fn tab_empty_history_does_nothing_on_prev_next() -> io::Result<()> { let fs = simple_fs(0, NUM_NESTED, NUM_DIRS, 0, NAME_LEN)?; let path = fs.path(); let mut tab = Tab::new(Location::Path(path.into()), TabConfig::default()); // Tab's location shouldn't change if GoPrev or GoNext is triggered debug!("Emitting Message::GoPrevious",); tab.update(Message::GoPrevious, Modifiers::empty()); assert_eq_tab_path(&tab, path); debug!("Emitting Message::GoNext",); tab.update(Message::GoNext, Modifiers::empty()); assert_eq_tab_path(&tab, path); Ok(()) } #[test] fn tab_locationup_moves_up_hierarchy() -> io::Result<()> { let fs = simple_fs(0, NUM_NESTED, NUM_DIRS, 0, NAME_LEN)?; let path = fs.path(); let mut next_dir = filter_dirs(path)? .next() .expect("should be at least one directory"); let mut tab = Tab::new(Location::Path(next_dir.clone()), TabConfig::default()); // This will eventually yield false once root is hit while next_dir.pop() { debug!("Emitting Message::LocationUp",); tab.update(Message::LocationUp, Modifiers::empty()); assert_eq_tab_path(&tab, &next_dir); } Ok(()) } #[test] fn sort_long_number_file_names() -> io::Result<()> { let fs = empty_fs()?; let path = fs.path(); // Create files with names 255 characters long that only contain a single number // Example: 0000...0 for 255 characters // https://en.wikipedia.org/wiki/Filename#Comparison_of_filename_limitations let mut base_nums: Vec<_> = ('0'..'9').collect(); fastrand::shuffle(&mut base_nums); debug!("Shuffled numbers for paths: {base_nums:?}"); let paths: Vec<_> = base_nums .iter() .map(|&base| path.join(std::iter::repeat(base).take(255).collect::())) .collect(); for (file, &base) in paths.iter().zip(base_nums.iter()) { trace!("Creating long file name for {base}"); fs::File::create(file)?; } debug!("Creating tab for directory of long file names"); Tab::new(Location::Path(path.into()), TabConfig::default()); Ok(()) } } #[derive(Clone)] pub struct ArcElementWrapper(pub Arc>>); impl Widget for ArcElementWrapper { fn size(&self) -> Size { self.0.lock().unwrap().as_widget().size() } fn size_hint(&self) -> Size { self.0.lock().unwrap().as_widget().size_hint() } fn layout( &self, tree: &mut tree::Tree, renderer: &cosmic::Renderer, limits: &cosmic::iced_core::layout::Limits, ) -> cosmic::iced_core::layout::Node { self.0 .lock() .unwrap() .as_widget_mut() .layout(tree, renderer, limits) } fn draw( &self, tree: &tree::Tree, renderer: &mut cosmic::Renderer, theme: &cosmic::Theme, style: &cosmic::iced_core::renderer::Style, layout: cosmic::iced_core::Layout<'_>, cursor: cosmic::iced_core::mouse::Cursor, viewport: &Rectangle, ) { self.0 .lock() .unwrap() .as_widget() .draw(tree, renderer, theme, style, layout, cursor, viewport) } fn tag(&self) -> tree::Tag { self.0.lock().unwrap().as_widget().tag() } fn state(&self) -> tree::State { self.0.lock().unwrap().as_widget().state() } fn children(&self) -> Vec { self.0.lock().unwrap().as_widget().children() } fn diff(&mut self, tree: &mut tree::Tree) { self.0.lock().unwrap().as_widget_mut().diff(tree) } fn operate( &self, state: &mut tree::Tree, layout: cosmic::iced_core::Layout<'_>, renderer: &cosmic::Renderer, operation: &mut dyn widget::Operation>, ) { self.0 .lock() .unwrap() .as_widget() .operate(state, layout, renderer, operation) } fn on_event( &mut self, _state: &mut tree::Tree, _event: cosmic::iced::Event, _layout: cosmic::iced_core::Layout<'_>, _cursor: cosmic::iced_core::mouse::Cursor, _renderer: &cosmic::Renderer, _clipboard: &mut dyn cosmic::iced_core::Clipboard, _shell: &mut cosmic::iced_core::Shell<'_, M>, _viewport: &Rectangle, ) -> cosmic::iced_core::event::Status { self.0.lock().unwrap().as_widget_mut().on_event( _state, _event, _layout, _cursor, _renderer, _clipboard, _shell, _viewport, ) } fn mouse_interaction( &self, _state: &tree::Tree, _layout: cosmic::iced_core::Layout<'_>, _cursor: cosmic::iced_core::mouse::Cursor, _viewport: &Rectangle, _renderer: &cosmic::Renderer, ) -> cosmic::iced_core::mouse::Interaction { self.0 .lock() .unwrap() .as_widget() .mouse_interaction(_state, _layout, _cursor, _viewport, _renderer) } fn overlay<'a>( &'a mut self, _state: &'a mut tree::Tree, _layout: cosmic::iced_core::Layout<'_>, _renderer: &cosmic::Renderer, ) -> Option> { // TODO None } fn id(&self) -> Option { self.0.lock().unwrap().as_widget().id() } fn set_id(&mut self, _id: Id) { self.0.lock().unwrap().as_widget_mut().set_id(_id) } fn drag_destinations( &self, _state: &tree::Tree, _layout: cosmic::iced_core::Layout<'_>, renderer: &cosmic::Renderer, _dnd_rectangles: &mut cosmic::iced_core::clipboard::DndDestinationRectangles, ) { self.0.lock().unwrap().as_widget().drag_destinations( _state, _layout, renderer, _dnd_rectangles, ) } } impl From> for Element<'static, Message> { fn from(wrapper: ArcElementWrapper) -> Self { Element::new(wrapper) } }