trying another time

Author: jameswexler1

internal/ui/model.go

@@ -1453,10 +1453,8 @@ func removeMessageFromSlice(messages []store.Message, messageID string) []store.
 func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	if token, ok := shiftedEnterTokenFromMsg(msg); ok {
 		updated, cmd := m.handleSpecialKeyToken(token)
-		if next, ok := updated.(Model); ok {
-			return next.withPreviewCmd(cmd)
-		}
-		return updated, cmd
+		next := updated.(Model)
+		return next.withPreviewCmd(cmd)
 	}
 
 	switch msg := msg.(type) {
@@ -1557,16 +1555,12 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		return m.handlePickedSticker(msg)
 	case mediaPreviewReadyMsg:
 		updated, cmd := m.handleMediaPreviewReady(msg)
-		if next, ok := updated.(Model); ok {
-			return next.withPreviewCmd(cmd)
-		}
-		return updated, cmd
+		next := updated.(Model)
+		return next.withPreviewCmd(cmd)
 	case mediaDownloadedMsg:
 		updated, cmd := m.handleMediaDownloaded(msg)
-		if next, ok := updated.(Model); ok {
-			return next.withPreviewCmd(cmd)
-		}
-		return updated, cmd
+		next := updated.(Model)
+		return next.withPreviewCmd(cmd)
 	case mediaSavedMsg:
 		return m.handleMediaSaved(msg)
 	case audioStartedMsg:
@@ -1681,10 +1675,8 @@ func (m Model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		return next, batchCmds(cmd, nextPoll)
 	case tea.KeyMsg:
 		updated, cmd := m.handleKey(msg)
-		if next, ok := updated.(Model); ok {
-			return next.withPreviewCmd(cmd)
-		}
-		return updated, cmd
+		next := updated.(Model)
+		return next.withPreviewCmd(cmd)
 	}
 
 	return m, nil

internal/ui/model_test.go

@@ -7550,6 +7550,26 @@ func TestPendingMediaOverlayDoesNotRenderLowResolutionFallback(t *testing.T) {
 	}
 }
 
+func TestKeyMsgRunsPreviewSyncForVisibleOverlay(t *testing.T) {
+	localPath := filepath.Join(t.TempDir(), "photo.jpg")
+	if err := os.WriteFile(localPath, []byte("fake"), 0o644); err != nil {
+		t.Fatalf("WriteFile() error = %v", err)
+	}
+	model := mediaTestModel(localPath, media.BackendUeberzugPP)
+	cacheOverlayPreview(t, &model, localPath)
+	model.mode = ModeNormal
+	model.focus = FocusMessages
+
+	updated, cmd := model.Update(tea.KeyMsg{Type: tea.KeyRunes, Runes: []rune("x")})
+	model = updated.(Model)
+	if cmd == nil {
+		t.Fatal("KeyMsg returned nil command, want overlay sync from withPreviewCmd")
+	}
+	if !model.overlaySyncPending || model.overlayPendingSignature == "" {
+		t.Fatalf("overlay pending = %v signature %q, want key-driven preview sync", model.overlaySyncPending, model.overlayPendingSignature)
+	}
+}
+
 func TestSnapshotReloadPausesTerminalOverlays(t *testing.T) {
 	localPath := filepath.Join(t.TempDir(), "photo.jpg")
 	if err := os.WriteFile(localPath, []byte("fake"), 0o644); err != nil {