Fix clipboard copying on Wayland

Author: replydev

src/utils.rs

@@ -1,5 +1,6 @@
 use base64::{engine::general_purpose, Engine as _};
 use copypasta_ext::prelude::*;
+use copypasta_ext::wayland_bin::WaylandBinClipboardContext;
 use copypasta_ext::x11_bin::ClipboardContext as BinClipboardContext;
 use copypasta_ext::x11_fork::ClipboardContext as ForkClipboardContext;
 use crossterm::style::Print;
@@ -90,15 +91,8 @@ pub fn verified_password(message: &str, minimum_length: usize) -> String {
     }
 }
 
-fn in_ssh_shell() -> bool {
-    return !env::var("SSH_CONNECTION")
-        .unwrap_or_default()
-        .trim()
-        .is_empty();
-}
-
 pub fn copy_string_to_clipboard(content: String) -> Result<CopyType, ()> {
-    if in_ssh_shell()
+    if env_var_set("SSH_CONNECTION")
         && crossterm::execute!(
             io::stdout(),
             Print(format!(
@@ -111,6 +105,12 @@ pub fn copy_string_to_clipboard(content: String) -> Result<CopyType, ()> {
         // We do not use copypasta_ext::osc52 module because we have enabled terminal raw mode, so we print with crossterm utilities
         // Check https://github.com/timvisee/rust-clipboard-ext/blob/371df19d2f961882a21c957f396d1e24548d1f28/src/osc52.rs#L92
         Ok(CopyType::OSC52)
+    } else if env_var_set("WAYLAND_DISPLAY")
+        && WaylandBinClipboardContext::new()
+            .and_then(|mut ctx| ctx.set_contents(content.clone()))
+            .is_ok()
+    {
+        Ok(CopyType::Native)
     } else if BinClipboardContext::new()
         .and_then(|mut ctx| ctx.set_contents(content.clone()))
         .is_ok()
@@ -123,3 +123,9 @@ pub fn copy_string_to_clipboard(content: String) -> Result<CopyType, ()> {
         Err(())
     }
 }
+
+fn env_var_set(env_var: &str) -> bool {
+    env::var(env_var)
+        .map(|v| !v.trim().is_empty())
+        .unwrap_or(false)
+}