From 36e884c4b58e2d81fc44b9f3201543632cde33fa Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 13 Nov 2024 06:58:13 +0800
Subject: refactor: no side effect `--save-session` (#987)

---
 src/config/mod.rs     | 10 ++++++++++
 src/config/session.rs | 11 ++++++++++-
 2 files changed, 20 insertions(+), 1 deletion(-)

(limited to 'src/config')

diff --git a/src/config/mod.rs b/src/config/mod.rs
index 2455588..aa0fe00 100644
--- a/src/config/mod.rs
+++ b/src/config/mod.rs
@@ -1108,6 +1108,16 @@ impl Config {
         self.last_message = None;
         Ok(())
     }
+
+    pub fn set_append_conversation(&mut self) -> Result<()> {
+        if let Some(session) = self.session.as_mut() {
+            session.set_append_conversation();
+        } else {
+            bail!("No session")
+        }
+        Ok(())
+    }
+
     pub fn list_sessions(&self) -> Vec<String> {
         list_file_names(self.sessions_dir(), ".yaml")
     }
diff --git a/src/config/session.rs b/src/config/session.rs
index 31a6e02..c072d1c 100644
--- a/src/config/session.rs
+++ b/src/config/session.rs
@@ -50,6 +50,8 @@ pub struct Session {
     #[serde(skip)]
     dirty: bool,
     #[serde(skip)]
+    append_conversation: bool,
+    #[serde(skip)]
     compressing: bool,
 }
 
@@ -276,6 +278,10 @@ impl Session {
         }
     }
 
+    pub fn set_append_conversation(&mut self) {
+        self.append_conversation = true;
+    }
+
     pub fn set_compress_threshold(&mut self, value: Option<usize>) {
         if self.compress_threshold != value {
             self.compress_threshold = value;
@@ -308,7 +314,10 @@ impl Session {
     }
 
     pub fn exit(&mut self, session_dir: &Path, is_repl: bool) -> Result<()> {
-        let save_session = self.save_session();
+        let mut save_session = self.save_session();
+        if self.append_conversation {
+            save_session = Some(true);
+        }
         if self.dirty && save_session != Some(false) {
             let mut session_name = self.name().to_string();
             if save_session.is_none() {
-- 
cgit v1.2.3