How to focus a cell of a TreeGrid when it is being selected?

Flow
1

So this is a piece of the code responsible for the setup of my TreeGrid. As it is shown I’ve created listeners for each letter so that whenever I press a letter the item which label starts with that letter is selected. But the struggle I am facing is that even if the correct item is selected I cannot use the usual shortcuts for TreeGrid (the arrows for example which expand and collapse the grid) which I’ve concluded is due to the item not being on focus. So my question is - Is there a way to dynamically focus the cell/item which I’ve currently selected ?

import java.util.List;

import com.dhl.evo.web.components.navigation.tree.ItemContainer;
import com.vaadin.flow.component.Key;
import com.vaadin.flow.component.Shortcuts;
import com.vaadin.flow.component.treegrid.TreeGrid;
import com.wexlog.wexsuite.commons.util.WsStringUtil;

class Scratch {
    
    private final TreeGrid<ItemContainer> treeGrid = new TreeGrid<>();
    private String lastSearchedKey = "";
    private int lastFoundIndex = -1;
    
    private void setUpTree(){
        for (char c = 'a'; c <= 'z'; c++) {
            String key = String.valueOf(c);
            Shortcuts.addShortcutListener(treeGrid, () -> handleKeyPress(key), Key.of(key));
        }
    }

    private void handleKeyPress(String _pressedKey) {
        if (WsStringUtil.getNotNullString(_pressedKey).length() == 1) {
            String lowerCasePressedKey = _pressedKey.toLowerCase();
            if (!lowerCasePressedKey.equals(lastSearchedKey)) {
                lastSearchedKey = lowerCasePressedKey;
                lastFoundIndex = -1;
            }
            List<ItemContainer> visibleItems = getVisibleItems();
            List<ItemContainer> matchingItems = visibleItems.stream()
                .filter(item -> item.getLabel().toLowerCase().startsWith(lowerCasePressedKey))
                .toList();
            if (matchingItems.isEmpty()) {
                return;
            }
            lastFoundIndex = (lastFoundIndex + 1) % matchingItems.size();
            ItemContainer itemToSelect = matchingItems.get(lastFoundIndex);
            treeGrid.select(itemToSelect);
            int scrollToIndex = visibleItems.indexOf(itemToSelect);
            if (scrollToIndex != -1) {
                treeGrid.scrollToIndex(scrollToIndex);
            }
        }
    }

}
2

Unfortunately, Grid does not provide any API for setting focus in a particular cell, and programmatically selecting an item also doesn’t do that (although the user selecting a row by clicking a cell does move focus to that cell).

I think the only way you might achieve that right now is to use javascript. Here’s an old workaround for focusing the first cell in the first row: Add a way to focus first cell programmaticaly · Issue #1254 · vaadin/flow-components · GitHub It’s probably a bit outdated, so you might have to change the selectors a bit, and of course change the specific cell to be dynamically set.

You could also create a feature request ticket for programmatically setting cell focus here: GitHub · Where software is built