fractal/components/avatar/
data.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
use gtk::{gdk, glib, prelude::*, subclass::prelude::*};
use tracing::warn;

use super::AvatarImage;
use crate::{
    application::Application,
    utils::notifications::{paintable_as_notification_icon, string_as_notification_icon},
};

mod imp {
    use std::cell::RefCell;

    use super::*;

    #[derive(Debug, Default, glib::Properties)]
    #[properties(wrapper_type = super::AvatarData)]
    pub struct AvatarData {
        /// The data of the user-defined image.
        #[property(get, set = Self::set_image, explicit_notify, nullable)]
        image: RefCell<Option<AvatarImage>>,
        /// The display name used as a fallback for this avatar.
        #[property(get, set = Self::set_display_name, explicit_notify)]
        display_name: RefCell<String>,
    }

    #[glib::object_subclass]
    impl ObjectSubclass for AvatarData {
        const NAME: &'static str = "AvatarData";
        type Type = super::AvatarData;
    }

    #[glib::derived_properties]
    impl ObjectImpl for AvatarData {}

    impl AvatarData {
        /// Set the data of the user-defined image.
        fn set_image(&self, image: Option<AvatarImage>) {
            if *self.image.borrow() == image {
                return;
            }

            self.image.replace(image);
            self.obj().notify_image();
        }

        /// Set the display name used as a fallback for this avatar.
        fn set_display_name(&self, display_name: String) {
            if *self.display_name.borrow() == display_name {
                return;
            }

            self.display_name.replace(display_name);
            self.obj().notify_display_name();
        }
    }
}

glib::wrapper! {
    /// Data about a User’s or Room’s avatar.
    pub struct AvatarData(ObjectSubclass<imp::AvatarData>);
}

impl AvatarData {
    /// Construct a new empty `AvatarData`.
    pub fn new() -> Self {
        glib::Object::new()
    }

    /// Constructs an `AvatarData` with the given image data.
    pub(crate) fn with_image(image: AvatarImage) -> Self {
        glib::Object::builder().property("image", image).build()
    }

    /// Get this avatar as a notification icon.
    ///
    /// Returns `None` if an error occurred while generating the icon.
    pub(crate) async fn as_notification_icon(&self) -> Option<gdk::Texture> {
        let Some(window) = Application::default().active_window() else {
            warn!("Could not generate icon for notification: no active window");
            return None;
        };
        let Some(renderer) = window.renderer() else {
            warn!("Could not generate icon for notification: no renderer");
            return None;
        };
        let scale_factor = window.scale_factor();

        if let Some(image) = self.image() {
            match image.load_small_paintable().await {
                Ok(Some(paintable)) => {
                    let texture = paintable_as_notification_icon(
                        paintable.upcast_ref(),
                        scale_factor,
                        &renderer,
                    );
                    return Some(texture);
                }
                // No paintable, we will try to generate the fallback.
                Ok(None) => {}
                // Could not get the paintable, we will try to generate the fallback.
                Err(error) => {
                    warn!("Could not generate icon for notification: {error}");
                }
            }
        }

        let texture = string_as_notification_icon(
            &self.display_name(),
            scale_factor,
            &window.create_pango_layout(None),
            &renderer,
        );
        Some(texture)
    }
}

impl Default for AvatarData {
    fn default() -> Self {
        Self::new()
    }
}