bezel-ui 0.2.7

//! [`titlebar`] — the strip a window with no system titlebar moves itself by.
//!
//! The strip does not move the window. A [`grip`] in it does, and everything
//! else in the bar is an ordinary element.
//!
//! That split is forced by how the platform asks. Windows answers
//! `WM_NCHITTEST` out of a flat list of control areas and takes the first one
//! the pointer falls in, parent before child — so a bar that is itself one
//! drag area turns every control in it into a window handle, unless each
//! control blocks the mouse, and blocking the mouse also takes the scroll
//! wheel from everything behind it. Naming the drag surface instead is what
//! AppKit does with a drag gesture on a view, and it costs no control
//! anything.
//!
//! Three platforms, three mechanisms, all of them on the grip: AppKit drags by
//! itself, Linux is told to with `start_window_move` on the first *motion*
//! after a press rather than on the press, and Windows implements neither and
//! reads `WindowControlArea::Drag` back out of the hit test.
//!
//! The macOS traffic lights need [`Theme::TRAFFIC_LIGHT_INSET`] of leading
//! room, which is nothing until the window goes full screen and AppKit takes
//! them away. Off macOS the buttons are the app's to paint: [`controls`] is
//! the cluster, and the frame around it — border, corners, shadow, resize
//! edges — is [`crate::window::frame`].
//!
//! The window it belongs to opens with `appears_transparent: true`, and on
//! Linux asks for `WindowDecorations::Client`. On macOS AppKit still drags the
//! window by its transparent titlebar, and a [`grip`] there is redundant.
//!
//! ```ignore
//! titlebar::titlebar("titlebar", true, window)
//!     .px(px(8.0))
//!     .child(/* … */)
//!     .child(titlebar::grip("titlebar-grip", &self.drag, window))
//!     .child(titlebar::controls(CaptionSide::Right, window, cx))
//! ```

use std::{cell::Cell, rc::Rc};

use gpui::{
    App, Div, ElementId, MAX_BUTTONS_PER_SIDE, MouseButton, Stateful, Window, WindowButton,
    WindowButtonLayout, WindowControlArea, WindowControls, div, prelude::*, px, rgb,
};

use theme::Theme;

/// Appearance of app-drawn caption buttons; native window actions stay the same.
#[derive(Clone, Copy, Debug, Default, PartialEq, Eq)]
pub enum CaptionStyle {
    #[default]
    Rectangular,
    /// Left-aligned close, minimise, zoom discs within full-size caption targets.
    Lights,
}

impl gpui::Global for CaptionStyle {}

impl CaptionStyle {
    /// Resolves button order and permissions. Only rectangular captions use the desktop layout.
    pub fn button_layout(
        self,
        desktop: Option<WindowButtonLayout>,
        allowed: WindowControls,
    ) -> WindowButtonLayout {
        let mut layout = match self {
            Self::Rectangular => desktop.unwrap_or(TRAILING),
            Self::Lights => WindowButtonLayout {
                left: [
                    Some(WindowButton::Close),
                    Some(WindowButton::Minimize),
                    Some(WindowButton::Maximize),
                ],
                right: [None; MAX_BUTTONS_PER_SIDE],
            },
        };
        for button in layout.left.iter_mut().chain(layout.right.iter_mut()) {
            if matches!(button, Some(WindowButton::Minimize) if !allowed.minimize)
                || matches!(button, Some(WindowButton::Maximize) if !allowed.maximize)
            {
                *button = None;
            }
        }
        layout
    }
}

/// Full left strip for three lights, including the space around their discs.
/// Targets keep the platform caption width and reach the left edge without padding.
pub const LIGHTS_WIDTH: f32 = 3.0 * Theme::CAPTION_BUTTON_WIDTH;

/// Width occupied by left-side lights for this window, accounting for permissions
/// and fullscreen. Reserve it only when positioning controls as an overlay;
/// [`controls`] already occupies this space in normal layout.
pub fn lights_width(window: &Window) -> f32 {
    if window.is_fullscreen() {
        return 0.0;
    }
    CaptionStyle::Lights
        .button_layout(None, window.window_controls())
        .left
        .into_iter()
        .flatten()
        .count() as f32
        * Theme::CAPTION_BUTTON_WIDTH
}

pub(crate) fn caption_style(cx: &App) -> CaptionStyle {
    cx.try_global::<CaptionStyle>().copied().unwrap_or_default()
}

const CAPTION_GROUP: &str = "caption-controls";

/// Whether the press on a [`grip`] is still a candidate for a window move.
///
/// Shaped like [`crate::scroll::FollowState`] and for the same reason: it
/// mutates through `&self`, so the element carries the whole gesture and the
/// view holds one field.
#[derive(Clone, Default)]
pub struct DragState(Rc<Cell<bool>>);

/// The strip: full width, [`Theme::TITLEBAR_HEIGHT`] tall, and inert. What it
/// holds is the caller's, including [`grip`], without which the window has no
/// handle off macOS.
///
/// `traffic_lights` reserves the leading inset for the macOS buttons — pass it
/// on the one strip they sit over, and it stands down in full screen, where
/// they are gone and the gap would be a hole.
pub fn titlebar(id: impl Into<ElementId>, traffic_lights: bool, window: &Window) -> Stateful<Div> {
    div()
        .id(id)
        .w_full()
        .h(px(Theme::TITLEBAR_HEIGHT))
        .flex()
        .flex_row()
        .items_center()
        .when(traffic_lights && !window.is_fullscreen(), |bar| {
            bar.pl(px(Theme::TRAFFIC_LIGHT_INSET))
        })
}

/// The bare stretch of a titlebar that drags its window, zooms it on a double
/// click and opens the desktop's window menu on a right press.
///
/// Takes the free space in the bar, so it is the room the content leaves. A
/// bar whose content fills it has no handle — the window can then only be
/// moved by its own edges, where the system has any.
///
/// Nothing needs to opt out of it: a control beside a grip is not inside it,
/// and the platform hit test only ever lands in one of them.
pub fn grip(id: impl Into<ElementId>, drag: &DragState, window: &Window) -> Stateful<Div> {
    let (armed, disarm, release) = (drag.0.clone(), drag.0.clone(), drag.0.clone());
    let moving = drag.0.clone();
    let zoomable = window.window_controls().maximize && window.is_resizable();
    div()
        .id(id)
        .flex_1()
        .self_stretch()
        .on_mouse_down(MouseButton::Left, move |_, _, _| armed.set(true))
        .on_mouse_up(MouseButton::Left, move |_, _, _| release.set(false))
        // A press that leaves the grip is not a window move either — without
        // this the flag survives, and the next stray motion over it drags the
        // window with no button held.
        .on_mouse_down_out(move |_, _, _| disarm.set(false))
        .on_mouse_move(move |_, window, _| {
            if moving.replace(false) {
                window.start_window_move();
            }
        })
        // What Windows moves by: the area answers `WM_NCHITTEST` with
        // `HTCAPTION`, and the system drag, the edge snap and the double-click
        // zoom all follow from that. `start_window_move` above is the Linux
        // path and is not implemented there at all.
        .window_control_area(WindowControlArea::Drag)
        // The window menu the desktop hangs off its own titlebar, where the
        // compositor says there is one. On the press, which is where a context
        // menu belongs, and a no-op on macOS and on Windows, where the system
        // menu comes from the caption hit test instead.
        .when(window.window_controls().window_menu, |grip| {
            grip.on_mouse_down(MouseButton::Right, |event, window, _| {
                window.show_window_menu(event.position);
            })
        })
        .on_click(move |click, window, _| {
            if click.click_count() == 2 {
                // macOS runs whatever the user set the gesture to — zoom,
                // minimise or nothing — and Windows zooms from `HTCAPTION`
                // without being asked. Linux has neither, and only where the
                // window can be zoomed at all.
                match cfg!(any(target_os = "linux", target_os = "freebsd")) {
                    true if zoomable => window.zoom_window(),
                    true => {}
                    false => window.titlebar_double_click(),
                }
            }
        })
}

/// Which end of the bar a caption cluster sits at.
#[derive(Clone, Copy, Debug, PartialEq, Eq)]
pub enum CaptionSide {
    Left,
    Right,
}

/// The caption buttons, for a window whose system caption is gone —
/// `appears_transparent` on Windows, `Decorations::Client` on Linux.
///
/// Call it at both ends. Rectangular captions follow `App::button_layout`,
/// falling back to the right. Lights always use close/minimise/zoom on the left.
///
/// Empty in full screen, and short whatever `Window::window_controls` says the
/// compositor will refuse. Close is never refused.
///
/// The cluster takes its own width in the bar, so nothing is reserved for it
/// the way [`Theme::TRAFFIC_LIGHT_INSET`] is reserved for AppKit's lights —
/// those are painted over the client area by someone else, and these are not.
///
/// ```ignore
/// titlebar::titlebar("titlebar", true, window)
///     .child(titlebar::controls(CaptionSide::Left, window, cx))
///     .child(titlebar::grip("titlebar-grip", &self.drag, window))
///     .child(titlebar::controls(CaptionSide::Right, window, cx))
/// ```
pub fn controls(side: CaptionSide, window: &Window, cx: &App) -> Div {
    let row = div()
        .group(CAPTION_GROUP)
        .flex()
        .flex_row()
        .flex_none()
        .items_center()
        .h_full();
    if window.is_fullscreen() {
        return row;
    }

    let layout = caption_style(cx).button_layout(cx.button_layout(), window.window_controls());
    let buttons = match side {
        CaptionSide::Left => layout.left,
        CaptionSide::Right => layout.right,
    };

    buttons.into_iter().flatten().fold(row, |row, button| {
        row.child(caption_button(button, window, cx))
    })
}

/// What a platform with no layout of its own gets: all three, trailing.
const TRAILING: WindowButtonLayout = WindowButtonLayout {
    left: [None; MAX_BUTTONS_PER_SIDE],
    right: [
        Some(WindowButton::Minimize),
        Some(WindowButton::Maximize),
        Some(WindowButton::Close),
    ],
};

/// The caption glyphs are drawn to their own scale, not the type ladder's —
/// Windows sets them at 10px whatever the shell font is doing.
const CAPTION_GLYPH: f32 = 10.0;

/// One caption button: the glyph, the hover wash, and the hitbox the platform
/// reads.
///
/// The click is wired everywhere but Windows, where answering `WM_NCHITTEST`
/// with `HTCLOSE` and friends has already handed the press to the system —
/// acting on it here as well would minimise and restore in one gesture.
fn caption_button(button: WindowButton, window: &Window, cx: &App) -> Stateful<Div> {
    let theme = Theme::of(cx);
    let (area, glyph, hover) = match button {
        WindowButton::Close => (WindowControlArea::Close, icons::glyph::X, theme.danger),
        WindowButton::Minimize => (
            WindowControlArea::Min,
            icons::glyph::Minus,
            theme.element_hover,
        ),
        // The restore mark is two offset squares, which is what `copy` draws.
        WindowButton::Maximize => (
            WindowControlArea::Max,
            match window.is_maximized() {
                true => icons::glyph::Copy,
                false => icons::glyph::Square,
            },
            theme.element_hover,
        ),
    };

    div()
        .id(button.id())
        .debug_selector(move || format!("caption-{}", button.id()))
        .flex()
        .flex_none()
        .items_center()
        .justify_center()
        .w(px(Theme::CAPTION_BUTTON_WIDTH))
        .h_full()
        .map(|control| match caption_style(cx) {
            CaptionStyle::Rectangular => control.hover(|button| button.bg(hover)).child(
                icons::icon(glyph)
                    .size(px(CAPTION_GLYPH))
                    .text_color(theme.text),
            ),
            CaptionStyle::Lights => {
                let (color, mark) = match button {
                    WindowButton::Close => (0xff5f57, icons::glyph::X),
                    WindowButton::Minimize => (0xfebc2e, icons::glyph::Minus),
                    WindowButton::Maximize => (0x28c840, icons::glyph::Plus),
                };
                control.child(
                    div()
                        .size(px(12.0))
                        .rounded_full()
                        .bg(rgb(if window.is_window_active() {
                            color
                        } else {
                            0x999999
                        }))
                        .flex()
                        .items_center()
                        .justify_center()
                        .child(
                            div()
                                .invisible()
                                .group_hover(CAPTION_GROUP, |mark| mark.visible())
                                .child(icons::icon(mark).size(px(8.0)).text_color(rgb(0x333333))),
                        ),
                )
            }
        })
        .window_control_area(area)
        .when(!cfg!(target_os = "windows"), |control| {
            control.on_click(move |_, window, _| match button {
                WindowButton::Close => window.remove_window(),
                WindowButton::Minimize => window.minimize_window(),
                WindowButton::Maximize => window.zoom_window(),
            })
        })
}