Re-enable mouse tracking after the external editor closes

Opening the description in $EDITOR with ^e and then quitting left the
mouse unresponsive: clicks and wheel scrolling stopped working while
keyboard navigation still did.

The editor is launched with tea.ExecProcess, which releases the terminal
before running the command and calls Bubble Tea's RestoreTerminal when it
returns. RestoreTerminal re-enables the alt-screen, bracketed paste, and
focus reporting, but it does not re-enable mouse tracking. The editor
(e.g. vim) disables mouse reporting on exit, so once control returns to
the TUI the terminal no longer emits mouse events.

Re-arm mouse mode when the editor-finished message arrives by batching
tea.EnableMouseCellMotion with the handler's command. That re-enables
cell-motion and SGR mouse reporting, matching the WithMouseCellMotion
option the program starts with, on every editor-return path (success or
error).

Author: skarim

internal/tui/submitview/model.go

@@ -214,7 +214,13 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		return m.handleMouse(msg)
 
 	case editorFinishedMsg:
-		return m.handleEditorFinished(msg)
+		updated, cmd := m.handleEditorFinished(msg)
+		// After tea.ExecProcess runs the external editor, Bubble Tea's
+		// RestoreTerminal re-enables the alt-screen, bracketed paste, and focus
+		// reporting but NOT mouse tracking, so the terminal stops emitting mouse
+		// events once the editor closes. Re-enable cell-motion mouse mode (which
+		// also re-arms SGR mode) to match the program's startup options.
+		return updated, tea.Batch(cmd, tea.EnableMouseCellMotion)
 	}
 
 	return m, nil

internal/tui/submitview/preview_test.go

@@ -2,6 +2,7 @@ package submitview
 
 import (
 	"os"
+	"reflect"
 	"testing"
 
 	tea "github.com/charmbracelet/bubbletea"
@@ -66,6 +67,55 @@ func TestHandleEditorFinished_UpdatesDescription(t *testing.T) {
 	assert.True(t, os.IsNotExist(statErr))
 }
 
+// cmdMessages flattens the message(s) a command (possibly a tea.Batch) emits.
+func cmdMessages(cmd tea.Cmd) []tea.Msg {
+	if cmd == nil {
+		return nil
+	}
+	switch msg := cmd().(type) {
+	case tea.BatchMsg:
+		var out []tea.Msg
+		for _, c := range msg {
+			out = append(out, cmdMessages(c)...)
+		}
+		return out
+	default:
+		return []tea.Msg{msg}
+	}
+}
+
+func containsType(msgs []tea.Msg, want tea.Msg) bool {
+	wt := reflect.TypeOf(want)
+	for _, m := range msgs {
+		if reflect.TypeOf(m) == wt {
+			return true
+		}
+	}
+	return false
+}
+
+func TestEditorFinished_ReenablesMouse(t *testing.T) {
+	// Bubble Tea's RestoreTerminal (run after the external editor exits) does not
+	// re-enable mouse tracking, so the editor-finished handler must re-arm it or
+	// the mouse stops working after the editor closes.
+	want := tea.EnableMouseCellMotion()
+
+	t.Run("success", func(t *testing.T) {
+		m := testModel(t, newNodes())
+		f, err := os.CreateTemp(t.TempDir(), "ed-*.md")
+		require.NoError(t, err)
+		require.NoError(t, f.Close())
+		_, cmd := m.Update(editorFinishedMsg{path: f.Name()})
+		assert.True(t, containsType(cmdMessages(cmd), want), "mouse tracking is re-enabled after a clean editor exit")
+	})
+
+	t.Run("editor error", func(t *testing.T) {
+		m := testModel(t, newNodes())
+		_, cmd := m.Update(editorFinishedMsg{path: "/no/such/file", err: assert.AnError})
+		assert.True(t, containsType(cmdMessages(cmd), want), "mouse tracking is re-enabled even when the editor errored")
+	})
+}
+
 func TestRenderMarkdown(t *testing.T) {
 	assert.Contains(t, renderMarkdown("", 40), "no description")
 	// Glamour styles each word as a separate ANSI span, so assert on the words