notedeck/crates/notedeck_columns/src/ui/note/context.rs

use egui::{Rect, Vec2};
use enostr::{NoteId, Pubkey};
use nostrdb::{Note, NoteKey};
use tracing::error;

#[derive(Clone)]
#[allow(clippy::enum_variant_names)]
pub enum NoteContextSelection {
    CopyText,
    CopyPubkey,
    CopyNoteId,
    CopyNoteJSON,
}

impl NoteContextSelection {
    pub fn process(&self, ui: &mut egui::Ui, note: &Note<'_>) {
        match self {
            NoteContextSelection::CopyText => {
                ui.output_mut(|w| {
                    w.copied_text = note.content().to_string();
                });
            }
            NoteContextSelection::CopyPubkey => {
                ui.output_mut(|w| {
                    if let Some(bech) = Pubkey::new(*note.pubkey()).to_bech() {
                        w.copied_text = bech;
                    }
                });
            }
            NoteContextSelection::CopyNoteId => {
                ui.output_mut(|w| {
                    if let Some(bech) = NoteId::new(*note.id()).to_bech() {
                        w.copied_text = bech;
                    }
                });
            }
            NoteContextSelection::CopyNoteJSON => {
                ui.output_mut(|w| match note.json() {
                    Ok(json) => w.copied_text = json,
                    Err(err) => error!("error copying note json: {err}"),
                });
            }
        }
    }
}

pub struct NoteContextButton {
    put_at: Option<Rect>,
    note_key: NoteKey,
}

impl egui::Widget for NoteContextButton {
    fn ui(self, ui: &mut egui::Ui) -> egui::Response {
        let r = if let Some(r) = self.put_at {
            r
        } else {
            let mut place = ui.available_rect_before_wrap();
            let size = Self::max_width();
            place.set_width(size);
            place.set_height(size);
            place
        };

        Self::show(ui, self.note_key, r)
    }
}

impl NoteContextButton {
    pub fn new(note_key: NoteKey) -> Self {
        let put_at: Option<Rect> = None;
        NoteContextButton { note_key, put_at }
    }

    pub fn place_at(mut self, rect: Rect) -> Self {
        self.put_at = Some(rect);
        self
    }

    pub fn max_width() -> f32 {
        Self::max_radius() * 3.0 + Self::max_distance_between_circles() * 2.0
    }

    pub fn size() -> Vec2 {
        let width = Self::max_width();
        egui::vec2(width, width)
    }

    fn max_radius() -> f32 {
        8.0
    }

    fn min_radius() -> f32 {
        Self::max_radius() / Self::expansion_multiple()
    }

    fn max_distance_between_circles() -> f32 {
        2.0
    }

    fn expansion_multiple() -> f32 {
        2.0
    }

    fn min_distance_between_circles() -> f32 {
        Self::max_distance_between_circles() / Self::expansion_multiple()
    }

    pub fn show(ui: &mut egui::Ui, note_key: NoteKey, put_at: Rect) -> egui::Response {
        #[cfg(feature = "profiling")]
        puffin::profile_function!();

        let id = ui.id().with(("more_options_anim", note_key));

        let min_radius = Self::min_radius();
        let anim_speed = 0.05;
        let response = ui.interact(put_at, id, egui::Sense::click());

        let hovered = response.hovered();
        let animation_progress = ui.ctx().animate_bool_with_time(id, hovered, anim_speed);

        if hovered {
            ui.ctx().set_cursor_icon(egui::CursorIcon::PointingHand);
        }

        let min_distance = Self::min_distance_between_circles();
        let cur_distance = min_distance
            + (Self::max_distance_between_circles() - min_distance) * animation_progress;

        let cur_radius = min_radius + (Self::max_radius() - min_radius) * animation_progress;

        let center = put_at.center();
        let left_circle_center = center - egui::vec2(cur_distance + cur_radius, 0.0);
        let right_circle_center = center + egui::vec2(cur_distance + cur_radius, 0.0);

        let translated_radius = (cur_radius - 1.0) / 2.0;

        // This works in both themes
        let color = ui.style().visuals.noninteractive().fg_stroke.color;

        // Draw circles
        let painter = ui.painter_at(put_at);
        painter.circle_filled(left_circle_center, translated_radius, color);
        painter.circle_filled(center, translated_radius, color);
        painter.circle_filled(right_circle_center, translated_radius, color);

        response
    }

    pub fn menu(
        ui: &mut egui::Ui,
        button_response: egui::Response,
    ) -> Option<NoteContextSelection> {
        #[cfg(feature = "profiling")]
        puffin::profile_function!();

        let mut context_selection: Option<NoteContextSelection> = None;

        stationary_arbitrary_menu_button(ui, button_response, |ui| {
            ui.set_max_width(200.0);
            if ui.button("Copy text").clicked() {
                context_selection = Some(NoteContextSelection::CopyText);
                ui.close_menu();
            }
            if ui.button("Copy user public key").clicked() {
                context_selection = Some(NoteContextSelection::CopyPubkey);
                ui.close_menu();
            }
            if ui.button("Copy note id").clicked() {
                context_selection = Some(NoteContextSelection::CopyNoteId);
                ui.close_menu();
            }
            if ui.button("Copy note json").clicked() {
                context_selection = Some(NoteContextSelection::CopyNoteJSON);
                ui.close_menu();
            }
        });

        context_selection
    }
}

fn stationary_arbitrary_menu_button<R>(
    ui: &mut egui::Ui,
    button_response: egui::Response,
    add_contents: impl FnOnce(&mut egui::Ui) -> R,
) -> egui::InnerResponse<Option<R>> {
    let bar_id = ui.id();
    let mut bar_state = egui::menu::BarState::load(ui.ctx(), bar_id);

    let inner = bar_state.bar_menu(&button_response, add_contents);

    bar_state.store(ui.ctx(), bar_id);
    egui::InnerResponse::new(inner.map(|r| r.inner), button_response)
}