@atlaskit/editor-tables 2.2.2 → 2.2.4

package/CHANGELOG.md

@@ -1,5 +1,19 @@
 # @atlaskit/editor-tables
 
+## 2.2.4
+
+### Patch Changes
+
+- [`4080eb013ef`](https://bitbucket.org/atlassian/atlassian-frontend/commits/4080eb013ef) - [ux] Fix bug where pasting inside a table did not respect the table's headers. Pasted cells are formatted based on the table they are being pasted into.
+- [`4f6a895f1d5`](https://bitbucket.org/atlassian/atlassian-frontend/commits/4f6a895f1d5) - Set selectable property for selectable nodes
+- [`4f6a895f1d5`](https://bitbucket.org/atlassian/atlassian-frontend/commits/4f6a895f1d5) - Fix arrow shift selection coming from outside of the table
+
+## 2.2.3
+
+### Patch Changes
+
+- [`dbef481f7df`](https://bitbucket.org/atlassian/atlassian-frontend/commits/dbef481f7df) - [ux] ED-15724: fix the merged row(2nd row or later) cannot been recongnized as row selection and cannot be deleted.
+
 ## 2.2.2
 
 ### Patch Changes

package/dist/cjs/cell-selection.js

@@ -265,9 +265,25 @@ var CellSelection = /*#__PURE__*/function (_Selection) {
         return false;
       }
 
+      var start = this.$anchorCell.start(-1);
+
       var map = _tableMap.TableMap.get(this.$anchorCell.node(-1));
 
-      var start = this.$anchorCell.start(-1);
+      var rowAtAnchorCell = map.rowCount(this.$anchorCell.pos - start);
+      var rowAtHeadCell = map.rowCount(this.$headCell.pos - start);
+      var isSelectionSameRow = rowAtAnchorCell === rowAtHeadCell; // if anchor and head in the same line, counting how many cells
+      // should be in the row except merged cell
+
+      var maxColumnInSelectedRow = map.getMaxColInRow(this.$anchorCell); // if selected cells less than table max column amount, and
+      // the anchor/head not in a merged cell
+      // it should be select maxColumnInSelectedRow to be TRUE
+
+      if (isSelectionSameRow && this.ranges.length <= map.width && !map.isPosMerged(this.$anchorCell.pos - start) && !map.isPosMerged(this.$headCell.pos - start)) {
+        return this.ranges.length === maxColumnInSelectedRow;
+      } // If anchor and head in different row, it should be always in first and
+      // last column to select the whole row.
+
+
       var anchorLeft = map.colCount(this.$anchorCell.pos - start);
       var headLeft = map.colCount(this.$headCell.pos - start);
 

package/dist/cjs/pm-plugins/input.js

@@ -100,12 +100,27 @@ function shiftArrow(axis, dir) {
     var sel = state.selection;
 
     if (!(sel instanceof _cellSelection.CellSelection)) {
+      var _maybeTableCell$paren;
+
       var end = view ? atEndOfCell(view, axis, dir) : null;
 
       if (end === null) {
         return false;
       }
 
+      var _state$selection = state.selection,
+          _$head = _state$selection.$head,
+          $anchor = _state$selection.$anchor;
+
+      var maybeTableCell = _$head.blockRange($anchor); // Make sure the selection is coming from the same cell
+
+
+      var sameCell = ['tableCell', 'tableHeader'].includes((maybeTableCell === null || maybeTableCell === void 0 ? void 0 : (_maybeTableCell$paren = maybeTableCell.parent) === null || _maybeTableCell$paren === void 0 ? void 0 : _maybeTableCell$paren.type.name) || '');
+
+      if (!sameCell) {
+        return false;
+      }
+
       sel = new _cellSelection.CellSelection(state.doc.resolve(end));
     }
 

package/dist/cjs/table-map.js

@@ -143,6 +143,29 @@ var TableMap = /*#__PURE__*/function () {
       }
 
       throw new RangeError('No cell with offset ' + pos + ' found');
+    } // Find the top side of the cell at the given position.
+
+  }, {
+    key: "rowCount",
+    value: function rowCount(pos) {
+      if (this.width <= 0) {
+        throw new RangeError('Wrong table width found');
+      }
+
+      for (var i = 0; i < this.map.length; i++) {
+        if (this.map[i] === pos) {
+          return Math.floor(i / this.width);
+        }
+      }
+
+      throw new RangeError('No cell with offset ' + pos + ' found');
+    }
+  }, {
+    key: "isPosMerged",
+    value: function isPosMerged(pos) {
+      return this.map.filter(function (cellPos) {
+        return cellPos === pos;
+      }).length > 1;
     } // :: (number, string, number) → ?number
     // Find the next cell in the given direction, starting from the cell
     // at `pos`, if any.
@@ -239,6 +262,15 @@ var TableMap = /*#__PURE__*/function () {
 
         rowStart = rowEnd;
       }
+    }
+  }, {
+    key: "getMaxColInRow",
+    value: function getMaxColInRow(pos) {
+      var parentRowNode = pos.parent;
+
+      if (parentRowNode.type.name === 'tableRow') {
+        return parentRowNode.childCount;
+      }
     } // :: (Node) → TableMap
     // Find the table map for the given table node.
 

package/dist/cjs/utils/copy-paste.js

@@ -22,16 +22,41 @@ var _tableMap = require("../table-map");
 
 var _colspan = require("./colspan");
 
+var _selectionRect = require("./selection-rect");
+
 var _tableNodeTypes = require("./table-node-types");
 
+var _toggleHeader = require("./toggle-header");
+
 function ownKeys(object, enumerableOnly) { var keys = Object.keys(object); if (Object.getOwnPropertySymbols) { var symbols = Object.getOwnPropertySymbols(object); enumerableOnly && (symbols = symbols.filter(function (sym) { return Object.getOwnPropertyDescriptor(object, sym).enumerable; })), keys.push.apply(keys, symbols); } return keys; }
 
 function _objectSpread(target) { for (var i = 1; i < arguments.length; i++) { var source = null != arguments[i] ? arguments[i] : {}; i % 2 ? ownKeys(Object(source), !0).forEach(function (key) { (0, _defineProperty2.default)(target, key, source[key]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(target, Object.getOwnPropertyDescriptors(source)) : ownKeys(Object(source)).forEach(function (key) { Object.defineProperty(target, key, Object.getOwnPropertyDescriptor(source, key)); }); } return target; }
 
 // Utilities to help with copying and pasting table cells
-// : (Slice) → ?{width: number, height: number, rows: [Fragment]}
+
+/**
+ * Replace any header cells with table cells.
+ *
+ * @param schema
+ * @param cells
+ * @returns Fragment with header cells converted to table cells
+ */
+function stripHeaderType(schema, cells) {
+  var newCells = [];
+  cells.forEach(function (cell) {
+    var _cellNodeType$createA;
+
+    // Convert to cell type if not already
+    var cellNodeType = (0, _tableNodeTypes.tableNodeTypes)(schema).cell;
+    var tableCell = cell.type === cellNodeType ? cell : (_cellNodeType$createA = cellNodeType.createAndFill(cell.attrs, cell.content, cell.marks)) !== null && _cellNodeType$createA !== void 0 ? _cellNodeType$createA : cell;
+    newCells.push(tableCell);
+  });
+  return _prosemirrorModel.Fragment.from(newCells);
+} // : (Slice) → ?{width: number, height: number, rows: [Fragment]}
 // Get a rectangular area of cells from a slice, or null if the outer
 // nodes of the slice aren't table cells or rows.
+
+
 function pastedCells(slice) {
   if (!slice.size) {
     return null;
@@ -78,7 +103,10 @@ function pastedCells(slice) {
     return null;
   }
 
-  return ensureRectangular(schema, rows);
+  var rowsWithoutHeaders = rows.map(function (row) {
+    return stripHeaderType(schema, row);
+  });
+  return ensureRectangular(schema, rowsWithoutHeaders);
 } // : (Schema, [Fragment]) → {width: number, height: number, rows: [Fragment]}
 // Compute the width and height of a set of cells, and make sure each
 // row has the same number of cells.
@@ -364,12 +392,45 @@ function isolateVertical(tr, map, table, start, top, bottom, left, mapFrom) {
   }
 
   return found;
+}
+
+function applyHeaderCells(tr, tableMap, state, tableStart, table, headerRowEnabled, headerColumnEnabled) {
+  var schema = state.schema;
+
+  var setMarkup = function setMarkup(tr, row, col, headerEnabled) {
+    var cellPos = tableStart + tableMap.positionAt(row, col, table);
+    var cell = tr.doc.nodeAt(cellPos);
+    var newType = headerEnabled ? schema.nodes.tableHeader : schema.nodes.tableCell;
+    var isCellTypeChanged = newType !== (cell === null || cell === void 0 ? void 0 : cell.type);
+    var isCellTypeValid = [schema.nodes.tableCell, schema.nodes.tableHeader].includes(cell === null || cell === void 0 ? void 0 : cell.type);
+
+    if (isCellTypeChanged && isCellTypeValid) {
+      tr.setNodeMarkup(cellPos, newType, cell === null || cell === void 0 ? void 0 : cell.attrs, cell === null || cell === void 0 ? void 0 : cell.marks);
+    }
+  }; // For row === 0 && col === 0 it is enabled if either are enabled
+
+
+  setMarkup(tr, 0, 0, headerColumnEnabled || headerRowEnabled); // Header Column
+
+  for (var col = 1; col < tableMap.width; col++) {
+    setMarkup(tr, 0, col, headerRowEnabled);
+  } // Header Row
+
+
+  for (var row = 1; row < tableMap.height; row++) {
+    setMarkup(tr, row, 0, headerColumnEnabled);
+  }
 } // Insert the given set of cells (as returned by `pastedCells`) into a
 // table, at the position pointed at by rect.
 
 
 function insertCells(state, dispatch, tableStart, rect, cells) {
   var table = state.doc;
+  var newRect = (0, _selectionRect.selectedRect)(state);
+  var types = (0, _tableNodeTypes.tableNodeTypes)(state.schema); // Get if the header row and column are enabled on the original table
+
+  var headerRowEnabled = (0, _toggleHeader.isHeaderEnabledByType)('row', newRect, types);
+  var headerColumnEnabled = (0, _toggleHeader.isHeaderEnabledByType)('column', newRect, types);
 
   if (tableStart) {
     table = state.doc.nodeAt(tableStart - 1);
@@ -425,6 +486,7 @@ function insertCells(state, dispatch, tableStart, rect, cells) {
   }
 
   recomp();
+  applyHeaderCells(tr, map, state, tableStart, table, headerRowEnabled, headerColumnEnabled);
   tr.setSelection(new _cellSelection.CellSelection(tr.doc.resolve(tableStart + map.positionAt(top, left, table)), tr.doc.resolve(tableStart + map.positionAt(bottom - 1, right - 1, table))));
   dispatch(tr);
 }

package/dist/cjs/utils/toggle-header.js

@@ -3,6 +3,7 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
+exports.isHeaderEnabledByType = isHeaderEnabledByType;
 exports.toggleHeader = toggleHeader;
 
 var _tableMap = require("../table-map");

package/dist/cjs/version.json

@@ -1,5 +1,5 @@
 {
 	"name": "@atlaskit/editor-tables",
-	"version": "2.2.2",
+	"version": "2.2.4",
 	"sideEffects": false
 }

package/dist/es2019/cell-selection.js

@@ -245,8 +245,23 @@ export class CellSelection extends Selection {
       return false;
     }
 
-    const map = TableMap.get(this.$anchorCell.node(-1));
     const start = this.$anchorCell.start(-1);
+    const map = TableMap.get(this.$anchorCell.node(-1));
+    const rowAtAnchorCell = map.rowCount(this.$anchorCell.pos - start);
+    const rowAtHeadCell = map.rowCount(this.$headCell.pos - start);
+    const isSelectionSameRow = rowAtAnchorCell === rowAtHeadCell; // if anchor and head in the same line, counting how many cells
+    // should be in the row except merged cell
+
+    let maxColumnInSelectedRow = map.getMaxColInRow(this.$anchorCell); // if selected cells less than table max column amount, and
+    // the anchor/head not in a merged cell
+    // it should be select maxColumnInSelectedRow to be TRUE
+
+    if (isSelectionSameRow && this.ranges.length <= map.width && !map.isPosMerged(this.$anchorCell.pos - start) && !map.isPosMerged(this.$headCell.pos - start)) {
+      return this.ranges.length === maxColumnInSelectedRow;
+    } // If anchor and head in different row, it should be always in first and
+    // last column to select the whole row.
+
+
     const anchorLeft = map.colCount(this.$anchorCell.pos - start);
     const headLeft = map.colCount(this.$headCell.pos - start);
 

package/dist/es2019/pm-plugins/input.js

@@ -82,12 +82,28 @@ function shiftArrow(axis, dir) {
     let sel = state.selection;
 
     if (!(sel instanceof CellSelection)) {
+      var _maybeTableCell$paren;
+
       const end = view ? atEndOfCell(view, axis, dir) : null;
 
       if (end === null) {
         return false;
       }
 
+      const {
+        selection: {
+          $head,
+          $anchor
+        }
+      } = state;
+      const maybeTableCell = $head.blockRange($anchor); // Make sure the selection is coming from the same cell
+
+      const sameCell = ['tableCell', 'tableHeader'].includes((maybeTableCell === null || maybeTableCell === void 0 ? void 0 : (_maybeTableCell$paren = maybeTableCell.parent) === null || _maybeTableCell$paren === void 0 ? void 0 : _maybeTableCell$paren.type.name) || '');
+
+      if (!sameCell) {
+        return false;
+      }
+
       sel = new CellSelection(state.doc.resolve(end));
     }
 

package/dist/es2019/table-map.js

@@ -122,6 +122,25 @@ export class TableMap {
     }
 
     throw new RangeError('No cell with offset ' + pos + ' found');
+  } // Find the top side of the cell at the given position.
+
+
+  rowCount(pos) {
+    if (this.width <= 0) {
+      throw new RangeError('Wrong table width found');
+    }
+
+    for (let i = 0; i < this.map.length; i++) {
+      if (this.map[i] === pos) {
+        return Math.floor(i / this.width);
+      }
+    }
+
+    throw new RangeError('No cell with offset ' + pos + ' found');
+  }
+
+  isPosMerged(pos) {
+    return this.map.filter(cellPos => cellPos === pos).length > 1;
   } // :: (number, string, number) → ?number
   // Find the next cell in the given direction, starting from the cell
   // at `pos`, if any.
@@ -215,6 +234,14 @@ export class TableMap {
 
       rowStart = rowEnd;
     }
+  }
+
+  getMaxColInRow(pos) {
+    let parentRowNode = pos.parent;
+
+    if (parentRowNode.type.name === 'tableRow') {
+      return parentRowNode.childCount;
+    }
   } // :: (Node) → TableMap
   // Find the table map for the given table node.
 

package/dist/es2019/utils/copy-paste.js

@@ -14,11 +14,34 @@ import { Transform } from 'prosemirror-transform';
 import { CellSelection } from '../cell-selection';
 import { TableMap } from '../table-map';
 import { removeColSpan } from './colspan';
-import { tableNodeTypes } from './table-node-types'; // Utilities to help with copying and pasting table cells
-// : (Slice) → ?{width: number, height: number, rows: [Fragment]}
+import { selectedRect } from './selection-rect';
+import { tableNodeTypes } from './table-node-types';
+import { isHeaderEnabledByType } from './toggle-header'; // Utilities to help with copying and pasting table cells
+
+/**
+ * Replace any header cells with table cells.
+ *
+ * @param schema
+ * @param cells
+ * @returns Fragment with header cells converted to table cells
+ */
+
+function stripHeaderType(schema, cells) {
+  const newCells = [];
+  cells.forEach(cell => {
+    var _cellNodeType$createA;
+
+    // Convert to cell type if not already
+    const cellNodeType = tableNodeTypes(schema).cell;
+    const tableCell = cell.type === cellNodeType ? cell : (_cellNodeType$createA = cellNodeType.createAndFill(cell.attrs, cell.content, cell.marks)) !== null && _cellNodeType$createA !== void 0 ? _cellNodeType$createA : cell;
+    newCells.push(tableCell);
+  });
+  return Fragment.from(newCells);
+} // : (Slice) → ?{width: number, height: number, rows: [Fragment]}
 // Get a rectangular area of cells from a slice, or null if the outer
 // nodes of the slice aren't table cells or rows.
 
+
 export function pastedCells(slice) {
   if (!slice.size) {
     return null;
@@ -69,7 +92,8 @@ export function pastedCells(slice) {
     return null;
   }
 
-  return ensureRectangular(schema, rows);
+  const rowsWithoutHeaders = rows.map(row => stripHeaderType(schema, row));
+  return ensureRectangular(schema, rowsWithoutHeaders);
 } // : (Schema, [Fragment]) → {width: number, height: number, rows: [Fragment]}
 // Compute the width and height of a set of cells, and make sure each
 // row has the same number of cells.
@@ -356,12 +380,47 @@ function isolateVertical(tr, map, table, start, top, bottom, left, mapFrom) {
   }
 
   return found;
+}
+
+function applyHeaderCells(tr, tableMap, state, tableStart, table, headerRowEnabled, headerColumnEnabled) {
+  const {
+    schema
+  } = state;
+
+  const setMarkup = (tr, row, col, headerEnabled) => {
+    const cellPos = tableStart + tableMap.positionAt(row, col, table);
+    const cell = tr.doc.nodeAt(cellPos);
+    const newType = headerEnabled ? schema.nodes.tableHeader : schema.nodes.tableCell;
+    const isCellTypeChanged = newType !== (cell === null || cell === void 0 ? void 0 : cell.type);
+    const isCellTypeValid = [schema.nodes.tableCell, schema.nodes.tableHeader].includes(cell === null || cell === void 0 ? void 0 : cell.type);
+
+    if (isCellTypeChanged && isCellTypeValid) {
+      tr.setNodeMarkup(cellPos, newType, cell === null || cell === void 0 ? void 0 : cell.attrs, cell === null || cell === void 0 ? void 0 : cell.marks);
+    }
+  }; // For row === 0 && col === 0 it is enabled if either are enabled
+
+
+  setMarkup(tr, 0, 0, headerColumnEnabled || headerRowEnabled); // Header Column
+
+  for (let col = 1; col < tableMap.width; col++) {
+    setMarkup(tr, 0, col, headerRowEnabled);
+  } // Header Row
+
+
+  for (let row = 1; row < tableMap.height; row++) {
+    setMarkup(tr, row, 0, headerColumnEnabled);
+  }
 } // Insert the given set of cells (as returned by `pastedCells`) into a
 // table, at the position pointed at by rect.
 
 
 export function insertCells(state, dispatch, tableStart, rect, cells) {
   let table = state.doc;
+  const newRect = selectedRect(state);
+  const types = tableNodeTypes(state.schema); // Get if the header row and column are enabled on the original table
+
+  const headerRowEnabled = isHeaderEnabledByType('row', newRect, types);
+  const headerColumnEnabled = isHeaderEnabledByType('column', newRect, types);
 
   if (tableStart) {
     table = state.doc.nodeAt(tableStart - 1);
@@ -420,6 +479,7 @@ export function insertCells(state, dispatch, tableStart, rect, cells) {
   }
 
   recomp();
+  applyHeaderCells(tr, map, state, tableStart, table, headerRowEnabled, headerColumnEnabled);
   tr.setSelection(new CellSelection(tr.doc.resolve(tableStart + map.positionAt(top, left, table)), tr.doc.resolve(tableStart + map.positionAt(bottom - 1, right - 1, table))));
   dispatch(tr);
 }

package/dist/es2019/utils/toggle-header.js

@@ -2,8 +2,7 @@ import { Rect } from '../table-map';
 import { selectedRect } from './selection-rect';
 import { tableNodeTypes } from './table-node-types';
 import { isInTable } from './tables';
-
-function isHeaderEnabledByType(type, rect, types) {
+export function isHeaderEnabledByType(type, rect, types) {
   // Get cell positions for first row or first column
   const cellPositions = rect.map.cellsInRect({
     left: 0,
@@ -23,7 +22,6 @@ function isHeaderEnabledByType(type, rect, types) {
   return true;
 } // Toggles between row/column header and normal cells (Only applies to first row/column).
 
-
 export function toggleHeader(type) {
   return function (state, dispatch) {
     if (!isInTable(state)) {

package/dist/es2019/version.json

@@ -1,5 +1,5 @@
 {
 	"name": "@atlaskit/editor-tables",
-	"version": "2.2.2",
+	"version": "2.2.4",
 	"sideEffects": false
 }

package/dist/esm/cell-selection.js

@@ -246,8 +246,23 @@ export var CellSelection = /*#__PURE__*/function (_Selection) {
         return false;
       }
 
-      var map = TableMap.get(this.$anchorCell.node(-1));
       var start = this.$anchorCell.start(-1);
+      var map = TableMap.get(this.$anchorCell.node(-1));
+      var rowAtAnchorCell = map.rowCount(this.$anchorCell.pos - start);
+      var rowAtHeadCell = map.rowCount(this.$headCell.pos - start);
+      var isSelectionSameRow = rowAtAnchorCell === rowAtHeadCell; // if anchor and head in the same line, counting how many cells
+      // should be in the row except merged cell
+
+      var maxColumnInSelectedRow = map.getMaxColInRow(this.$anchorCell); // if selected cells less than table max column amount, and
+      // the anchor/head not in a merged cell
+      // it should be select maxColumnInSelectedRow to be TRUE
+
+      if (isSelectionSameRow && this.ranges.length <= map.width && !map.isPosMerged(this.$anchorCell.pos - start) && !map.isPosMerged(this.$headCell.pos - start)) {
+        return this.ranges.length === maxColumnInSelectedRow;
+      } // If anchor and head in different row, it should be always in first and
+      // last column to select the whole row.
+
+
       var anchorLeft = map.colCount(this.$anchorCell.pos - start);
       var headLeft = map.colCount(this.$headCell.pos - start);
 

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

@@ -82,12 +82,27 @@ function shiftArrow(axis, dir) {
     var sel = state.selection;
 
     if (!(sel instanceof CellSelection)) {
+      var _maybeTableCell$paren;
+
       var end = view ? atEndOfCell(view, axis, dir) : null;
 
       if (end === null) {
         return false;
       }
 
+      var _state$selection = state.selection,
+          _$head = _state$selection.$head,
+          $anchor = _state$selection.$anchor;
+
+      var maybeTableCell = _$head.blockRange($anchor); // Make sure the selection is coming from the same cell
+
+
+      var sameCell = ['tableCell', 'tableHeader'].includes((maybeTableCell === null || maybeTableCell === void 0 ? void 0 : (_maybeTableCell$paren = maybeTableCell.parent) === null || _maybeTableCell$paren === void 0 ? void 0 : _maybeTableCell$paren.type.name) || '');
+
+      if (!sameCell) {
+        return false;
+      }
+
       sel = new CellSelection(state.doc.resolve(end));
     }
 

package/dist/esm/table-map.js

@@ -130,6 +130,29 @@ export var TableMap = /*#__PURE__*/function () {
       }
 
       throw new RangeError('No cell with offset ' + pos + ' found');
+    } // Find the top side of the cell at the given position.
+
+  }, {
+    key: "rowCount",
+    value: function rowCount(pos) {
+      if (this.width <= 0) {
+        throw new RangeError('Wrong table width found');
+      }
+
+      for (var i = 0; i < this.map.length; i++) {
+        if (this.map[i] === pos) {
+          return Math.floor(i / this.width);
+        }
+      }
+
+      throw new RangeError('No cell with offset ' + pos + ' found');
+    }
+  }, {
+    key: "isPosMerged",
+    value: function isPosMerged(pos) {
+      return this.map.filter(function (cellPos) {
+        return cellPos === pos;
+      }).length > 1;
     } // :: (number, string, number) → ?number
     // Find the next cell in the given direction, starting from the cell
     // at `pos`, if any.
@@ -226,6 +249,15 @@ export var TableMap = /*#__PURE__*/function () {
 
         rowStart = rowEnd;
       }
+    }
+  }, {
+    key: "getMaxColInRow",
+    value: function getMaxColInRow(pos) {
+      var parentRowNode = pos.parent;
+
+      if (parentRowNode.type.name === 'tableRow') {
+        return parentRowNode.childCount;
+      }
     } // :: (Node) → TableMap
     // Find the table map for the given table node.
 

package/dist/esm/utils/copy-paste.js

@@ -20,11 +20,34 @@ import { Transform } from 'prosemirror-transform';
 import { CellSelection } from '../cell-selection';
 import { TableMap } from '../table-map';
 import { removeColSpan } from './colspan';
-import { tableNodeTypes } from './table-node-types'; // Utilities to help with copying and pasting table cells
-// : (Slice) → ?{width: number, height: number, rows: [Fragment]}
+import { selectedRect } from './selection-rect';
+import { tableNodeTypes } from './table-node-types';
+import { isHeaderEnabledByType } from './toggle-header'; // Utilities to help with copying and pasting table cells
+
+/**
+ * Replace any header cells with table cells.
+ *
+ * @param schema
+ * @param cells
+ * @returns Fragment with header cells converted to table cells
+ */
+
+function stripHeaderType(schema, cells) {
+  var newCells = [];
+  cells.forEach(function (cell) {
+    var _cellNodeType$createA;
+
+    // Convert to cell type if not already
+    var cellNodeType = tableNodeTypes(schema).cell;
+    var tableCell = cell.type === cellNodeType ? cell : (_cellNodeType$createA = cellNodeType.createAndFill(cell.attrs, cell.content, cell.marks)) !== null && _cellNodeType$createA !== void 0 ? _cellNodeType$createA : cell;
+    newCells.push(tableCell);
+  });
+  return Fragment.from(newCells);
+} // : (Slice) → ?{width: number, height: number, rows: [Fragment]}
 // Get a rectangular area of cells from a slice, or null if the outer
 // nodes of the slice aren't table cells or rows.
 
+
 export function pastedCells(slice) {
   if (!slice.size) {
     return null;
@@ -71,7 +94,10 @@ export function pastedCells(slice) {
     return null;
   }
 
-  return ensureRectangular(schema, rows);
+  var rowsWithoutHeaders = rows.map(function (row) {
+    return stripHeaderType(schema, row);
+  });
+  return ensureRectangular(schema, rowsWithoutHeaders);
 } // : (Schema, [Fragment]) → {width: number, height: number, rows: [Fragment]}
 // Compute the width and height of a set of cells, and make sure each
 // row has the same number of cells.
@@ -354,12 +380,45 @@ function isolateVertical(tr, map, table, start, top, bottom, left, mapFrom) {
   }
 
   return found;
+}
+
+function applyHeaderCells(tr, tableMap, state, tableStart, table, headerRowEnabled, headerColumnEnabled) {
+  var schema = state.schema;
+
+  var setMarkup = function setMarkup(tr, row, col, headerEnabled) {
+    var cellPos = tableStart + tableMap.positionAt(row, col, table);
+    var cell = tr.doc.nodeAt(cellPos);
+    var newType = headerEnabled ? schema.nodes.tableHeader : schema.nodes.tableCell;
+    var isCellTypeChanged = newType !== (cell === null || cell === void 0 ? void 0 : cell.type);
+    var isCellTypeValid = [schema.nodes.tableCell, schema.nodes.tableHeader].includes(cell === null || cell === void 0 ? void 0 : cell.type);
+
+    if (isCellTypeChanged && isCellTypeValid) {
+      tr.setNodeMarkup(cellPos, newType, cell === null || cell === void 0 ? void 0 : cell.attrs, cell === null || cell === void 0 ? void 0 : cell.marks);
+    }
+  }; // For row === 0 && col === 0 it is enabled if either are enabled
+
+
+  setMarkup(tr, 0, 0, headerColumnEnabled || headerRowEnabled); // Header Column
+
+  for (var col = 1; col < tableMap.width; col++) {
+    setMarkup(tr, 0, col, headerRowEnabled);
+  } // Header Row
+
+
+  for (var row = 1; row < tableMap.height; row++) {
+    setMarkup(tr, row, 0, headerColumnEnabled);
+  }
 } // Insert the given set of cells (as returned by `pastedCells`) into a
 // table, at the position pointed at by rect.
 
 
 export function insertCells(state, dispatch, tableStart, rect, cells) {
   var table = state.doc;
+  var newRect = selectedRect(state);
+  var types = tableNodeTypes(state.schema); // Get if the header row and column are enabled on the original table
+
+  var headerRowEnabled = isHeaderEnabledByType('row', newRect, types);
+  var headerColumnEnabled = isHeaderEnabledByType('column', newRect, types);
 
   if (tableStart) {
     table = state.doc.nodeAt(tableStart - 1);
@@ -414,6 +473,7 @@ export function insertCells(state, dispatch, tableStart, rect, cells) {
   }
 
   recomp();
+  applyHeaderCells(tr, map, state, tableStart, table, headerRowEnabled, headerColumnEnabled);
   tr.setSelection(new CellSelection(tr.doc.resolve(tableStart + map.positionAt(top, left, table)), tr.doc.resolve(tableStart + map.positionAt(bottom - 1, right - 1, table))));
   dispatch(tr);
 }

package/dist/esm/utils/toggle-header.js

@@ -2,8 +2,7 @@ import { Rect } from '../table-map';
 import { selectedRect } from './selection-rect';
 import { tableNodeTypes } from './table-node-types';
 import { isInTable } from './tables';
-
-function isHeaderEnabledByType(type, rect, types) {
+export function isHeaderEnabledByType(type, rect, types) {
   // Get cell positions for first row or first column
   var cellPositions = rect.map.cellsInRect({
     left: 0,
@@ -23,7 +22,6 @@ function isHeaderEnabledByType(type, rect, types) {
   return true;
 } // Toggles between row/column header and normal cells (Only applies to first row/column).
 
-
 export function toggleHeader(type) {
   return function (state, dispatch) {
     if (!isInTable(state)) {

package/dist/esm/version.json

@@ -1,5 +1,5 @@
 {
 	"name": "@atlaskit/editor-tables",
-	"version": "2.2.2",
+	"version": "2.2.4",
 	"sideEffects": false
 }

package/dist/types/table-map.d.ts

@@ -1,4 +1,4 @@
-import { Node as PMNode } from 'prosemirror-model';
+import { Node as PMNode, ResolvedPos } from 'prosemirror-model';
 import { Axis } from './types';
 export declare class Rect {
     left: number;
@@ -54,9 +54,12 @@ export declare class TableMap {
     constructor(width: number, height: number, map: number[], problems?: TableProblem[] | null);
     findCell(pos: number): Rect;
     colCount(pos: number): number;
+    rowCount(pos: number): number;
+    isPosMerged(pos: number): boolean;
     nextCell(pos: number, axis: Axis, dir: number): number | null;
     rectBetween(a: number, b: number): Rect;
     cellsInRect(rect: Rect): number[];
     positionAt(row: number, col: number, table: PMNode): number;
+    getMaxColInRow(pos: ResolvedPos): number | undefined;
     static get(table: PMNode): TableMap;
 }

package/dist/types/utils/toggle-header.d.ts

@@ -1,3 +1,6 @@
 import { Command } from '../types';
+import { SelectionRect } from './selection-rect';
+import { TableNodeCache } from './table-node-types';
 export declare type ToggleType = 'column' | 'row';
+export declare function isHeaderEnabledByType(type: ToggleType, rect: SelectionRect, types: TableNodeCache): boolean;
 export declare function toggleHeader(type: ToggleType): Command;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@atlaskit/editor-tables",
-  "version": "2.2.2",
+  "version": "2.2.4",
   "description": "A package that contains common classes and utility functions for editor tables",
   "publishConfig": {
     "registry": "https://registry.npmjs.org/"
@@ -29,8 +29,8 @@
     "prosemirror-view": "1.23.2"
   },
   "devDependencies": {
-    "@atlaskit/adf-schema": "^24.0.0",
-    "@atlaskit/editor-test-helpers": "^17.2.0",
+    "@atlaskit/adf-schema": "^25.1.0",
+    "@atlaskit/editor-test-helpers": "^18.0.0",
     "@atlassian/atlassian-frontend-prettier-config-1.0.1": "npm:@atlassian/atlassian-frontend-prettier-config@1.0.1"
   },
   "techstack": {
@@ -49,4 +49,4 @@
     }
   },
   "prettier": "@atlassian/atlassian-frontend-prettier-config-1.0.1"
-}
+}

package/report.api.md

@@ -1,4 +1,4 @@
-<!-- API Report Version: 2.2 -->
+<!-- API Report Version: 2.3 -->
 
 ## API Report File for "@atlaskit/editor-tables"
 
@@ -8,6 +8,7 @@
 ### Table of contents
 
 - [Main Entry Types](#main-entry-types)
+- [Peer Dependencies](#peer-dependencies)
 
 ### Main Entry Types
 
@@ -132,8 +133,12 @@ export class TableMap {
   // (undocumented)
   static get(table: Node_2): TableMap;
   // (undocumented)
+  getMaxColInRow(pos: ResolvedPos): number | undefined;
+  // (undocumented)
   height: number;
   // (undocumented)
+  isPosMerged(pos: number): boolean;
+  // (undocumented)
   map: number[];
   // (undocumented)
   nextCell(pos: number, axis: Axis, dir: number): null | number;
@@ -144,6 +149,8 @@ export class TableMap {
   // (undocumented)
   rectBetween(a: number, b: number): Rect;
   // (undocumented)
+  rowCount(pos: number): number;
+  // (undocumented)
   width: number;
 }
 
@@ -205,3 +212,13 @@ export const uuid: {
 ```
 
 <!--SECTION END: Main Entry Types-->
+
+### Peer Dependencies
+
+<!--SECTION START: Peer Dependencies-->
+
+```json
+{}
+```
+
+<!--SECTION END: Peer Dependencies-->