From d4b530653675b809ad5839c0a2c297f323a7799a Mon Sep 17 00:00:00 2001
From: Kujtim Hoxha <kujtimii.h@gmail.com>
Date: Fri, 25 Jul 2025 17:13:50 +0200
Subject: [PATCH] chore: fix editor

---
 internal/tui/components/chat/editor/editor.go | 6 +++---
 internal/tui/page/chat/chat.go                | 4 ++++
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/internal/tui/components/chat/editor/editor.go b/internal/tui/components/chat/editor/editor.go
index 4e5f0bc431eb466cea5c6c7d436234c7a5e8531b..fda718f290a2bf4eea089bb6f2804531aa224f00 100644
--- a/internal/tui/components/chat/editor/editor.go
+++ b/internal/tui/components/chat/editor/editor.go
@@ -80,7 +80,7 @@ const (
 	maxAttachments = 5
 )
 
-type openEditorMsg struct {
+type OpenEditorMsg struct {
 	Text string
 }
 
@@ -119,7 +119,7 @@ func (m *editorCmp) openEditor(value string) tea.Cmd {
 			return util.ReportWarn("Message is empty")
 		}
 		os.Remove(tmpfile.Name())
-		return openEditorMsg{
+		return OpenEditorMsg{
 			Text: strings.TrimSpace(string(content)),
 		}
 	})
@@ -204,7 +204,7 @@ func (m *editorCmp) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 				m.completionsStartIndex = 0
 			}
 		}
-	case openEditorMsg:
+	case OpenEditorMsg:
 		m.textarea.SetValue(msg.Text)
 		m.textarea.MoveToEnd()
 	case tea.KeyPressMsg:
diff --git a/internal/tui/page/chat/chat.go b/internal/tui/page/chat/chat.go
index 770644357ab2c65911f4531e5edbf60735eda3ec..253f60ee5c733045bae4ee272d64f4bf8c18a2bb 100644
--- a/internal/tui/page/chat/chat.go
+++ b/internal/tui/page/chat/chat.go
@@ -179,6 +179,10 @@ func (p *chatPage) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	case CancelTimerExpiredMsg:
 		p.isCanceling = false
 		return p, nil
+	case editor.OpenEditorMsg:
+		u, cmd := p.editor.Update(msg)
+		p.editor = u.(editor.Editor)
+		return p, cmd
 	case chat.SendMsg:
 		return p, p.sendMessage(msg.Text, msg.Attachments)
 	case chat.SessionSelectedMsg: