@atlaskit/editor-plugin-table 7.16.11 → 7.16.13

package/src/pm-plugins/view-mode-sort/index.ts

@@ -20,238 +20,234 @@ import type tablePlugin from '../../plugin';
 import { getPluginState } from '../plugin-factory';
 
 import {
-  IS_DISABLED_CLASS_NAME,
-  SORT_INDEX_DATA_ATTRIBUTE,
-  SORTING_ICON_CLASS_NAME,
+	IS_DISABLED_CLASS_NAME,
+	SORT_INDEX_DATA_ATTRIBUTE,
+	SORTING_ICON_CLASS_NAME,
 } from './consts';
 import { tableViewModeSortPluginKey as key } from './plugin-key';
 import type { ViewModeSortPluginState } from './types';
 import { getTableElements, toggleSort } from './utils';
 
 export const createPlugin = (
-  editorViewModeAPI: ExtractInjectionAPI<typeof tablePlugin>['editorViewMode'],
+	editorViewModeAPI: ExtractInjectionAPI<typeof tablePlugin>['editorViewMode'],
 ) => {
-  return new SafePlugin({
-    state: {
-      init: () => ({
-        decorations: DecorationSet.empty,
-        sort: {},
-        allTables: [],
-      }),
-      apply(tr, pluginState: ViewModeSortPluginState, oldState) {
-        // TODO - move this mode check to plugin creation if possible. Right now it's here because the initial state
-        // does not appear correct when the plugin is created.
-        const { mode } = editorViewModeAPI?.sharedState.currentState() || {};
-        if (mode !== 'view') {
-          return pluginState;
-        }
-        let { decorations, sort, allTables } = pluginState;
-
-        const sortMeta = tr.getMeta('tableSortMeta');
-
-        let hoverTableMeta = tr.getMeta('mouseEnterTable');
-        let removeTableMeta = tr.getMeta('removeTable');
-        let tableId = '';
-
-        // Remove the table from the state
-        if (removeTableMeta) {
-          allTables = allTables.filter(([id]) => id !== removeTableMeta);
-        } else {
-          tableId = hoverTableMeta?.[0];
-        }
-
-        sort = { ...sort, ...sortMeta };
-
-        const isTableInState = allTables.some(([id]) => id === tableId);
-
-        // Update the table in the state
-        if (hoverTableMeta) {
-          allTables = allTables.filter(([id]) => id !== hoverTableMeta[0]);
-          allTables.push(hoverTableMeta);
-        }
-
-        /**
-         * Create decorations for the sorting icons
-         */
-        const decs: Decoration[] = [];
-
-        // TODO - add support for keyboard only users
-        if ((hoverTableMeta && !isTableInState) || sortMeta) {
-          allTables.forEach((table) => {
-            const [tableId, _node, pos] = table;
-            const tableNode = tr.doc.nodeAt(tr.mapping.map(pos));
-            if (!tableNode || tableNode.type.name !== 'table') {
-              return pluginState;
-            }
-            const map = TableMap.get(tableNode);
-            const hasMergedCells = new Set(map.map).size !== map.map.length;
-            map.mapByRow[0].forEach((cell, index) => {
-              // return pluginState;
-              decs.push(
-                Decoration.widget(cell + pos + 2, () => {
-                  const element = document.createElement('div');
-                  element.setAttribute(SORT_INDEX_DATA_ATTRIBUTE, `${index}`);
-                  element.classList.add(SORTING_ICON_CLASS_NAME);
-                  if (hasMergedCells) {
-                    element.classList.add(IS_DISABLED_CLASS_NAME);
-                  }
-
-                  let sortOrdered;
-                  if (index === sort[tableId]?.index) {
-                    sortOrdered = sort[tableId]?.direction;
-                  } else {
-                    sortOrdered = SortOrder.NO_ORDER;
-                  }
-
-                  const { getIntl } = getPluginState(oldState);
-
-                  ReactDOM.render(
-                    createElement(
-                      RawIntlProvider,
-                      { value: getIntl() },
-                      createElement(SortingIcon, {
-                        isSortingAllowed: !hasMergedCells,
-                        sortOrdered,
-                        onClick: () => {},
-                        onKeyDown: () => {},
-                      }),
-                    ),
-                    element,
-                  );
-                  return element;
-                }),
-              );
-            });
-          });
-          decorations = DecorationSet.create(tr.doc, decs);
-        }
-
-        /**
-         * Map the decorations to the new document if there are changes
-         */
-        if (tr.docChanged) {
-          decorations = decorations.map(tr.mapping, tr.doc);
-          allTables = allTables.map((table) => {
-            return [table[0], table[1], tr.mapping.map(table[2])];
-          });
-        }
-
-        return {
-          decorations,
-          sort,
-          allTables,
-        };
-      },
-    },
-    key,
-    appendTransaction: (trs, oldState, newState) => {
-      // return newState.tr;
-      const { mode } = editorViewModeAPI?.sharedState.currentState() || {};
-      if (mode !== 'view') {
-        return newState.tr;
-      }
-
-      let allTables = key.getState(newState)?.allTables || [];
-
-      /**
-       * If incoming changes have affected a table node, remove the sorting. This prevents the
-       * table from breaking if changes like merged cells are incoming.
-       */
-      for (const tr of trs) {
-        const hoverTableMeta = tr.getMeta('mouseEnterTable');
-        if (hoverTableMeta) {
-          allTables = allTables.filter(([id]) => id !== hoverTableMeta[0]);
-          allTables.push(hoverTableMeta);
-        }
-        const isRemote = tr.getMeta('isRemote');
-        const isDocChanged = tr.docChanged;
-        const isChangesIncoming = isRemote && isDocChanged;
-
-        const oldPluginState = key.getState(oldState);
-        const newPluginState = key.getState(newState);
-
-        for (const table of allTables) {
-          const [tableId, node, pos] = table;
-          const {
-            order: oldOrder,
-            direction: oldDirection,
-            index: oldIndex,
-          } = oldPluginState?.sort?.[tableId] || {};
-
-          if (isChangesIncoming) {
-            const maybeTableNode = tr.doc.nodeAt(pos);
-            const isTableNodeChanged =
-              maybeTableNode?.attrs?.localId !== tableId ||
-              !node.eq(maybeTableNode);
-
-            if (isTableNodeChanged) {
-              const newtr = newState.tr;
-              newtr.setMeta('tableSortMeta', {
-                [tableId]: {},
-              });
-              newtr.setMeta('removeTable', tableId);
-
-              // Unsort the table here
-              if (oldOrder !== undefined) {
-                const { rows, tbody } = getTableElements(tableId);
-                if (!rows || !tbody) {
-                  return newtr;
-                }
-                const sortedOrder = [...oldOrder].sort(
-                  (a, b) => a.value - b.value,
-                );
-                sortedOrder.forEach((index, i) => {
-                  tbody.appendChild(rows[index.index + 1]);
-                });
-                return newtr;
-              }
-            }
-          }
-
-          /**
-           * Sort the table if the sort order has changed
-           */
-          const {
-            order: newOrder,
-            direction: newDirection,
-            index: newIndex,
-          } = newPluginState?.sort?.[tableId] || {};
-          const orderChanged =
-            oldDirection !== newDirection || oldIndex !== newIndex;
-
-          if (orderChanged) {
-            if (!isRemote && newDirection !== SortOrder.NO_ORDER) {
-              const { rows, tbody } = getTableElements(tableId);
-              if (rows && newOrder) {
-                newOrder.forEach((index, i) => {
-                  tbody?.appendChild(rows[index.value + 1]);
-                });
-              }
-            }
-          }
-        }
-      }
-      return newState.tr;
-    },
-    props: {
-      handleDOMEvents: {
-        keydown: (view, event) => {
-          // TODO - fix the focus issue here, where toggling sort with a keypress loses focus
-          if (event.key === 'Enter' || event.key === ' ') {
-            const pluginState = key.getState(view.state)?.sort || {};
-            toggleSort(view, event, pluginState);
-          }
-        },
-
-        click: (view, event) => {
-          const pluginState = key.getState(view.state)?.sort || {};
-          toggleSort(view, event, pluginState);
-        },
-      },
-      decorations(state) {
-        const decs = key.getState(state)?.decorations || DecorationSet.empty;
-        return decs;
-      },
-    },
-  });
+	return new SafePlugin({
+		state: {
+			init: () => ({
+				decorations: DecorationSet.empty,
+				sort: {},
+				allTables: [],
+			}),
+			apply(tr, pluginState: ViewModeSortPluginState, oldState) {
+				// TODO - move this mode check to plugin creation if possible. Right now it's here because the initial state
+				// does not appear correct when the plugin is created.
+				const { mode } = editorViewModeAPI?.sharedState.currentState() || {};
+				if (mode !== 'view') {
+					return pluginState;
+				}
+				let { decorations, sort, allTables } = pluginState;
+
+				const sortMeta = tr.getMeta('tableSortMeta');
+
+				let hoverTableMeta = tr.getMeta('mouseEnterTable');
+				let removeTableMeta = tr.getMeta('removeTable');
+				let tableId = '';
+
+				// Remove the table from the state
+				if (removeTableMeta) {
+					allTables = allTables.filter(([id]) => id !== removeTableMeta);
+				} else {
+					tableId = hoverTableMeta?.[0];
+				}
+
+				sort = { ...sort, ...sortMeta };
+
+				const isTableInState = allTables.some(([id]) => id === tableId);
+
+				// Update the table in the state
+				if (hoverTableMeta) {
+					allTables = allTables.filter(([id]) => id !== hoverTableMeta[0]);
+					allTables.push(hoverTableMeta);
+				}
+
+				/**
+				 * Create decorations for the sorting icons
+				 */
+				const decs: Decoration[] = [];
+
+				// TODO - add support for keyboard only users
+				if ((hoverTableMeta && !isTableInState) || sortMeta) {
+					allTables.forEach((table) => {
+						const [tableId, _node, pos] = table;
+						const tableNode = tr.doc.nodeAt(tr.mapping.map(pos));
+						if (!tableNode || tableNode.type.name !== 'table') {
+							return pluginState;
+						}
+						const map = TableMap.get(tableNode);
+						const hasMergedCells = new Set(map.map).size !== map.map.length;
+						map.mapByRow[0].forEach((cell, index) => {
+							// return pluginState;
+							decs.push(
+								Decoration.widget(cell + pos + 2, () => {
+									const element = document.createElement('div');
+									element.setAttribute(SORT_INDEX_DATA_ATTRIBUTE, `${index}`);
+									element.classList.add(SORTING_ICON_CLASS_NAME);
+									if (hasMergedCells) {
+										element.classList.add(IS_DISABLED_CLASS_NAME);
+									}
+
+									let sortOrdered;
+									if (index === sort[tableId]?.index) {
+										sortOrdered = sort[tableId]?.direction;
+									} else {
+										sortOrdered = SortOrder.NO_ORDER;
+									}
+
+									const { getIntl } = getPluginState(oldState);
+
+									ReactDOM.render(
+										createElement(
+											RawIntlProvider,
+											{ value: getIntl() },
+											createElement(SortingIcon, {
+												isSortingAllowed: !hasMergedCells,
+												sortOrdered,
+												onClick: () => {},
+												onKeyDown: () => {},
+											}),
+										),
+										element,
+									);
+									return element;
+								}),
+							);
+						});
+					});
+					decorations = DecorationSet.create(tr.doc, decs);
+				}
+
+				/**
+				 * Map the decorations to the new document if there are changes
+				 */
+				if (tr.docChanged) {
+					decorations = decorations.map(tr.mapping, tr.doc);
+					allTables = allTables.map((table) => {
+						return [table[0], table[1], tr.mapping.map(table[2])];
+					});
+				}
+
+				return {
+					decorations,
+					sort,
+					allTables,
+				};
+			},
+		},
+		key,
+		appendTransaction: (trs, oldState, newState) => {
+			// return newState.tr;
+			const { mode } = editorViewModeAPI?.sharedState.currentState() || {};
+			if (mode !== 'view') {
+				return newState.tr;
+			}
+
+			let allTables = key.getState(newState)?.allTables || [];
+
+			/**
+			 * If incoming changes have affected a table node, remove the sorting. This prevents the
+			 * table from breaking if changes like merged cells are incoming.
+			 */
+			for (const tr of trs) {
+				const hoverTableMeta = tr.getMeta('mouseEnterTable');
+				if (hoverTableMeta) {
+					allTables = allTables.filter(([id]) => id !== hoverTableMeta[0]);
+					allTables.push(hoverTableMeta);
+				}
+				const isRemote = tr.getMeta('isRemote');
+				const isDocChanged = tr.docChanged;
+				const isChangesIncoming = isRemote && isDocChanged;
+
+				const oldPluginState = key.getState(oldState);
+				const newPluginState = key.getState(newState);
+
+				for (const table of allTables) {
+					const [tableId, node, pos] = table;
+					const {
+						order: oldOrder,
+						direction: oldDirection,
+						index: oldIndex,
+					} = oldPluginState?.sort?.[tableId] || {};
+
+					if (isChangesIncoming) {
+						const maybeTableNode = tr.doc.nodeAt(pos);
+						const isTableNodeChanged =
+							maybeTableNode?.attrs?.localId !== tableId || !node.eq(maybeTableNode);
+
+						if (isTableNodeChanged) {
+							const newtr = newState.tr;
+							newtr.setMeta('tableSortMeta', {
+								[tableId]: {},
+							});
+							newtr.setMeta('removeTable', tableId);
+
+							// Unsort the table here
+							if (oldOrder !== undefined) {
+								const { rows, tbody } = getTableElements(tableId);
+								if (!rows || !tbody) {
+									return newtr;
+								}
+								const sortedOrder = [...oldOrder].sort((a, b) => a.value - b.value);
+								sortedOrder.forEach((index, i) => {
+									tbody.appendChild(rows[index.index + 1]);
+								});
+								return newtr;
+							}
+						}
+					}
+
+					/**
+					 * Sort the table if the sort order has changed
+					 */
+					const {
+						order: newOrder,
+						direction: newDirection,
+						index: newIndex,
+					} = newPluginState?.sort?.[tableId] || {};
+					const orderChanged = oldDirection !== newDirection || oldIndex !== newIndex;
+
+					if (orderChanged) {
+						if (!isRemote && newDirection !== SortOrder.NO_ORDER) {
+							const { rows, tbody } = getTableElements(tableId);
+							if (rows && newOrder) {
+								newOrder.forEach((index, i) => {
+									tbody?.appendChild(rows[index.value + 1]);
+								});
+							}
+						}
+					}
+				}
+			}
+			return newState.tr;
+		},
+		props: {
+			handleDOMEvents: {
+				keydown: (view, event) => {
+					// TODO - fix the focus issue here, where toggling sort with a keypress loses focus
+					if (event.key === 'Enter' || event.key === ' ') {
+						const pluginState = key.getState(view.state)?.sort || {};
+						toggleSort(view, event, pluginState);
+					}
+				},
+
+				click: (view, event) => {
+					const pluginState = key.getState(view.state)?.sort || {};
+					toggleSort(view, event, pluginState);
+				},
+			},
+			decorations(state) {
+				const decs = key.getState(state)?.decorations || DecorationSet.empty;
+				return decs;
+			},
+		},
+	});
 };

package/src/pm-plugins/view-mode-sort/plugin-key.ts

@@ -2,5 +2,6 @@ import { PluginKey } from '@atlaskit/editor-prosemirror/state';
 
 import type { ViewModeSortPluginState } from './types';
 
-export const tableViewModeSortPluginKey =
-  new PluginKey<ViewModeSortPluginState>('tableViewModeSortPlugin');
+export const tableViewModeSortPluginKey = new PluginKey<ViewModeSortPluginState>(
+	'tableViewModeSortPlugin',
+);

package/src/pm-plugins/view-mode-sort/types.ts

@@ -3,21 +3,21 @@ import type { Node as PMNode } from '@atlaskit/editor-prosemirror/model';
 import type { DecorationSet } from '@atlaskit/editor-prosemirror/view';
 
 export type TableSortMeta = Record<
-  string,
-  {
-    index: number;
-    order: {
-      index: number;
-      value: number;
-    }[];
-    direction: SortOrder;
-  }
+	string,
+	{
+		index: number;
+		order: {
+			index: number;
+			value: number;
+		}[];
+		direction: SortOrder;
+	}
 >;
 
 export interface ViewModeSortPluginState {
-  decorations: DecorationSet;
-  sort: TableSortMeta;
-  allTables: HoverTableMeta[];
+	decorations: DecorationSet;
+	sort: TableSortMeta;
+	allTables: HoverTableMeta[];
 }
 
 export type HoverTableMeta = [string, PMNode, number];