How to expand to first found leaf or expand all when filtering

[chattling]

I'm trying to implement it but unable to figure out how.
I have hierarchical data with subRows. I use global filter with filterFromLeafRows to keep the parent nodes displayed. All expanding and filtering works perfectly.
However I want to implement a feature when after filtering is done the table will auto-expand either to 1st found leaf, or, if its too complex, auto-expand all rows.

Table instance is defined this way (react):

const table = useReactTable({
    data: tasks,
    columns,
    state: {
      globalFilter,
    },
    getSubRows: (row) => row.subRows,
    getCoreRowModel: getCoreRowModel(),
    getExpandedRowModel: getExpandedRowModel(),
    getFilteredRowModel: getFilteredRowModel(),
    onGlobalFilterChange: setGlobalFilter,
    globalFilterFn: (row, columnId, filterValue) => {
      return row.getValue(columnId).toLowerCase().includes(filterValue.toLowerCase())
    },
    filterFromLeafRows: true,
  })

[chattling]

I found the solution. First I debounced filter value to improve performance. Then I added logic to globalFilterFn, if the filter match, I get all the parent rows (those are the expandable rows), and if they not yet expanded, I expand each:

  const table = useReactTable({
    data: tasks,
    columns,
    state: {
      globalFilter: debounceGlobalFilter,
    },
    getSubRows: (row) => row.subRows,
    getCoreRowModel: getCoreRowModel(),
    getExpandedRowModel: getExpandedRowModel(),
    getFilteredRowModel: getFilteredRowModel(),
    onGlobalFilterChange: setGlobalFilter,
    globalFilterFn: (row, columnId, filterValue) => {
      if (row.getValue(columnId).toLowerCase().includes(filterValue.toLowerCase())) {
        row.getParentRows().forEach((parentRow) => !parentRow.getIsExpanded() && parentRow.toggleExpanded())
      }
      return row.getValue(columnId).toLowerCase().includes(filterValue.toLowerCase())
    },
    filterFromLeafRows: true,
  })