Track spreadsheet cell selection for Find/Replace

Add tracking of active spreadsheet cell editor and selected text in EditTextWindow. Implement GetSpreadsheetSelectedText to retrieve selection from the cell editor, last known selection, or cell value. Update Find/Replace to use this logic in spreadsheet mode. Handle PreparingCellForEdit to attach selection tracking and ensure cleanup on edit end and selection changes.

closes #663

Author: TheJoeFin

Text-Grab/Views/EditTextWindow.xaml

@@ -824,6 +824,7 @@
                     HorizontalGridLinesBrush="{DynamicResource ControlStrokeColorDefaultBrush}"
                     HorizontalScrollBarVisibility="Auto"
                     LoadingRow="SpreadsheetDataGrid_LoadingRow"
+                    PreparingCellForEdit="SpreadsheetDataGrid_PreparingCellForEdit"
                     PreviewKeyDown="SpreadsheetDataGrid_PreviewKeyDown"
                     PreviewMouseLeftButtonDown="SpreadsheetDataGrid_PreviewMouseLeftButtonDown"
                     PreviewMouseRightButtonDown="SpreadsheetDataGrid_PreviewMouseRightButtonDown"

Text-Grab/Views/EditTextWindow.xaml.cs

@@ -101,6 +101,8 @@ public partial class EditTextWindow : Wpf.Ui.Controls.FluentWindow
     private readonly DataTable spreadsheetTable = new();
     private readonly List<DataGridColumn> trackedSpreadsheetColumns = [];
     private List<(int RowIndex, int ColumnIndex)> selectedSpreadsheetCellCoordinates = [];
+    private System.Windows.Controls.TextBox? activeSpreadsheetCellEditor;
+    private string lastSpreadsheetCellEditorSelectedText = string.Empty;
     private EtwEditorMode editorMode = EtwEditorMode.Text;
     private SpellCheckMode spellCheckMode = SpellCheckMode.Auto;
     private bool isSyncingTextFromSpreadsheet = false;
@@ -1097,6 +1099,8 @@ private void SpreadsheetDataGrid_BeginningEdit(object sender, DataGridBeginningE
 
     private void SpreadsheetDataGrid_CellEditEnding(object sender, DataGridCellEditEndingEventArgs e)
     {
+        DetachSpreadsheetCellEditor();
+
         if (e.EditAction == DataGridEditAction.Cancel)
         {
             pendingSpreadsheetUndoState = null;
@@ -1112,6 +1116,32 @@ private void SpreadsheetDataGrid_CellEditEnding(object sender, DataGridCellEditE
             DispatcherPriority.Background);
     }
 
+    private void SpreadsheetDataGrid_PreparingCellForEdit(object sender, DataGridPreparingCellForEditEventArgs e)
+    {
+        DetachSpreadsheetCellEditor();
+
+        if (e.EditingElement is not System.Windows.Controls.TextBox cellTextBox)
+            return;
+
+        activeSpreadsheetCellEditor = cellTextBox;
+        lastSpreadsheetCellEditorSelectedText = cellTextBox.SelectedText;
+        cellTextBox.SelectionChanged += SpreadsheetCellEditor_SelectionChanged;
+    }
+
+    private void SpreadsheetCellEditor_SelectionChanged(object sender, RoutedEventArgs e)
+    {
+        if (sender is System.Windows.Controls.TextBox cellTextBox)
+            lastSpreadsheetCellEditorSelectedText = cellTextBox.SelectedText;
+    }
+
+    private void DetachSpreadsheetCellEditor()
+    {
+        if (activeSpreadsheetCellEditor is not null)
+            activeSpreadsheetCellEditor.SelectionChanged -= SpreadsheetCellEditor_SelectionChanged;
+
+        activeSpreadsheetCellEditor = null;
+    }
+
     private void SpreadsheetDataGrid_CurrentCellChanged(object sender, EventArgs e)
     {
         if (editorMode == EtwEditorMode.Spreadsheet)
@@ -1241,6 +1271,9 @@ private void SpreadsheetDataGrid_PreviewMouseRightButtonDown(object sender, Mous
 
     private void SpreadsheetDataGrid_SelectedCellsChanged(object sender, SelectedCellsChangedEventArgs e)
     {
+        if (activeSpreadsheetCellEditor is null)
+            lastSpreadsheetCellEditorSelectedText = string.Empty;
+
         UpdateSelectedSpreadsheetCellCoordinates();
 
         if (editorMode == EtwEditorMode.Spreadsheet)
@@ -1826,6 +1859,28 @@ private bool IsSpreadsheetCellEditorFocused()
             && FindVisualParent<System.Windows.Controls.TextBox>(focusedElement) is not null;
     }
 
+    private string GetSpreadsheetSelectedText()
+    {
+        if (activeSpreadsheetCellEditor is System.Windows.Controls.TextBox cellTextBox)
+            return cellTextBox.SelectedText;
+
+        if (lastSpreadsheetCellEditorSelectedText.Length > 0)
+            return lastSpreadsheetCellEditorSelectedText;
+
+        List<(int RowIndex, int ColumnIndex)> targetCells = GetSelectedOrCurrentSpreadsheetCellCoordinates();
+        if (targetCells.Count != 1)
+            return string.Empty;
+
+        (int rowIndex, int columnIndex) = targetCells[0];
+        if (rowIndex < 0 || rowIndex >= spreadsheetTable.Rows.Count
+            || columnIndex < 0 || columnIndex >= spreadsheetTable.Columns.Count)
+        {
+            return string.Empty;
+        }
+
+        return spreadsheetTable.Rows[rowIndex][columnIndex]?.ToString() ?? string.Empty;
+    }
+
     internal static bool ShouldHandleSpreadsheetDeleteKey(int selectedCellCount, bool isCellEditorFocused)
     {
         return !isCellEditorFocused && selectedCellCount > 0;
@@ -3461,10 +3516,11 @@ private void LaunchFindAndReplace()
         findAndReplaceWindow.TextEditWindow = this;
         findAndReplaceWindow.Show();
 
+        string selectedText = IsSpreadsheetMode ? GetSpreadsheetSelectedText() : PassedTextControl.SelectedText;
 
-        if (PassedTextControl.SelectedText.Length > 0)
+        if (selectedText.Length > 0)
         {
-            findAndReplaceWindow.SetFindText(PassedTextControl.SelectedText.Trim());
+            findAndReplaceWindow.SetFindText(selectedText.Trim());
             findAndReplaceWindow.SearchForText();
         }
     }