tui: call handleResize on states before transitioning

pkg/tui/chat.go

@@ -214,6 +214,16 @@ func (m chatModel) Init() tea.Cmd {
 	)
 }
 
+func (m *chatModel) handleResize(width, height int) {
+	m.width, m.height = width, height
+	m.content.Width = width
+	m.input.SetWidth(width - m.input.FocusedStyle.Base.GetHorizontalFrameSize())
+	if len(m.messages) > 0 {
+		m.rebuildMessageCache()
+		m.updateContent()
+	}
+}
+
 func (m chatModel) Update(msg tea.Msg) (chatModel, tea.Cmd) {
 	var cmds []tea.Cmd
 	switch msg := msg.(type) {
@@ -222,12 +232,7 @@ func (m chatModel) Update(msg tea.Msg) (chatModel, tea.Cmd) {
 			cmds = append(cmds, m.loadConversation(m.opts.convShortname))
 		}
 	case tea.WindowSizeMsg:
-		m.width = msg.Width
-		m.height = msg.Height
-		m.content.Width = msg.Width
-		m.input.SetWidth(msg.Width - m.input.FocusedStyle.Base.GetHorizontalBorderSize())
-		m.rebuildMessageCache()
-		m.updateContent()
+		m.handleResize(msg.Width, msg.Height)
 	case msgTempfileEditorClosed:
 		contents := string(msg)
 		switch m.editorTarget {

pkg/tui/conversations.go

@@ -66,13 +66,18 @@ func (m conversationsModel) Init() tea.Cmd {
 	return nil
 }
 
+func (m *conversationsModel) handleResize(width, height int) {
+	m.width, m.height = width, height
+	m.content.Width = width
+}
+
 func (m conversationsModel) Update(msg tea.Msg) (conversationsModel, tea.Cmd) {
 	var cmds []tea.Cmd
 	switch msg := msg.(type) {
 	case msgChangeState:
 		cmds = append(cmds, m.loadConversations())
 	case tea.WindowSizeMsg:
-		m.content.Width = msg.Width
+		m.handleResize(msg.Width, msg.Height)
 	case msgConversationsLoaded:
 		m.conversations = msg
 		m.content.SetContent(m.renderConversationList())

pkg/tui/tui.go

@@ -125,12 +125,15 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 			return m, cmd
 		}
 	case msgChangeState:
+		switch msg {
+		case stateChat:
+			m.chat.handleResize(m.width, m.height)
+		case stateConversations:
+			m.conversations.handleResize(m.width, m.height)
+		}
 		m.state = state(msg)
 	case tea.WindowSizeMsg:
-		w, h := msg.Width, msg.Height
-		m.width, m.height = w, h
-		m.chat.width, m.chat.height = w, h
-		m.conversations.width, m.conversations.height = w, h
+		m.width, m.height = msg.Width, msg.Height
 	case msgError:
 		m.err = msg
 	}