notedeck/crates/notedeck_ui/src/mention.rs

use crate::ProfilePreview;
use egui::Sense;
use enostr::Pubkey;
use nostrdb::{Ndb, Transaction};
use notedeck::{name::get_display_name, Images, NoteAction};

pub struct Mention<'a> {
    ndb: &'a Ndb,
    img_cache: &'a mut Images,
    txn: &'a Transaction,
    pk: &'a [u8; 32],
    selectable: bool,
    size: Option<f32>,
}

impl<'a> Mention<'a> {
    pub fn new(
        ndb: &'a Ndb,
        img_cache: &'a mut Images,
        txn: &'a Transaction,
        pk: &'a [u8; 32],
    ) -> Self {
        let size = None;
        let selectable = true;
        Mention {
            ndb,
            img_cache,
            txn,
            pk,
            selectable,
            size,
        }
    }

    pub fn selectable(mut self, selectable: bool) -> Self {
        self.selectable = selectable;
        self
    }

    pub fn size(mut self, size: f32) -> Self {
        self.size = Some(size);
        self
    }

    pub fn show(self, ui: &mut egui::Ui) -> Option<NoteAction> {
        mention_ui(
            self.ndb,
            self.img_cache,
            self.txn,
            self.pk,
            ui,
            self.size,
            self.selectable,
        )
    }
}

#[allow(clippy::too_many_arguments)]
#[profiling::function]
fn mention_ui(
    ndb: &Ndb,
    img_cache: &mut Images,
    txn: &Transaction,
    pk: &[u8; 32],
    ui: &mut egui::Ui,
    size: Option<f32>,
    selectable: bool,
) -> Option<NoteAction> {
    let link_color = ui.visuals().hyperlink_color;

    let profile = ndb.get_profile_by_pubkey(txn, pk).ok();

    let name: String = format!(
        "@{}",
        get_display_name(profile.as_ref()).username_or_displayname()
    );

    let mut text = egui::RichText::new(name).color(link_color);
    if let Some(size) = size {
        text = text.size(size);
    }

    let resp = ui
        .add(
            egui::Label::new(text)
                .sense(Sense::click())
                .selectable(selectable),
        )
        .on_hover_cursor(egui::CursorIcon::PointingHand);

    let note_action = if resp.clicked() {
        Some(NoteAction::Profile(Pubkey::new(*pk)))
    } else {
        None
    };

    if let Some(rec) = profile.as_ref() {
        resp.on_hover_ui_at_pointer(|ui| {
            ui.set_max_width(300.0);
            ui.add(ProfilePreview::new(rec, img_cache));
        });
    }

    note_action
}