mirror of
https://github.com/tiennm99/claude-code-usage-bubble.git
synced 2026-06-06 14:12:14 +00:00
tiennm99
526786b902
Phase 2 lite. Replaces the horizontal pill (two stacked progress bars) with a stadium-shaped bubble: a circle "head" on the left showing the 5h percentage as a big glyph surrounded by a stroked progress ring, plus a "tail" extending right with the 7d label, a thin progress bar, and the 7d countdown. The bubble's primary metric (5h window) is now glanceable from across the room — a thick ring sweeping around a big number reads at a much greater distance than two thin horizontal bars. The 7d window remains visible as supporting context. The expanded panel (left-click) still shows both windows in full. Implementation notes: - Hybrid render: tiny-skia (already a Cargo dep for tray badge) paints the AA shape into a Pixmap. The pixmap is copied byte-for-byte into the 32bpp BI_RGB DIB; GDI overlays ClearType text on top; UpdateLayeredWindow blits with per-pixel alpha as before. - Stadium outline: corner_radius = height/2 so point_in_rounded_rect exactly approximates the capsule shape for hit-test. - Pulse animation on ≥95% applies to both the ring sweep (5h) and the tail bar fill (7d) independently. - Codex teal #10A37F and Claude orange #D97757 carry across the ring, the tail bar, and the tray badge sweep via crate::usage_color. Removed (dead after pipeline swap): - per-pixel paint_background / paint_accent_stripe / paint_bars / paint_one_bar / apply_alpha_mask / row_band / rgb_to_dib / blend - BarLayout struct + compute_layout - old paint_text_layer / draw_label / draw_percent / draw_countdown - Breakpoint struct + breakpoint_for_width_logical (font sizes now derive from head_diameter directly) - luminance / use_dark_text_over (text was over bar fills; new tail bar carries no overlaid text) - constants ACCENT_STRIPE_W_LOGICAL, LABEL_PAD_LOGICAL, PERCENT_TEMPLATE Build: cargo build --release clean. Clippy 13 warnings (was 11); the 2 new ones are field-assign-after-Default::default() on tiny-skia Stroke setup, matching the existing pattern in src/tray/badge.rs. Known follow-up: BubbleState.session_text + BubbleConfig.session_text plumbing is now unused (head shows percent only, no 5h countdown on the bubble). Removing it is a multi-file chain through app.rs and panel.rs; deferred.
1513 lines
49 KiB
Rust
1513 lines
49 KiB
Rust
// Floating stadium-shaped bubble window.
|
||
//
|
||
// Top-level window with WS_POPUP + WS_EX_LAYERED + WS_EX_TOPMOST + WS_EX_NOACTIVATE.
|
||
// The shape is a stadium (rounded-rect with corner_radius = height/2). The left
|
||
// half is the "head" — a stroked progress ring around the 5h percentage glyph.
|
||
// The right half is the "tail" — small "7d" label, thin progress bar, countdown.
|
||
//
|
||
// Painting is hybrid: tiny-skia renders the shape (AA fills + AA stroked arc)
|
||
// into a Pixmap; the Pixmap is copied byte-for-byte into a 32bpp BI_RGB DIB;
|
||
// GDI then overlays ClearType text on top; UpdateLayeredWindow blits the result
|
||
// to the screen with per-pixel alpha. WM_NCHITTEST returns HTCAPTION inside the
|
||
// stadium so the OS handles drag for free.
|
||
|
||
use std::collections::HashMap;
|
||
use std::ffi::c_void;
|
||
use std::sync::{Mutex, MutexGuard, OnceLock};
|
||
|
||
use tiny_skia::{FillRule, LineCap, Paint, PathBuilder, Pixmap, Rect, Stroke, Transform};
|
||
use windows::core::PCWSTR;
|
||
use windows::Win32::Foundation::*;
|
||
use windows::Win32::Graphics::Gdi::*;
|
||
use windows::Win32::System::LibraryLoader::{GetModuleFileNameW, GetModuleHandleW};
|
||
use windows::Win32::UI::HiDpi::*;
|
||
use windows::Win32::UI::Shell::{
|
||
ExtractIconExW, SHAppBarMessage, ABE_BOTTOM, ABE_LEFT, ABE_RIGHT, ABE_TOP, ABM_GETTASKBARPOS,
|
||
APPBARDATA,
|
||
};
|
||
use windows::Win32::UI::WindowsAndMessaging::*;
|
||
|
||
use crate::os::dpi::scale as scale_to_dpi;
|
||
use crate::os::{to_utf16_nul as wide_str, Rgb as Color};
|
||
|
||
const TIMER_FULLSCREEN_CHECK: usize = 5;
|
||
const TIMER_PULSE: usize = 6;
|
||
const PULSE_INTERVAL_MS: u32 = 80;
|
||
use crate::usage::ProviderId;
|
||
|
||
// ---------- Public types & API ----------
|
||
|
||
// Width clamps in logical pixels. Height is derived per width (see
|
||
// `bubble_height_logical`) — aspect tapers from 3:1 at the small end toward
|
||
// 2.6:1 at the large end so the bars look proportionally chunkier as the
|
||
// bubble grows.
|
||
pub const MIN_BUBBLE_SIZE: i32 = 140;
|
||
pub const MAX_BUBBLE_SIZE: i32 = 360;
|
||
pub const DEFAULT_BUBBLE_SIZE: i32 = 200;
|
||
const RESIZE_STEP: i32 = 20;
|
||
const SNAP_ZONE_LOGICAL: i32 = 12;
|
||
const CORNER_SNAP_ZONE_LOGICAL: i32 = 32;
|
||
const CORNER_INSET_LOGICAL: i32 = 12;
|
||
const TASKBAR_GAP_LOGICAL: i32 = 4;
|
||
const PEER_ALIGN_TOLERANCE_LOGICAL: i32 = 8;
|
||
const CLASS_NAME: &str = "ClaudeCodeUsageBubble";
|
||
const FULLSCREEN_POLL_MS: u32 = 1500;
|
||
|
||
/// (num, den) such that bubble_height = (width * den) / num. 3:1 below 200,
|
||
/// 2.8:1 below 280, 2.6:1 above — the bubble gets a touch taller as it
|
||
/// grows so the wider bars don't look anaemic.
|
||
fn aspect_at_width(w_logical: i32) -> (i32, i32) {
|
||
if w_logical <= 200 {
|
||
(3, 1)
|
||
} else if w_logical <= 280 {
|
||
(14, 5) // 2.8 : 1
|
||
} else {
|
||
(13, 5) // 2.6 : 1
|
||
}
|
||
}
|
||
|
||
pub struct BubbleConfig {
|
||
pub model: ProviderId,
|
||
pub size_logical: i32,
|
||
pub position: Option<(i32, i32)>,
|
||
pub session_pct: Option<f64>,
|
||
pub session_text: String,
|
||
pub weekly_pct: Option<f64>,
|
||
pub weekly_text: String,
|
||
pub is_dark: bool,
|
||
}
|
||
|
||
fn bubble_height_logical(width_logical: i32) -> i32 {
|
||
let (num, den) = aspect_at_width(width_logical);
|
||
((width_logical * den) / num).max(20)
|
||
}
|
||
|
||
/// Owner-supplied event callbacks. The bubble window proc is a leaf — it
|
||
/// doesn't know about `app`. The owner installs these once at startup so the
|
||
/// proc can dispatch UI events back without an upward `crate::app::` reach.
|
||
pub struct Callbacks {
|
||
pub on_click: fn(HWND, ProviderId),
|
||
pub on_right_click: fn(HWND, ProviderId, POINT),
|
||
pub on_moved: fn(ProviderId, (i32, i32)),
|
||
pub on_resized: fn(ProviderId, i32),
|
||
pub on_menu_command: fn(u32, HWND),
|
||
pub on_settings_changed: fn(),
|
||
}
|
||
|
||
static CALLBACKS: OnceLock<Callbacks> = OnceLock::new();
|
||
|
||
/// Install the owner's callbacks. Called once by `app::run` before any
|
||
/// bubble is created. Subsequent calls are silently ignored.
|
||
pub fn install_callbacks(cb: Callbacks) {
|
||
let _ = CALLBACKS.set(cb);
|
||
}
|
||
|
||
fn dispatch<F: FnOnce(&Callbacks)>(f: F) {
|
||
if let Some(cb) = CALLBACKS.get() {
|
||
f(cb);
|
||
} else {
|
||
log::warn!("bubble event dispatched before install_callbacks; event dropped");
|
||
}
|
||
}
|
||
|
||
/// Register the bubble window class. Idempotent; safe to call before the first
|
||
/// `create()` from the UI thread.
|
||
pub fn register_class() {
|
||
static REGISTERED: OnceLock<()> = OnceLock::new();
|
||
REGISTERED.get_or_init(|| unsafe {
|
||
let class_w = wide_str(CLASS_NAME);
|
||
let hinstance = GetModuleHandleW(PCWSTR::null()).unwrap_or_default();
|
||
let wc = WNDCLASSEXW {
|
||
cbSize: std::mem::size_of::<WNDCLASSEXW>() as u32,
|
||
style: CS_HREDRAW | CS_VREDRAW,
|
||
lpfnWndProc: Some(wnd_proc),
|
||
hInstance: HINSTANCE(hinstance.0),
|
||
hCursor: LoadCursorW(HINSTANCE::default(), IDC_SIZEALL).unwrap_or_default(),
|
||
hbrBackground: HBRUSH(std::ptr::null_mut()),
|
||
lpszClassName: PCWSTR::from_raw(class_w.as_ptr()),
|
||
..Default::default()
|
||
};
|
||
if RegisterClassExW(&wc) == 0 {
|
||
log::error!("bubble RegisterClassExW returned 0");
|
||
}
|
||
});
|
||
}
|
||
|
||
/// Create a bubble window. Returns the HWND. The caller (app::run) owns the
|
||
/// message-loop dispatch.
|
||
pub fn create(config: BubbleConfig) -> HWND {
|
||
register_class();
|
||
let initial_size_logical = config
|
||
.size_logical
|
||
.clamp(MIN_BUBBLE_SIZE, MAX_BUBBLE_SIZE);
|
||
let dpi_for_create = crate::os::dpi::for_system();
|
||
let width_px = scale_to_dpi(initial_size_logical, dpi_for_create);
|
||
let height_px = scale_to_dpi(bubble_height_logical(initial_size_logical), dpi_for_create);
|
||
let (x, y) = config
|
||
.position
|
||
.unwrap_or_else(|| default_position(width_px, height_px, config.model));
|
||
let hwnd = unsafe {
|
||
let class_w = wide_str(CLASS_NAME);
|
||
let title_w = wide_str("Claude Code Usage Bubble");
|
||
let hinstance = GetModuleHandleW(PCWSTR::null()).unwrap_or_default();
|
||
CreateWindowExW(
|
||
WS_EX_TOOLWINDOW | WS_EX_LAYERED | WS_EX_TOPMOST | WS_EX_NOACTIVATE,
|
||
PCWSTR::from_raw(class_w.as_ptr()),
|
||
PCWSTR::from_raw(title_w.as_ptr()),
|
||
WS_POPUP,
|
||
x,
|
||
y,
|
||
width_px,
|
||
height_px,
|
||
HWND::default(),
|
||
HMENU::default(),
|
||
hinstance,
|
||
None,
|
||
)
|
||
.unwrap_or_default()
|
||
};
|
||
|
||
if hwnd == HWND::default() {
|
||
log::error!("bubble CreateWindowExW failed");
|
||
return hwnd;
|
||
}
|
||
|
||
// Embed app icon in window non-client (mostly cosmetic; toolwindows
|
||
// don't show captions but the icon helps in dev tooling). The HICONs
|
||
// are extracted once at process startup and reused across every bubble
|
||
// create() so we don't leak a pair per toggle cycle.
|
||
let (large_icon, small_icon) = app_icons();
|
||
unsafe {
|
||
if !large_icon.is_invalid() {
|
||
let _ = SendMessageW(
|
||
hwnd,
|
||
WM_SETICON,
|
||
WPARAM(ICON_BIG as usize),
|
||
LPARAM(large_icon.0 as isize),
|
||
);
|
||
}
|
||
if !small_icon.is_invalid() {
|
||
let _ = SendMessageW(
|
||
hwnd,
|
||
WM_SETICON,
|
||
WPARAM(ICON_SMALL as usize),
|
||
LPARAM(small_icon.0 as isize),
|
||
);
|
||
}
|
||
}
|
||
|
||
let dpi = unsafe { GetDpiForWindow(hwnd).max(96) };
|
||
lock_bubbles().insert(
|
||
hwnd.0 as isize,
|
||
BubbleState {
|
||
model: config.model,
|
||
size_logical: initial_size_logical,
|
||
dpi,
|
||
session_pct: config.session_pct,
|
||
session_text: config.session_text,
|
||
weekly_pct: config.weekly_pct,
|
||
weekly_text: config.weekly_text,
|
||
is_dark: config.is_dark,
|
||
drag_start_pos: None,
|
||
hidden_by_fullscreen: false,
|
||
user_hidden: false,
|
||
pulse_phase: 0,
|
||
pulse_timer_armed: false,
|
||
},
|
||
);
|
||
|
||
log::info!(
|
||
"bubble create model={:?} pos=({x},{y}) size={width_px}x{height_px} dpi={dpi}",
|
||
config.model
|
||
);
|
||
|
||
// Defense in depth: settings::load already validates positions against
|
||
// currently-connected monitors, but a monitor unplug between load and
|
||
// create (or a partially-off-screen saved position) is still possible.
|
||
clamp_into_work_area(hwnd);
|
||
|
||
render(hwnd);
|
||
unsafe {
|
||
let _ = ShowWindow(hwnd, SW_SHOWNOACTIVATE);
|
||
// Periodic fullscreen-foreground check.
|
||
SetTimer(hwnd, TIMER_FULLSCREEN_CHECK, FULLSCREEN_POLL_MS, None);
|
||
}
|
||
|
||
hwnd
|
||
}
|
||
|
||
pub fn destroy(hwnd: HWND) {
|
||
unsafe {
|
||
let _ = KillTimer(hwnd, TIMER_FULLSCREEN_CHECK);
|
||
let _ = KillTimer(hwnd, TIMER_PULSE);
|
||
let _ = DestroyWindow(hwnd);
|
||
}
|
||
}
|
||
|
||
/// Extract the EXE's own icon pair once per process. Stored as raw pointer
|
||
/// values because `HICON` is `!Send`/`!Sync`; reconstituted for each caller.
|
||
/// The pair is intentionally never destroyed — Windows tears them down on
|
||
/// process exit, and one pair per process is bounded leak rather than the
|
||
/// O(bubble-toggles) leak we'd get from extracting per `create()`.
|
||
fn app_icons() -> (HICON, HICON) {
|
||
static ICONS: OnceLock<(isize, isize)> = OnceLock::new();
|
||
let (big, small) = *ICONS.get_or_init(|| unsafe {
|
||
let mut large = HICON::default();
|
||
let mut small = HICON::default();
|
||
let mut exe = [0u16; 260];
|
||
GetModuleFileNameW(HMODULE::default(), &mut exe);
|
||
let _ = ExtractIconExW(
|
||
PCWSTR::from_raw(exe.as_ptr()),
|
||
0,
|
||
Some(&mut large),
|
||
Some(&mut small),
|
||
1,
|
||
);
|
||
if large.is_invalid() && small.is_invalid() {
|
||
log::warn!("ExtractIconExW yielded null handles; bubbles will be iconless");
|
||
}
|
||
(large.0 as isize, small.0 as isize)
|
||
});
|
||
(HICON(big as *mut _), HICON(small as *mut _))
|
||
}
|
||
|
||
pub fn update_data(
|
||
hwnd: HWND,
|
||
session_pct: Option<f64>,
|
||
session_text: String,
|
||
weekly_pct: Option<f64>,
|
||
weekly_text: String,
|
||
) {
|
||
{
|
||
let mut bubbles = lock_bubbles();
|
||
let Some(b) = bubbles.get_mut(&(hwnd.0 as isize)) else {
|
||
return;
|
||
};
|
||
b.session_pct = session_pct;
|
||
b.session_text = session_text;
|
||
b.weekly_pct = weekly_pct;
|
||
b.weekly_text = weekly_text;
|
||
}
|
||
sync_pulse_timer(hwnd);
|
||
render(hwnd);
|
||
}
|
||
|
||
fn any_pct_in_alarm(b: &BubbleState) -> bool {
|
||
b.session_pct.is_some_and(|p| p >= 95.0) || b.weekly_pct.is_some_and(|p| p >= 95.0)
|
||
}
|
||
|
||
fn sync_pulse_timer(hwnd: HWND) {
|
||
let (should_be_armed, currently_armed) = {
|
||
let bubbles = lock_bubbles();
|
||
let Some(b) = bubbles.get(&(hwnd.0 as isize)) else {
|
||
return;
|
||
};
|
||
(any_pct_in_alarm(b), b.pulse_timer_armed)
|
||
};
|
||
if should_be_armed == currently_armed {
|
||
return;
|
||
}
|
||
unsafe {
|
||
if should_be_armed {
|
||
SetTimer(hwnd, TIMER_PULSE, PULSE_INTERVAL_MS, None);
|
||
} else {
|
||
let _ = KillTimer(hwnd, TIMER_PULSE);
|
||
}
|
||
}
|
||
if let Some(b) = lock_bubbles().get_mut(&(hwnd.0 as isize)) {
|
||
b.pulse_timer_armed = should_be_armed;
|
||
if !should_be_armed {
|
||
b.pulse_phase = 0;
|
||
}
|
||
}
|
||
}
|
||
|
||
pub fn update_dark_mode(hwnd: HWND, is_dark: bool) {
|
||
{
|
||
let mut bubbles = lock_bubbles();
|
||
let Some(b) = bubbles.get_mut(&(hwnd.0 as isize)) else {
|
||
return;
|
||
};
|
||
b.is_dark = is_dark;
|
||
}
|
||
render(hwnd);
|
||
}
|
||
|
||
pub fn set_user_visible(hwnd: HWND, visible: bool) {
|
||
{
|
||
let mut bubbles = lock_bubbles();
|
||
let Some(b) = bubbles.get_mut(&(hwnd.0 as isize)) else {
|
||
return;
|
||
};
|
||
b.user_hidden = !visible;
|
||
}
|
||
unsafe {
|
||
let cmd = if visible { SW_SHOWNOACTIVATE } else { SW_HIDE };
|
||
let _ = ShowWindow(hwnd, cmd);
|
||
}
|
||
// A layered window's composited surface is dropped while hidden, so
|
||
// ShowWindow(SW_SHOWNOACTIVATE) on its own renders blank until the next
|
||
// UpdateLayeredWindow. The cached BubbleState (pcts + texts) hasn't gone
|
||
// anywhere, so just re-paint from it so the bubble pops back with the
|
||
// last good data instead of empty placeholders.
|
||
if visible {
|
||
render(hwnd);
|
||
}
|
||
}
|
||
|
||
pub fn position(hwnd: HWND) -> Option<(i32, i32)> {
|
||
let mut r = RECT::default();
|
||
unsafe {
|
||
if GetWindowRect(hwnd, &mut r).is_err() {
|
||
return None;
|
||
}
|
||
}
|
||
Some((r.left, r.top))
|
||
}
|
||
|
||
pub fn model(hwnd: HWND) -> Option<ProviderId> {
|
||
lock_bubbles()
|
||
.get(&(hwnd.0 as isize))
|
||
.map(|b| b.model)
|
||
}
|
||
|
||
pub fn size_logical(hwnd: HWND) -> Option<i32> {
|
||
lock_bubbles()
|
||
.get(&(hwnd.0 as isize))
|
||
.map(|b| b.size_logical)
|
||
}
|
||
|
||
// ---------- State ----------
|
||
|
||
struct BubbleState {
|
||
model: ProviderId,
|
||
size_logical: i32,
|
||
dpi: u32,
|
||
session_pct: Option<f64>,
|
||
session_text: String,
|
||
weekly_pct: Option<f64>,
|
||
weekly_text: String,
|
||
is_dark: bool,
|
||
drag_start_pos: Option<(i32, i32)>,
|
||
hidden_by_fullscreen: bool,
|
||
user_hidden: bool,
|
||
/// Frame counter for the ≥95% pulse animation. Increments on each
|
||
/// TIMER_PULSE tick when at least one bar is in the alarm band.
|
||
pulse_phase: u32,
|
||
/// Whether TIMER_PULSE is currently armed for this bubble.
|
||
pulse_timer_armed: bool,
|
||
}
|
||
|
||
fn bubbles() -> &'static Mutex<HashMap<isize, BubbleState>> {
|
||
static BUBBLES: OnceLock<Mutex<HashMap<isize, BubbleState>>> = OnceLock::new();
|
||
BUBBLES.get_or_init(|| Mutex::new(HashMap::new()))
|
||
}
|
||
|
||
fn lock_bubbles() -> MutexGuard<'static, HashMap<isize, BubbleState>> {
|
||
bubbles().lock().expect("bubble state mutex poisoned")
|
||
}
|
||
|
||
// ---------- Window proc ----------
|
||
|
||
unsafe extern "system" fn wnd_proc(
|
||
hwnd: HWND,
|
||
msg: u32,
|
||
wparam: WPARAM,
|
||
lparam: LPARAM,
|
||
) -> LRESULT {
|
||
match msg {
|
||
WM_NCHITTEST => hit_test(hwnd, lparam),
|
||
WM_ENTERSIZEMOVE => {
|
||
let mut r = RECT::default();
|
||
let _ = GetWindowRect(hwnd, &mut r);
|
||
if let Some(b) = lock_bubbles().get_mut(&(hwnd.0 as isize)) {
|
||
b.drag_start_pos = Some((r.left, r.top));
|
||
}
|
||
LRESULT(0)
|
||
}
|
||
WM_EXITSIZEMOVE => {
|
||
// WM_NCLBUTTONUP isn't reliably delivered for HTCAPTION drags; instead
|
||
// we infer click-vs-drag from whether the window actually moved.
|
||
let start = {
|
||
let mut bubbles = lock_bubbles();
|
||
let start = bubbles
|
||
.get(&(hwnd.0 as isize))
|
||
.and_then(|b| b.drag_start_pos);
|
||
if let Some(b) = bubbles.get_mut(&(hwnd.0 as isize)) {
|
||
b.drag_start_pos = None;
|
||
}
|
||
start
|
||
};
|
||
let mut current = RECT::default();
|
||
let _ = GetWindowRect(hwnd, &mut current);
|
||
let moved = match start {
|
||
Some((sx, sy)) => (current.left - sx).abs() >= 3 || (current.top - sy).abs() >= 3,
|
||
None => false,
|
||
};
|
||
if moved {
|
||
snap_to_edge(hwnd);
|
||
if let Some(model) = model(hwnd) {
|
||
if let Some(pos) = position(hwnd) {
|
||
dispatch(|cb| (cb.on_moved)(model, pos));
|
||
}
|
||
}
|
||
} else if let Some(model) = model(hwnd) {
|
||
dispatch(|cb| (cb.on_click)(hwnd, model));
|
||
}
|
||
LRESULT(0)
|
||
}
|
||
WM_NCRBUTTONUP => {
|
||
if let Some(model) = model(hwnd) {
|
||
let pt = lparam_to_point(lparam);
|
||
dispatch(|cb| (cb.on_right_click)(hwnd, model, pt));
|
||
}
|
||
LRESULT(0)
|
||
}
|
||
WM_MOUSEWHEEL => {
|
||
let modifiers = (wparam.0 & 0xFFFF) as u32;
|
||
const MK_CONTROL: u32 = 0x0008;
|
||
if modifiers & MK_CONTROL != 0 {
|
||
let delta = ((wparam.0 >> 16) & 0xFFFF) as i16;
|
||
let step = if delta > 0 { RESIZE_STEP } else { -RESIZE_STEP };
|
||
resize_step(hwnd, step);
|
||
LRESULT(0)
|
||
} else {
|
||
DefWindowProcW(hwnd, msg, wparam, lparam)
|
||
}
|
||
}
|
||
WM_DPICHANGED => {
|
||
let new_dpi = ((wparam.0 >> 16) & 0xFFFF) as u32;
|
||
if let Some(b) = lock_bubbles().get_mut(&(hwnd.0 as isize)) {
|
||
b.dpi = new_dpi;
|
||
}
|
||
let rect_ptr = lparam.0 as *const RECT;
|
||
if !rect_ptr.is_null() {
|
||
let r = *rect_ptr;
|
||
let _ = SetWindowPos(
|
||
hwnd,
|
||
HWND::default(),
|
||
r.left,
|
||
r.top,
|
||
r.right - r.left,
|
||
r.bottom - r.top,
|
||
SWP_NOZORDER | SWP_NOACTIVATE,
|
||
);
|
||
}
|
||
render(hwnd);
|
||
LRESULT(0)
|
||
}
|
||
WM_TIMER => {
|
||
match wparam.0 {
|
||
w if w == TIMER_FULLSCREEN_CHECK => check_fullscreen(hwnd),
|
||
w if w == TIMER_PULSE => {
|
||
if let Some(b) = lock_bubbles().get_mut(&(hwnd.0 as isize)) {
|
||
b.pulse_phase = b.pulse_phase.wrapping_add(1);
|
||
}
|
||
render(hwnd);
|
||
}
|
||
_ => {}
|
||
}
|
||
LRESULT(0)
|
||
}
|
||
WM_COMMAND => {
|
||
dispatch(|cb| (cb.on_menu_command)(wparam.0 as u32, hwnd));
|
||
LRESULT(0)
|
||
}
|
||
WM_SETTINGCHANGE => {
|
||
// Taskbar move / auto-hide toggle / DPI change / theme toggle
|
||
// all post this. Re-clamp into the new work area (bubble must
|
||
// not end up hidden behind the new taskbar position) and ask
|
||
// the app to re-read the light/dark setting — Windows fires
|
||
// this message when the user flips the OS theme in Settings.
|
||
clamp_into_work_area(hwnd);
|
||
dispatch(|cb| (cb.on_settings_changed)());
|
||
LRESULT(0)
|
||
}
|
||
WM_DESTROY => {
|
||
lock_bubbles().remove(&(hwnd.0 as isize));
|
||
LRESULT(0)
|
||
}
|
||
_ => DefWindowProcW(hwnd, msg, wparam, lparam),
|
||
}
|
||
}
|
||
|
||
fn hit_test(hwnd: HWND, lparam: LPARAM) -> LRESULT {
|
||
let pt = lparam_to_point(lparam);
|
||
let mut r = RECT::default();
|
||
unsafe {
|
||
if GetWindowRect(hwnd, &mut r).is_err() {
|
||
return LRESULT(HTNOWHERE as isize);
|
||
}
|
||
}
|
||
let w = r.right - r.left;
|
||
let h = r.bottom - r.top;
|
||
let radius = corner_radius_px(h);
|
||
// Local coordinates relative to top-left of the bubble.
|
||
let lx = pt.x - r.left;
|
||
let ly = pt.y - r.top;
|
||
if point_in_rounded_rect(lx, ly, w, h, radius) {
|
||
LRESULT(HTCAPTION as isize)
|
||
} else {
|
||
LRESULT(HTTRANSPARENT as isize)
|
||
}
|
||
}
|
||
|
||
fn corner_radius_px(height_px: i32) -> i32 {
|
||
height_px / 2
|
||
}
|
||
|
||
fn point_in_rounded_rect(x: i32, y: i32, w: i32, h: i32, r: i32) -> bool {
|
||
if x < 0 || y < 0 || x >= w || y >= h {
|
||
return false;
|
||
}
|
||
// The straight horizontal and vertical strips are always inside; only the
|
||
// four corner squares need the circular falloff check.
|
||
let in_x_strip = x >= r && x < w - r;
|
||
let in_y_strip = y >= r && y < h - r;
|
||
if in_x_strip || in_y_strip {
|
||
return true;
|
||
}
|
||
let cx = if x < r { r } else { w - 1 - r };
|
||
let cy = if y < r { r } else { h - 1 - r };
|
||
let dx = x - cx;
|
||
let dy = y - cy;
|
||
dx * dx + dy * dy <= r * r
|
||
}
|
||
|
||
fn lparam_to_point(lparam: LPARAM) -> POINT {
|
||
let lo = (lparam.0 & 0xFFFF) as i16 as i32;
|
||
let hi = ((lparam.0 >> 16) & 0xFFFF) as i16 as i32;
|
||
POINT { x: lo, y: hi }
|
||
}
|
||
|
||
// ---------- Resize / snap ----------
|
||
|
||
fn resize_step(hwnd: HWND, delta: i32) {
|
||
let (new_logical, dpi) = {
|
||
let mut bubbles = lock_bubbles();
|
||
let Some(b) = bubbles.get_mut(&(hwnd.0 as isize)) else {
|
||
return;
|
||
};
|
||
let new_logical = (b.size_logical + delta).clamp(MIN_BUBBLE_SIZE, MAX_BUBBLE_SIZE);
|
||
if new_logical == b.size_logical {
|
||
return;
|
||
}
|
||
b.size_logical = new_logical;
|
||
(new_logical, b.dpi)
|
||
};
|
||
let width_px = scale_to_dpi(new_logical, dpi);
|
||
let height_px = scale_to_dpi(bubble_height_logical(new_logical), dpi);
|
||
let mut r = RECT::default();
|
||
unsafe {
|
||
let _ = GetWindowRect(hwnd, &mut r);
|
||
// Resize centered on existing center.
|
||
let cx = (r.left + r.right) / 2;
|
||
let cy = (r.top + r.bottom) / 2;
|
||
let new_x = cx - width_px / 2;
|
||
let new_y = cy - height_px / 2;
|
||
let _ = SetWindowPos(
|
||
hwnd,
|
||
HWND::default(),
|
||
new_x,
|
||
new_y,
|
||
width_px,
|
||
height_px,
|
||
SWP_NOZORDER | SWP_NOACTIVATE,
|
||
);
|
||
}
|
||
render(hwnd);
|
||
if let Some(m) = model(hwnd) {
|
||
dispatch(|cb| (cb.on_resized)(m, new_logical));
|
||
}
|
||
}
|
||
|
||
fn snap_to_edge(hwnd: HWND) {
|
||
let dpi = lock_bubbles()
|
||
.get(&(hwnd.0 as isize))
|
||
.map(|b| b.dpi)
|
||
.unwrap_or(96);
|
||
let edge_zone = scale_to_dpi(SNAP_ZONE_LOGICAL, dpi);
|
||
let corner_zone = scale_to_dpi(CORNER_SNAP_ZONE_LOGICAL, dpi);
|
||
let corner_inset = scale_to_dpi(CORNER_INSET_LOGICAL, dpi);
|
||
let taskbar_gap = scale_to_dpi(TASKBAR_GAP_LOGICAL, dpi);
|
||
let peer_tolerance = scale_to_dpi(PEER_ALIGN_TOLERANCE_LOGICAL, dpi);
|
||
|
||
let mut r = RECT::default();
|
||
let monitor;
|
||
unsafe {
|
||
if GetWindowRect(hwnd, &mut r).is_err() {
|
||
return;
|
||
}
|
||
monitor = MonitorFromWindow(hwnd, MONITOR_DEFAULTTONEAREST);
|
||
}
|
||
if monitor.is_invalid() {
|
||
return;
|
||
}
|
||
let mut info = MONITORINFO {
|
||
cbSize: std::mem::size_of::<MONITORINFO>() as u32,
|
||
..Default::default()
|
||
};
|
||
unsafe {
|
||
if !GetMonitorInfoW(monitor, &mut info).as_bool() {
|
||
return;
|
||
}
|
||
}
|
||
let wa = info.rcWork;
|
||
let w = r.right - r.left;
|
||
let h = r.bottom - r.top;
|
||
let mut nx = r.left;
|
||
let mut ny = r.top;
|
||
|
||
// 1. Corner snap — if the bubble's nearest-corner distance is under the
|
||
// 32-px corner zone, slam it into the corner with the 12-px inset.
|
||
let snapped_to_corner = try_corner_snap(&mut nx, &mut ny, &wa, w, h, corner_zone, corner_inset);
|
||
|
||
if !snapped_to_corner {
|
||
// 2. Edge snap (existing behavior) — also handles taskbar-adjacency
|
||
// when the taskbar steals from the work area on the same edge.
|
||
let taskbar = read_taskbar();
|
||
snap_to_work_area_edges(&mut nx, &mut ny, &wa, w, h, edge_zone);
|
||
if let Some(tb) = taskbar {
|
||
snap_alongside_taskbar(&mut nx, &mut ny, &tb, &wa, w, h, edge_zone, taskbar_gap);
|
||
}
|
||
|
||
// 3. Peer vertical alignment — when the other bubble is within ±8 px
|
||
// on Y, snap the dragged bubble to share its baseline.
|
||
align_with_peer(hwnd, &mut ny, peer_tolerance);
|
||
}
|
||
|
||
// Clamp into the work area in any case (so the bubble can't be lost off-screen).
|
||
nx = nx.clamp(wa.left, (wa.right - w).max(wa.left));
|
||
ny = ny.clamp(wa.top, (wa.bottom - h).max(wa.top));
|
||
|
||
if nx != r.left || ny != r.top {
|
||
unsafe {
|
||
let _ = SetWindowPos(
|
||
hwnd,
|
||
HWND::default(),
|
||
nx,
|
||
ny,
|
||
0,
|
||
0,
|
||
SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE,
|
||
);
|
||
}
|
||
}
|
||
}
|
||
|
||
fn try_corner_snap(
|
||
nx: &mut i32,
|
||
ny: &mut i32,
|
||
wa: &RECT,
|
||
w: i32,
|
||
h: i32,
|
||
zone: i32,
|
||
inset: i32,
|
||
) -> bool {
|
||
// Distance from each work-area corner to the bubble's nearest corner.
|
||
let tl = (*nx - wa.left).abs() + (*ny - wa.top).abs();
|
||
let tr = (wa.right - (*nx + w)).abs() + (*ny - wa.top).abs();
|
||
let bl = (*nx - wa.left).abs() + (wa.bottom - (*ny + h)).abs();
|
||
let br = (wa.right - (*nx + w)).abs() + (wa.bottom - (*ny + h)).abs();
|
||
let min = tl.min(tr).min(bl).min(br);
|
||
if min > zone * 2 {
|
||
return false;
|
||
}
|
||
if min == tl {
|
||
*nx = wa.left + inset;
|
||
*ny = wa.top + inset;
|
||
} else if min == tr {
|
||
*nx = wa.right - inset - w;
|
||
*ny = wa.top + inset;
|
||
} else if min == bl {
|
||
*nx = wa.left + inset;
|
||
*ny = wa.bottom - inset - h;
|
||
} else {
|
||
*nx = wa.right - inset - w;
|
||
*ny = wa.bottom - inset - h;
|
||
}
|
||
true
|
||
}
|
||
|
||
fn snap_to_work_area_edges(nx: &mut i32, ny: &mut i32, wa: &RECT, w: i32, h: i32, zone: i32) {
|
||
if (*nx - wa.left).abs() < zone {
|
||
*nx = wa.left;
|
||
} else if (wa.right - (*nx + w)).abs() < zone {
|
||
*nx = wa.right - w;
|
||
}
|
||
if (*ny - wa.top).abs() < zone {
|
||
*ny = wa.top;
|
||
} else if (wa.bottom - (*ny + h)).abs() < zone {
|
||
*ny = wa.bottom - h;
|
||
}
|
||
}
|
||
|
||
struct Taskbar {
|
||
rect: RECT,
|
||
edge: u32,
|
||
}
|
||
|
||
fn read_taskbar() -> Option<Taskbar> {
|
||
let mut abd = APPBARDATA {
|
||
cbSize: std::mem::size_of::<APPBARDATA>() as u32,
|
||
..Default::default()
|
||
};
|
||
let res = unsafe { SHAppBarMessage(ABM_GETTASKBARPOS, &mut abd) };
|
||
if res == 0 {
|
||
return None;
|
||
}
|
||
Some(Taskbar {
|
||
rect: abd.rc,
|
||
edge: abd.uEdge,
|
||
})
|
||
}
|
||
|
||
fn snap_alongside_taskbar(
|
||
nx: &mut i32,
|
||
ny: &mut i32,
|
||
tb: &Taskbar,
|
||
wa: &RECT,
|
||
w: i32,
|
||
h: i32,
|
||
zone: i32,
|
||
gap: i32,
|
||
) {
|
||
// Only snap on the taskbar's docked edge. The bubble docks against the
|
||
// inner face of the taskbar with a 4-px gap so it visually leans on it.
|
||
match tb.edge {
|
||
e if e == ABE_BOTTOM => {
|
||
let target = tb.rect.top - gap - h;
|
||
if (*ny - target).abs() < zone {
|
||
*ny = target.max(wa.top);
|
||
}
|
||
}
|
||
e if e == ABE_TOP => {
|
||
let target = tb.rect.bottom + gap;
|
||
if (*ny - target).abs() < zone {
|
||
*ny = target.min(wa.bottom - h);
|
||
}
|
||
}
|
||
e if e == ABE_LEFT => {
|
||
let target = tb.rect.right + gap;
|
||
if (*nx - target).abs() < zone {
|
||
*nx = target.min(wa.right - w);
|
||
}
|
||
}
|
||
e if e == ABE_RIGHT => {
|
||
let target = tb.rect.left - gap - w;
|
||
if (*nx - target).abs() < zone {
|
||
*nx = target.max(wa.left);
|
||
}
|
||
}
|
||
_ => {}
|
||
}
|
||
}
|
||
|
||
fn clamp_into_work_area(hwnd: HWND) {
|
||
let mut r = RECT::default();
|
||
let monitor;
|
||
unsafe {
|
||
if GetWindowRect(hwnd, &mut r).is_err() {
|
||
return;
|
||
}
|
||
monitor = MonitorFromWindow(hwnd, MONITOR_DEFAULTTONEAREST);
|
||
}
|
||
if monitor.is_invalid() {
|
||
return;
|
||
}
|
||
let mut info = MONITORINFO {
|
||
cbSize: std::mem::size_of::<MONITORINFO>() as u32,
|
||
..Default::default()
|
||
};
|
||
unsafe {
|
||
if !GetMonitorInfoW(monitor, &mut info).as_bool() {
|
||
return;
|
||
}
|
||
}
|
||
let wa = info.rcWork;
|
||
let w = r.right - r.left;
|
||
let h = r.bottom - r.top;
|
||
let nx = r.left.clamp(wa.left, (wa.right - w).max(wa.left));
|
||
let mut ny = r.top.clamp(wa.top, (wa.bottom - h).max(wa.top));
|
||
|
||
// When both bubbles get clamped to the same bottom-right corner (e.g.,
|
||
// saved positions were on a disconnected monitor and the validator missed
|
||
// them), keep the Codex-above-Claude stagger that `default_position` uses
|
||
// so they don't visually stack.
|
||
let is_codex = lock_bubbles()
|
||
.get(&(hwnd.0 as isize))
|
||
.is_some_and(|b| matches!(b.model, ProviderId::ChatGpt));
|
||
if is_codex && nx == wa.right - w && ny == wa.bottom - h {
|
||
const STAGGER_GAP: i32 = 24;
|
||
ny = (ny - h - STAGGER_GAP).max(wa.top);
|
||
}
|
||
|
||
if nx != r.left || ny != r.top {
|
||
log::warn!(
|
||
"clamp_into_work_area moved bubble from ({}, {}) to ({nx}, {ny})",
|
||
r.left,
|
||
r.top
|
||
);
|
||
unsafe {
|
||
let _ = SetWindowPos(
|
||
hwnd,
|
||
HWND::default(),
|
||
nx,
|
||
ny,
|
||
0,
|
||
0,
|
||
SWP_NOSIZE | SWP_NOZORDER | SWP_NOACTIVATE,
|
||
);
|
||
}
|
||
}
|
||
}
|
||
|
||
fn align_with_peer(this_hwnd: HWND, ny: &mut i32, tolerance: i32) {
|
||
let bubbles = lock_bubbles();
|
||
for (id, _) in bubbles.iter() {
|
||
if *id == this_hwnd.0 as isize {
|
||
continue;
|
||
}
|
||
let peer_hwnd = HWND(*id as *mut c_void);
|
||
let mut pr = RECT::default();
|
||
unsafe {
|
||
if GetWindowRect(peer_hwnd, &mut pr).is_err() {
|
||
continue;
|
||
}
|
||
}
|
||
if (*ny - pr.top).abs() <= tolerance {
|
||
*ny = pr.top;
|
||
return;
|
||
}
|
||
}
|
||
}
|
||
|
||
// ---------- Fullscreen detection ----------
|
||
|
||
fn check_fullscreen(bubble_hwnd: HWND) {
|
||
let fg = unsafe { GetForegroundWindow() };
|
||
if fg == HWND::default() || fg == bubble_hwnd {
|
||
return;
|
||
}
|
||
let mut fr = RECT::default();
|
||
unsafe {
|
||
if GetWindowRect(fg, &mut fr).is_err() {
|
||
return;
|
||
}
|
||
}
|
||
let monitor = unsafe { MonitorFromWindow(fg, MONITOR_DEFAULTTONEAREST) };
|
||
if monitor.is_invalid() {
|
||
return;
|
||
}
|
||
let mut info = MONITORINFO {
|
||
cbSize: std::mem::size_of::<MONITORINFO>() as u32,
|
||
..Default::default()
|
||
};
|
||
let ok = unsafe { GetMonitorInfoW(monitor, &mut info).as_bool() };
|
||
if !ok {
|
||
return;
|
||
}
|
||
let mr = info.rcMonitor;
|
||
let is_fullscreen =
|
||
fr.left <= mr.left && fr.top <= mr.top && fr.right >= mr.right && fr.bottom >= mr.bottom;
|
||
|
||
let (was_hidden_by_fs, user_hidden) = {
|
||
let bubbles = lock_bubbles();
|
||
let Some(b) = bubbles.get(&(bubble_hwnd.0 as isize)) else {
|
||
return;
|
||
};
|
||
(b.hidden_by_fullscreen, b.user_hidden)
|
||
};
|
||
|
||
if is_fullscreen && !was_hidden_by_fs {
|
||
unsafe {
|
||
let _ = ShowWindow(bubble_hwnd, SW_HIDE);
|
||
}
|
||
if let Some(b) = lock_bubbles().get_mut(&(bubble_hwnd.0 as isize)) {
|
||
b.hidden_by_fullscreen = true;
|
||
}
|
||
} else if !is_fullscreen && was_hidden_by_fs {
|
||
if !user_hidden {
|
||
unsafe {
|
||
let _ = ShowWindow(bubble_hwnd, SW_SHOWNOACTIVATE);
|
||
}
|
||
// Re-paint so the layered surface has the cached data again
|
||
// (see comment in `set_user_visible`).
|
||
render(bubble_hwnd);
|
||
}
|
||
if let Some(b) = lock_bubbles().get_mut(&(bubble_hwnd.0 as isize)) {
|
||
b.hidden_by_fullscreen = false;
|
||
}
|
||
}
|
||
}
|
||
|
||
// ---------- Painting ----------
|
||
|
||
// Sized for the widest countdown across all shipped locales. Korean
|
||
// "999시간" (3 digits + 2 CJK chars for the hour suffix) is the current
|
||
// worst case; ASCII-only "999d" was too narrow and let CJK text spill
|
||
// out of the column. Update this when adding a locale with a longer
|
||
// suffix.
|
||
const COUNTDOWN_TEMPLATE: &str = "999시간";
|
||
|
||
/// Geometry for the bubble's "circle head + pill tail" shape, in DPI-scaled pixels.
|
||
///
|
||
/// The outline is a stadium (rounded rect with `corner_radius = canvas_h / 2`).
|
||
/// The left `head_diameter × canvas_h` square holds the 5h progress ring + big
|
||
/// percent glyph. The rest is the tail: 7d label, thin bar, countdown.
|
||
struct BubbleLayout {
|
||
canvas_w: i32,
|
||
canvas_h: i32,
|
||
corner_radius: i32,
|
||
head_diameter: i32,
|
||
ring_cx: f32,
|
||
ring_cy: f32,
|
||
ring_radius: f32,
|
||
ring_stroke_w: f32,
|
||
head_label_rect: RECT,
|
||
head_pct_rect: RECT,
|
||
tail_label_rect: RECT,
|
||
tail_bar_rect: RECT,
|
||
tail_countdown_rect: RECT,
|
||
big_font_px: i32,
|
||
small_font_px: i32,
|
||
main_font_px: i32,
|
||
}
|
||
|
||
fn compute_bubble_layout(size_logical: i32, dpi: u32, mem_dc: HDC) -> BubbleLayout {
|
||
let width_px = scale_to_dpi(size_logical, dpi);
|
||
let height_px = scale_to_dpi(bubble_height_logical(size_logical), dpi);
|
||
let head_diameter = height_px;
|
||
|
||
let head_pad = scale_to_dpi(6, dpi);
|
||
let ring_stroke_w = scale_to_dpi(3, dpi).max(2) as f32;
|
||
let ring_cx = (head_diameter as f32) / 2.0;
|
||
let ring_cy = (height_px as f32) / 2.0;
|
||
// Ring centerline: midway between outer and inner edge, then keep stroke
|
||
// inside the head padding. ring_radius is the centerline radius.
|
||
let ring_outer = (head_diameter as f32) / 2.0 - (head_pad as f32);
|
||
let ring_radius = ring_outer - ring_stroke_w / 2.0;
|
||
|
||
let big_font_px = (head_diameter * 35 / 100).max(scale_to_dpi(12, dpi));
|
||
let small_font_px = ((big_font_px * 45) / 100).max(scale_to_dpi(8, dpi));
|
||
let main_font_px = small_font_px;
|
||
|
||
let head_label_h = small_font_px + scale_to_dpi(2, dpi);
|
||
let head_pct_h = big_font_px + scale_to_dpi(2, dpi);
|
||
let head_total_h = head_label_h + head_pct_h;
|
||
let head_text_top = (height_px - head_total_h) / 2;
|
||
let head_label_rect = RECT {
|
||
left: scale_to_dpi(4, dpi),
|
||
top: head_text_top,
|
||
right: head_diameter - scale_to_dpi(4, dpi),
|
||
bottom: head_text_top + head_label_h,
|
||
};
|
||
let head_pct_rect = RECT {
|
||
left: scale_to_dpi(4, dpi),
|
||
top: head_text_top + head_label_h,
|
||
right: head_diameter - scale_to_dpi(4, dpi),
|
||
bottom: head_text_top + head_total_h,
|
||
};
|
||
|
||
let tail_left = head_diameter;
|
||
let tail_right = width_px - scale_to_dpi(12, dpi);
|
||
let pad = scale_to_dpi(6, dpi);
|
||
|
||
let countdown_w = measure_text_w(mem_dc, COUNTDOWN_TEMPLATE, main_font_px);
|
||
let label_w = measure_text_w(mem_dc, "7d", small_font_px);
|
||
|
||
let tail_label_left = tail_left + pad;
|
||
let tail_label_right = tail_label_left + label_w;
|
||
let tail_countdown_right = tail_right;
|
||
let tail_countdown_left = tail_countdown_right - countdown_w;
|
||
let tail_bar_left = tail_label_right + pad;
|
||
let tail_bar_right =
|
||
(tail_countdown_left - pad).max(tail_bar_left + scale_to_dpi(20, dpi));
|
||
let tail_bar_h = scale_to_dpi(6, dpi);
|
||
let tail_bar_top = (height_px - tail_bar_h) / 2;
|
||
|
||
BubbleLayout {
|
||
canvas_w: width_px,
|
||
canvas_h: height_px,
|
||
corner_radius: height_px / 2,
|
||
head_diameter,
|
||
ring_cx,
|
||
ring_cy,
|
||
ring_radius,
|
||
ring_stroke_w,
|
||
head_label_rect,
|
||
head_pct_rect,
|
||
tail_label_rect: RECT {
|
||
left: tail_label_left,
|
||
top: 0,
|
||
right: tail_label_right,
|
||
bottom: height_px,
|
||
},
|
||
tail_bar_rect: RECT {
|
||
left: tail_bar_left,
|
||
top: tail_bar_top,
|
||
right: tail_bar_right,
|
||
bottom: tail_bar_top + tail_bar_h,
|
||
},
|
||
tail_countdown_rect: RECT {
|
||
left: tail_countdown_left,
|
||
top: 0,
|
||
right: tail_countdown_right,
|
||
bottom: height_px,
|
||
},
|
||
big_font_px,
|
||
small_font_px,
|
||
main_font_px,
|
||
}
|
||
}
|
||
|
||
/// Render the bubble's shape into a fresh tiny-skia `Pixmap`. The Pixmap is
|
||
/// premultiplied RGBA at one byte per channel — the caller copies it into the
|
||
/// GDI DIB section, then GDI text is overlaid on top.
|
||
fn paint_bubble_pixmap(layout: &BubbleLayout, inputs: &PaintInputs) -> Option<Pixmap> {
|
||
let mut pixmap = Pixmap::new(layout.canvas_w as u32, layout.canvas_h as u32)?;
|
||
pixmap.fill(tiny_skia::Color::TRANSPARENT);
|
||
|
||
let bg = if inputs.is_dark {
|
||
Color::from_hex("#1F1F1F")
|
||
} else {
|
||
Color::from_hex("#F3F3F3")
|
||
};
|
||
let track = if inputs.is_dark {
|
||
Color::from_hex("#3A3A3A")
|
||
} else {
|
||
Color::from_hex("#D6D6D6")
|
||
};
|
||
|
||
// ---- Stadium background ----
|
||
{
|
||
let mut paint = Paint::default();
|
||
paint.set_color(rgb_to_skia(bg));
|
||
paint.anti_alias = true;
|
||
let r = (layout.canvas_h as f32) / 2.0;
|
||
let w = layout.canvas_w as f32;
|
||
let h = layout.canvas_h as f32;
|
||
|
||
// Two end-cap circles + middle rect. Overlap is fine — same color.
|
||
let mut pb = PathBuilder::new();
|
||
pb.push_circle(r, r, r);
|
||
pb.push_circle(w - r, r, r);
|
||
if let Some(p) = pb.finish() {
|
||
pixmap.fill_path(&p, &paint, FillRule::Winding, Transform::identity(), None);
|
||
}
|
||
if let Some(rect) = Rect::from_xywh(r, 0.0, (w - 2.0 * r).max(0.0), h) {
|
||
pixmap.fill_rect(rect, &paint, Transform::identity(), None);
|
||
}
|
||
}
|
||
|
||
// ---- Ring (5h) ----
|
||
{
|
||
// Track: full circle in muted color.
|
||
let mut paint = Paint::default();
|
||
paint.set_color(rgb_to_skia(track));
|
||
paint.anti_alias = true;
|
||
let mut stroke = Stroke::default();
|
||
stroke.width = layout.ring_stroke_w;
|
||
let mut pb = PathBuilder::new();
|
||
pb.push_circle(layout.ring_cx, layout.ring_cy, layout.ring_radius);
|
||
if let Some(p) = pb.finish() {
|
||
pixmap.stroke_path(&p, &paint, &stroke, Transform::identity(), None);
|
||
}
|
||
|
||
// Active sweep arc.
|
||
if let Some(pct) = inputs.session_pct {
|
||
let sweep = (pct.clamp(0.0, 100.0) / 100.0) as f32;
|
||
if sweep > 0.0 {
|
||
let mut color = crate::usage_color::bar_fill_color(inputs.model, inputs.is_dark, pct);
|
||
if pct >= 95.0 {
|
||
let t = pulse_triangle(inputs.pulse_phase);
|
||
color = brighten(color, t);
|
||
}
|
||
let mut paint = Paint::default();
|
||
paint.set_color(rgb_to_skia(color));
|
||
paint.anti_alias = true;
|
||
let mut stroke = Stroke::default();
|
||
stroke.width = layout.ring_stroke_w;
|
||
stroke.line_cap = LineCap::Round;
|
||
if let Some(path) =
|
||
build_arc(layout.ring_cx, layout.ring_cy, layout.ring_radius, sweep)
|
||
{
|
||
pixmap.stroke_path(&path, &paint, &stroke, Transform::identity(), None);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
// ---- Tail bar (7d) ----
|
||
{
|
||
let bar_x = layout.tail_bar_rect.left as f32;
|
||
let bar_y = layout.tail_bar_rect.top as f32;
|
||
let bar_w = (layout.tail_bar_rect.right - layout.tail_bar_rect.left) as f32;
|
||
let bar_h = (layout.tail_bar_rect.bottom - layout.tail_bar_rect.top) as f32;
|
||
let cap = bar_h * 0.5;
|
||
if bar_w > 0.0 && bar_h > 0.0 {
|
||
paint_pill(&mut pixmap, bar_x, bar_y, bar_w, bar_h, cap, track);
|
||
if let Some(pct) = inputs.weekly_pct {
|
||
let frac = (pct.clamp(0.0, 100.0) / 100.0) as f32;
|
||
let fill_w = bar_w * frac;
|
||
if fill_w > 0.0 {
|
||
let mut color =
|
||
crate::usage_color::bar_fill_color(inputs.model, inputs.is_dark, pct);
|
||
if pct >= 95.0 {
|
||
let t = pulse_triangle(inputs.pulse_phase);
|
||
color = brighten(color, t);
|
||
}
|
||
let clipped_w = fill_w.min(bar_w);
|
||
paint_pill(&mut pixmap, bar_x, bar_y, clipped_w, bar_h, cap, color);
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
Some(pixmap)
|
||
}
|
||
|
||
/// Fill a horizontal pill at `(x, y, w, h)` with circular end caps of radius
|
||
/// `cap`. Used for both the track and the fill of the tail's 7d bar.
|
||
fn paint_pill(pixmap: &mut Pixmap, x: f32, y: f32, w: f32, h: f32, cap: f32, color: Color) {
|
||
let mut paint = Paint::default();
|
||
paint.set_color(rgb_to_skia(color));
|
||
paint.anti_alias = true;
|
||
let mut pb = PathBuilder::new();
|
||
pb.push_circle(x + cap, y + h * 0.5, cap);
|
||
pb.push_circle(x + w - cap, y + h * 0.5, cap);
|
||
if let Some(p) = pb.finish() {
|
||
pixmap.fill_path(&p, &paint, FillRule::Winding, Transform::identity(), None);
|
||
}
|
||
if let Some(rect) = Rect::from_xywh(x + cap, y, (w - 2.0 * cap).max(0.0), h) {
|
||
pixmap.fill_rect(rect, &paint, Transform::identity(), None);
|
||
}
|
||
}
|
||
|
||
fn rgb_to_skia(c: Color) -> tiny_skia::Color {
|
||
tiny_skia::Color::from_rgba8(c.r, c.g, c.b, 0xFF)
|
||
}
|
||
|
||
/// Build a clockwise arc path starting at 12 o'clock, sweeping `sweep_fraction`
|
||
/// of a full turn. Sampled — tiny-skia 0.11 lacks a direct arc primitive.
|
||
fn build_arc(cx: f32, cy: f32, radius: f32, sweep_fraction: f32) -> Option<tiny_skia::Path> {
|
||
let segments = ((sweep_fraction * 64.0).ceil() as usize).max(1);
|
||
let mut pb = PathBuilder::new();
|
||
let start_angle: f32 = -std::f32::consts::FRAC_PI_2;
|
||
let total = sweep_fraction * std::f32::consts::TAU;
|
||
for i in 0..=segments {
|
||
let t = i as f32 / segments as f32;
|
||
let a = start_angle + t * total;
|
||
let x = cx + a.cos() * radius;
|
||
let y = cy + a.sin() * radius;
|
||
if i == 0 {
|
||
pb.move_to(x, y);
|
||
} else {
|
||
pb.line_to(x, y);
|
||
}
|
||
}
|
||
pb.finish()
|
||
}
|
||
|
||
/// Copy a premultiplied-RGBA `Pixmap` into the 32bpp BI_RGB DIB the bubble
|
||
/// uses for `UpdateLayeredWindow`. The DIB stores BGRA bytes (little-endian
|
||
/// `0xAARRGGBB` when read as u32); tiny-skia's premultiplied alpha is exactly
|
||
/// the format `AC_SRC_ALPHA` expects.
|
||
fn copy_pixmap_to_dib(pixmap: &Pixmap, dst: &mut [u32]) {
|
||
let src = pixmap.data();
|
||
let pixel_count = (pixmap.width() * pixmap.height()) as usize;
|
||
for i in 0..pixel_count {
|
||
let r = src[i * 4];
|
||
let g = src[i * 4 + 1];
|
||
let b = src[i * 4 + 2];
|
||
let a = src[i * 4 + 3];
|
||
dst[i] = ((a as u32) << 24) | ((r as u32) << 16) | ((g as u32) << 8) | (b as u32);
|
||
}
|
||
}
|
||
|
||
fn measure_text_w(hdc: HDC, text: &str, font_height_px: i32) -> i32 {
|
||
use windows::Win32::Foundation::SIZE;
|
||
let font_name = wide_str("Segoe UI");
|
||
let mut w: Vec<u16> = text.encode_utf16().collect();
|
||
unsafe {
|
||
let font = CreateFontW(
|
||
-font_height_px,
|
||
0,
|
||
0,
|
||
0,
|
||
FW_NORMAL.0 as i32,
|
||
0,
|
||
0,
|
||
0,
|
||
DEFAULT_CHARSET.0 as u32,
|
||
OUT_DEFAULT_PRECIS.0 as u32,
|
||
CLIP_DEFAULT_PRECIS.0 as u32,
|
||
CLEARTYPE_QUALITY.0 as u32,
|
||
(FF_SWISS.0 | DEFAULT_PITCH.0) as u32,
|
||
PCWSTR::from_raw(font_name.as_ptr()),
|
||
);
|
||
let old = SelectObject(hdc, font);
|
||
let mut size = SIZE::default();
|
||
let _ = GetTextExtentPoint32W(hdc, &mut w, &mut size);
|
||
SelectObject(hdc, old);
|
||
let _ = DeleteObject(font);
|
||
size.cx
|
||
}
|
||
}
|
||
|
||
struct PaintInputs {
|
||
model: ProviderId,
|
||
session_pct: Option<f64>,
|
||
session_text: String,
|
||
weekly_pct: Option<f64>,
|
||
weekly_text: String,
|
||
is_dark: bool,
|
||
pulse_phase: u32,
|
||
}
|
||
|
||
fn render(hwnd: HWND) {
|
||
let (size_logical, dpi, inputs) = {
|
||
let bubbles = lock_bubbles();
|
||
let Some(b) = bubbles.get(&(hwnd.0 as isize)) else {
|
||
return;
|
||
};
|
||
(
|
||
b.size_logical,
|
||
b.dpi,
|
||
PaintInputs {
|
||
model: b.model,
|
||
session_pct: b.session_pct,
|
||
session_text: b.session_text.clone(),
|
||
weekly_pct: b.weekly_pct,
|
||
weekly_text: b.weekly_text.clone(),
|
||
is_dark: b.is_dark,
|
||
pulse_phase: b.pulse_phase,
|
||
},
|
||
)
|
||
};
|
||
|
||
unsafe {
|
||
let screen_dc = GetDC(hwnd);
|
||
if screen_dc.is_invalid() {
|
||
return;
|
||
}
|
||
let mem_dc = CreateCompatibleDC(screen_dc);
|
||
if mem_dc.is_invalid() {
|
||
ReleaseDC(hwnd, screen_dc);
|
||
return;
|
||
}
|
||
let layout = compute_bubble_layout(size_logical, dpi, mem_dc);
|
||
|
||
let bmi = BITMAPINFO {
|
||
bmiHeader: BITMAPINFOHEADER {
|
||
biSize: std::mem::size_of::<BITMAPINFOHEADER>() as u32,
|
||
biWidth: layout.canvas_w,
|
||
biHeight: -layout.canvas_h,
|
||
biPlanes: 1,
|
||
biBitCount: 32,
|
||
biCompression: 0,
|
||
..Default::default()
|
||
},
|
||
..Default::default()
|
||
};
|
||
let mut bits: *mut c_void = std::ptr::null_mut();
|
||
let dib = CreateDIBSection(mem_dc, &bmi, DIB_RGB_COLORS, &mut bits, None, 0)
|
||
.unwrap_or_default();
|
||
if dib.is_invalid() || bits.is_null() {
|
||
let _ = DeleteDC(mem_dc);
|
||
ReleaseDC(hwnd, screen_dc);
|
||
return;
|
||
}
|
||
let old_bmp = SelectObject(mem_dc, dib);
|
||
|
||
let pixel_count = (layout.canvas_w * layout.canvas_h) as usize;
|
||
let pixels = std::slice::from_raw_parts_mut(bits as *mut u32, pixel_count);
|
||
|
||
// Paint shape via tiny-skia (AA), then copy into the DIB. GDI text
|
||
// overlays on top of the resulting bitmap.
|
||
if let Some(pixmap) = paint_bubble_pixmap(&layout, &inputs) {
|
||
copy_pixmap_to_dib(&pixmap, pixels);
|
||
} else {
|
||
pixels.fill(0);
|
||
}
|
||
paint_bubble_text(mem_dc, &layout, &inputs);
|
||
|
||
let mut wr = RECT::default();
|
||
let _ = GetWindowRect(hwnd, &mut wr);
|
||
let pt_dst = POINT {
|
||
x: wr.left,
|
||
y: wr.top,
|
||
};
|
||
let pt_src = POINT { x: 0, y: 0 };
|
||
let sz = SIZE {
|
||
cx: layout.canvas_w,
|
||
cy: layout.canvas_h,
|
||
};
|
||
let blend = BLENDFUNCTION {
|
||
BlendOp: 0,
|
||
BlendFlags: 0,
|
||
SourceConstantAlpha: 255,
|
||
AlphaFormat: 1, // AC_SRC_ALPHA
|
||
};
|
||
let _ = UpdateLayeredWindow(
|
||
hwnd,
|
||
screen_dc,
|
||
Some(&pt_dst),
|
||
Some(&sz),
|
||
mem_dc,
|
||
Some(&pt_src),
|
||
COLORREF(0),
|
||
Some(&blend),
|
||
ULW_ALPHA,
|
||
);
|
||
|
||
SelectObject(mem_dc, old_bmp);
|
||
let _ = DeleteObject(dib);
|
||
let _ = DeleteDC(mem_dc);
|
||
ReleaseDC(hwnd, screen_dc);
|
||
}
|
||
}
|
||
|
||
/// Triangle wave in [0, 1] with period 24 ticks. 0 at phase=0,12; 1 at phase=6,18.
|
||
fn pulse_triangle(phase: u32) -> f64 {
|
||
let p = (phase % 24) as i32;
|
||
let dist = (p - 12).abs(); // 0..12
|
||
1.0 - (dist as f64 / 12.0)
|
||
}
|
||
|
||
/// Linearly brighten `c` toward white by `t` in [0, 1].
|
||
fn brighten(c: Color, t: f64) -> Color {
|
||
// Map t to a smaller brightness delta — the pulse should be a subtle nudge.
|
||
let t = t.clamp(0.0, 1.0) * 0.30;
|
||
Color::new(
|
||
((c.r as f64) + (255.0 - c.r as f64) * t).round() as u8,
|
||
((c.g as f64) + (255.0 - c.g as f64) * t).round() as u8,
|
||
((c.b as f64) + (255.0 - c.b as f64) * t).round() as u8,
|
||
)
|
||
}
|
||
|
||
/// Paint the new bubble's text overlay via GDI: small "5h" label + big "%"
|
||
/// glyph in the head, small "7d" label + countdown on the tail.
|
||
fn paint_bubble_text(hdc: HDC, layout: &BubbleLayout, inputs: &PaintInputs) {
|
||
let text_color = if inputs.is_dark {
|
||
Color::from_hex("#EAEAEA")
|
||
} else {
|
||
Color::from_hex("#1F1F1F")
|
||
};
|
||
let muted_color = if inputs.is_dark {
|
||
Color::from_hex("#888888")
|
||
} else {
|
||
Color::from_hex("#6E6E6E")
|
||
};
|
||
|
||
let font_name = wide_str("Segoe UI");
|
||
unsafe {
|
||
let big_font = create_font(layout.big_font_px, &font_name, FW_SEMIBOLD.0 as i32);
|
||
let small_font = create_font(layout.small_font_px, &font_name, FW_NORMAL.0 as i32);
|
||
let main_font = create_font(layout.main_font_px, &font_name, FW_NORMAL.0 as i32);
|
||
SetBkMode(hdc, TRANSPARENT);
|
||
|
||
let prev_font = SelectObject(hdc, small_font);
|
||
|
||
// Head: "5h" label (muted, centered horizontally).
|
||
SetTextColor(hdc, COLORREF(muted_color.into_colorref()));
|
||
draw_text_in_rect(hdc, &layout.head_label_rect, "5h", DT_CENTER);
|
||
|
||
// Head: big "X%" glyph centered.
|
||
SelectObject(hdc, big_font);
|
||
SetTextColor(hdc, COLORREF(text_color.into_colorref()));
|
||
let pct_text = match inputs.session_pct {
|
||
Some(p) => format!("{:.0}%", p),
|
||
None => String::from("—"),
|
||
};
|
||
draw_text_in_rect(hdc, &layout.head_pct_rect, &pct_text, DT_CENTER);
|
||
|
||
// Tail: "7d" label (muted, left-aligned).
|
||
SelectObject(hdc, small_font);
|
||
SetTextColor(hdc, COLORREF(muted_color.into_colorref()));
|
||
draw_text_in_rect(hdc, &layout.tail_label_rect, "7d", DT_LEFT);
|
||
|
||
// Tail: countdown (right-aligned).
|
||
SelectObject(hdc, main_font);
|
||
SetTextColor(hdc, COLORREF(text_color.into_colorref()));
|
||
if !inputs.weekly_text.is_empty() {
|
||
draw_text_in_rect(hdc, &layout.tail_countdown_rect, &inputs.weekly_text, DT_RIGHT);
|
||
}
|
||
|
||
SelectObject(hdc, prev_font);
|
||
let _ = DeleteObject(big_font);
|
||
let _ = DeleteObject(small_font);
|
||
let _ = DeleteObject(main_font);
|
||
}
|
||
}
|
||
|
||
/// Draw `text` into `rect` with the given horizontal alignment flag, vertically
|
||
/// centered. The DT_NOCLIP flag preserves ascenders/descenders that would
|
||
/// otherwise be clipped by tight rects.
|
||
fn draw_text_in_rect(hdc: HDC, rect: &RECT, text: &str, halign: DRAW_TEXT_FORMAT) {
|
||
let mut buf = wide_str(text);
|
||
let len_no_nul = buf.len().saturating_sub(1);
|
||
let mut r = *rect;
|
||
unsafe {
|
||
let _ = DrawTextW(
|
||
hdc,
|
||
&mut buf[..len_no_nul],
|
||
&mut r,
|
||
halign | DT_VCENTER | DT_SINGLELINE | DT_NOCLIP,
|
||
);
|
||
}
|
||
}
|
||
|
||
fn create_font(height_px: i32, name_w: &[u16], weight: i32) -> HFONT {
|
||
unsafe {
|
||
CreateFontW(
|
||
-height_px,
|
||
0,
|
||
0,
|
||
0,
|
||
weight,
|
||
0,
|
||
0,
|
||
0,
|
||
DEFAULT_CHARSET.0 as u32,
|
||
OUT_DEFAULT_PRECIS.0 as u32,
|
||
CLIP_DEFAULT_PRECIS.0 as u32,
|
||
CLEARTYPE_QUALITY.0 as u32,
|
||
(FF_SWISS.0 | DEFAULT_PITCH.0) as u32,
|
||
PCWSTR::from_raw(name_w.as_ptr()),
|
||
)
|
||
}
|
||
}
|
||
|
||
// ---------- Helpers ----------
|
||
|
||
fn default_position(width_px: i32, height_px: i32, model: ProviderId) -> (i32, i32) {
|
||
// Bottom-right of primary work area, with a 24-pixel gap from the edges.
|
||
// Stagger the Codex bubble above the Claude one if both are enabled.
|
||
unsafe {
|
||
let monitor = MonitorFromPoint(POINT { x: 0, y: 0 }, MONITOR_DEFAULTTOPRIMARY);
|
||
let mut info = MONITORINFO {
|
||
cbSize: std::mem::size_of::<MONITORINFO>() as u32,
|
||
..Default::default()
|
||
};
|
||
let wa = if GetMonitorInfoW(monitor, &mut info).as_bool() {
|
||
info.rcWork
|
||
} else {
|
||
RECT {
|
||
left: 0,
|
||
top: 0,
|
||
right: 1920,
|
||
bottom: 1080,
|
||
}
|
||
};
|
||
let gap = 24;
|
||
let stagger = match model {
|
||
ProviderId::Claude => 0,
|
||
ProviderId::ChatGpt => height_px + gap,
|
||
};
|
||
let x = wa.right - width_px - gap;
|
||
let y = wa.bottom - height_px - gap - stagger;
|
||
(x, y)
|
||
}
|
||
}
|