From 00ce1575715329a21726a506b13c0c2aef7f3f82 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 4 May 2023 22:42:50 +0800
Subject: feat: add `.copy` command (#119)

---
 src/utils/mod.rs | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'src/utils')

diff --git a/src/utils/mod.rs b/src/utils/mod.rs
index 64fc01c..ec57fed 100644
--- a/src/utils/mod.rs
+++ b/src/utils/mod.rs
@@ -2,6 +2,7 @@ mod tiktoken;
 
 pub use self::tiktoken::{cl100k_base_singleton, count_tokens, text_to_tokens, tokens_to_text};
 
+use arboard::Clipboard;
 use chrono::prelude::*;
 use crossterm::style::{Color, Stylize};
 use std::io::{stdout, Write};
@@ -35,6 +36,11 @@ pub fn mask_text(text: &str, head: usize, tail: usize) -> String {
     format!("{}...{}", &text[0..head], &text[text.len() - tail..])
 }
 
+pub fn copy(src: &str) -> Result<(), arboard::Error> {
+    let mut clipboard = Clipboard::new()?;
+    clipboard.set_text(src)
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;
-- 
cgit v1.2.3