From 34c22ed63b743a594fe4913d7f3d6ca882ca7e38 Mon Sep 17 00:00:00 2001
From: Jonas Widen <jonas.widen@widens.eu>
Date: Sun, 16 Mar 2025 19:30:15 +0100
Subject: [PATCH] Restore chat history

---
 lua/gemini/chat.lua | 48 +++++++++++++++++++++++++++------------------
 1 file changed, 29 insertions(+), 19 deletions(-)

diff --git a/lua/gemini/chat.lua b/lua/gemini/chat.lua
index 45ab353..3b67cd2 100644
--- a/lua/gemini/chat.lua
+++ b/lua/gemini/chat.lua
@@ -113,29 +113,39 @@ end
 
 function M.update_content(content, is_new_chat, is_thinking)
     vim.api.nvim_buf_set_option(state.bufnr, 'modifiable', true)
-
     local lines = vim.split(content, "\n")
-    
-    -- Only add separator if:
-    -- 1. Not the first message
-    -- 2. Not a thinking message
-    -- 3. There's existing content in the buffer
-    if not is_thinking and not is_new_chat and vim.api.nvim_buf_line_count(state.bufnr) > 0 then
-        local separator = "━━━━━━━━━━━━━━━━━━━━━━━━━━"
-        table.insert(lines, 1, separator)
-    end
 
-    if is_new_chat then
-        -- If it's a new chat but we have history, append to existing content
-        if #state.conversation_history > 0 and not is_thinking then
-            local current_lines = vim.api.nvim_buf_get_lines(state.bufnr, 0, -1, false)
-            for i = #current_lines, 1, -1 do
-                table.insert(lines, 1, current_lines[i])
-            end
-        end
+    if is_thinking then
+        -- For thinking message, just replace/append without affecting history
         vim.api.nvim_buf_set_lines(state.bufnr, 0, -1, false, lines)
     else
-        vim.api.nvim_buf_set_lines(state.bufnr, -1, -1, false, lines)
+        if is_new_chat then
+            -- For new messages, show the complete history
+            local display_lines = {}
+            for i, msg in ipairs(state.conversation_history) do
+                if i > 1 then  -- Add separator before messages (except first)
+                    table.insert(display_lines, "━━━━━━━━━━━━━━━━━━━━━━━━━━")
+                end
+                local msg_lines = vim.split(msg.content, "\n")
+                for _, line in ipairs(msg_lines) do
+                    table.insert(display_lines, line)
+                end
+            end
+            -- Add the new content
+            if #state.conversation_history > 0 then
+                table.insert(display_lines, "━━━━━━━━━━━━━━━━━━━━━━━━━━")
+            end
+            for _, line in ipairs(lines) do
+                table.insert(display_lines, line)
+            end
+            vim.api.nvim_buf_set_lines(state.bufnr, 0, -1, false, display_lines)
+        else
+            -- For subsequent messages in the same chat
+            if #state.conversation_history > 0 then
+                table.insert(lines, 1, "━━━━━━━━━━━━━━━━━━━━━━━━━━")
+            end
+            vim.api.nvim_buf_set_lines(state.bufnr, -1, -1, false, lines)
+        end
     end
 
     vim.api.nvim_buf_set_option(state.bufnr, 'modifiable', false)