From aadffb17a3cbbfc5e3c0dad4c8587eff79aaae9b Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Sun, 5 Mar 2023 19:16:41 +0800
Subject: feat: repl command replaces `.editor` with `.multiline` (#18)
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

You can use `.multiline` command to put multiline content

```
〉.multiline {
::: abc
::: cdf
::: }
```
It's much easier than `.editor`.
---
 src/repl.rs | 65 ++++++++++++++++++++++++++++++++++++++++++++++---------------
 1 file changed, 49 insertions(+), 16 deletions(-)

(limited to 'src/repl.rs')

diff --git a/src/repl.rs b/src/repl.rs
index 2a5947f..b507d89 100644
--- a/src/repl.rs
+++ b/src/repl.rs
@@ -9,7 +9,7 @@ use crossbeam::sync::WaitGroup;
 use reedline::{
     default_emacs_keybindings, ColumnarMenu, DefaultCompleter, DefaultPrompt, DefaultPromptSegment,
     Emacs, FileBackedHistory, KeyCode, KeyModifiers, Keybindings, Reedline, ReedlineEvent,
-    ReedlineMenu, Signal,
+    ReedlineMenu, Signal, ValidationResult, Validator,
 };
 use std::cell::RefCell;
 use std::fs::File;
@@ -21,10 +21,10 @@ use std::thread::spawn;
 const REPL_COMMANDS: [(&str, &str); 10] = [
     (".role", "Specifies the role the AI will play"),
     (".clear role", "Clear the currently selected role"),
-    (".editor", "Enter editor mode"),
-    (".copy", "Copy last reply message"),
     (".history", "Print the history"),
     (".clear history", "Clear the history"),
+    (".multiline", "Enter multiline editor mode"),
+    (".copy", "Copy last reply message"),
     (".info", "Print the information"),
     (".help", "Print this help message"),
     (".exit", "Exit the REPL"),
@@ -51,7 +51,9 @@ impl Repl {
             .with_menu(menu)
             .with_edit_mode(edit_mode)
             .with_quick_completions(true)
-            .with_partial_completions(true);
+            .with_partial_completions(true)
+            .with_validator(Box::new(ReplValidator))
+            .with_ansi_colors(true);
         let prompt = Self::create_prompt();
         Ok(Self { editor, prompt })
     }
@@ -132,14 +134,16 @@ impl Repl {
                     Some(name) => handler.handle(ReplCmd::SetRole(name.to_string()))?,
                     None => dump("Usage: .role <name>", 2),
                 },
-                ".editor" => {
-                    dump(
-                        "// Entering editor mode (Ctrl+D to finish, Ctrl+C to cancel)",
-                        1,
-                    );
-                    let content = term::edit()?;
-                    dump("", 1);
-                    handler.handle(ReplCmd::Submit(content))?;
+                ".info" => {
+                    handler.handle(ReplCmd::Info)?;
+                }
+                ".multiline" => {
+                    let text = args.unwrap_or_default().to_string();
+                    if text.starts_with('{') && text.ends_with('}') {
+                        handler.handle(ReplCmd::Submit(text))?;
+                    } else {
+                        dump("Usage: .multiline { put your content here }", 2);
+                    }
                 }
                 ".copy" => {
                     let reply = handler.get_reply();
@@ -147,12 +151,9 @@ impl Repl {
                         dump("No reply messages that can be copied", 1)
                     } else {
                         copy(&reply)?;
-                        dump("Pasted", 1);
+                        dump("Copied", 1);
                     }
                 }
-                ".info" => {
-                    handler.handle(ReplCmd::Info)?;
-                }
                 _ => dump_unknown_command(),
             }
         } else {
@@ -207,6 +208,38 @@ impl Repl {
         ))
     }
 }
+pub struct ReplValidator;
+
+impl Validator for ReplValidator {
+    fn validate(&self, line: &str) -> ValidationResult {
+        if line.split('"').count() % 2 == 0 || incomplete_brackets(line) {
+            ValidationResult::Incomplete
+        } else {
+            ValidationResult::Complete
+        }
+    }
+}
+
+fn incomplete_brackets(line: &str) -> bool {
+    let mut balance: Vec<char> = Vec::new();
+    if !line.trim_start().starts_with(".multiline") {
+        return false;
+    }
+
+    for c in line.chars() {
+        if c == '{' {
+            balance.push('}');
+        } else if c == '}' {
+            if let Some(last) = balance.last() {
+                if last == &c {
+                    balance.pop();
+                }
+            }
+        }
+    }
+
+    !balance.is_empty()
+}
 
 pub struct ReplCmdHandler {
     client: ChatGptClient,
-- 
cgit v1.2.3