ZIm/crates/gpui/src/platform/windows/window.rs
Junkui Zhang e0751c467b init
2025-08-04 16:51:15 +08:00

1352 lines
45 KiB
Rust

#![deny(unsafe_op_in_unsafe_fn)]
use std::{
cell::RefCell,
num::NonZeroIsize,
path::PathBuf,
rc::{Rc, Weak},
str::FromStr,
sync::{Arc, Once},
time::{Duration, Instant},
};
use ::util::ResultExt;
use anyhow::{Context as _, Result};
use async_task::Runnable;
use futures::channel::oneshot::{self, Receiver};
use raw_window_handle as rwh;
use smallvec::SmallVec;
use windows::{
Win32::{
Foundation::*,
Graphics::Gdi::*,
System::{Com::*, LibraryLoader::*, Ole::*, SystemServices::*},
UI::{Controls::*, HiDpi::*, Input::KeyboardAndMouse::*, Shell::*, WindowsAndMessaging::*},
},
core::*,
};
use crate::*;
pub(crate) struct WindowsWindow(pub Rc<WindowsWindowStatePtr>);
pub struct WindowsWindowState {
pub origin: Point<Pixels>,
pub logical_size: Size<Pixels>,
pub min_size: Option<Size<Pixels>>,
pub fullscreen_restore_bounds: Bounds<Pixels>,
pub border_offset: WindowBorderOffset,
pub appearance: WindowAppearance,
pub scale_factor: f32,
pub restore_from_minimized: Option<Box<dyn FnMut(RequestFrameOptions)>>,
pub callbacks: Callbacks,
pub input_handler: Option<PlatformInputHandler>,
pub pending_surrogate: Option<u16>,
pub last_reported_modifiers: Option<Modifiers>,
pub last_reported_capslock: Option<Capslock>,
pub system_key_handled: bool,
pub hovered: bool,
pub renderer: DirectXRenderer,
pub keydown_time: Option<std::time::Instant>,
pub click_state: ClickState,
pub system_settings: WindowsSystemSettings,
pub current_cursor: Option<HCURSOR>,
pub nc_button_pressed: Option<u32>,
pub display: WindowsDisplay,
fullscreen: Option<StyleAndBounds>,
initial_placement: Option<WindowOpenStatus>,
hwnd: HWND,
}
pub(crate) struct WindowsWindowStatePtr {
hwnd: HWND,
this: Weak<Self>,
drop_target_helper: IDropTargetHelper,
pub(crate) state: RefCell<WindowsWindowState>,
pub(crate) handle: AnyWindowHandle,
pub(crate) hide_title_bar: bool,
pub(crate) is_movable: bool,
pub(crate) executor: ForegroundExecutor,
pub(crate) windows_version: WindowsVersion,
pub(crate) validation_number: usize,
pub(crate) main_receiver: flume::Receiver<Runnable>,
pub(crate) main_thread_id_win32: u32,
}
impl WindowsWindowState {
fn new(
hwnd: HWND,
cs: &CREATESTRUCTW,
current_cursor: Option<HCURSOR>,
display: WindowsDisplay,
min_size: Option<Size<Pixels>>,
appearance: WindowAppearance,
disable_direct_composition: bool,
) -> Result<Self> {
let scale_factor = {
let monitor_dpi = unsafe { GetDpiForWindow(hwnd) } as f32;
monitor_dpi / USER_DEFAULT_SCREEN_DPI as f32
};
let origin = logical_point(cs.x as f32, cs.y as f32, scale_factor);
let logical_size = {
let physical_size = size(DevicePixels(cs.cx), DevicePixels(cs.cy));
physical_size.to_pixels(scale_factor)
};
let fullscreen_restore_bounds = Bounds {
origin,
size: logical_size,
};
let border_offset = WindowBorderOffset::default();
let restore_from_minimized = None;
let renderer = DirectXRenderer::new(hwnd, disable_direct_composition)
.context("Creating DirectX renderer")?;
let callbacks = Callbacks::default();
let input_handler = None;
let pending_surrogate = None;
let last_reported_modifiers = None;
let last_reported_capslock = None;
let system_key_handled = false;
let hovered = false;
let click_state = ClickState::new();
let system_settings = WindowsSystemSettings::new(display);
let nc_button_pressed = None;
let fullscreen = None;
let initial_placement = None;
let keydown_time = None;
Ok(Self {
origin,
logical_size,
fullscreen_restore_bounds,
border_offset,
appearance,
scale_factor,
restore_from_minimized,
min_size,
callbacks,
input_handler,
pending_surrogate,
last_reported_modifiers,
last_reported_capslock,
system_key_handled,
hovered,
renderer,
keydown_time,
click_state,
system_settings,
current_cursor,
nc_button_pressed,
display,
fullscreen,
initial_placement,
hwnd,
})
}
#[inline]
pub(crate) fn is_fullscreen(&self) -> bool {
self.fullscreen.is_some()
}
pub(crate) fn is_maximized(&self) -> bool {
!self.is_fullscreen() && unsafe { IsZoomed(self.hwnd) }.as_bool()
}
fn bounds(&self) -> Bounds<Pixels> {
Bounds {
origin: self.origin,
size: self.logical_size,
}
}
// Calculate the bounds used for saving and whether the window is maximized.
fn calculate_window_bounds(&self) -> (Bounds<Pixels>, bool) {
let placement = unsafe {
let mut placement = WINDOWPLACEMENT {
length: std::mem::size_of::<WINDOWPLACEMENT>() as u32,
..Default::default()
};
GetWindowPlacement(self.hwnd, &mut placement).log_err();
placement
};
(
calculate_client_rect(
placement.rcNormalPosition,
self.border_offset,
self.scale_factor,
),
placement.showCmd == SW_SHOWMAXIMIZED.0 as u32,
)
}
fn window_bounds(&self) -> WindowBounds {
let (bounds, maximized) = self.calculate_window_bounds();
if self.is_fullscreen() {
WindowBounds::Fullscreen(self.fullscreen_restore_bounds)
} else if maximized {
WindowBounds::Maximized(bounds)
} else {
WindowBounds::Windowed(bounds)
}
}
/// get the logical size of the app's drawable area.
///
/// Currently, GPUI uses the logical size of the app to handle mouse interactions (such as
/// whether the mouse collides with other elements of GPUI).
fn content_size(&self) -> Size<Pixels> {
self.logical_size
}
}
impl WindowsWindowStatePtr {
fn new(context: &WindowCreateContext, hwnd: HWND, cs: &CREATESTRUCTW) -> Result<Rc<Self>> {
let state = RefCell::new(WindowsWindowState::new(
hwnd,
cs,
context.current_cursor,
context.display,
context.min_size,
context.appearance,
context.disable_direct_composition,
)?);
Ok(Rc::new_cyclic(|this| Self {
hwnd,
this: this.clone(),
drop_target_helper: context.drop_target_helper.clone(),
state,
handle: context.handle,
hide_title_bar: context.hide_title_bar,
is_movable: context.is_movable,
executor: context.executor.clone(),
windows_version: context.windows_version,
validation_number: context.validation_number,
main_receiver: context.main_receiver.clone(),
main_thread_id_win32: context.main_thread_id_win32,
}))
}
fn toggle_fullscreen(&self) {
let Some(state_ptr) = self.this.upgrade() else {
log::error!("Unable to toggle fullscreen: window has been dropped");
return;
};
self.executor
.spawn(async move {
let mut lock = state_ptr.state.borrow_mut();
let StyleAndBounds {
style,
x,
y,
cx,
cy,
} = if let Some(state) = lock.fullscreen.take() {
state
} else {
let (window_bounds, _) = lock.calculate_window_bounds();
lock.fullscreen_restore_bounds = window_bounds;
let style =
WINDOW_STYLE(unsafe { get_window_long(state_ptr.hwnd, GWL_STYLE) } as _);
let mut rc = RECT::default();
unsafe { GetWindowRect(state_ptr.hwnd, &mut rc) }.log_err();
let _ = lock.fullscreen.insert(StyleAndBounds {
style,
x: rc.left,
y: rc.top,
cx: rc.right - rc.left,
cy: rc.bottom - rc.top,
});
let style = style
& !(WS_THICKFRAME
| WS_SYSMENU
| WS_MAXIMIZEBOX
| WS_MINIMIZEBOX
| WS_CAPTION);
let physical_bounds = lock.display.physical_bounds();
StyleAndBounds {
style,
x: physical_bounds.left().0,
y: physical_bounds.top().0,
cx: physical_bounds.size.width.0,
cy: physical_bounds.size.height.0,
}
};
drop(lock);
unsafe { set_window_long(state_ptr.hwnd, GWL_STYLE, style.0 as isize) };
unsafe {
SetWindowPos(
state_ptr.hwnd,
None,
x,
y,
cx,
cy,
SWP_FRAMECHANGED | SWP_NOACTIVATE | SWP_NOZORDER,
)
}
.log_err();
})
.detach();
}
fn set_window_placement(&self) -> Result<()> {
let Some(open_status) = self.state.borrow_mut().initial_placement.take() else {
return Ok(());
};
match open_status.state {
WindowOpenState::Maximized => unsafe {
SetWindowPlacement(self.hwnd, &open_status.placement)?;
ShowWindowAsync(self.hwnd, SW_MAXIMIZE).ok()?;
},
WindowOpenState::Fullscreen => {
unsafe { SetWindowPlacement(self.hwnd, &open_status.placement)? };
self.toggle_fullscreen();
}
WindowOpenState::Windowed => unsafe {
SetWindowPlacement(self.hwnd, &open_status.placement)?;
},
}
Ok(())
}
}
#[derive(Default)]
pub(crate) struct Callbacks {
pub(crate) request_frame: Option<Box<dyn FnMut(RequestFrameOptions)>>,
pub(crate) input: Option<Box<dyn FnMut(crate::PlatformInput) -> DispatchEventResult>>,
pub(crate) active_status_change: Option<Box<dyn FnMut(bool)>>,
pub(crate) hovered_status_change: Option<Box<dyn FnMut(bool)>>,
pub(crate) resize: Option<Box<dyn FnMut(Size<Pixels>, f32)>>,
pub(crate) moved: Option<Box<dyn FnMut()>>,
pub(crate) should_close: Option<Box<dyn FnMut() -> bool>>,
pub(crate) close: Option<Box<dyn FnOnce()>>,
pub(crate) hit_test_window_control: Option<Box<dyn FnMut() -> Option<WindowControlArea>>>,
pub(crate) appearance_changed: Option<Box<dyn FnMut()>>,
}
struct WindowCreateContext {
inner: Option<Result<Rc<WindowsWindowStatePtr>>>,
handle: AnyWindowHandle,
hide_title_bar: bool,
display: WindowsDisplay,
is_movable: bool,
min_size: Option<Size<Pixels>>,
executor: ForegroundExecutor,
current_cursor: Option<HCURSOR>,
windows_version: WindowsVersion,
drop_target_helper: IDropTargetHelper,
validation_number: usize,
main_receiver: flume::Receiver<Runnable>,
main_thread_id_win32: u32,
appearance: WindowAppearance,
disable_direct_composition: bool,
}
impl WindowsWindow {
pub(crate) fn new(
handle: AnyWindowHandle,
params: WindowParams,
creation_info: WindowCreationInfo,
) -> Result<Self> {
let WindowCreationInfo {
icon,
executor,
current_cursor,
windows_version,
drop_target_helper,
validation_number,
main_receiver,
main_thread_id_win32,
disable_direct_composition,
} = creation_info;
let classname = register_wnd_class(icon);
let hide_title_bar = params
.titlebar
.as_ref()
.map(|titlebar| titlebar.appears_transparent)
.unwrap_or(true);
let windowname = HSTRING::from(
params
.titlebar
.as_ref()
.and_then(|titlebar| titlebar.title.as_ref())
.map(|title| title.as_ref())
.unwrap_or(""),
);
let (mut dwexstyle, dwstyle) = if params.kind == WindowKind::PopUp {
(WS_EX_TOOLWINDOW, WINDOW_STYLE(0x0))
} else {
(
WS_EX_APPWINDOW,
WS_THICKFRAME | WS_SYSMENU | WS_MAXIMIZEBOX | WS_MINIMIZEBOX,
)
};
if !disable_direct_composition {
dwexstyle |= WS_EX_NOREDIRECTIONBITMAP;
}
let hinstance = get_module_handle();
let display = if let Some(display_id) = params.display_id {
// if we obtain a display_id, then this ID must be valid.
WindowsDisplay::new(display_id).unwrap()
} else {
WindowsDisplay::primary_monitor().unwrap()
};
let appearance = system_appearance().unwrap_or_default();
let mut context = WindowCreateContext {
inner: None,
handle,
hide_title_bar,
display,
is_movable: params.is_movable,
min_size: params.window_min_size,
executor,
current_cursor,
windows_version,
drop_target_helper,
validation_number,
main_receiver,
main_thread_id_win32,
appearance,
disable_direct_composition,
};
let lpparam = Some(&context as *const _ as *const _);
let creation_result = unsafe {
CreateWindowExW(
dwexstyle,
classname,
&windowname,
dwstyle,
CW_USEDEFAULT,
CW_USEDEFAULT,
CW_USEDEFAULT,
CW_USEDEFAULT,
None,
None,
Some(hinstance.into()),
lpparam,
)
};
// We should call `?` on state_ptr first, then call `?` on hwnd.
// Or, we will lose the error info reported by `WindowsWindowState::new`
let state_ptr = context.inner.take().unwrap()?;
let hwnd = creation_result?;
register_drag_drop(state_ptr.clone())?;
configure_dwm_dark_mode(hwnd, appearance);
state_ptr.state.borrow_mut().border_offset.update(hwnd)?;
let placement = retrieve_window_placement(
hwnd,
display,
params.bounds,
state_ptr.state.borrow().scale_factor,
state_ptr.state.borrow().border_offset,
)?;
if params.show {
unsafe { SetWindowPlacement(hwnd, &placement)? };
} else {
state_ptr.state.borrow_mut().initial_placement = Some(WindowOpenStatus {
placement,
state: WindowOpenState::Windowed,
});
}
Ok(Self(state_ptr))
}
}
impl rwh::HasWindowHandle for WindowsWindow {
fn window_handle(&self) -> std::result::Result<rwh::WindowHandle<'_>, rwh::HandleError> {
let raw = rwh::Win32WindowHandle::new(unsafe {
NonZeroIsize::new_unchecked(self.0.hwnd.0 as isize)
})
.into();
Ok(unsafe { rwh::WindowHandle::borrow_raw(raw) })
}
}
// todo(windows)
impl rwh::HasDisplayHandle for WindowsWindow {
fn display_handle(&self) -> std::result::Result<rwh::DisplayHandle<'_>, rwh::HandleError> {
unimplemented!()
}
}
impl Drop for WindowsWindow {
fn drop(&mut self) {
// clone this `Rc` to prevent early release of the pointer
let this = self.0.clone();
self.0
.executor
.spawn(async move {
let handle = this.hwnd;
unsafe {
RevokeDragDrop(handle).log_err();
DestroyWindow(handle).log_err();
}
})
.detach();
}
}
impl PlatformWindow for WindowsWindow {
fn bounds(&self) -> Bounds<Pixels> {
self.0.state.borrow().bounds()
}
fn is_maximized(&self) -> bool {
self.0.state.borrow().is_maximized()
}
fn window_bounds(&self) -> WindowBounds {
self.0.state.borrow().window_bounds()
}
/// get the logical size of the app's drawable area.
///
/// Currently, GPUI uses the logical size of the app to handle mouse interactions (such as
/// whether the mouse collides with other elements of GPUI).
fn content_size(&self) -> Size<Pixels> {
self.0.state.borrow().content_size()
}
fn resize(&mut self, size: Size<Pixels>) {
let hwnd = self.0.hwnd;
let bounds =
crate::bounds(self.bounds().origin, size).to_device_pixels(self.scale_factor());
let rect = calculate_window_rect(bounds, self.0.state.borrow().border_offset);
self.0
.executor
.spawn(async move {
unsafe {
SetWindowPos(
hwnd,
None,
bounds.origin.x.0,
bounds.origin.y.0,
rect.right - rect.left,
rect.bottom - rect.top,
SWP_NOMOVE,
)
.context("unable to set window content size")
.log_err();
}
})
.detach();
}
fn scale_factor(&self) -> f32 {
self.0.state.borrow().scale_factor
}
fn appearance(&self) -> WindowAppearance {
self.0.state.borrow().appearance
}
fn display(&self) -> Option<Rc<dyn PlatformDisplay>> {
Some(Rc::new(self.0.state.borrow().display))
}
fn mouse_position(&self) -> Point<Pixels> {
let scale_factor = self.scale_factor();
let point = unsafe {
let mut point: POINT = std::mem::zeroed();
GetCursorPos(&mut point)
.context("unable to get cursor position")
.log_err();
ScreenToClient(self.0.hwnd, &mut point).ok().log_err();
point
};
logical_point(point.x as f32, point.y as f32, scale_factor)
}
fn modifiers(&self) -> Modifiers {
current_modifiers()
}
fn capslock(&self) -> Capslock {
current_capslock()
}
fn set_input_handler(&mut self, input_handler: PlatformInputHandler) {
self.0.state.borrow_mut().input_handler = Some(input_handler);
}
fn take_input_handler(&mut self) -> Option<PlatformInputHandler> {
self.0.state.borrow_mut().input_handler.take()
}
fn prompt(
&self,
level: PromptLevel,
msg: &str,
detail: Option<&str>,
answers: &[PromptButton],
) -> Option<Receiver<usize>> {
let (done_tx, done_rx) = oneshot::channel();
let msg = msg.to_string();
let detail_string = match detail {
Some(info) => Some(info.to_string()),
None => None,
};
let handle = self.0.hwnd;
let answers = answers.to_vec();
self.0
.executor
.spawn(async move {
unsafe {
let mut config = TASKDIALOGCONFIG::default();
config.cbSize = std::mem::size_of::<TASKDIALOGCONFIG>() as _;
config.hwndParent = handle;
let title;
let main_icon;
match level {
crate::PromptLevel::Info => {
title = windows::core::w!("Info");
main_icon = TD_INFORMATION_ICON;
}
crate::PromptLevel::Warning => {
title = windows::core::w!("Warning");
main_icon = TD_WARNING_ICON;
}
crate::PromptLevel::Critical => {
title = windows::core::w!("Critical");
main_icon = TD_ERROR_ICON;
}
};
config.pszWindowTitle = title;
config.Anonymous1.pszMainIcon = main_icon;
let instruction = HSTRING::from(msg);
config.pszMainInstruction = PCWSTR::from_raw(instruction.as_ptr());
let hints_encoded;
if let Some(ref hints) = detail_string {
hints_encoded = HSTRING::from(hints);
config.pszContent = PCWSTR::from_raw(hints_encoded.as_ptr());
};
let mut button_id_map = Vec::with_capacity(answers.len());
let mut buttons = Vec::new();
let mut btn_encoded = Vec::new();
for (index, btn) in answers.iter().enumerate() {
let encoded = HSTRING::from(btn.label().as_ref());
let button_id = if btn.is_cancel() {
IDCANCEL.0
} else {
index as i32 - 100
};
button_id_map.push(button_id);
buttons.push(TASKDIALOG_BUTTON {
nButtonID: button_id,
pszButtonText: PCWSTR::from_raw(encoded.as_ptr()),
});
btn_encoded.push(encoded);
}
config.cButtons = buttons.len() as _;
config.pButtons = buttons.as_ptr();
config.pfCallback = None;
let mut res = std::mem::zeroed();
let _ = TaskDialogIndirect(&config, Some(&mut res), None, None)
.context("unable to create task dialog")
.log_err();
let clicked = button_id_map
.iter()
.position(|&button_id| button_id == res)
.unwrap();
let _ = done_tx.send(clicked);
}
})
.detach();
Some(done_rx)
}
fn activate(&self) {
let hwnd = self.0.hwnd;
let this = self.0.clone();
self.0
.executor
.spawn(async move {
this.set_window_placement().log_err();
unsafe { SetActiveWindow(hwnd).log_err() };
unsafe { SetFocus(Some(hwnd)).log_err() };
// todo(windows)
// crate `windows 0.56` reports true as Err
unsafe { SetForegroundWindow(hwnd).as_bool() };
})
.detach();
}
fn is_active(&self) -> bool {
self.0.hwnd == unsafe { GetActiveWindow() }
}
fn is_hovered(&self) -> bool {
self.0.state.borrow().hovered
}
fn set_title(&mut self, title: &str) {
unsafe { SetWindowTextW(self.0.hwnd, &HSTRING::from(title)) }
.inspect_err(|e| log::error!("Set title failed: {e}"))
.ok();
}
fn set_background_appearance(&self, background_appearance: WindowBackgroundAppearance) {
let hwnd = self.0.hwnd;
match background_appearance {
WindowBackgroundAppearance::Opaque => {
// ACCENT_DISABLED
set_window_composition_attribute(hwnd, None, 0);
}
WindowBackgroundAppearance::Transparent => {
// Use ACCENT_ENABLE_TRANSPARENTGRADIENT for transparent background
set_window_composition_attribute(hwnd, None, 2);
}
WindowBackgroundAppearance::Blurred => {
// Enable acrylic blur
// ACCENT_ENABLE_ACRYLICBLURBEHIND
set_window_composition_attribute(hwnd, Some((0, 0, 0, 0)), 4);
}
}
}
fn minimize(&self) {
unsafe { ShowWindowAsync(self.0.hwnd, SW_MINIMIZE).ok().log_err() };
}
fn zoom(&self) {
unsafe {
if IsWindowVisible(self.0.hwnd).as_bool() {
ShowWindowAsync(self.0.hwnd, SW_MAXIMIZE).ok().log_err();
} else if let Some(status) = self.0.state.borrow_mut().initial_placement.as_mut() {
status.state = WindowOpenState::Maximized;
}
}
}
fn toggle_fullscreen(&self) {
if unsafe { IsWindowVisible(self.0.hwnd).as_bool() } {
self.0.toggle_fullscreen();
} else if let Some(status) = self.0.state.borrow_mut().initial_placement.as_mut() {
status.state = WindowOpenState::Fullscreen;
}
}
fn is_fullscreen(&self) -> bool {
self.0.state.borrow().is_fullscreen()
}
fn on_request_frame(&self, callback: Box<dyn FnMut(RequestFrameOptions)>) {
self.0.state.borrow_mut().callbacks.request_frame = Some(callback);
}
fn on_input(&self, callback: Box<dyn FnMut(PlatformInput) -> DispatchEventResult>) {
self.0.state.borrow_mut().callbacks.input = Some(callback);
}
fn on_active_status_change(&self, callback: Box<dyn FnMut(bool)>) {
self.0.state.borrow_mut().callbacks.active_status_change = Some(callback);
}
fn on_hover_status_change(&self, callback: Box<dyn FnMut(bool)>) {
self.0.state.borrow_mut().callbacks.hovered_status_change = Some(callback);
}
fn on_resize(&self, callback: Box<dyn FnMut(Size<Pixels>, f32)>) {
self.0.state.borrow_mut().callbacks.resize = Some(callback);
}
fn on_moved(&self, callback: Box<dyn FnMut()>) {
self.0.state.borrow_mut().callbacks.moved = Some(callback);
}
fn on_should_close(&self, callback: Box<dyn FnMut() -> bool>) {
self.0.state.borrow_mut().callbacks.should_close = Some(callback);
}
fn on_close(&self, callback: Box<dyn FnOnce()>) {
self.0.state.borrow_mut().callbacks.close = Some(callback);
}
fn on_hit_test_window_control(&self, callback: Box<dyn FnMut() -> Option<WindowControlArea>>) {
self.0.state.borrow_mut().callbacks.hit_test_window_control = Some(callback);
}
fn on_appearance_changed(&self, callback: Box<dyn FnMut()>) {
self.0.state.borrow_mut().callbacks.appearance_changed = Some(callback);
}
fn draw(&self, scene: &Scene) {
self.0.state.borrow_mut().renderer.draw(scene).log_err();
}
fn sprite_atlas(&self) -> Arc<dyn PlatformAtlas> {
self.0.state.borrow().renderer.sprite_atlas()
}
fn get_raw_handle(&self) -> HWND {
self.0.hwnd
}
fn gpu_specs(&self) -> Option<GpuSpecs> {
self.0.state.borrow().renderer.gpu_specs().log_err()
}
fn update_ime_position(&self, _bounds: Bounds<ScaledPixels>) {
// There is no such thing on Windows.
}
}
#[implement(IDropTarget)]
struct WindowsDragDropHandler(pub Rc<WindowsWindowStatePtr>);
impl WindowsDragDropHandler {
fn handle_drag_drop(&self, input: PlatformInput) {
let mut lock = self.0.state.borrow_mut();
if let Some(mut func) = lock.callbacks.input.take() {
drop(lock);
func(input);
self.0.state.borrow_mut().callbacks.input = Some(func);
}
}
}
#[allow(non_snake_case)]
impl IDropTarget_Impl for WindowsDragDropHandler_Impl {
fn DragEnter(
&self,
pdataobj: windows::core::Ref<IDataObject>,
_grfkeystate: MODIFIERKEYS_FLAGS,
pt: &POINTL,
pdweffect: *mut DROPEFFECT,
) -> windows::core::Result<()> {
unsafe {
let idata_obj = pdataobj.ok()?;
let config = FORMATETC {
cfFormat: CF_HDROP.0,
ptd: std::ptr::null_mut() as _,
dwAspect: DVASPECT_CONTENT.0,
lindex: -1,
tymed: TYMED_HGLOBAL.0 as _,
};
let cursor_position = POINT { x: pt.x, y: pt.y };
if idata_obj.QueryGetData(&config as _) == S_OK {
*pdweffect = DROPEFFECT_COPY;
let Some(mut idata) = idata_obj.GetData(&config as _).log_err() else {
return Ok(());
};
if idata.u.hGlobal.is_invalid() {
return Ok(());
}
let hdrop = idata.u.hGlobal.0 as *mut HDROP;
let mut paths = SmallVec::<[PathBuf; 2]>::new();
with_file_names(*hdrop, |file_name| {
if let Some(path) = PathBuf::from_str(&file_name).log_err() {
paths.push(path);
}
});
ReleaseStgMedium(&mut idata);
let mut cursor_position = cursor_position;
ScreenToClient(self.0.hwnd, &mut cursor_position)
.ok()
.log_err();
let scale_factor = self.0.state.borrow().scale_factor;
let input = PlatformInput::FileDrop(FileDropEvent::Entered {
position: logical_point(
cursor_position.x as f32,
cursor_position.y as f32,
scale_factor,
),
paths: ExternalPaths(paths),
});
self.handle_drag_drop(input);
} else {
*pdweffect = DROPEFFECT_NONE;
}
self.0
.drop_target_helper
.DragEnter(self.0.hwnd, idata_obj, &cursor_position, *pdweffect)
.log_err();
}
Ok(())
}
fn DragOver(
&self,
_grfkeystate: MODIFIERKEYS_FLAGS,
pt: &POINTL,
pdweffect: *mut DROPEFFECT,
) -> windows::core::Result<()> {
let mut cursor_position = POINT { x: pt.x, y: pt.y };
unsafe {
*pdweffect = DROPEFFECT_COPY;
self.0
.drop_target_helper
.DragOver(&cursor_position, *pdweffect)
.log_err();
ScreenToClient(self.0.hwnd, &mut cursor_position)
.ok()
.log_err();
}
let scale_factor = self.0.state.borrow().scale_factor;
let input = PlatformInput::FileDrop(FileDropEvent::Pending {
position: logical_point(
cursor_position.x as f32,
cursor_position.y as f32,
scale_factor,
),
});
self.handle_drag_drop(input);
Ok(())
}
fn DragLeave(&self) -> windows::core::Result<()> {
unsafe {
self.0.drop_target_helper.DragLeave().log_err();
}
let input = PlatformInput::FileDrop(FileDropEvent::Exited);
self.handle_drag_drop(input);
Ok(())
}
fn Drop(
&self,
pdataobj: windows::core::Ref<IDataObject>,
_grfkeystate: MODIFIERKEYS_FLAGS,
pt: &POINTL,
pdweffect: *mut DROPEFFECT,
) -> windows::core::Result<()> {
let idata_obj = pdataobj.ok()?;
let mut cursor_position = POINT { x: pt.x, y: pt.y };
unsafe {
*pdweffect = DROPEFFECT_COPY;
self.0
.drop_target_helper
.Drop(idata_obj, &cursor_position, *pdweffect)
.log_err();
ScreenToClient(self.0.hwnd, &mut cursor_position)
.ok()
.log_err();
}
let scale_factor = self.0.state.borrow().scale_factor;
let input = PlatformInput::FileDrop(FileDropEvent::Submit {
position: logical_point(
cursor_position.x as f32,
cursor_position.y as f32,
scale_factor,
),
});
self.handle_drag_drop(input);
Ok(())
}
}
#[derive(Debug, Clone, Copy)]
pub(crate) struct ClickState {
button: MouseButton,
last_click: Instant,
last_position: Point<DevicePixels>,
double_click_spatial_tolerance_width: i32,
double_click_spatial_tolerance_height: i32,
double_click_interval: Duration,
pub(crate) current_count: usize,
}
impl ClickState {
pub fn new() -> Self {
let double_click_spatial_tolerance_width = unsafe { GetSystemMetrics(SM_CXDOUBLECLK) };
let double_click_spatial_tolerance_height = unsafe { GetSystemMetrics(SM_CYDOUBLECLK) };
let double_click_interval = Duration::from_millis(unsafe { GetDoubleClickTime() } as u64);
ClickState {
button: MouseButton::Left,
last_click: Instant::now(),
last_position: Point::default(),
double_click_spatial_tolerance_width,
double_click_spatial_tolerance_height,
double_click_interval,
current_count: 0,
}
}
/// update self and return the needed click count
pub fn update(&mut self, button: MouseButton, new_position: Point<DevicePixels>) -> usize {
if self.button == button && self.is_double_click(new_position) {
self.current_count += 1;
} else {
self.current_count = 1;
}
self.last_click = Instant::now();
self.last_position = new_position;
self.button = button;
self.current_count
}
pub fn system_update(&mut self, wparam: usize) {
match wparam {
// SPI_SETDOUBLECLKWIDTH
29 => {
self.double_click_spatial_tolerance_width =
unsafe { GetSystemMetrics(SM_CXDOUBLECLK) }
}
// SPI_SETDOUBLECLKHEIGHT
30 => {
self.double_click_spatial_tolerance_height =
unsafe { GetSystemMetrics(SM_CYDOUBLECLK) }
}
// SPI_SETDOUBLECLICKTIME
32 => {
self.double_click_interval =
Duration::from_millis(unsafe { GetDoubleClickTime() } as u64)
}
_ => {}
}
}
#[inline]
fn is_double_click(&self, new_position: Point<DevicePixels>) -> bool {
let diff = self.last_position - new_position;
self.last_click.elapsed() < self.double_click_interval
&& diff.x.0.abs() <= self.double_click_spatial_tolerance_width
&& diff.y.0.abs() <= self.double_click_spatial_tolerance_height
}
}
struct StyleAndBounds {
style: WINDOW_STYLE,
x: i32,
y: i32,
cx: i32,
cy: i32,
}
#[repr(C)]
struct WINDOWCOMPOSITIONATTRIBDATA {
attrib: u32,
pv_data: *mut std::ffi::c_void,
cb_data: usize,
}
#[repr(C)]
struct AccentPolicy {
accent_state: u32,
accent_flags: u32,
gradient_color: u32,
animation_id: u32,
}
type Color = (u8, u8, u8, u8);
#[derive(Debug, Default, Clone, Copy)]
pub(crate) struct WindowBorderOffset {
pub(crate) width_offset: i32,
pub(crate) height_offset: i32,
}
impl WindowBorderOffset {
pub(crate) fn update(&mut self, hwnd: HWND) -> anyhow::Result<()> {
let window_rect = unsafe {
let mut rect = std::mem::zeroed();
GetWindowRect(hwnd, &mut rect)?;
rect
};
let client_rect = unsafe {
let mut rect = std::mem::zeroed();
GetClientRect(hwnd, &mut rect)?;
rect
};
self.width_offset =
(window_rect.right - window_rect.left) - (client_rect.right - client_rect.left);
self.height_offset =
(window_rect.bottom - window_rect.top) - (client_rect.bottom - client_rect.top);
Ok(())
}
}
struct WindowOpenStatus {
placement: WINDOWPLACEMENT,
state: WindowOpenState,
}
enum WindowOpenState {
Maximized,
Fullscreen,
Windowed,
}
fn register_wnd_class(icon_handle: HICON) -> PCWSTR {
const CLASS_NAME: PCWSTR = w!("Zed::Window");
static ONCE: Once = Once::new();
ONCE.call_once(|| {
let wc = WNDCLASSW {
lpfnWndProc: Some(wnd_proc),
hIcon: icon_handle,
lpszClassName: PCWSTR(CLASS_NAME.as_ptr()),
style: CS_HREDRAW | CS_VREDRAW,
hInstance: get_module_handle().into(),
hbrBackground: unsafe { CreateSolidBrush(COLORREF(0x00000000)) },
..Default::default()
};
unsafe { RegisterClassW(&wc) };
});
CLASS_NAME
}
unsafe extern "system" fn wnd_proc(
hwnd: HWND,
msg: u32,
wparam: WPARAM,
lparam: LPARAM,
) -> LRESULT {
if msg == WM_NCCREATE {
let cs = lparam.0 as *const CREATESTRUCTW;
let cs = unsafe { &*cs };
let ctx = cs.lpCreateParams as *mut WindowCreateContext;
let ctx = unsafe { &mut *ctx };
let creation_result = WindowsWindowStatePtr::new(ctx, hwnd, cs);
if creation_result.is_err() {
ctx.inner = Some(creation_result);
return LRESULT(0);
}
let weak = Box::new(Rc::downgrade(creation_result.as_ref().unwrap()));
unsafe { set_window_long(hwnd, GWLP_USERDATA, Box::into_raw(weak) as isize) };
ctx.inner = Some(creation_result);
return unsafe { DefWindowProcW(hwnd, msg, wparam, lparam) };
}
let ptr = unsafe { get_window_long(hwnd, GWLP_USERDATA) } as *mut Weak<WindowsWindowStatePtr>;
if ptr.is_null() {
return unsafe { DefWindowProcW(hwnd, msg, wparam, lparam) };
}
let inner = unsafe { &*ptr };
let r = if let Some(state) = inner.upgrade() {
handle_msg(hwnd, msg, wparam, lparam, state)
} else {
unsafe { DefWindowProcW(hwnd, msg, wparam, lparam) }
};
if msg == WM_NCDESTROY {
unsafe { set_window_long(hwnd, GWLP_USERDATA, 0) };
unsafe { drop(Box::from_raw(ptr)) };
}
r
}
pub(crate) fn try_get_window_inner(hwnd: HWND) -> Option<Rc<WindowsWindowStatePtr>> {
if hwnd.is_invalid() {
return None;
}
let ptr = unsafe { get_window_long(hwnd, GWLP_USERDATA) } as *mut Weak<WindowsWindowStatePtr>;
if !ptr.is_null() {
let inner = unsafe { &*ptr };
inner.upgrade()
} else {
None
}
}
fn get_module_handle() -> HMODULE {
unsafe {
let mut h_module = std::mem::zeroed();
GetModuleHandleExW(
GET_MODULE_HANDLE_EX_FLAG_FROM_ADDRESS | GET_MODULE_HANDLE_EX_FLAG_UNCHANGED_REFCOUNT,
windows::core::w!("ZedModule"),
&mut h_module,
)
.expect("Unable to get module handle"); // this should never fail
h_module
}
}
fn register_drag_drop(state_ptr: Rc<WindowsWindowStatePtr>) -> Result<()> {
let window_handle = state_ptr.hwnd;
let handler = WindowsDragDropHandler(state_ptr);
// The lifetime of `IDropTarget` is handled by Windows, it won't release until
// we call `RevokeDragDrop`.
// So, it's safe to drop it here.
let drag_drop_handler: IDropTarget = handler.into();
unsafe {
RegisterDragDrop(window_handle, &drag_drop_handler)
.context("unable to register drag-drop event")?;
}
Ok(())
}
fn calculate_window_rect(bounds: Bounds<DevicePixels>, border_offset: WindowBorderOffset) -> RECT {
// NOTE:
// The reason we're not using `AdjustWindowRectEx()` here is
// that the size reported by this function is incorrect.
// You can test it, and there are similar discussions online.
// See: https://stackoverflow.com/questions/12423584/how-to-set-exact-client-size-for-overlapped-window-winapi
//
// So we manually calculate these values here.
let mut rect = RECT {
left: bounds.left().0,
top: bounds.top().0,
right: bounds.right().0,
bottom: bounds.bottom().0,
};
let left_offset = border_offset.width_offset / 2;
let top_offset = border_offset.height_offset / 2;
let right_offset = border_offset.width_offset - left_offset;
let bottom_offset = border_offset.height_offset - top_offset;
rect.left -= left_offset;
rect.top -= top_offset;
rect.right += right_offset;
rect.bottom += bottom_offset;
rect
}
fn calculate_client_rect(
rect: RECT,
border_offset: WindowBorderOffset,
scale_factor: f32,
) -> Bounds<Pixels> {
let left_offset = border_offset.width_offset / 2;
let top_offset = border_offset.height_offset / 2;
let right_offset = border_offset.width_offset - left_offset;
let bottom_offset = border_offset.height_offset - top_offset;
let left = rect.left + left_offset;
let top = rect.top + top_offset;
let right = rect.right - right_offset;
let bottom = rect.bottom - bottom_offset;
let physical_size = size(DevicePixels(right - left), DevicePixels(bottom - top));
Bounds {
origin: logical_point(left as f32, top as f32, scale_factor),
size: physical_size.to_pixels(scale_factor),
}
}
fn retrieve_window_placement(
hwnd: HWND,
display: WindowsDisplay,
initial_bounds: Bounds<Pixels>,
scale_factor: f32,
border_offset: WindowBorderOffset,
) -> Result<WINDOWPLACEMENT> {
let mut placement = WINDOWPLACEMENT {
length: std::mem::size_of::<WINDOWPLACEMENT>() as u32,
..Default::default()
};
unsafe { GetWindowPlacement(hwnd, &mut placement)? };
// the bounds may be not inside the display
let bounds = if display.check_given_bounds(initial_bounds) {
initial_bounds
} else {
display.default_bounds()
};
let bounds = bounds.to_device_pixels(scale_factor);
placement.rcNormalPosition = calculate_window_rect(bounds, border_offset);
Ok(placement)
}
fn set_window_composition_attribute(hwnd: HWND, color: Option<Color>, state: u32) {
let mut version = unsafe { std::mem::zeroed() };
let status = unsafe { windows::Wdk::System::SystemServices::RtlGetVersion(&mut version) };
if !status.is_ok() || version.dwBuildNumber < 17763 {
return;
}
unsafe {
type SetWindowCompositionAttributeType =
unsafe extern "system" fn(HWND, *mut WINDOWCOMPOSITIONATTRIBDATA) -> BOOL;
let module_name = PCSTR::from_raw(c"user32.dll".as_ptr() as *const u8);
if let Some(user32) = GetModuleHandleA(module_name)
.context("Unable to get user32.dll handle")
.log_err()
{
let func_name = PCSTR::from_raw(c"SetWindowCompositionAttribute".as_ptr() as *const u8);
let set_window_composition_attribute: SetWindowCompositionAttributeType =
std::mem::transmute(GetProcAddress(user32, func_name));
let mut color = color.unwrap_or_default();
let is_acrylic = state == 4;
if is_acrylic && color.3 == 0 {
color.3 = 1;
}
let accent = AccentPolicy {
accent_state: state,
accent_flags: if is_acrylic { 0 } else { 2 },
gradient_color: (color.0 as u32)
| ((color.1 as u32) << 8)
| ((color.2 as u32) << 16)
| ((color.3 as u32) << 24),
animation_id: 0,
};
let mut data = WINDOWCOMPOSITIONATTRIBDATA {
attrib: 0x13,
pv_data: &accent as *const _ as *mut _,
cb_data: std::mem::size_of::<AccentPolicy>(),
};
let _ = set_window_composition_attribute(hwnd, &mut data as *mut _ as _);
}
}
}
#[cfg(test)]
mod tests {
use super::ClickState;
use crate::{DevicePixels, MouseButton, point};
use std::time::Duration;
#[test]
fn test_double_click_interval() {
let mut state = ClickState::new();
assert_eq!(
state.update(MouseButton::Left, point(DevicePixels(0), DevicePixels(0))),
1
);
assert_eq!(
state.update(MouseButton::Right, point(DevicePixels(0), DevicePixels(0))),
1
);
assert_eq!(
state.update(MouseButton::Left, point(DevicePixels(0), DevicePixels(0))),
1
);
assert_eq!(
state.update(MouseButton::Left, point(DevicePixels(0), DevicePixels(0))),
2
);
state.last_click -= Duration::from_millis(700);
assert_eq!(
state.update(MouseButton::Left, point(DevicePixels(0), DevicePixels(0))),
1
);
}
#[test]
fn test_double_click_spatial_tolerance() {
let mut state = ClickState::new();
assert_eq!(
state.update(MouseButton::Left, point(DevicePixels(-3), DevicePixels(0))),
1
);
assert_eq!(
state.update(MouseButton::Left, point(DevicePixels(0), DevicePixels(3))),
2
);
assert_eq!(
state.update(MouseButton::Right, point(DevicePixels(3), DevicePixels(2))),
1
);
assert_eq!(
state.update(MouseButton::Right, point(DevicePixels(10), DevicePixels(0))),
1
);
}
}