Added support for --to-clipboard in Windows

conradludgate · Aug 7, 2020 · 3a74de1 · 3a74de1
1 parent c7ffab8
commit 3a74de1
Show file tree

Hide file tree

Showing 3 changed files with 43 additions and 1 deletion.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -29,6 +29,9 @@ log = "0.4.11"
 [target.'cfg(target_os = "macos")'.dependencies]
 pasteboard = "0.1.1"
 
+[target.'cfg(target_os = "windows")'.dependencies]
+clipboard-win = "4.0.2"
+
 [dependencies.image]
 version = "0.23"
 default-features = false

diff --git a/src/bin.rs b/src/bin.rs
@@ -14,6 +14,8 @@ use syntect::util::LinesWithEndings;
 use {image::ImageOutputFormat, pasteboard::Pasteboard};
 #[cfg(target_os = "linux")]
 use {image::ImageOutputFormat, std::process::Command};
+#[cfg(target_os = "windows")]
+use {image::ImageOutputFormat, clipboard_win::{set_clipboard, formats::RawData}};
 
 pub mod blur;
 pub mod config;
@@ -49,7 +51,16 @@ pub fn dump_image_to_clipboard(image: &DynamicImage) -> Result<(), Error> {
     Ok(())
 }
 
-#[cfg(not(any(target_os = "linux", target_os = "macos")))]
+#[cfg(target_os = "windows")]
+pub fn dump_image_to_clipboard(image: &DynamicImage) -> Result<(), Error> {
+    let mut temp: Vec<u8> = Vec::new();
+    image.write_to(&mut temp, ImageOutputFormat::Png)?;
+    set_clipboard(RawData(49488), &temp)
+        .map_err(|e| format_err!("Failed to access clipboard {}", e))?;
+    Ok(())
+}
+
+#[cfg(not(any(target_os = "linux", target_os = "macos", target_os = "windows")))]
 pub fn dump_image_to_clipboard(_image: &DynamicImage) -> Result<(), Error> {
     Err(format_err!(
         "This feature hasn't been implemented for your system"