From a055d4bee697c2dbf600afbd768e840b0f040247 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 5 Mar 2023 00:19:48 +0800
Subject: feat: add `.copy` command (#10)

---
 src/repl.rs | 31 ++++++++++++++++++++-----------
 1 file changed, 20 insertions(+), 11 deletions(-)

(limited to 'src/repl.rs')

diff --git a/src/repl.rs b/src/repl.rs
index 127ec72..81a5474 100644
--- a/src/repl.rs
+++ b/src/repl.rs
@@ -2,6 +2,7 @@ use crate::client::ChatGptClient;
 use crate::config::{Config, Role};
 use crate::editor;
 use crate::render::{self, MarkdownRender};
+use crate::utils::{copy, dump};
 use anyhow::{anyhow, Result};
 use crossbeam::channel::{unbounded, Sender};
 use crossbeam::sync::WaitGroup;
@@ -12,15 +13,15 @@ use reedline::{
 };
 use std::cell::RefCell;
 use std::fs::File;
-use std::io::{stdout, Write};
 use std::sync::atomic::{AtomicBool, Ordering};
 use std::sync::Arc;
 use std::thread::spawn;
 
-const REPL_COMMANDS: [(&str, &str); 8] = [
+const REPL_COMMANDS: [(&str, &str); 9] = [
     (".clear", "Clear the screen"),
     (".clear-history", "Clear the history"),
     (".clear-role", "Clear the role status"),
+    (".copy", "Copy last reply message"),
     (".editor", "Enter multiline editor"),
     (".exit", "Exit the REPL"),
     (".help", "Print this help message"),
@@ -140,6 +141,15 @@ impl Repl {
                     dump("", 1);
                     handler.handle(ReplCmd::Input(content))?;
                 }
+                ".copy" => {
+                    let reply = handler.get_reply();
+                    if reply.is_empty() {
+                        dump("No reply messages that can be copied", 1)
+                    } else {
+                        copy(&reply)?;
+                        dump("Pasted", 1);
+                    }
+                }
                 _ => dump_unknown_command(),
             }
         } else {
@@ -200,7 +210,7 @@ pub struct ReplCmdHandler {
 
 struct ReplCmdHandlerState {
     prompt: String,
-    output: String,
+    reply: String,
     save_file: Option<File>,
 }
 
@@ -217,7 +227,7 @@ impl ReplCmdHandler {
         let state = RefCell::new(ReplCmdHandlerState {
             prompt,
             save_file,
-            output: String::new(),
+            reply: String::new(),
         });
         Ok(Self {
             client,
@@ -231,7 +241,7 @@ impl ReplCmdHandler {
         match cmd {
             ReplCmd::Input(input) => {
                 if input.is_empty() {
-                    self.state.borrow_mut().output.clear();
+                    self.state.borrow_mut().reply.clear();
                     return Ok(());
                 }
                 let prompt = self.state.borrow().prompt.to_string();
@@ -261,7 +271,7 @@ impl ReplCmdHandler {
                     &receiver.output,
                 );
                 wg.wait();
-                self.state.borrow_mut().output = receiver.output;
+                self.state.borrow_mut().reply = receiver.output;
             }
             ReplCmd::SetRole(name) => match self.config.find_role(&name) {
                 Some(v) => {
@@ -278,6 +288,10 @@ impl ReplCmdHandler {
         }
         Ok(())
     }
+
+    fn get_reply(&self) -> String {
+        self.state.borrow().reply.to_string()
+    }
 }
 
 pub struct ReplyReceiver {
@@ -322,11 +336,6 @@ pub enum RenderStreamEvent {
     Done,
 }
 
-pub fn dump<T: ToString>(text: T, newlines: usize) {
-    print!("{}{}", text.to_string(), "\n".repeat(newlines));
-    let _ = stdout().flush();
-}
-
 enum ReplCmd {
     UnsetRole,
     Input(String),
-- 
cgit v1.2.3