Chore: simplify search

tui.go

@@ -389,50 +389,23 @@ func performSearch(term string) {
 		colorText()
 		return
 	}
-
+	// Get formatted text and search directly in it to avoid mapping issues
-	// Get formatted text
 	formattedText := textView.GetText(true)
-	plainText, mapping := stripTags(formattedText)
+	searchTermLower := strings.ToLower(searchText)
-
+	formattedTextLower := strings.ToLower(formattedText)
-	// Find all occurrences of the search term in plain text
+	// Find all occurrences of the search term in the formatted text directly
-	plainSearchResults := []int{}
+	formattedSearchResults := []int{}
-
+	searchStart := 0
-	start := 0
 	for {
-		// Use case-insensitive search
+		pos := strings.Index(formattedTextLower[searchStart:], searchTermLower)
-		index := strings.Index(strings.ToLower(plainText[start:]), strings.ToLower(searchText))
+		if pos == -1 {
-		if index == -1 {
 			break
 		}
-
+		absolutePos := searchStart + pos
-		absoluteIndex := start + index
+		formattedSearchResults = append(formattedSearchResults, absolutePos)
-		plainSearchResults = append(plainSearchResults, absoluteIndex)
+		searchStart = absolutePos + len(searchText)
-		start = absoluteIndex + len(searchText) // Advance past the last match
 	}
-
+	if len(formattedSearchResults) == 0 {
-	if len(plainSearchResults) > 0 {
-		searchResults = make([]int, len(plainSearchResults))
-		searchResultLengths = make([]int, len(plainSearchResults))
-
-		for i, p_start := range plainSearchResults {
-			p_end_exclusive := p_start + len(searchText)
-
-			f_start := mapping[p_start]
-			var f_end_exclusive int
-			if p_end_exclusive < len(mapping) {
-				f_end_exclusive = mapping[p_end_exclusive]
-			} else {
-				// Reached the end of the text
-				f_end_exclusive = len(formattedText)
-			}
-
-			searchResults[i] = f_start
-			searchResultLengths[i] = f_end_exclusive - f_start
-		}
-
-		searchIndex = 0
-		highlightCurrentMatch()
-	} else {
 		// No matches found
 		searchResults = nil
 		searchResultLengths = nil
@@ -440,7 +413,17 @@ func performSearch(term string) {
 		if err := notifyUser("search", notification); err != nil {
 			logger.Error("failed to send notification", "error", err)
 		}
+		return
 	}
+	// Store the formatted text positions and lengths for accurate highlighting
+	searchResults = formattedSearchResults
+	// Create lengths array - all matches have the same length as the search term
+	searchResultLengths = make([]int, len(formattedSearchResults))
+	for i := range searchResultLengths {
+		searchResultLengths[i] = len(searchText)
+	}
+	searchIndex = 0
+	highlightCurrentMatch()
 }
 
 // highlightCurrentMatch highlights the current search match and scrolls to it
@@ -665,10 +648,7 @@ func init() {
 	textArea.SetBorder(true).SetTitle("input")
 	textView = tview.NewTextView().
 		SetDynamicColors(true).
-		SetRegions(true).
+		SetRegions(true)
-		SetChangedFunc(func() {
-			app.Draw()
-		})
 
 	flex = tview.NewFlex().SetDirection(tview.FlexRow).
 		AddItem(textView, 0, 40, false).
@@ -734,9 +714,6 @@ func init() {
 	position = tview.NewTextView().
 		SetDynamicColors(true).
 		SetTextAlign(tview.AlignCenter)
-	position.SetChangedFunc(func() {
-		app.Draw()
-	})
 	// Initially set up flex without search bar
 	flex = tview.NewFlex().SetDirection(tview.FlexRow).
 		AddItem(textView, 0, 40, false).
@@ -1410,7 +1387,6 @@ func init() {
 			return nil
 		}
 
-
 		if isASCII(string(event.Rune())) && !botRespMode {
 			return event
 		}