feat(custom): image widget

src/image.rs

@@ -68,7 +68,7 @@ impl<'a> ImageProvider<'a> {
             None if PathBuf::from(input_name).exists() => {
                 Ok(ImageLocation::Local(PathBuf::from(input_name)))
             }
-            None => match get_desktop_icon_name(&input_name) {
+            None => match get_desktop_icon_name(input_name) {
                 Some(input) => Self::get_location(input, theme, size),
                 None => Err(Report::msg("Unknown image type")),
             },

src/modules/custom.rs

@@ -1,12 +1,13 @@
 use crate::config::CommonConfig;
 use crate::dynamic_string::DynamicString;
+use crate::image::ImageProvider;
 use crate::modules::{Module, ModuleInfo, ModuleUpdateEvent, ModuleWidget, WidgetContext};
 use crate::popup::{ButtonGeometry, Popup};
 use crate::script::Script;
 use crate::{send_async, try_send};
 use color_eyre::{Report, Result};
 use gtk::prelude::*;
-use gtk::{Button, Label, Orientation};
+use gtk::{Button, IconTheme, Label, Orientation};
 use serde::Deserialize;
 use tokio::spawn;
 use tokio::sync::mpsc::{Receiver, Sender};
@@ -46,6 +47,8 @@ pub struct Widget {
     class: Option<String>,
     on_click: Option<String>,
     orientation: Option<String>,
+    src: Option<String>,
+    size: Option<i32>,
 }
 
 /// Supported GTK widget types
@@ -55,6 +58,7 @@ pub enum WidgetType {
     Box,
     Label,
     Button,
+    Image,
 }
 
 impl Widget {
@@ -64,6 +68,7 @@ impl Widget {
             WidgetType::Box => parent.add(&self.into_box(&tx, bar_orientation)),
             WidgetType::Label => parent.add(&self.into_label()),
             WidgetType::Button => parent.add(&self.into_button(tx, bar_orientation)),
+            WidgetType::Image => parent.add(&self.into_image()),
         }
     }
 
@@ -157,6 +162,32 @@ impl Widget {
 
         button
     }
+
+    fn into_image(self) -> gtk::Image {
+        let mut builder = gtk::Image::builder();
+
+        if let Some(name) = self.name {
+            builder = builder.name(&name);
+        }
+
+        let gtk_image = builder.build();
+
+        if let Some(src) = self.src {
+            let theme = IconTheme::new();
+            let size = self.size.unwrap_or(32);
+            if let Err(err) = ImageProvider::parse(src, &theme, size)
+                .and_then(|image| image.load_into_image(gtk_image.clone()))
+            {
+                error!("{err:?}");
+            }
+        }
+
+        if let Some(class) = self.class {
+            gtk_image.style_context().add_class(&class);
+        }
+
+        gtk_image
+    }
 }
 
 #[derive(Debug)]

src/modules/music/mod.rs

@@ -1,6 +1,5 @@
 mod config;
 
-use std::path::PathBuf;
 use crate::clients::music::{self, MusicClient, PlayerState, PlayerUpdate, Status, Track};
 use crate::image::ImageProvider;
 use crate::modules::{Module, ModuleInfo, ModuleUpdateEvent, ModuleWidget, WidgetContext};
@@ -11,6 +10,7 @@ use glib::Continue;
 use gtk::prelude::*;
 use gtk::{Button, IconTheme, Label, Orientation, Scale};
 use regex::Regex;
+use std::path::PathBuf;
 use std::sync::Arc;
 use std::time::Duration;
 use tokio::spawn;
@@ -306,14 +306,18 @@ impl Module<Button> for MusicModule {
                     let new_cover = update.song.cover_path;
                     if prev_cover != new_cover {
                         prev_cover = new_cover.clone();
-                        let res = match new_cover.map(|cover_path| ImageProvider::parse(cover_path, &icon_theme, 128))
+                        let res = match new_cover
+                            .map(|cover_path| ImageProvider::parse(cover_path, &icon_theme, 128))
                         {
                             Some(Ok(image)) => image.load_into_image(album_image.clone()),
                             Some(Err(err)) => {
                                 album_image.set_from_pixbuf(None);
                                 Err(err)
                             }
-                            None => Ok(album_image.set_from_pixbuf(None)),
+                            None => {
+                                album_image.set_from_pixbuf(None);
+                                Ok(())
+                            }
                         };
                         if let Err(err) = res {
                             error!("{err:?}");