From 06e714da56f687116f9e24ced579a547cb15fa81 Mon Sep 17 00:00:00 2001
From: zyl <z@zyl.gay>
Date: Thu, 23 Jan 2025 18:35:18 -0800
Subject: [PATCH] Implement winapi dropping files

---
 Cargo.toml            |  1 +
 src/native/windows.rs | 23 ++++++++++++++++++++++-
 2 files changed, 23 insertions(+), 1 deletion(-)

diff --git a/Cargo.toml b/Cargo.toml
index 47662de6..d6f7f8b3 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -34,6 +34,7 @@ winapi = { version = "0.3", features = [
     "windowsx",
     "winbase",
     "hidusage",
+    "shellapi",
 ] }
 
 [target.'cfg(target_os = "android")'.dependencies]
diff --git a/src/native/windows.rs b/src/native/windows.rs
index 8be60ed1..a200442d 100644
--- a/src/native/windows.rs
+++ b/src/native/windows.rs
@@ -1,3 +1,5 @@
+use std::{ffi::OsString, os::windows::ffi::OsStringExt, path::PathBuf};
+
 use crate::{
     conf::{Conf, Icon},
     event::{KeyMods, MouseButton},
@@ -8,13 +10,14 @@ use crate::{
 use winapi::{
     shared::{
         hidusage::{HID_USAGE_GENERIC_MOUSE, HID_USAGE_PAGE_GENERIC},
-        minwindef::{DWORD, HIWORD, LOWORD, LPARAM, LRESULT, UINT, WPARAM},
+        minwindef::{DWORD, HIWORD, LOWORD, LPARAM, LRESULT, MAX_PATH, TRUE, UINT, WPARAM},
         ntdef::NULL,
         windef::{HBRUSH, HCURSOR, HDC, HICON, HWND, POINT, RECT},
         windowsx::{GET_X_LPARAM, GET_Y_LPARAM},
     },
     um::{
         libloaderapi::{GetModuleHandleW, GetProcAddress},
+        shellapi::{DragAcceptFiles, DragQueryFileW, HDROP},
         shellscalingapi::*,
         wingdi::*,
         winuser::*,
@@ -511,6 +514,22 @@ unsafe extern "system" fn win32_wndproc(
         WM_EXITSIZEMOVE | WM_EXITMENULOOP => {
             KillTimer(hwnd, &mut payload.modal_resizing_timer as *mut _ as usize);
         }
+        WM_DROPFILES => {
+            let hdrop = wparam as HDROP;
+            let mut path: [u16; MAX_PATH] = std::mem::uninitialized();
+            let num_drops = DragQueryFileW(hdrop, u32::MAX, std::ptr::null_mut(), 0);
+
+            let mut d = crate::native_display().lock().unwrap();
+            for i in 0..num_drops {
+                let path_len =
+                    DragQueryFileW(hdrop, i, path.as_mut_ptr(), MAX_PATH as u32) as usize;
+                if path_len > 0 {
+                    let path: PathBuf = OsString::from_wide(&path[0..path_len]).into();
+                    d.dropped_files.bytes.push(std::fs::read(&path).unwrap());
+                    d.dropped_files.paths.push(path);
+                }
+            }
+        }
         _ => {}
     }
 
@@ -693,6 +712,8 @@ unsafe fn create_window(
     let dc = GetDC(hwnd);
     assert!(dc.is_null() == false);
 
+    DragAcceptFiles(hwnd, TRUE);
+
     (hwnd, dc)
 }