Add storing clipboard data logic

Author: unsecretised

src/app.rs

@@ -1,8 +1,9 @@
-use crate::commands::Function;
+use crate::commands::{ClipBoardContentType, ClipboardContent, Function};
 use crate::config::Config;
 use crate::macos::{focus_this_app, transform_process_to_ui_element};
 use crate::{macos, utils::get_installed_apps};
 
+use arboard::{Clipboard, ImageData};
 use global_hotkey::{GlobalHotKeyEvent, HotKeyState};
 use iced::futures::SinkExt;
 use iced::{
@@ -116,6 +117,8 @@ pub enum Message {
     WindowFocusChanged(Id, bool),
     ClearSearchQuery,
     ReloadConfig,
+    ClipboardHistory(ClipBoardContentType),
+    ShowClipboardHistory,
     _Nothing,
 }
 
@@ -148,6 +151,7 @@ pub struct Tile {
     frontmost: Option<Retained<NSRunningApplication>>,
     config: Config,
     open_hotkey_id: u32,
+    clipboard_content: Vec<ClipboardContent>,
 }
 
 impl Tile {
@@ -197,6 +201,7 @@ impl Tile {
                 config: config.clone(),
                 theme: config.theme.to_owned().to_iced_theme(),
                 open_hotkey_id: keybind_id,
+                clipboard_content: vec![],
             },
             Task::batch([open.map(|_| Message::OpenWindow)]),
         )
@@ -347,6 +352,16 @@ impl Tile {
                 }
             }
 
+            Message::ClipboardHistory(clip_content) => {
+                self.clipboard_content
+                    .push(ClipboardContent::from_content_type(clip_content));
+                Task::none()
+            }
+
+            Message::ShowClipboardHistory => {
+                Task::none()
+            },
+
             Message::_Nothing => Task::none(),
         }
     }
@@ -391,6 +406,7 @@ impl Tile {
         Subscription::batch([
             Subscription::run(handle_hotkeys),
             Subscription::run(handle_hot_reloading),
+            Subscription::run(handle_clipboard_history),
             window::close_events().map(Message::HideWindow),
             keyboard::listen().filter_map(|event| {
                 if let keyboard::Event::KeyPressed { key, .. } = event {
@@ -494,3 +510,31 @@ fn handle_hotkeys() -> impl futures::Stream<Item = Message> {
         }
     })
 }
+
+fn handle_clipboard_history() -> impl futures::Stream<Item = Message> {
+    stream::channel(100, async |mut output| {
+        let mut clipboard = Clipboard::new().unwrap();
+        let mut prev_byte_rep: Option<ClipBoardContentType> = None;
+
+        loop {
+            let byte_rep = if let Ok(a) = clipboard.get_image() {
+                Some(ClipBoardContentType::Image(a))
+            } else if let Ok(a) = clipboard.get_text() {
+                Some(ClipBoardContentType::Text(a))
+            } else {
+                None
+            };
+
+            if byte_rep != prev_byte_rep
+                && let Some(content) = &byte_rep
+            {
+                output
+                    .send(Message::ClipboardHistory(content.to_owned()))
+                    .await
+                    .ok();
+                prev_byte_rep = byte_rep;
+            }
+            tokio::time::sleep(Duration::from_millis(10)).await;
+        }
+    })
+}

src/commands.rs

@@ -1,10 +1,10 @@
 use std::process::Command;
 
-use arboard::Clipboard;
+use arboard::{Clipboard, ImageData};
 use objc2_app_kit::NSWorkspace;
 use objc2_foundation::NSURL;
 
-use crate::config::Config;
+use crate::{config::Config, utils::get_time_since_epoch};
 
 #[derive(Debug, Clone)]
 pub enum Function {
@@ -63,3 +63,39 @@ impl Function {
         }
     }
 }
+
+#[derive(Debug, Clone)]
+pub struct ClipboardContent {
+    pub content_type: ClipBoardContentType,
+    pub copied_since_epoch: i64,
+}
+
+#[derive(Debug, Clone)]
+pub enum ClipBoardContentType {
+    Text(String),
+    Image(ImageData<'static>),
+}
+
+impl PartialEq for ClipBoardContentType {
+    fn eq(&self, other: &Self) -> bool {
+        if let Self::Text(a) = self
+            && let Self::Text(b) = other
+        {
+            return a == b;
+        } else if let Self::Image(image_data) = self
+            && let Self::Image(other_image_data) = other
+        {
+            return image_data.bytes == other_image_data.bytes;
+        }
+        false
+    }
+}
+
+impl ClipboardContent {
+    pub fn from_content_type(content_type: ClipBoardContentType) -> ClipboardContent {
+        Self {
+            content_type,
+            copied_since_epoch: get_time_since_epoch(),
+        }
+    }
+}

src/utils.rs

@@ -3,6 +3,7 @@ use std::{
     io::Write,
     path::{Path, PathBuf},
     process::exit,
+    time::SystemTime,
 };
 
 use global_hotkey::hotkey::Code;
@@ -287,3 +288,10 @@ pub fn to_key_code(key_str: &str) -> Option<Code> {
         _ => None,
     }
 }
+
+pub fn get_time_since_epoch() -> i64 {
+    SystemTime::now()
+        .duration_since(SystemTime::UNIX_EPOCH)
+        .unwrap()
+        .as_millis() as i64
+}