feat: quick record navigation

Author: jonas-grgt

ui/pages/consume_page/consume_page.go

@@ -97,10 +97,13 @@ func (m *Model) Update(msg tea.Msg) tea.Cmd {
 					selectedRow := m.rows[m.table.Cursor()]
 					selectedRecord := m.recordForRow(selectedRow)
 					m.consuming = false
+					recordIndex := m.recordIndexForRow(selectedRow)
 					return m.navigator.ToRecordDetailsPage(
 						tabs.LoadRecordDetailPageMsg{
 							Record:    &selectedRecord,
 							TopicName: m.readDetails.TopicName,
+							Records:   m.records,
+							Index:     recordIndex,
 						})
 				}
 			}
@@ -148,6 +151,18 @@ func (m *Model) recordForRow(row table.Row) kadmin.ConsumerRecord {
 	panic(fmt.Sprintf("Record not found for row: %v", row))
 }
 
+func (m *Model) recordIndexForRow(row table.Row) int {
+	offset, _ := strconv.ParseInt(row[3], 10, 64)
+	partition, _ := strconv.ParseInt(row[2], 10, 32)
+	for i, rec := range m.records {
+		if rec.Partition == partition &&
+			rec.Offset == offset {
+			return i
+		}
+	}
+	panic(fmt.Sprintf("Record not found for row: %v", row))
+}
+
 func (m *Model) createRows() []table.Row {
 	var rows []table.Row
 	for _, rec := range m.records {

ui/pages/consume_page/consume_page_test.go

@@ -653,18 +653,11 @@ func TestConsumptionPage(t *testing.T) {
 		msgs := tests.ExecuteBatchCmd(cmd)
 		assert.Len(t, msgs, 1)
 		assert.IsType(t, tabs.ToRecordDetailsPageCalledMsg{}, msgs[0])
-		assert.Equal(t, tabs.LoadRecordDetailPageMsg{
-			Record: &kadmin.ConsumerRecord{
-				Key:       "key-9",
-				Payload:   serdes.DesData{},
-				Err:       nil,
-				Partition: int64(9),
-				Offset:    int64(9),
-				Headers:   nil,
-				Timestamp: now.Add(time.Duration(9) * time.Second),
-			},
-			TopicName: "topic1",
-		}, msgs[0].(tabs.ToRecordDetailsPageCalledMsg).Msg)
+		msg := msgs[0].(tabs.ToRecordDetailsPageCalledMsg).Msg
+		assert.Equal(t, "key-9", msg.Record.Key)
+		assert.Equal(t, "topic1", msg.TopicName)
+		assert.Len(t, msg.Records, 10)
+		assert.Equal(t, 9, msg.Index)
 
 		m.Update(tests.Key(tea.KeyF3))
 		m.Update(tests.Key(tea.KeyLeft))
@@ -677,18 +670,11 @@ func TestConsumptionPage(t *testing.T) {
 		msgs = tests.ExecuteBatchCmd(cmd)
 		assert.Len(t, msgs, 1)
 		assert.IsType(t, tabs.ToRecordDetailsPageCalledMsg{}, msgs[0])
-		assert.Equal(t, tabs.LoadRecordDetailPageMsg{
-			Record: &kadmin.ConsumerRecord{
-				Key:       "key-9",
-				Payload:   serdes.DesData{},
-				Err:       nil,
-				Partition: int64(9),
-				Offset:    int64(9),
-				Headers:   nil,
-				Timestamp: now.Add(time.Duration(9) * time.Second),
-			},
-			TopicName: "topic1",
-		}, msgs[0].(tabs.ToRecordDetailsPageCalledMsg).Msg)
+		msg = msgs[0].(tabs.ToRecordDetailsPageCalledMsg).Msg
+		assert.Equal(t, "key-9", msg.Record.Key)
+		assert.Equal(t, "topic1", msg.TopicName)
+		assert.Len(t, msg.Records, 10)
+		assert.Equal(t, 9, msg.Index)
 	})
 
 	t.Run("Search by key", func(t *testing.T) {
@@ -761,7 +747,7 @@ func TestConsumptionPage(t *testing.T) {
 		m.Update(kadmin.ConsumerRecordReceived{
 			Records: records,
 		})
-	
+
 		m.View(tests.NewKontext(), tests.Renderer)
 
 		kb := tests.NewKeyboard(m)

ui/pages/record_details_page/record_details_page.go

@@ -39,6 +39,8 @@ const (
 type Model struct {
 	notifierCmdbar *cmdbar.NotifierCmdBar
 	record         *kadmin.ConsumerRecord
+	records        []kadmin.ConsumerRecord
+	recordIndex    int
 	recordVp       *viewport.Model
 	headerValueVp  *viewport.Model
 	topicName      string
@@ -68,6 +70,12 @@ type CopyErrorMsg struct {
 	Err error
 }
 
+type NavigateToNextRecordMsg struct {
+}
+
+type NavigateToPrevRecordMsg struct {
+}
+
 func (m *Model) View(ktx *kontext.ProgramKtx, renderer *ui.Renderer) string {
 
 	notifierCmdbarView := m.notifierCmdbar.View(ktx, renderer)
@@ -102,6 +110,10 @@ func (m *Model) Update(msg tea.Msg) tea.Cmd {
 		switch msg.String() {
 		case "esc":
 			return ui.PublishMsg(nav.LoadCachedConsumptionPageMsg{})
+		case "ctrl+n":
+			cmds = m.handleNavigateToNext(cmds)
+		case "ctrl+p":
+			cmds = m.handleNavigateToPrev(cmds)
 		case "h", "left", "right":
 			if len(m.record.Headers) >= 1 {
 				m.focus = !m.focus
@@ -117,6 +129,10 @@ func (m *Model) Update(msg tea.Msg) tea.Cmd {
 		default:
 			cmds = m.updatedFocussedArea(msg, cmds)
 		}
+	case NavigateToNextRecordMsg:
+		cmds = m.loadRecordAtIndex(m.recordIndex+1, cmds)
+	case NavigateToPrevRecordMsg:
+		cmds = m.loadRecordAtIndex(m.recordIndex-1, cmds)
 	}
 
 	return tea.Batch(cmds...)
@@ -274,6 +290,66 @@ func (m *Model) handleCopy(cmds []tea.Cmd) []tea.Cmd {
 	return cmds
 }
 
+func (m *Model) handleNavigateToNext(cmds []tea.Cmd) []tea.Cmd {
+	if m.recordIndex >= len(m.records)-1 {
+		m.notifierCmdbar.Notifier.ShowError(fmt.Errorf("no more records"))
+		return cmds
+	}
+	cmds = append(cmds, ui.PublishMsg(NavigateToNextRecordMsg{}))
+	return cmds
+}
+
+func (m *Model) handleNavigateToPrev(cmds []tea.Cmd) []tea.Cmd {
+	if m.recordIndex <= 0 {
+		m.notifierCmdbar.Notifier.ShowError(fmt.Errorf("no previous records"))
+		return cmds
+	}
+	cmds = append(cmds, ui.PublishMsg(NavigateToPrevRecordMsg{}))
+	return cmds
+}
+
+func (m *Model) loadRecordAtIndex(index int, cmds []tea.Cmd) []tea.Cmd {
+	if index < 0 || index >= len(m.records) {
+		return cmds
+	}
+	m.record = &m.records[index]
+	m.recordIndex = index
+	m.resetViews()
+	m.rebuildHeaderRows()
+	m.updateMetaInfo()
+	return cmds
+}
+
+func (m *Model) rebuildHeaderRows() {
+	sort.SliceStable(m.record.Headers, func(i, j int) bool {
+		return m.record.Headers[i].Key < m.record.Headers[j].Key
+	})
+	m.headerRows = nil
+	for _, header := range m.record.Headers {
+		m.headerRows = append(m.headerRows, table.Row{header.Key})
+	}
+}
+
+func (m *Model) resetViews() {
+	m.recordVp = nil
+	m.schemaVp = nil
+}
+
+func (m *Model) updateMetaInfo() {
+	key := m.record.Key
+	if key == "" {
+		key = "<null>"
+	}
+	m.metaInfo = fmt.Sprintf("key: %s\ntimestamp: %s", key, m.record.Timestamp.Format(time.UnixDate))
+	if m.record.Err != nil {
+		m.err = m.record.Err
+		m.notifierCmdbar.Notifier.ShowError(m.record.Err)
+	} else {
+		m.err = nil
+		m.payload = ui.PrettyPrintJson(m.record.Payload.Value)
+	}
+}
+
 func (m *Model) updatedFocussedArea(msg tea.Msg, cmds []tea.Cmd) []tea.Cmd {
 	// only update the component if no error is present
 	if m.err != nil {
@@ -314,6 +390,13 @@ func (m *Model) Shortcuts() []statusbar.Shortcut {
 			{"Copy " + whatToCopy, "c"},
 		}
 
+		if len(m.records) > 1 {
+			shortcuts = append(shortcuts, []statusbar.Shortcut{
+				{"Next Record", "ctrl+n"},
+				{"Prev Record", "ctrl+p"},
+			}...)
+		}
+
 		if m.config.ActiveCluster().HasSchemaRegistry() && m.focus == mainViewFocus {
 			shortcuts = append(shortcuts, statusbar.Shortcut{
 				Name:       "Toggle Record/Schema",
@@ -337,6 +420,8 @@ func (m *Model) Title() string {
 func New(
 	record *kadmin.ConsumerRecord,
 	topicName string,
+	records []kadmin.ConsumerRecord,
+	recordIndex int,
 	clipWriter clipper.Writer,
 	ktx *kontext.ProgramKtx,
 ) *Model {
@@ -402,6 +487,8 @@ func New(
 
 	return &Model{
 		record:         record,
+		records:        records,
+		recordIndex:    recordIndex,
 		topicName:      topicName,
 		headerKeyTable: &headersTable,
 		focus:          mainViewFocus,