How to select a first table cell node of a new just inserted row at the end of a table?

[EugeneVorobyev]

Hi everyone,

I am trying to recreate a functionality like Google Docs tables:

when TAB key is pressed and current position is at the last cell of the last row
then a new row is inserted and the first cell of the new row is selected

I've added a command listener for TAB key, that inserts a new row if the selection is the last cell of a table:

      editor.registerCommand(
        KEY_TAB_COMMAND,
        (event) => {
          const selection = $getSelection();
          if (!$isRangeSelection(selection) || !selection.isCollapsed()) {
            return false;
          }

          const currentCellNode = $findCellNode(selection.anchor.getNode());
          if (currentCellNode === null) {
            return false;
          }

          const tableNode =
            $getTableNodeFromLexicalNodeOrThrow(currentCellNode);
          const table = $getElementForTableNode(editor, tableNode);
          const currentCords = tableNode.getCordsFromCellNode(
            currentCellNode,
            table
          );

          if (
            currentCords.x === table.columns - 1 &&
            currentCords.y === table.rows - 1
          ) {
            insertTableRowAtEnd();
            stopEvent(event);

            // TODO: select the first cell of the new row

            return true;
          }

          // pass handling to regular TAB handler of table plugin
          return false;
        },
        COMMAND_PRIORITY_CRITICAL
      );

where insertTableRowAtEnd() is just:

$insertTableRow__EXPERIMENTAL();

The problem is that when I add a new row the state of table is not updated. So helpers (including table observer) don't detect new elements. I guess it's by design, because it's not reconciled yet.

But I can't wrap my head around how to enable such behavior. Should I add another listener to detect new row insertion?

Will appreciate any help! Thanks.