@atlaskit/editor-plugin-table 5.7.10 → 5.8.1

package/dist/es2019/ui/FloatingDragMenu/styles.js

@@ -0,0 +1,62 @@
+import { css } from '@emotion/react';
+import { tableBackgroundBorderColor } from '@atlaskit/adf-schema';
+import { N60A, N90 } from '@atlaskit/theme/colors';
+import { TableCssClassName as ClassName } from '../../types';
+import { dragMenuDropdownWidth } from '../consts';
+export const cellColourPreviewStyles = selectedColor => css`
+  &::before {
+    background: ${selectedColor};
+  }
+`;
+export const dragMenuBackgroundColorStyles = css`
+  .${ClassName.DRAG_SUBMENU} {
+    border-radius: ${"var(--ds-border-radius, 3px)"};
+    background: ${"var(--ds-surface-overlay, white)"};
+    box-shadow: ${`var(--ds-shadow-overlay, ${`0 4px 8px -2px ${N60A}, 0 0 1px ${N60A}`})`};
+    display: block;
+    position: absolute;
+    top: 0;
+    left: ${dragMenuDropdownWidth}px;
+    padding: ${"var(--ds-space-100, 8px)"};
+
+    > div {
+      padding: 0;
+    }
+  }
+
+  .${ClassName.DRAG_SUBMENU_ICON} {
+    display: flex;
+
+    &::before {
+      content: '';
+      display: block;
+      border: 1px solid ${tableBackgroundBorderColor};
+      border-radius: ${"var(--ds-border-radius, 3px)"};
+      width: 14px;
+      height: 14px;
+    }
+
+    &::after {
+      content: '›';
+      margin-left: ${"var(--ds-space-050, 4px)"};
+      line-height: 14px;
+      color: ${`var(--ds-icon, ${N90})`};
+    }
+  }
+`;
+export const toggleStyles = css`
+  display: flex;
+  input[type='checkbox'] {
+    width: 30px;
+    height: 14px;
+    pointer-events: initial;
+    cursor: pointer;
+  }
+  > label {
+    margin: 0px;
+    pointer-events: none;
+    > span {
+      pointer-events: none;
+    }
+  }
+`;

package/dist/es2019/ui/consts.js

@@ -1,5 +1,6 @@
 import { tableCellBorderWidth, tableMarginTop } from '@atlaskit/editor-common/styles';
 import { akEditorTableBorder, akEditorTableBorderSelected, akEditorTableCellBlanketDeleted, akEditorTableCellBlanketSelected, akEditorTableHeaderCellBackground, akEditorTableToolbar, akEditorTableToolbarSize, akEditorUnitZIndex, akRichMediaResizeZIndex } from '@atlaskit/editor-shared-styles';
+import { getBooleanFF } from '@atlaskit/platform-feature-flags';
 import { B200, N0, N20, N200, N20A, N300, R300, R400, R75 } from '@atlaskit/theme/colors';
 import { RESIZE_HANDLE_AREA_DECORATION_GAP } from '../types';
 
@@ -49,7 +50,7 @@ export const resizeHandlerAreaWidth = RESIZE_HANDLE_AREA_DECORATION_GAP / 3;
 export const resizeLineWidth = 2;
 export const resizeHandlerZIndex = columnControlsZIndex + akRichMediaResizeZIndex;
 export const contextualMenuTriggerSize = 16;
-export const contextualMenuDropdownWidth = 180;
+export const contextualMenuDropdownWidth = getBooleanFF('platform.editor.table.drag-and-drop') ? 250 : 180;
 export const stickyRowZIndex = resizeHandlerZIndex + 2;
 export const stickyRowOffsetTop = 8;
 export const stickyHeaderBorderBottomWidth = 1;

package/dist/esm/commands/column-resize.js

@@ -0,0 +1,149 @@
+import { tableCellMinWidth } from '@atlaskit/editor-common/styles';
+import { TableMap } from '@atlaskit/editor-tables';
+import { findCellClosestToPos, findCellRectClosestToPos, findTableClosestToPos, getSelectionRect, isSelectionType, nextCell } from '@atlaskit/editor-tables/src/utils';
+import { getBooleanFF } from '@atlaskit/platform-feature-flags';
+import { getPluginState as getTableResizingPluginState, createCommand as tableResizingPluginCreateCommand } from '../pm-plugins/table-resizing/plugin-factory';
+import { currentColWidth, getResizeState, getTableMaxWidth, resizeColumn, updateControls } from '../pm-plugins/table-resizing/utils';
+import { updateColumnWidths } from '../transforms';
+import { getSelectedColumnIndexes } from '../utils';
+import { addResizeHandleDecorations, showResizeHandleLine } from './index';
+var updateResizeHandleAndStatePosition = function updateResizeHandleAndStatePosition(rowIndex, columnIndex, nextResizeHandlePos) {
+  return function (state, dispatch) {
+    addResizeHandleDecorations(rowIndex, columnIndex, true)(state, dispatch);
+    // Currently only 'right' position is used in showResizeHandleLine
+    showResizeHandleLine({
+      left: 0,
+      right: columnIndex
+    })(state, dispatch);
+    tableResizingPluginCreateCommand({
+      type: 'SET_RESIZE_HANDLE_POSITION',
+      data: {
+        resizeHandlePos: nextResizeHandlePos
+      }
+    })(state, dispatch);
+    return false;
+  };
+};
+export var initiateKeyboardColumnResizing = function initiateKeyboardColumnResizing(state, dispatch, view) {
+  if (!getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+    return false;
+  }
+  var selection = state.selection;
+  var selectionRect = isSelectionType(selection, 'cell') ? getSelectionRect(selection) : findCellRectClosestToPos(selection.$from);
+  var cell = findCellClosestToPos(selection.$from);
+  if (selectionRect && cell && view) {
+    var cellAttrs = cell.node.attrs;
+    var width = currentColWidth(view, cell.pos, cellAttrs);
+    updateResizeHandleAndStatePosition(selectionRect.top, selectionRect.right, cell.pos)(state, dispatch);
+    tableResizingPluginCreateCommand({
+      type: 'SET_DRAGGING',
+      data: {
+        dragging: {
+          startX: 0,
+          startWidth: width
+        }
+      }
+    })(state, dispatch);
+    return true;
+  }
+  return false;
+};
+export var activateNextResizeArea = function activateNextResizeArea(direction) {
+  return function (state, dispatch) {
+    if (!getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+      return false;
+    }
+    var _ref = getTableResizingPluginState(state) || {},
+      resizeHandlePos = _ref.resizeHandlePos,
+      dragging = _ref.dragging;
+    // If No resizing has initiated, skip to regular handler
+    if (!resizeHandlePos || !dragging) {
+      return false;
+    }
+    var selection = state.selection;
+    var cell = findCellClosestToPos(selection.$from);
+    if (!cell) {
+      // cursor position may be changed by mouse to be outside of table;
+      return false;
+    }
+    var $currentCell = state.doc.resolve(resizeHandlePos);
+    if (!$currentCell) {
+      return false;
+    }
+    var tableNode = $currentCell.node(-1);
+    var closestTable = findTableClosestToPos($currentCell);
+    var tableMap = TableMap.get(tableNode);
+    if (!tableNode || !closestTable || !tableMap) {
+      return false;
+    }
+    var currentCellRect = tableMap.findCell($currentCell.pos - $currentCell.start(-1));
+    var $nextCell = nextCell($currentCell, 'horiz', direction);
+    if ($nextCell) {
+      // we are somewhere in between the side columns of the table
+      var offset = $nextCell.pos - $nextCell.start(-1);
+      var rectForNextCell = tableMap.findCell(offset);
+      updateResizeHandleAndStatePosition(rectForNextCell.top, rectForNextCell.right, $nextCell.pos)(state, dispatch);
+    } else {
+      // current position is in the one of the side columns of the table(left or right)
+      if (currentCellRect.left === 0) {
+        var lastCellInCurrentRow = tableMap.positionAt(currentCellRect.top, tableMap.width - 1, tableNode) + closestTable.start;
+        var $lastCell = state.doc.resolve(lastCellInCurrentRow);
+        updateResizeHandleAndStatePosition(currentCellRect.top, tableMap.width, $lastCell.pos)(state, dispatch);
+      } else if (tableMap.width === currentCellRect.right) {
+        var firsCellInCurrentRow = tableMap.positionAt(currentCellRect.top, 0, tableNode) + closestTable.start;
+        var _$nextCell = state.doc.resolve(firsCellInCurrentRow);
+        updateResizeHandleAndStatePosition(currentCellRect.top, 1, _$nextCell.pos)(state, dispatch);
+      }
+    }
+    return true;
+  };
+};
+export var changeColumnWidthByStep = function changeColumnWidthByStep(stepSize, getEditorContainerWidth) {
+  return function (state, dispatch, view) {
+    if (!getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+      return false;
+    }
+    var _getTableResizingPlug = getTableResizingPluginState(state),
+      resizeHandlePos = _getTableResizingPlug.resizeHandlePos,
+      dragging = _getTableResizingPlug.dragging;
+    if (!view || !resizeHandlePos || !dragging) {
+      return false;
+    }
+    var $cell = state.doc.resolve(resizeHandlePos);
+    var tableStartPosition = $cell.start(-1);
+    var originalTable = $cell.node(-1);
+    var map = TableMap.get(originalTable);
+    var domAtPos = view.domAtPos.bind(view);
+    var colIndex = map.colCount($cell.pos - tableStartPosition) + ($cell.nodeAfter ? $cell.nodeAfter.attrs.colspan : 1) - 1;
+    var dom = domAtPos(tableStartPosition).node;
+    if (dom && dom.nodeName !== 'TABLE') {
+      dom = dom.closest('table');
+    }
+    var maxSize = getTableMaxWidth({
+      table: originalTable,
+      tableStart: tableStartPosition,
+      state: state,
+      layout: originalTable.attrs.layout,
+      getEditorContainerWidth: getEditorContainerWidth
+    });
+    var initialResizeState = getResizeState({
+      minWidth: tableCellMinWidth,
+      maxSize: maxSize,
+      table: originalTable,
+      tableRef: dom,
+      start: tableStartPosition,
+      domAtPos: domAtPos
+    });
+    updateControls()(state);
+    var selectionRect = getSelectionRect(state.selection);
+    var selectedColumns = selectionRect ? getSelectedColumnIndexes(selectionRect) : [];
+    // only selected (or selected - 1) columns should be distributed
+    var resizingSelectedColumns = selectedColumns.indexOf(colIndex) > -1 || selectedColumns.indexOf(colIndex + 1) > -1;
+    var newResizeState = resizeColumn(initialResizeState, colIndex, stepSize, dom, resizingSelectedColumns ? selectedColumns : undefined);
+    var tr = updateColumnWidths(newResizeState, originalTable, tableStartPosition)(state.tr);
+    if (dispatch) {
+      dispatch(tr);
+    }
+    return true;
+  };
+};

package/dist/esm/commands/go-to-next-cell.js

@@ -4,7 +4,10 @@ import { INPUT_METHOD } from '@atlaskit/editor-common/analytics';
 import { findParentNodeOfType } from '@atlaskit/editor-prosemirror/utils';
 import { TableMap } from '@atlaskit/editor-tables/table-map';
 import { goToNextCell as baseGotoNextCell, findTable } from '@atlaskit/editor-tables/utils';
+import { getBooleanFF } from '@atlaskit/platform-feature-flags';
 import { insertRowWithAnalytics } from '../commands-with-analytics';
+import { getPluginState as getResizePluginState, createCommand as tableResizingPluginCreateCommand } from '../pm-plugins/table-resizing/plugin-factory';
+import { hideResizeHandleLine } from './hover';
 var TAB_FORWARD_DIRECTION = 1;
 var TAB_BACKWARD_DIRECTION = -1;
 export var goToNextCell = function goToNextCell(editorAnalyticsAPI) {
@@ -14,6 +17,18 @@ export var goToNextCell = function goToNextCell(editorAnalyticsAPI) {
       if (!table) {
         return false;
       }
+      if (getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+        var _getResizePluginState;
+        var isColumnResizing = !!((_getResizePluginState = getResizePluginState(state)) !== null && _getResizePluginState !== void 0 && _getResizePluginState.dragging);
+        if (isColumnResizing) {
+          tableResizingPluginCreateCommand({
+            type: 'STOP_RESIZING'
+          }, function (originalTr) {
+            return (state.tr || originalTr).setMeta('scrollIntoView', false);
+          })(state, dispatch);
+          hideResizeHandleLine()(state, dispatch);
+        }
+      }
       var map = TableMap.get(table.node);
       var _state$schema$nodes = state.schema.nodes,
         tableCell = _state$schema$nodes.tableCell,

package/dist/esm/commands/selection.js

@@ -3,7 +3,9 @@ import { Selection, TextSelection } from '@atlaskit/editor-prosemirror/state';
 import { CellSelection } from '@atlaskit/editor-tables/cell-selection';
 import { TableMap } from '@atlaskit/editor-tables/table-map';
 import { findTable, isColumnSelected, isRowSelected, isTableSelected, selectedRect } from '@atlaskit/editor-tables/utils';
+import { getBooleanFF } from '@atlaskit/platform-feature-flags';
 import { selectColumn, selectRow } from '../commands/misc';
+import { getPluginState as getResizePluginState } from '../pm-plugins/table-resizing/plugin-factory';
 import { getClosestSelectionRect } from '../toolbar';
 export var TableSelectionDirection = /*#__PURE__*/function (TableSelectionDirection) {
   TableSelectionDirection["TopToBottom"] = "TopToBottom";
@@ -192,9 +194,14 @@ var arrowLeftFromText = function arrowLeftFromText(editorSelectionAPI) {
       var table = findTable(selection);
       if (table) {
         var $from = selection.$from;
+        var isColumnResizing = false;
+        if (getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+          var columResizePluginState = getResizePluginState(state) || {};
+          isColumnResizing = Boolean(columResizePluginState === null || columResizePluginState === void 0 ? void 0 : columResizePluginState.dragging);
+        }
         if (isSelectionAtStartOfTable($from, selection) && $from.parent.type.name === 'paragraph' && $from.depth === table.depth + 3 &&
         // + 3 for: row, cell & paragraph nodes
-        editorSelectionAPI) {
+        editorSelectionAPI && !isColumnResizing) {
           var selectionState = editorSelectionAPI.sharedState.currentState() || {};
           if ((selectionState === null || selectionState === void 0 ? void 0 : selectionState.selectionRelativeToNode) === RelativeSelectionPos.Before) {
             // we have a text selection at start of first table cell, directly inside a top level paragraph,
@@ -221,8 +228,14 @@ var arrowRightFromText = function arrowRightFromText(editorSelectionAPI) {
       var table = findTable(selection);
       if (table) {
         var $to = selection.$to;
-        if (isSelectionAtEndOfTable($to, selection) && $to.parent.type.name === 'paragraph' && $to.depth === table.depth + 3 // + 3 for: row, cell & paragraph nodes
-        ) {
+        var isColumnResizing = false;
+        if (getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+          var columResizePluginState = getResizePluginState(state) || {};
+          isColumnResizing = Boolean(columResizePluginState === null || columResizePluginState === void 0 ? void 0 : columResizePluginState.dragging);
+        }
+        if (isSelectionAtEndOfTable($to, selection) && $to.parent.type.name === 'paragraph' && $to.depth === table.depth + 3 &&
+        // + 3 for: row, cell & paragraph nodes
+        !isColumnResizing) {
           // we have a text selection at end of last table cell, directly inside a top level paragraph,
           // and want to set a full table cell selection
           return selectFullTable(editorSelectionAPI)({

package/dist/esm/plugin.js

@@ -355,7 +355,8 @@ var tablesPlugin = function tablesPlugin(_ref) {
             canDrag: dragAndDropState === null || dragAndDropState === void 0 ? void 0 : dragAndDropState.canDrag,
             getEditorContainerWidth: defaultGetEditorContainerWidth,
             editorAnalyticsAPI: editorAnalyticsAPI,
-            stickyHeaders: stickyHeader
+            stickyHeaders: stickyHeader,
+            pluginConfig: pluginConfig
           }), allowControls && !isDragAndDropEnabled && !isResizing && /*#__PURE__*/React.createElement(FloatingDeleteButton, {
             editorView: editorView,
             selection: editorView.state.selection,

package/dist/esm/pm-plugins/keymap.js

@@ -1,9 +1,11 @@
 import { ACTION, ACTION_SUBJECT, ACTION_SUBJECT_ID, EVENT_TYPE, INPUT_METHOD } from '@atlaskit/editor-common/analytics';
-import { addColumnAfter, addColumnBefore, addRowAfter, addRowBefore, backspace, bindKeymapWithCommand, deleteColumn, deleteRow, moveColumnLeft, moveColumnRight, moveRowDown, moveRowUp, nextCell, previousCell, toggleTable } from '@atlaskit/editor-common/keymaps';
+import { addColumnAfter, addColumnBefore, addRowAfter, addRowBefore, backspace, bindKeymapWithCommand, decreaseMediaSize, deleteColumn, deleteRow, increaseMediaSize, moveColumnLeft, moveColumnRight, moveLeft, moveRight, moveRowDown, moveRowUp, nextCell, previousCell, startColumnResizing, toggleTable } from '@atlaskit/editor-common/keymaps';
 import { chainCommands } from '@atlaskit/editor-prosemirror/commands';
 import { keymap } from '@atlaskit/editor-prosemirror/keymap';
+import { getBooleanFF } from '@atlaskit/platform-feature-flags';
 import { createTable, goToNextCell, moveCursorBackward, triggerUnlessTableHeader } from '../commands';
 import { addRowAroundSelection, deleteSelectedRowsOrColumnsWithAnalyticsViaShortcut, deleteTableIfSelectedWithAnalytics, emptyMultipleCellsWithAnalytics } from '../commands-with-analytics';
+import { activateNextResizeArea, changeColumnWidthByStep, initiateKeyboardColumnResizing } from '../commands/column-resize';
 import { addColumnAfter as addColumnAfterCommand, addColumnBefore as addColumnBeforeCommand } from '../commands/insert';
 import { moveSourceWithAnalyticsViaShortcut } from '../pm-plugins/drag-and-drop/commands-with-analytics';
 import { withEditorAnalyticsAPI } from '../utils/analytics';
@@ -42,6 +44,13 @@ export function keymapPlugin(getEditorContainerWidth, editorAnalyticsAPI, dragAn
     bindKeymapWithCommand(deleteColumn.common, deleteSelectedRowsOrColumnsWithAnalyticsViaShortcut(editorAnalyticsAPI), list);
     bindKeymapWithCommand(deleteRow.common, deleteSelectedRowsOrColumnsWithAnalyticsViaShortcut(editorAnalyticsAPI), list);
   }
+  if (getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+    bindKeymapWithCommand(startColumnResizing.common, initiateKeyboardColumnResizing, list);
+    bindKeymapWithCommand(moveRight.common, activateNextResizeArea(1), list);
+    bindKeymapWithCommand(moveLeft.common, activateNextResizeArea(-1), list);
+    bindKeymapWithCommand(decreaseMediaSize.common, changeColumnWidthByStep(-10, getEditorContainerWidth), list);
+    bindKeymapWithCommand(increaseMediaSize.common, changeColumnWidthByStep(10, getEditorContainerWidth), list);
+  }
   return keymap(list);
 }
 export default keymapPlugin;

package/dist/esm/pm-plugins/table-resizing/event-handlers.js

@@ -11,7 +11,7 @@ import { getSelectedColumnIndexes } from '../../utils';
 import { META_KEYS } from '../table-analytics';
 import { evenColumns, setDragging, stopResizing } from './commands';
 import { getPluginState } from './plugin-factory';
-import { currentColWidth, getLayoutSize, getResizeState, pointsAtCell, resizeColumn, updateControls } from './utils';
+import { currentColWidth, getLayoutSize, getResizeState, getTableMaxWidth, pointsAtCell, resizeColumn, updateControls } from './utils';
 export var handleMouseDown = function handleMouseDown(view, event, localResizeHandlePos, getEditorContainerWidth, getEditorFeatureFlags, editorAnalyticsAPI) {
   var state = view.state,
     dispatch = view.dispatch;
@@ -35,13 +35,24 @@ export var handleMouseDown = function handleMouseDown(view, event, localResizeHa
   if (dom && dom.nodeName !== 'TABLE') {
     dom = dom.closest('table');
   }
-  var containerWidth = getEditorContainerWidth();
-  var parentWidth = getParentNodeWidth(start, state, containerWidth);
-  var maxSize = getBooleanFF('platform.editor.custom-table-width') ? parentWidth ||
-  // its safe to reference table width from node as this will not have changed
-  originalTable.attrs.width || getLayoutSize(dom.getAttribute('data-layout'), containerWidth.width, {}) : parentWidth || getLayoutSize(dom.getAttribute('data-layout'), containerWidth.width, {});
-  if (originalTable.attrs.isNumberColumnEnabled) {
-    maxSize -= akEditorTableNumberColumnWidth;
+  var maxSize;
+  if (!getBooleanFF('platform.editor.a11y-column-resizing_emcvz')) {
+    var containerWidth = getEditorContainerWidth();
+    var parentWidth = getParentNodeWidth(start, state, containerWidth);
+    maxSize = getBooleanFF('platform.editor.custom-table-width') ? parentWidth ||
+    // its safe to reference table width from node as this will not have changed
+    originalTable.attrs.width || getLayoutSize(dom.getAttribute('data-layout'), containerWidth.width, {}) : parentWidth || getLayoutSize(dom.getAttribute('data-layout'), containerWidth.width, {});
+    if (originalTable.attrs.isNumberColumnEnabled) {
+      maxSize -= akEditorTableNumberColumnWidth;
+    }
+  } else {
+    maxSize = getTableMaxWidth({
+      table: originalTable,
+      tableStart: start,
+      state: state,
+      layout: originalTable.attrs.layout,
+      getEditorContainerWidth: getEditorContainerWidth
+    });
   }
   var resizeState = getResizeState({
     minWidth: tableCellMinWidth,

package/dist/esm/pm-plugins/table-resizing/utils/index.js

@@ -3,7 +3,7 @@ export { contentWidth } from './content-width';
 export { getColumnStateFromDOM, getFreeSpace, getCellsRefsInColumn, calculateColumnWidth, addContainerLeftRightPadding } from './column-state';
 export { growColumn, shrinkColumn, reduceSpace } from './resize-logic';
 export { getResizeState, updateColgroup, getTotalWidth, evenAllColumnsWidths, bulkColumnsResize, areColumnsEven, adjustColumnsWidths } from './resize-state';
-export { getLayoutSize, getDefaultLayoutMaxWidth, pointsAtCell, currentColWidth, domCellAround } from './misc';
+export { getLayoutSize, getDefaultLayoutMaxWidth, pointsAtCell, currentColWidth, domCellAround, getTableMaxWidth } from './misc';
 export { updateControls, isClickNear, getResizeCellPos } from './dom';
 export { scale, scaleWithParent, scaleTable, previewScaleTable } from './scale-table';
 export { resizeColumn } from './resize-column';

package/dist/esm/types.js

@@ -99,6 +99,8 @@ export var TableCssClassName = _objectSpread(_objectSpread({}, TableSharedCssCla
   DRAG_HANDLE_DISABLED: "".concat(tablePrefixSelector, "-drag-handle-disabled"),
   /** minimised handle class */
   DRAG_HANDLE_MINIMISED: "".concat(tablePrefixSelector, "-drag-handle-minimised"),
+  DRAG_SUBMENU: "".concat(tablePrefixSelector, "-drag-submenu"),
+  DRAG_SUBMENU_ICON: "".concat(tablePrefixSelector, "-drag-submenu-icon"),
   /** Other classes */
   NUMBERED_COLUMN: "".concat(tablePrefixSelector, "-numbered-column"),
   NUMBERED_COLUMN_BUTTON: "".concat(tablePrefixSelector, "-numbered-column__button"),

package/dist/esm/ui/FloatingDragMenu/DragMenu.js

@@ -1,16 +1,31 @@
+import _slicedToArray from "@babel/runtime/helpers/slicedToArray";
 import _toConsumableArray from "@babel/runtime/helpers/toConsumableArray";
+/* eslint-disable @atlaskit/design-system/prefer-primitives */
 /** @jsx jsx */
+import { useState } from 'react';
 import { jsx } from '@emotion/react';
+import { INPUT_METHOD } from '@atlaskit/editor-common/analytics';
+import { DropdownMenuSharedCssClassName } from '@atlaskit/editor-common/styles';
+import { backgroundPaletteTooltipMessages, cellBackgroundColorPalette, ColorPalette } from '@atlaskit/editor-common/ui-color';
 import { ArrowKeyNavigationType, DropdownMenu } from '@atlaskit/editor-common/ui-menu';
+import { closestElement } from '@atlaskit/editor-common/utils';
+import { hexToEditorBackgroundPaletteColor } from '@atlaskit/editor-palette';
 import { shortcutStyle } from '@atlaskit/editor-shared-styles/shortcut';
 import { TableMap } from '@atlaskit/editor-tables/table-map';
 import { findCellRectClosestToPos, getSelectionRect, isSelectionType } from '@atlaskit/editor-tables/utils';
+import EditorBackgroundColorIcon from '@atlaskit/icon/glyph/editor/background-color';
+import { getBooleanFF } from '@atlaskit/platform-feature-flags';
+import Toggle from '@atlaskit/toggle';
 import { clearHoverSelection, hoverColumns, hoverRows } from '../../commands';
+import { setColorWithAnalytics, toggleHeaderColumnWithAnalytics, toggleHeaderRowWithAnalytics, toggleNumberColumnWithAnalytics } from '../../commands-with-analytics';
 import { toggleDragMenu } from '../../pm-plugins/drag-and-drop/commands';
 import { getPluginState } from '../../pm-plugins/drag-and-drop/plugin-factory';
-import { getMergedCellsPositions, getSelectedColumnIndexes, getSelectedRowIndexes, hasMergedCellsInColumn, hasMergedCellsInRow } from '../../utils';
+import { getPluginState as getTablePluginState } from '../../pm-plugins/plugin-factory';
+import { TableCssClassName as ClassName } from '../../types';
+import { checkIfHeaderColumnEnabled, checkIfHeaderRowEnabled, checkIfNumberColumnEnabled, getMergedCellsPositions, getSelectedColumnIndexes, getSelectedRowIndexes, hasMergedCellsInColumn, hasMergedCellsInRow } from '../../utils';
 import { getDragMenuConfig } from '../../utils/drag-menu';
 import { dragMenuDropdownWidth } from '../consts';
+import { cellColourPreviewStyles, dragMenuBackgroundColorStyles, toggleStyles } from './styles';
 var groupedDragMenuConfig = [['add_row_above', 'add_row_below', 'add_column_left', 'add_column_right', 'distribute_columns', 'clear_cells', 'delete_row', 'delete_column'], ['move_column_left', 'move_column_right', 'move_row_up', 'move_row_down'], ['sort_column_asc', 'sort_column_desc']];
 var convertToDropdownItems = function convertToDropdownItems(dragMenuConfig) {
   var menuItemsArr = _toConsumableArray(Array(groupedDragMenuConfig.length)).map(function () {
@@ -33,7 +48,6 @@ var convertToDropdownItems = function convertToDropdownItems(dragMenuConfig) {
       isDisabled: item.disabled,
       elemBefore: item.icon ? jsx("div", {
         style: {
-          marginRight: "var(--ds-space-negative-075, -6px)",
           display: 'flex'
         }
       }, jsx(item.icon, {
@@ -57,6 +71,7 @@ var convertToDropdownItems = function convertToDropdownItems(dragMenuConfig) {
   };
 };
 export var DragMenu = function DragMenu(_ref) {
+  var _pluginConfig$allowBa;
   var _ref$direction = _ref.direction,
     direction = _ref$direction === void 0 ? 'row' : _ref$direction,
     index = _ref.index,
@@ -69,23 +84,136 @@ export var DragMenu = function DragMenu(_ref) {
     targetCellPosition = _ref.targetCellPosition,
     getEditorContainerWidth = _ref.getEditorContainerWidth,
     canDrag = _ref.canDrag,
-    editorAnalyticsAPI = _ref.editorAnalyticsAPI;
-  var tableMap = tableNode ? TableMap.get(tableNode) : undefined;
+    editorAnalyticsAPI = _ref.editorAnalyticsAPI,
+    pluginConfig = _ref.pluginConfig;
   var state = editorView.state,
     dispatch = editorView.dispatch;
   var selection = state.selection;
+  var tableMap = tableNode ? TableMap.get(tableNode) : undefined;
+  var _useState = useState(false),
+    _useState2 = _slicedToArray(_useState, 2),
+    isSubmenuOpen = _useState2[0],
+    setIsSubmenuOpen = _useState2[1];
   var selectionRect = isSelectionType(selection, 'cell') ? getSelectionRect(selection) : findCellRectClosestToPos(selection.$from);
   var hasMergedCells = direction === 'row' ? hasMergedCellsInRow : hasMergedCellsInColumn;
   var shouldMoveDisabled = index !== undefined && hasMergedCells(index)(selection);
   var hasMergedCellsInTable = getMergedCellsPositions(state.tr).length > 0;
+  var allowBackgroundColor = (_pluginConfig$allowBa = pluginConfig === null || pluginConfig === void 0 ? void 0 : pluginConfig.allowBackgroundColor) !== null && _pluginConfig$allowBa !== void 0 ? _pluginConfig$allowBa : false;
   var dragMenuConfig = getDragMenuConfig(direction, getEditorContainerWidth, !!canDrag && !shouldMoveDisabled, hasMergedCellsInTable, editorView, tableMap, index, targetCellPosition, selectionRect, editorAnalyticsAPI);
   var _convertToDropdownIte = convertToDropdownItems(dragMenuConfig),
     menuItems = _convertToDropdownIte.menuItems,
     menuCallback = _convertToDropdownIte.menuCallback;
+  var handleSubMenuRef = function handleSubMenuRef(ref) {
+    var parent = closestElement(editorView.dom, '.fabric-editor-popup-scroll-parent');
+    if (!(parent && ref)) {
+      return;
+    }
+    var boundariesRect = parent.getBoundingClientRect();
+    var rect = ref.getBoundingClientRect();
+    if (rect.left + rect.width > boundariesRect.width) {
+      ref.style.left = "-".concat(rect.width, "px");
+    }
+  };
+  var setColor = function setColor(color) {
+    var _getTablePluginState = getTablePluginState(editorView.state),
+      targetCellPosition = _getTablePluginState.targetCellPosition;
+    var state = editorView.state,
+      dispatch = editorView.dispatch;
+    setColorWithAnalytics(editorAnalyticsAPI)(INPUT_METHOD.CONTEXT_MENU, color, targetCellPosition)(state, dispatch);
+    closeMenu();
+    setIsSubmenuOpen(false);
+  };
+  var createBackgroundColorMenuItem = function createBackgroundColorMenuItem() {
+    var _node$attrs;
+    var _getTablePluginState2 = getTablePluginState(editorView.state),
+      targetCellPosition = _getTablePluginState2.targetCellPosition;
+    var node = targetCellPosition ? state.doc.nodeAt(targetCellPosition) : null;
+    var background = hexToEditorBackgroundPaletteColor((node === null || node === void 0 || (_node$attrs = node.attrs) === null || _node$attrs === void 0 ? void 0 : _node$attrs.background) || '#ffffff');
+    return {
+      content: 'Background color',
+      value: {
+        name: 'background'
+      },
+      elemBefore: jsx(EditorBackgroundColorIcon, {
+        label: 'background color',
+        size: "medium"
+      }),
+      elemAfter: jsx("div", {
+        className: DropdownMenuSharedCssClassName.SUBMENU,
+        css: dragMenuBackgroundColorStyles
+      }, jsx("div", {
+        css: cellColourPreviewStyles(background),
+        className: ClassName.DRAG_SUBMENU_ICON
+      }), isSubmenuOpen && jsx("div", {
+        className: ClassName.DRAG_SUBMENU,
+        ref: handleSubMenuRef
+      }, jsx(ColorPalette, {
+        cols: 7,
+        onClick: setColor,
+        selectedColor: background,
+        paletteOptions: {
+          palette: cellBackgroundColorPalette,
+          paletteColorTooltipMessages: backgroundPaletteTooltipMessages,
+          hexToPaletteColor: hexToEditorBackgroundPaletteColor
+        }
+      })))
+    };
+  };
+  var toggleHeaderColumn = function toggleHeaderColumn() {
+    toggleHeaderColumnWithAnalytics(editorAnalyticsAPI)(state, dispatch);
+  };
+  var toggleHeaderRow = function toggleHeaderRow() {
+    toggleHeaderRowWithAnalytics(editorAnalyticsAPI)(state, dispatch);
+  };
+  var toggleRowNumbers = function toggleRowNumbers() {
+    toggleNumberColumnWithAnalytics(editorAnalyticsAPI)(state, dispatch);
+  };
+  var createhHeaderRowColumnMenuItem = function createhHeaderRowColumnMenuItem(direction) {
+    return direction === 'column' ? {
+      content: 'Header column',
+      value: {
+        name: 'header_column'
+      },
+      elemAfter: jsx("div", {
+        css: toggleStyles
+      }, jsx(Toggle, {
+        id: "toggle-header-column",
+        onChange: toggleHeaderColumn,
+        isChecked: checkIfHeaderColumnEnabled(selection)
+      }))
+    } : {
+      content: 'Header row',
+      value: {
+        name: 'header_row'
+      },
+      elemAfter: jsx("div", {
+        css: toggleStyles
+      }, jsx(Toggle, {
+        id: "toggle-header-row",
+        onChange: toggleHeaderRow,
+        isChecked: checkIfHeaderRowEnabled(selection)
+      }))
+    };
+  };
+  var createRowNumbersMenuItem = function createRowNumbersMenuItem() {
+    return {
+      content: 'Row numbers',
+      value: {
+        name: 'row_numbers'
+      },
+      elemAfter: jsx("div", {
+        css: toggleStyles
+      }, jsx(Toggle, {
+        id: "toggle-row-numbers",
+        onChange: toggleRowNumbers,
+        isChecked: checkIfNumberColumnEnabled(selection)
+      }))
+    };
+  };
 
   /**
    * This function is to check if the menu should be closed or not.
-   * As when continously clicking on drag handle on different rows/columns,
+   * As when continously clicking on tyle handle on different rows/columns,
    * should open the menu corresponding to the position of the drag handle.
    * @returns true when the menu should be closed, false otherwise
    */
@@ -115,18 +243,42 @@ export var DragMenu = function DragMenu(_ref) {
     var _menuCallback$item$va;
     var item = _ref2.item;
     (_menuCallback$item$va = menuCallback[item.value.name]) === null || _menuCallback$item$va === void 0 || _menuCallback$item$va.call(menuCallback, state, dispatch);
-    closeMenu();
+    switch (item.value.name) {
+      case 'header_column':
+        toggleHeaderColumn();
+        break;
+      case 'header_row':
+        toggleHeaderRow();
+        break;
+      case 'row_numbers':
+        toggleRowNumbers();
+        break;
+      default:
+        break;
+    }
+    if (['header_column', 'header_row', 'row_numbers', 'background'].indexOf(item.value.name) <= -1) {
+      closeMenu();
+    }
   };
   var handleItemMouseEnter = function handleItemMouseEnter(_ref3) {
     var _item$value$name;
     var item = _ref3.item;
-    if (!selectionRect || !((_item$value$name = item.value.name) !== null && _item$value$name !== void 0 && _item$value$name.startsWith('delete'))) {
+    if (!selectionRect) {
+      return;
+    }
+    if (item.value.name === 'background' && !isSubmenuOpen) {
+      setIsSubmenuOpen(true);
+    }
+    if (!((_item$value$name = item.value.name) !== null && _item$value$name !== void 0 && _item$value$name.startsWith('delete'))) {
       return;
     }
     (item.value.name === 'delete_column' ? hoverColumns(getSelectedColumnIndexes(selectionRect), true) : hoverRows(getSelectedRowIndexes(selectionRect), true))(state, dispatch);
   };
   var handleItemMouseLeave = function handleItemMouseLeave(_ref4) {
     var item = _ref4.item;
+    if (item.value.name === 'background' && isSubmenuOpen) {
+      setIsSubmenuOpen(false);
+    }
     if (['sort_column_asc', 'sort_column_desc', 'delete_column', 'delete_row'].indexOf(item.value.name) > -1) {
       clearHoverSelection()(state, dispatch);
     }
@@ -134,6 +286,24 @@ export var DragMenu = function DragMenu(_ref) {
   if (!menuItems) {
     return null;
   }
+  if (allowBackgroundColor && getBooleanFF('platform.editor.table.new-cell-context-menu-styling')) {
+    menuItems[0].items.unshift(createBackgroundColorMenuItem());
+  }
+
+  // If first row, add toggle for Hearder row, default is true
+  // If first column, add toggle for Header column, default is false
+  if (getBooleanFF('platform.editor.table.new-cell-context-menu-styling') && index === 0) {
+    menuItems.push({
+      items: [createhHeaderRowColumnMenuItem(direction)]
+    });
+  }
+
+  // All rows, add toggle for numbered rows, default is false
+  if (getBooleanFF('platform.editor.table.new-cell-context-menu-styling') && direction === 'row') {
+    index === 0 ? menuItems[menuItems.length - 1].items.push(createRowNumbersMenuItem()) : menuItems.push({
+      items: [createRowNumbersMenuItem()]
+    });
+  }
   return jsx(DropdownMenu, {
     mountTo: mountPoint
     //This needs be removed when the a11y is completely handled