Contents

import { RowModel } from "@tanstack/react-table"
import { DataType, ExpandedStateObject } from "./types"

const filterExpandableRows = (expandedState: Record<string, boolean>) => {
  for (const expandedRow in expandedState) {
    if (expandedState[expandedRow] === false) {
      delete expandedState[expandedRow]
    }
  }
  return expandedState
}

export const updateExpandAndCollapseState = (
  tableRows: RowModel<DataType>,
  expanded: Record<string, boolean>,
  targetParent: string
) => {
  const updateExpandedRows: Record<string, boolean> = {};
  const rows = tableRows.rows;

  let isExpansionConsistent = true;
  const areRowsExpanded = new Set<boolean>();

  for (const row of rows) {
    const shouldBeUpdated = targetParent === undefined ? row.depth === 0 : targetParent === row.parentId;
    
    if (shouldBeUpdated) {
      const isExpanded = row.getIsExpanded();
      areRowsExpanded.add(isExpanded);

      updateExpandedRows[row.id] = !isExpansionConsistent ? true : !isExpanded;

      if (areRowsExpanded.size > 1) {
        isExpansionConsistent = false;
        // If expansion inconsistent, ensure all target rows are set to expand
        for (const key in updateExpandedRows) {
          updateExpandedRows[key] = true;
        }
      }
    }
  }

  return filterExpandableRows({
    ...(expanded as ExpandedStateObject),
    ...updateExpandedRows,
  });
};

Version data entries

3 entries across 3 versions & 1 rubygems

Version	Path
playbook_ui-13.21.0.pre.alpha.pbntr220improveexpansionspeed2451	app/pb_kits/playbook/pb_advanced_table/Utilities/ExpansionControlHelpers.tsx
playbook_ui-13.21.0.pre.alpha.pbntr220improveexpansionspeed2431	app/pb_kits/playbook/pb_advanced_table/Utilities/ExpansionControlHelpers.tsx
playbook_ui-13.21.0.pre.alpha.pbntr220improveexpansionspeed2415	app/pb_kits/playbook/pb_advanced_table/Utilities/ExpansionControlHelpers.tsx