@atlaskit/editor-plugin-show-diff/dist/es2019/pm-plugins/decorations/utils/tableCellEdgeAttrs.js

ShowDiff plugin for @atlaskit/editor-core

import { TableMap } from '@atlaskit/editor-tables/table-map';
export const getTableTopAndBottomCellEdgeAttrs = tableMap => {
  const edgeAttrsByOffset = new Map();
  const addLogicalRow = (offsets, edge) => {
    offsets.forEach((offset, columnIndex) => {
      var _edgeAttrsByOffset$ge;
      const edgeAttrs = (_edgeAttrsByOffset$ge = edgeAttrsByOffset.get(offset)) !== null && _edgeAttrsByOffset$ge !== void 0 ? _edgeAttrsByOffset$ge : {
        reachesBottom: false,
        reachesLeft: false,
        reachesRight: false,
        reachesTop: false
      };
      edgeAttrs.reachesBottom ||= edge === 'bottom';
      edgeAttrs.reachesLeft ||= columnIndex === 0;
      edgeAttrs.reachesRight ||= columnIndex === tableMap.width - 1;
      edgeAttrs.reachesTop ||= edge === 'top';
      edgeAttrsByOffset.set(offset, edgeAttrs);
    });
  };
  const topRow = tableMap.mapByRow[0];
  const bottomRow = tableMap.mapByRow[tableMap.height - 1];
  if (topRow) {
    addLogicalRow(topRow, 'top');
  }
  if (bottomRow) {
    addLogicalRow(bottomRow, 'bottom');
  }
  return edgeAttrsByOffset;
};
export const getRowCellEdgeAttrs = ({
  edgeAttrsByOffset,
  rowNode,
  rowStart
}) => {
  const cellEdgeAttrs = [];
  rowNode.content.forEach((cellNode, cellOffset) => {
    if (cellNode.type.name !== 'tableCell' && cellNode.type.name !== 'tableHeader') {
      return;
    }
    cellEdgeAttrs.push(edgeAttrsByOffset.get(rowStart + 1 + cellOffset));
  });
  return cellEdgeAttrs;
};
export const applyCellEdgeAttrs = (cell, edgeAttrs) => {
  if (!edgeAttrs) {
    return;
  }
  if (edgeAttrs.reachesTop) {
    cell.setAttribute('data-reaches-top', 'true');
  }
  if (edgeAttrs.reachesBottom) {
    cell.setAttribute('data-reaches-bottom', 'true');
  }
  if (edgeAttrs.reachesLeft) {
    cell.setAttribute('data-reaches-left', 'true');
  }
  if (edgeAttrs.reachesRight) {
    cell.setAttribute('data-reaches-right', 'true');
  }
};
export const applyTableCellEdgeAttrs = ({
  element,
  tableNode
}) => {
  try {
    const table = element instanceof HTMLTableElement ? element : element.querySelector('table');
    if (!(table instanceof HTMLTableElement)) {
      return;
    }
    const tableMap = TableMap.get(tableNode);
    const cells = Array.from(table.rows).flatMap(row => Array.from(row.cells));
    const cellsByOffset = new Map();
    let cellIndex = 0;
    tableNode.content.forEach((rowNode, rowStart) => {
      rowNode.content.forEach((cellNode, cellOffset) => {
        if (cellNode.type.name === 'tableCell' || cellNode.type.name === 'tableHeader') {
          const cell = cells[cellIndex];
          cellIndex++;
          if (cell) {
            cellsByOffset.set(rowStart + 1 + cellOffset, cell);
          }
        }
      });
    });
    getTableTopAndBottomCellEdgeAttrs(tableMap).forEach((edgeAttrs, offset) => {
      const cell = cellsByOffset.get(offset);
      if (cell) {
        applyCellEdgeAttrs(cell, edgeAttrs);
      }
    });
  } catch {
    // Table structure can be transient while widget DOM is being assembled.
  }
};