zoodex/src/ui/collatable_container/collated_grid.rs

use gtk4 :: { Button , FlowBox , Image , Justification , Label , SelectionMode } ;
use gtk4 :: Align :: * ;
use gtk4 :: Orientation :: * ;
use gtk4 :: gdk :: * ;
use gtk4 :: gio :: * ;
use gtk4 :: glib :: * ;
use gtk4 :: pango :: * ;
use gtk4 :: pango :: Weight :: * ;
use gtk4 :: prelude :: * ;
use std :: cell :: * ;
use std :: env :: * ;
use std :: iter :: * ;

use crate :: ui :: collatable_container :: * ;
use crate :: ui :: component :: * ;



pub struct CollatedMediaGrid < A : MediaAdapter > {
	media_widget_pairs : RefCell < Vec < ( A :: Overview , Button ) > > ,
	grid_widget : FlowBox ,
	on_media_selected : & 'static dyn Fn ( A :: Overview ) ,
}

impl < A : MediaAdapter > CollatedMediaGrid <A> {
	pub fn new ( on_media_selected : impl Fn ( A :: Overview ) + 'static ) -> Self {
		let grid_widget = view_expr ! {
			FlowBox {
				set_homogeneous : true ,
				set_selection_mode : SelectionMode :: None ,
				set_css_classes : & [ "collatable-container" ] ,
				set_orientation : Horizontal ,
			}
		} ;
		let media_widget_pairs = RefCell :: new ( Vec :: new () ) ;
		let on_media_selected = leak (on_media_selected) ;

		Self { media_widget_pairs , grid_widget , on_media_selected }
	}

	pub async fn set_media ( & self , media : Vec < A :: Overview > , sorting : A :: Sorting ) {
		// TODO: Check if we should use `MainContext :: invoke_local` here

		let mut widgets = Vec :: new () ;
		for media in media . as_slice () {
			widgets . push ( self . create_media_entry (media) . await ) ;
		}
		self . media_widget_pairs . replace ( zip ( media , widgets ) . collect () ) ;

		for ( _ , widget ) in self . sort_media_widget_pairs (sorting) {
			self . grid_widget . append ( & widget ) ;
		}
	}

	async fn create_media_entry ( & self , media : & A :: Overview ) -> Button {
		view_expr ! {
			Button {
				set_css_classes : & [ "flat" , "collection-item-button" ] ,

				connect_clicked : clone ! (
					# [ strong ] media ,
					# [ strong ( rename_to = on_media_selected ) ] self . on_media_selected ,
					move |_| on_media_selected ( media . clone () ) ,
				) ,

				set_child : Some ( & view_expr ! {
					gtk4 :: Box {
						set_css_classes : & [ "collection-item-box" ] ,
						set_valign : Center ,
						set_orientation : Vertical ,

						// Poster
						append_opt : & {
							let home_directory = var_os ("HOME") . unwrap () ;
							let xdg_data_home = var_os ("XDG_DATA_HOME") ;

							let data_dir = match xdg_data_home {
								Some (xdg_data_home) => concat_os_str ! ( xdg_data_home , "/zoodex" ) ,
								None => concat_os_str ! ( home_directory , "/.local/share/zoodex" ) ,
							} ;

							let poster_file_path = concat_os_str ! ( data_dir , "/posters/" , media . get_uuid () ) ;

							let poster_texture = spawn_blocking (
								move || Texture :: from_filename (poster_file_path) ,
							) . await . unwrap () ;

							match poster_texture {
								Ok (poster_texture) => Some ( view_expr ! {
									Image {
										set_paintable : Some ( & poster_texture ) ,
										set_pixel_size : 300 ,
										set_css_classes : & [ "collection-item-image" ] ,
									}
								} ) ,
								Err (error) => {
									if error . matches ( IOErrorEnum :: NotFound ) {
										None  // The file not existing simply means there is no poster for this piece of media
									} else {
										panic ! ( "{}" , error )  // Any other error means something unexpected went wrong
									}
								} ,
							}
						} ,

						// Name
						append : & view_expr ! {
							Label {
								set_attributes : Some ( & pango_attributes ! ( scale : SCALE_LARGE , weight : Bold ) ) ,
								set_justify : Justification :: Center ,
								set_max_width_chars : 1 ,  // Not the actual limit, used instead to wrap more aggressively
								set_wrap : true ,
								set_label : media . get_name () . as_str () ,
							}
						} ,

						// Original name
						append_opt : & media . get_original_name () . map ( |original_name| view_expr ! {
							Label {
								set_justify : Justification :: Center ,
								set_max_width_chars : 1 ,
								set_wrap : true ,
								set_label : original_name . as_str () ,
							}
						} ) ,

						// Details
						append : & view_expr ! {
							gtk4 :: Box {
								set_spacing : 20 ,
								set_halign : Center ,
								set_orientation : Horizontal ,

								// Release date
								append : & view_expr ! {
									Label { set_label : media . get_release_date () . split ('-') . next () . unwrap () }
								} ,

								// Runtime
								append_opt : & media . get_runtime_minutes () . map ( |runtime_minutes| view_expr ! {
									Label { set_label : format ! ( "{}m" , runtime_minutes ) . as_str () }
								} ) ,
							}
						} ,
					}
				} ) ,
			}
		}
	}

	pub fn set_sorting ( & self , sorting : A :: Sorting ) {
		self . grid_widget . remove_all () ;

		for ( _ , widget ) in self . sort_media_widget_pairs (sorting) {
			self . grid_widget . append ( & widget ) ;
		}
	}

	fn sort_media_widget_pairs ( & self , sorting : A :: Sorting ) -> Vec < ( A :: Overview , Button ) > {
		let mut sorted = Vec :: from (
			self . media_widget_pairs . borrow () . as_slice () ,
		) ;

		sorted . sort_by (
			| ( media_1 , _ ) , ( media_2 , _ ) | A :: compare_by ( media_1 , media_2 , sorting ) ,
		) ;

		// See it, say it, ...
		sorted
	}
}

impl < A : MediaAdapter > Component for CollatedMediaGrid <A> {
	fn get_widget ( & self ) -> & FlowBox { & self . grid_widget }
}