util.rs

  1use std::sync::OnceLock;
  2
  3use ::util::ResultExt;
  4use windows::Win32::{Foundation::*, System::Threading::*, UI::WindowsAndMessaging::*};
  5
  6use crate::*;
  7
  8pub(crate) trait HiLoWord {
  9    fn hiword(&self) -> u16;
 10    fn loword(&self) -> u16;
 11    fn signed_hiword(&self) -> i16;
 12    fn signed_loword(&self) -> i16;
 13}
 14
 15impl HiLoWord for WPARAM {
 16    fn hiword(&self) -> u16 {
 17        ((self.0 >> 16) & 0xFFFF) as u16
 18    }
 19
 20    fn loword(&self) -> u16 {
 21        (self.0 & 0xFFFF) as u16
 22    }
 23
 24    fn signed_hiword(&self) -> i16 {
 25        ((self.0 >> 16) & 0xFFFF) as i16
 26    }
 27
 28    fn signed_loword(&self) -> i16 {
 29        (self.0 & 0xFFFF) as i16
 30    }
 31}
 32
 33impl HiLoWord for LPARAM {
 34    fn hiword(&self) -> u16 {
 35        ((self.0 >> 16) & 0xFFFF) as u16
 36    }
 37
 38    fn loword(&self) -> u16 {
 39        (self.0 & 0xFFFF) as u16
 40    }
 41
 42    fn signed_hiword(&self) -> i16 {
 43        ((self.0 >> 16) & 0xFFFF) as i16
 44    }
 45
 46    fn signed_loword(&self) -> i16 {
 47        (self.0 & 0xFFFF) as i16
 48    }
 49}
 50
 51pub(crate) unsafe fn get_window_long(hwnd: HWND, nindex: WINDOW_LONG_PTR_INDEX) -> isize {
 52    #[cfg(target_pointer_width = "64")]
 53    unsafe {
 54        GetWindowLongPtrW(hwnd, nindex)
 55    }
 56    #[cfg(target_pointer_width = "32")]
 57    unsafe {
 58        GetWindowLongW(hwnd, nindex) as isize
 59    }
 60}
 61
 62pub(crate) unsafe fn set_window_long(
 63    hwnd: HWND,
 64    nindex: WINDOW_LONG_PTR_INDEX,
 65    dwnewlong: isize,
 66) -> isize {
 67    #[cfg(target_pointer_width = "64")]
 68    unsafe {
 69        SetWindowLongPtrW(hwnd, nindex, dwnewlong)
 70    }
 71    #[cfg(target_pointer_width = "32")]
 72    unsafe {
 73        SetWindowLongW(hwnd, nindex, dwnewlong as i32) as isize
 74    }
 75}
 76
 77pub(crate) struct OwnedHandle(HANDLE);
 78
 79impl OwnedHandle {
 80    pub(crate) fn new(handle: HANDLE) -> Self {
 81        Self(handle)
 82    }
 83
 84    #[inline(always)]
 85    pub(crate) fn to_raw(&self) -> HANDLE {
 86        self.0
 87    }
 88}
 89
 90impl Drop for OwnedHandle {
 91    fn drop(&mut self) {
 92        if !self.0.is_invalid() {
 93            unsafe { CloseHandle(self.0) }.log_err();
 94        }
 95    }
 96}
 97
 98pub(crate) fn create_event() -> windows::core::Result<OwnedHandle> {
 99    Ok(OwnedHandle::new(unsafe {
100        CreateEventW(None, false, false, None)?
101    }))
102}
103
104pub(crate) fn windows_credentials_target_name(url: &str) -> String {
105    format!("zed:url={}", url)
106}
107
108pub(crate) fn load_cursor(style: CursorStyle) -> HCURSOR {
109    static ARROW: OnceLock<HCURSOR> = OnceLock::new();
110    static IBEAM: OnceLock<HCURSOR> = OnceLock::new();
111    static CROSS: OnceLock<HCURSOR> = OnceLock::new();
112    static HAND: OnceLock<HCURSOR> = OnceLock::new();
113    static SIZEWE: OnceLock<HCURSOR> = OnceLock::new();
114    static SIZENS: OnceLock<HCURSOR> = OnceLock::new();
115    static NO: OnceLock<HCURSOR> = OnceLock::new();
116    let (lock, name) = match style {
117        CursorStyle::IBeam | CursorStyle::IBeamCursorForVerticalLayout => (&IBEAM, IDC_IBEAM),
118        CursorStyle::Crosshair => (&CROSS, IDC_CROSS),
119        CursorStyle::PointingHand | CursorStyle::DragLink => (&HAND, IDC_HAND),
120        CursorStyle::ResizeLeft | CursorStyle::ResizeRight | CursorStyle::ResizeLeftRight => {
121            (&SIZEWE, IDC_SIZEWE)
122        }
123        CursorStyle::ResizeUp | CursorStyle::ResizeDown | CursorStyle::ResizeUpDown => {
124            (&SIZENS, IDC_SIZENS)
125        }
126        CursorStyle::OperationNotAllowed => (&NO, IDC_NO),
127        _ => (&ARROW, IDC_ARROW),
128    };
129    *lock.get_or_init(|| {
130        HCURSOR(
131            unsafe { LoadImageW(None, name, IMAGE_CURSOR, 0, 0, LR_DEFAULTSIZE | LR_SHARED) }
132                .log_err()
133                .unwrap_or_default()
134                .0,
135        )
136    })
137}