fix: combine pipe and cli args, disable highlight when pipe (#23)

1 files changed, 7 insertions, 3 deletions

diff --git a/src/main.rs b/src/main.rs
index c7ae56e..0f12db5 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -45,9 +45,13 @@ fn main() -> Result<()> {
     config.borrow_mut().role = role;
     let client = ChatGptClient::init(config.clone())?;
     if atty::isnt(atty::Stream::Stdin) {
-        let mut text = String::new();
-        stdin().read_to_string(&mut text)?;
-        start_directive(client, config, &text)
+        let mut input = String::new();
+        config.borrow_mut().highlight = false;
+        stdin().read_to_string(&mut input)?;
+        if let Some(text) = text {
+            input = format!("{text}\n{input}");
+        }
+        start_directive(client, config, &input)
     } else {
         match text {
             Some(text) => start_directive(client, config, &text),