@atlaskit/editor-plugin-table 1.5.5 → 1.6.0

package/src/plugins/table/pm-plugins/table-resizing/utils/scale-table.ts

@@ -1,4 +1,5 @@
 import { Node as PMNode } from 'prosemirror-model';
+import { Transaction } from 'prosemirror-state';
 
 import { tableCellMinWidth } from '@atlaskit/editor-common/styles';
 import { akEditorTableNumberColumnWidth } from '@atlaskit/editor-shared-styles';
@@ -12,8 +13,11 @@ import {
   adjustColumnsWidths,
   getResizeState,
   getTotalWidth,
+  updateColgroup,
 } from '../utils/resize-state';
 import { ResizeState } from '../utils/types';
+import { hasTableBeenResized, insertColgroupFromNode } from './colgroup';
+import { updateColumnWidths } from '../../../transforms';
 
 export interface ScaleOptions {
   node: PMNode;
@@ -140,3 +144,78 @@ export function scaleTableTo(state: ResizeState, maxSize: number): ResizeState {
 
   return adjustColumnsWidths(newState, maxSize);
 }
+
+export const previewScaleTable = (
+  tableRef: HTMLTableElement | null | undefined,
+  options: ScaleOptions,
+  domAtPos: DomAtPos,
+) => {
+  const { node, start, parentWidth } = options;
+
+  if (!tableRef || !hasTableBeenResized(node)) {
+    return;
+  }
+
+  const resizeState = parentWidth
+    ? scaleWithParent(tableRef, parentWidth, node, start, domAtPos)
+    : scale(tableRef, options, domAtPos);
+
+  if (resizeState) {
+    updateColgroup(resizeState, tableRef);
+  }
+};
+
+// Scale the table to meet new requirements (col, layout change etc)
+export const scaleTable =
+  (
+    tableRef: HTMLTableElement | null | undefined,
+    options: ScaleOptions,
+    domAtPos: DomAtPos,
+  ) =>
+  (tr: Transaction) => {
+    if (!tableRef) {
+      return tr;
+    }
+
+    const { node, start, parentWidth, layoutChanged } = options;
+
+    // If a table has not been resized yet, columns should be auto.
+    if (hasTableBeenResized(node) === false) {
+      // If its not a re-sized table, we still want to re-create cols
+      // To force reflow of columns upon delete.
+      insertColgroupFromNode(tableRef, node);
+      return tr;
+    }
+
+    let resizeState;
+    if (parentWidth) {
+      resizeState = scaleWithParent(
+        tableRef,
+        parentWidth,
+        node,
+        start,
+        domAtPos,
+      );
+    } else {
+      resizeState = scale(tableRef, options, domAtPos);
+    }
+
+    if (resizeState) {
+      tr = updateColumnWidths(resizeState, node, start)(tr);
+
+      if (tr.docChanged) {
+        tr.setMeta('scrollIntoView', false);
+        // TODO: ED-8995
+        // We need to do this check to reduce the number of race conditions when working with tables.
+        // This metadata is been used in the sendTransaction function in the Collab plugin
+        /* Added !layoutChanged check here to solve unnecessary scroll bar after publish when click on breakout button multiple times and publish
+           scaleTable is only called once every time a breakout button is clicked, so it is safe not to add the meta 'scaleTable' to the tr.
+           Leaving the tr.setMeta('scaleTable', true) here for race conditions that we aren't aware of.
+         */
+        !layoutChanged && tr.setMeta('scaleTable', true);
+        return tr;
+      }
+    }
+
+    return tr;
+  };