fix: copy first row due to failed falsey check (supabase#40909)

* updated because of zero falsey value not copying

* test: add E2E test for copying first and second row values

Adds regression test to verify that users can copy cell values from
the first row (index 0) and second row. This prevents the bug where
falsy index checks would prevent copying from row 0.

Addresses FE-2178

Author: awaseem

apps/studio/components/grid/components/menu/RowContextMenu.tsx

@@ -22,15 +22,15 @@ export const RowContextMenu = ({ rows }: RowContextMenuProps) => {
 
   function onDeleteRow(p: RowContextMenuItemProps) {
     const rowIdx = p.props?.rowIdx
-    if (!rowIdx) return
+    if (rowIdx === undefined || rowIdx === null) return
 
     const row = rows[rowIdx]
     if (row) tableEditorSnap.onDeleteRows([row])
   }
 
   function onEditRowClick(p: RowContextMenuItemProps) {
     const rowIdx = p.props?.rowIdx
-    if (!rowIdx) return
+    if (rowIdx === undefined || rowIdx === null) return
 
     const row = rows[rowIdx]
     tableEditorSnap.onEditRow(row)
@@ -39,7 +39,7 @@ export const RowContextMenu = ({ rows }: RowContextMenuProps) => {
   const onCopyCellContent = useCallback(
     (p: RowContextMenuItemProps) => {
       const rowIdx = p.props?.rowIdx
-      if (!snap.selectedCellPosition || !rowIdx) return
+      if (!snap.selectedCellPosition || rowIdx === undefined || rowIdx === null) return
 
       const row = rows[rowIdx]
       const columnKey = snap.gridColumns[snap.selectedCellPosition.idx as number].key
@@ -56,7 +56,7 @@ export const RowContextMenu = ({ rows }: RowContextMenuProps) => {
   const onCopyRowContent = useCallback(
     (p: RowContextMenuItemProps) => {
       const rowIdx = p.props?.rowIdx
-      if (!rowIdx) return
+      if (rowIdx === undefined || rowIdx === null) return
 
       const row = rows[rowIdx]
       copyToClipboard(JSON.stringify(row))

e2e/studio/features/table-editor.spec.ts

@@ -717,4 +717,66 @@ test.describe('table editor', () => {
 
     await deleteTable(page, ref, tableNameDataActions)
   })
+
+  test('copying cell values from first and second row works', async ({ page, ref }) => {
+    const tableName = 'pw_table_copy_rows'
+    const colName = 'pw_column'
+
+    // Ensure we're on editor
+    if (!page.url().includes('/editor')) {
+      await page.goto(toUrl(`/project/${ref}/editor?schema=public`))
+      await waitForTableToLoad(page, ref)
+    }
+
+    // Create table and add two rows
+    await createTable(page, ref, tableName)
+    await page.getByRole('button', { name: `View ${tableName}`, exact: true }).click()
+    await page.waitForURL(/\/editor\/\d+\?schema=public$/)
+
+    // Insert first row with value 'first_row_value'
+    await page.getByTestId('table-editor-insert-new-row').click()
+    await page.getByRole('menuitem', { name: 'Insert row Insert a new row' }).click()
+    await page.getByTestId(`${colName}-input`).fill('first_row_value')
+    await page.getByTestId('action-bar-save-row').click()
+    await waitForApiResponse(page, 'pg-meta', ref, 'query?key=', { method: 'POST' })
+
+    // Insert second row with value 'second_row_value'
+    await page.getByTestId('table-editor-insert-new-row').click()
+    await page.getByRole('menuitem', { name: 'Insert row Insert a new row' }).click()
+    await page.getByTestId(`${colName}-input`).fill('second_row_value')
+    await page.getByTestId('action-bar-save-row').click()
+    await waitForApiResponse(page, 'pg-meta', ref, 'query?key=', { method: 'POST' })
+
+    // Wait for grid to be visible
+    await expect(page.getByRole('grid')).toBeVisible()
+
+    // Right-click on the first row's cell to open context menu
+    const firstRowCell = page.getByRole('gridcell', { name: 'first_row_value' })
+    await expect(firstRowCell).toBeVisible()
+    await firstRowCell.click({ button: 'right' })
+
+    // Click "Copy cell" from context menu
+    await page.getByRole('menuitem', { name: 'Copy cell' }).click()
+    await page.waitForTimeout(500)
+
+    // Verify first row value was copied
+    const firstCopiedValue = await page.evaluate(() => navigator.clipboard.readText())
+    expect(firstCopiedValue).toBe('first_row_value')
+
+    // Right-click on the second row's cell to open context menu
+    const secondRowCell = page.getByRole('gridcell', { name: 'second_row_value' })
+    await expect(secondRowCell).toBeVisible()
+    await secondRowCell.click({ button: 'right' })
+
+    // Click "Copy cell" from context menu
+    await page.getByRole('menuitem', { name: 'Copy cell' }).click()
+    await page.waitForTimeout(500)
+
+    // Verify second row value was copied
+    const secondCopiedValue = await page.evaluate(() => navigator.clipboard.readText())
+    expect(secondCopiedValue).toBe('second_row_value')
+
+    // Cleanup
+    await deleteTable(page, ref, tableName)
+  })
 })