desy-html 16.0.3 → 17.0.0

package/src/js/aria/treegrid.js

@@ -1,508 +1,508 @@
-'use strict';
-export function TreeGrid(treegridElem, doAllowRowFocus, doStartRowFocus) {
-  function initAttributes() {
-    // Make sure focusable elements are not in the tab order
-    // They will be added back in for the active row
-    setTabIndexOfFocusableElements(treegridElem, -1);
-
-    // Add tabindex="0" to first row, "-1" to other rows
-    // We will use the roving tabindex method since aria-activedescendant
-    var rows = getAllRows();
-    var index = rows.length;
-    var startRowIndex = doStartRowFocus ? 0 : -1;
-
-    while (index--) {
-      if (doAllowRowFocus) {
-        rows[index].tabIndex = index === startRowIndex ? 0 : -1;
-      } else {
-        setTabIndexForCellsInRow(rows[index], -1);
-        moveAriaExpandedToFirstCell(rows[index]);
-      }
-    }
-    // Here we collapse all items if parent not expanded (hidden class)
-    rows.forEach(r => {
-      if (isExpandable(r)) {
-        changeExpanded(isExpanded(r), r);
-      }
-    });
-
-    if (doStartRowFocus) {
-      return;
-    }
-
-    // Start with cell focus
-    var firstCell = getNavigableCols(rows[0])[0];
-    setTabIndexForCell(firstCell);
-  }
-
-  function setTabIndexForCell(cell, tabIndex) {
-    var focusable = getFocusableElements(cell)[0] || cell;
-    focusable.tabIndex = tabIndex;
-  }
-
-  function setTabIndexForCellsInRow(row, tabIndex) {
-    var cells = getNavigableCols(row);
-    var cellIndex = cells.length;
-    while (cellIndex--) {
-      setTabIndexForCell(cells[cellIndex], tabIndex);
-    }
-  }
-
-  function getAllRows() {
-    var nodeList = treegridElem.querySelectorAll('tbody > tr');
-    return Array.prototype.slice.call(nodeList);
-  }
-
-  function getFocusableElements(root) {
-    // textarea not supported as a cell widget as it's multiple lines
-    // and needs up/down keys
-    // These should all be descendants of a cell
-    var nodeList = root.querySelectorAll('a,button,input,td>[tabindex],th>[tabindex]');
-    return Array.prototype.slice.call(nodeList);
-  }
-
-  function setTabIndexOfFocusableElements(root, tabIndex) {
-    var focusableElements = getFocusableElements(root);
-    var index = focusableElements.length;
-    while (index--) {
-      focusableElements[index].tabIndex = tabIndex;
-    }
-  }
-
-  function getAllNavigableRows() {
-    var nodeList = treegridElem.querySelectorAll(
-      'tbody > tr:not([class~="hidden"])'
-    );
-    // Convert to array so that we can use array methods on it
-    return Array.prototype.slice.call(nodeList);
-  }
-
-  function getNavigableCols(currentRow) {
-    return Array.from(currentRow.querySelectorAll('td, th'));
-  }
-
-  function restrictIndex(index, numItems) {
-    if (index < 0) {
-      return 0;
-    }
-    return index >= numItems ? index - 1 : index;
-  }
-
-  function focus(elem) {
-    elem.tabIndex = 0; // Ensure focusable
-    elem.focus();
-  }
-
-  function focusCell(cell) {
-    // Check for focusable child such as link or textbox
-    // and use that if available
-    var focusableChildren = getFocusableElements(cell);
-    focus(focusableChildren[0] || cell);
-  }
-
-  // Restore tabIndex to what it should be when focus switches from
-  // one treegrid item to another
-  function onFocusIn(event) {
-    var newTreeGridFocus =
-      event.target !== window &&
-      treegridElem.contains(event.target) &&
-      event.target;
-
-    // The last row we considered focused
-    var oldCurrentRow = enableTabbingInActiveRowDescendants.tabbingRow;
-    if (oldCurrentRow) {
-      enableTabbingInActiveRowDescendants(false, oldCurrentRow);
-    }
-    if (
-      doAllowRowFocus &&
-      onFocusIn.prevTreeGridFocus &&
-      (onFocusIn.prevTreeGridFocus.localName === 'td' || onFocusIn.prevTreeGridFocus.localName === 'th')
-    ) {
-      // Was focused on td, remove tabIndex so that it's not focused on click
-      onFocusIn.prevTreeGridFocus.removeAttribute('tabindex');
-    }
-
-    if (newTreeGridFocus) {
-      // Stayed in treegrid
-      if (oldCurrentRow) {
-        // There will be a different current row that will be
-        // the tabbable one
-        oldCurrentRow.tabIndex = -1;
-      }
-
-      // The new row
-      var currentRow = getRowWithFocus();
-      if (currentRow) {
-        currentRow.tabIndex = 0;
-        // Items within current row are also tabbable
-        enableTabbingInActiveRowDescendants(true, currentRow);
-      }
-    }
-
-    onFocusIn.prevTreeGridFocus = newTreeGridFocus;
-  }
-
-  // Set whether interactive elements within a row are tabbable
-  function enableTabbingInActiveRowDescendants(isTabbingOn, row) {
-    if (row) {
-      setTabIndexOfFocusableElements(row, isTabbingOn ? 0 : -1);
-      if (isTabbingOn) {
-        enableTabbingInActiveRowDescendants.tabbingRow = row;
-      } else {
-        if (enableTabbingInActiveRowDescendants.tabbingRow === row) {
-          enableTabbingInActiveRowDescendants.tabbingRow = null;
-        }
-      }
-    }
-  }
-
-  // The row with focus is the row that either has focus or an element
-  // inside of it has focus
-  function getRowWithFocus() {
-    return getContainingRow(document.activeElement);
-  }
-
-  function getContainingRow(start) {
-    var possibleRow = start;
-    if (treegridElem.contains(possibleRow)) {
-      while (possibleRow !== treegridElem) {
-        if (possibleRow.localName === 'tr') {
-          return possibleRow;
-        }
-        possibleRow = possibleRow.parentElement;
-      }
-    }
-  }
-
-  function isRowFocused() {
-    return getRowWithFocus() === document.activeElement;
-  }
-
-  // Note: contenteditable not currently supported
-  function isEditableFocused() {
-    var focusedElem = document.activeElement;
-    return focusedElem.localName === 'input';
-  }
-
-  function getColWithFocus(currentRow) {
-    if (currentRow) {
-      var possibleCol = document.activeElement;
-      if (currentRow.contains(possibleCol)) {
-        while (possibleCol !== currentRow) {
-          if (possibleCol.localName === 'td' || possibleCol.localName === 'th') {
-            return possibleCol;
-          }
-          possibleCol = possibleCol.parentElement;
-        }
-      }
-    }
-  }
-
-  function getLevel(row) {
-    return row && parseInt(row.getAttribute('aria-level'));
-  }
-
-  // Move backwards (direction = -1) or forwards (direction = 1)
-  // If we also need to move down/up a level, requireLevelChange = true
-  // When
-  function moveByRow(direction, requireLevelChange) {
-    var currentRow = getRowWithFocus();
-    var requiredLevel =
-      requireLevelChange && currentRow && getLevel(currentRow) + direction;
-    var rows = getAllNavigableRows();
-    var numRows = rows.length;
-    var rowIndex = currentRow ? rows.indexOf(currentRow) : -1;
-    // When moving down a level, only allow moving to next row as the
-    // first child will never be farther than that
-    var maxDistance = requireLevelChange && direction === 1 ? 1 : NaN;
-
-    // Move in direction until required level is found
-    do {
-      if (maxDistance-- === 0) {
-        return; // Failed to find required level, return without focus change
-      }
-      rowIndex = restrictIndex(rowIndex + direction, numRows);
-    } while (requiredLevel && requiredLevel !== getLevel(rows[rowIndex]));
-
-    if (!focusSameColInDifferentRow(currentRow, rows[rowIndex])) {
-      focus(rows[rowIndex]);
-    }
-  }
-
-  function focusSameColInDifferentRow(fromRow, toRow) {
-    var currentCol = getColWithFocus(fromRow);
-    if (!currentCol) {
-      return;
-    }
-
-    var fromCols = getNavigableCols(fromRow);
-    var currentColIndex = fromCols.indexOf(currentCol);
-
-    if (currentColIndex < 0) {
-      return;
-    }
-
-    var toCols = getNavigableCols(toRow);
-    // Focus the first focusable element inside the <td> or <th>
-    focusCell(toCols[currentColIndex]);
-    return true;
-  }
-
-  function moveToExtreme(direction) {
-    var currentRow = getRowWithFocus();
-    if (!currentRow) {
-      return;
-    }
-    var currentCol = getColWithFocus(currentRow);
-    if (currentCol) {
-      moveToExtremeCol(direction, currentRow);
-    } else {
-      // Move to first/last row
-      moveToExtremeRow(direction);
-    }
-  }
-
-  function moveByCol(direction) {
-    var currentRow = getRowWithFocus();
-    if (!currentRow) {
-      return;
-    }
-    var cols = getNavigableCols(currentRow);
-    var numCols = cols.length;
-    var currentCol = getColWithFocus(currentRow);
-    var currentColIndex = cols.indexOf(currentCol);
-    // First right arrow moves to first column
-    var newColIndex =
-      currentCol || direction < 0 ? currentColIndex + direction : 0;
-    // Moving past beginning focuses row
-    if (doAllowRowFocus && newColIndex < 0) {
-      focus(currentRow);
-      return;
-    }
-    newColIndex = restrictIndex(newColIndex, numCols);
-    focusCell(cols[newColIndex]);
-  }
-
-  function moveToExtremeCol(direction, currentRow) {
-    // Move to first/last col
-    var cols = getNavigableCols(currentRow);
-    var desiredColIndex = direction < 0 ? 0 : cols.length - 1;
-    focusCell(cols[desiredColIndex]);
-  }
-
-  function moveToExtremeRow(direction) {
-    var rows = getAllNavigableRows();
-    var newRow = rows[direction > 0 ? rows.length - 1 : 0];
-    if (!focusSameColInDifferentRow(getRowWithFocus(), newRow)) {
-      focus(newRow);
-    }
-  }
-
-  function doPrimaryAction() {
-    var currentRow = getRowWithFocus();
-    if (!currentRow) {
-      return;
-    }
-
-    // If first col has focused, toggle expand/collapse
-    toggleExpanded(currentRow);
-  }
-
-  function toggleExpanded(row) {
-    var cols = getNavigableCols(row);
-    var currentCol = getColWithFocus(row);
-    if (currentCol === cols[0] && isExpandable(row)) {
-      changeExpanded(!isExpanded(row), row);
-    }
-  }
-
-  function changeExpanded(doExpand, row) {
-    var currentRow = row || getRowWithFocus();
-    if (!currentRow) {
-      return;
-    }
-    var currentLevel = getLevel(currentRow);
-    var rows = getAllRows();
-    var currentRowIndex = rows.indexOf(currentRow);
-    var didChange;
-    var doExpandLevel = [];
-    doExpandLevel[currentLevel + 1] = doExpand;
-
-    while (++currentRowIndex < rows.length) {
-      var nextRow = rows[currentRowIndex];
-      var rowLevel = getLevel(nextRow);
-      if (rowLevel <= currentLevel) {
-        break; // Next row is not a level down from current row
-      }
-      // Only expand the next level if this level is expanded
-      // and previous level is expanded
-      doExpandLevel[rowLevel + 1] =
-        doExpandLevel[rowLevel] && isExpanded(nextRow);
-      var willHideRow = !doExpandLevel[rowLevel];
-      var isRowHidden = nextRow.classList.contains('hidden');
-
-      if (willHideRow !== isRowHidden) {
-        if (willHideRow) {
-          nextRow.classList.add('hidden');
-        } else {
-          nextRow.classList.remove('hidden');
-        }
-        didChange = true;
-      }
-    }
-    if (didChange) {
-      setAriaExpanded(currentRow, doExpand);
-      return true;
-    }
-  }
-
-  // Mirror aria-expanded from the row to the first cell in that row
-  // (TBD is this a good idea? How else will screen reader user hear
-  // that the cell represents the opportunity to collapse/expand rows?)
-  function moveAriaExpandedToFirstCell(row) {
-    var expandedValue = row.getAttribute('aria-expanded');
-    var firstCell = getNavigableCols(row)[0];
-    if (expandedValue) {
-      firstCell.setAttribute('aria-expanded', expandedValue);
-      row.removeAttribute('aria-expanded');
-    }
-  }
-
-  function getAriaExpandedElem(row) {
-    return doAllowRowFocus ? row : getNavigableCols(row)[0];
-  }
-
-  function setAriaExpanded(row, doExpand) {
-    var elem = getAriaExpandedElem(row);
-    elem.setAttribute('aria-expanded', doExpand);
-  }
-
-  function isExpandable(row) {
-    var elem = getAriaExpandedElem(row);
-    return elem.hasAttribute('aria-expanded');
-  }
-
-  function isExpanded(row) {
-    var elem = getAriaExpandedElem(row);
-    return elem.getAttribute('aria-expanded') === 'true';
-  }
-
-  function onKeyDown(event) {
-    var ENTER = 13;
-    var UP = 38;
-    var DOWN = 40;
-    var LEFT = 37;
-    var RIGHT = 39;
-    var HOME = 36;
-    var END = 35;
-    var CTRL_HOME = -HOME;
-    var CTRL_END = -END;
-
-    var numModifiersPressed =
-      event.ctrlKey + event.altKey + event.shiftKey + event.metaKey;
-
-    var key = event.keyCode;
-
-    if (numModifiersPressed === 1 && event.ctrlKey) {
-      key = -key; // Treat as negative key value when ctrl pressed
-    } else if (numModifiersPressed) {
-      return;
-    }
-
-    switch (key) {
-      case DOWN:
-        moveByRow(1);
-        break;
-      case UP:
-        moveByRow(-1);
-        break;
-      case LEFT:
-        if (isEditableFocused()) {
-          return; // Leave key for editable area
-        }
-        if (isRowFocused()) {
-          changeExpanded(false) || moveByRow(-1, true);
-        } else {
-          moveByCol(-1);
-        }
-        break;
-      case RIGHT:
-        if (isEditableFocused()) {
-          return; // Leave key for editable area
-        }
-
-        // If row: try to expand
-        // If col or can't expand, move column to right
-        if (!isRowFocused() || !changeExpanded(true)) {
-          moveByCol(1);
-        }
-        break;
-      case CTRL_HOME:
-        moveToExtremeRow(-1);
-        break;
-      case HOME:
-        if (isEditableFocused()) {
-          return; // Leave key for editable area
-        }
-        moveToExtreme(-1);
-        break;
-      case CTRL_END:
-        moveToExtremeRow(1);
-        break;
-      case END:
-        if (isEditableFocused()) {
-          return; // Leave key for editable area
-        }
-        moveToExtreme(1);
-        break;
-      case ENTER:
-        doPrimaryAction();
-        break;
-      default:
-        return;
-    }
-
-    if (key === ENTER && (
-      (document.activeElement.tagName.toLowerCase() === 'a' && document.activeElement.hasAttribute('href')) ||
-      document.activeElement.tagName.toLowerCase() === 'button'
-    )) {
-      return;
-    }
-    // Important: don't use key for anything else, such as scrolling
-    event.preventDefault();
-  }
-
-  function onClick(iconElem) {
-    const row = iconElem.closest('tr');
-    if (!row) return;
-    if (!isExpandable(row)) return;
-
-    changeExpanded(!isExpanded(row), row);
-  }
-  // Double click on row toggles expansion
-  function onDoubleClick(event) {
-    var row = getContainingRow(event.target);
-    if (row) {
-      if (isExpandable(row)) {
-        changeExpanded(!isExpanded(row), row);
-      }
-      event.preventDefault();
-    }
-  }
-
-  initAttributes();
-  treegridElem.addEventListener('keydown', onKeyDown);
-  treegridElem.querySelectorAll('.c-treegrid__icon').forEach(icon => icon.addEventListener('click', () => onClick(icon)));
-  treegridElem.addEventListener('dblclick', e => {
-    if (!e.target.closest('.c-treegrid__icon')) {
-      onDoubleClick(e);
-    }
-  });
-
-  // Polyfill for focusin necessary for Firefox < 52
-  window.addEventListener(
-    window.onfocusin ? 'focusin' : 'focus',
-    onFocusIn,
-    true
-  );
+'use strict';
+export function TreeGrid(treegridElem, doAllowRowFocus, doStartRowFocus) {
+  function initAttributes() {
+    // Make sure focusable elements are not in the tab order
+    // They will be added back in for the active row
+    setTabIndexOfFocusableElements(treegridElem, -1);
+
+    // Add tabindex="0" to first row, "-1" to other rows
+    // We will use the roving tabindex method since aria-activedescendant
+    var rows = getAllRows();
+    var index = rows.length;
+    var startRowIndex = doStartRowFocus ? 0 : -1;
+
+    while (index--) {
+      if (doAllowRowFocus) {
+        rows[index].tabIndex = index === startRowIndex ? 0 : -1;
+      } else {
+        setTabIndexForCellsInRow(rows[index], -1);
+        moveAriaExpandedToFirstCell(rows[index]);
+      }
+    }
+    // Here we collapse all items if parent not expanded (hidden class)
+    rows.forEach(r => {
+      if (isExpandable(r)) {
+        changeExpanded(isExpanded(r), r);
+      }
+    });
+
+    if (doStartRowFocus) {
+      return;
+    }
+
+    // Start with cell focus
+    var firstCell = getNavigableCols(rows[0])[0];
+    setTabIndexForCell(firstCell);
+  }
+
+  function setTabIndexForCell(cell, tabIndex) {
+    var focusable = getFocusableElements(cell)[0] || cell;
+    focusable.tabIndex = tabIndex;
+  }
+
+  function setTabIndexForCellsInRow(row, tabIndex) {
+    var cells = getNavigableCols(row);
+    var cellIndex = cells.length;
+    while (cellIndex--) {
+      setTabIndexForCell(cells[cellIndex], tabIndex);
+    }
+  }
+
+  function getAllRows() {
+    var nodeList = treegridElem.querySelectorAll('tbody > tr');
+    return Array.prototype.slice.call(nodeList);
+  }
+
+  function getFocusableElements(root) {
+    // textarea not supported as a cell widget as it's multiple lines
+    // and needs up/down keys
+    // These should all be descendants of a cell
+    var nodeList = root.querySelectorAll('a,button,input,td>[tabindex],th>[tabindex]');
+    return Array.prototype.slice.call(nodeList);
+  }
+
+  function setTabIndexOfFocusableElements(root, tabIndex) {
+    var focusableElements = getFocusableElements(root);
+    var index = focusableElements.length;
+    while (index--) {
+      focusableElements[index].tabIndex = tabIndex;
+    }
+  }
+
+  function getAllNavigableRows() {
+    var nodeList = treegridElem.querySelectorAll(
+      'tbody > tr:not([class~="hidden"])'
+    );
+    // Convert to array so that we can use array methods on it
+    return Array.prototype.slice.call(nodeList);
+  }
+
+  function getNavigableCols(currentRow) {
+    return Array.from(currentRow.querySelectorAll('td, th'));
+  }
+
+  function restrictIndex(index, numItems) {
+    if (index < 0) {
+      return 0;
+    }
+    return index >= numItems ? index - 1 : index;
+  }
+
+  function focus(elem) {
+    elem.tabIndex = 0; // Ensure focusable
+    elem.focus();
+  }
+
+  function focusCell(cell) {
+    // Check for focusable child such as link or textbox
+    // and use that if available
+    var focusableChildren = getFocusableElements(cell);
+    focus(focusableChildren[0] || cell);
+  }
+
+  // Restore tabIndex to what it should be when focus switches from
+  // one treegrid item to another
+  function onFocusIn(event) {
+    var newTreeGridFocus =
+      event.target !== window &&
+      treegridElem.contains(event.target) &&
+      event.target;
+
+    // The last row we considered focused
+    var oldCurrentRow = enableTabbingInActiveRowDescendants.tabbingRow;
+    if (oldCurrentRow) {
+      enableTabbingInActiveRowDescendants(false, oldCurrentRow);
+    }
+    if (
+      doAllowRowFocus &&
+      onFocusIn.prevTreeGridFocus &&
+      (onFocusIn.prevTreeGridFocus.localName === 'td' || onFocusIn.prevTreeGridFocus.localName === 'th')
+    ) {
+      // Was focused on td, remove tabIndex so that it's not focused on click
+      onFocusIn.prevTreeGridFocus.removeAttribute('tabindex');
+    }
+
+    if (newTreeGridFocus) {
+      // Stayed in treegrid
+      if (oldCurrentRow) {
+        // There will be a different current row that will be
+        // the tabbable one
+        oldCurrentRow.tabIndex = -1;
+      }
+
+      // The new row
+      var currentRow = getRowWithFocus();
+      if (currentRow) {
+        currentRow.tabIndex = 0;
+        // Items within current row are also tabbable
+        enableTabbingInActiveRowDescendants(true, currentRow);
+      }
+    }
+
+    onFocusIn.prevTreeGridFocus = newTreeGridFocus;
+  }
+
+  // Set whether interactive elements within a row are tabbable
+  function enableTabbingInActiveRowDescendants(isTabbingOn, row) {
+    if (row) {
+      setTabIndexOfFocusableElements(row, isTabbingOn ? 0 : -1);
+      if (isTabbingOn) {
+        enableTabbingInActiveRowDescendants.tabbingRow = row;
+      } else {
+        if (enableTabbingInActiveRowDescendants.tabbingRow === row) {
+          enableTabbingInActiveRowDescendants.tabbingRow = null;
+        }
+      }
+    }
+  }
+
+  // The row with focus is the row that either has focus or an element
+  // inside of it has focus
+  function getRowWithFocus() {
+    return getContainingRow(document.activeElement);
+  }
+
+  function getContainingRow(start) {
+    var possibleRow = start;
+    if (treegridElem.contains(possibleRow)) {
+      while (possibleRow !== treegridElem) {
+        if (possibleRow.localName === 'tr') {
+          return possibleRow;
+        }
+        possibleRow = possibleRow.parentElement;
+      }
+    }
+  }
+
+  function isRowFocused() {
+    return getRowWithFocus() === document.activeElement;
+  }
+
+  // Note: contenteditable not currently supported
+  function isEditableFocused() {
+    var focusedElem = document.activeElement;
+    return focusedElem.localName === 'input';
+  }
+
+  function getColWithFocus(currentRow) {
+    if (currentRow) {
+      var possibleCol = document.activeElement;
+      if (currentRow.contains(possibleCol)) {
+        while (possibleCol !== currentRow) {
+          if (possibleCol.localName === 'td' || possibleCol.localName === 'th') {
+            return possibleCol;
+          }
+          possibleCol = possibleCol.parentElement;
+        }
+      }
+    }
+  }
+
+  function getLevel(row) {
+    return row && parseInt(row.getAttribute('aria-level'));
+  }
+
+  // Move backwards (direction = -1) or forwards (direction = 1)
+  // If we also need to move down/up a level, requireLevelChange = true
+  // When
+  function moveByRow(direction, requireLevelChange) {
+    var currentRow = getRowWithFocus();
+    var requiredLevel =
+      requireLevelChange && currentRow && getLevel(currentRow) + direction;
+    var rows = getAllNavigableRows();
+    var numRows = rows.length;
+    var rowIndex = currentRow ? rows.indexOf(currentRow) : -1;
+    // When moving down a level, only allow moving to next row as the
+    // first child will never be farther than that
+    var maxDistance = requireLevelChange && direction === 1 ? 1 : NaN;
+
+    // Move in direction until required level is found
+    do {
+      if (maxDistance-- === 0) {
+        return; // Failed to find required level, return without focus change
+      }
+      rowIndex = restrictIndex(rowIndex + direction, numRows);
+    } while (requiredLevel && requiredLevel !== getLevel(rows[rowIndex]));
+
+    if (!focusSameColInDifferentRow(currentRow, rows[rowIndex])) {
+      focus(rows[rowIndex]);
+    }
+  }
+
+  function focusSameColInDifferentRow(fromRow, toRow) {
+    var currentCol = getColWithFocus(fromRow);
+    if (!currentCol) {
+      return;
+    }
+
+    var fromCols = getNavigableCols(fromRow);
+    var currentColIndex = fromCols.indexOf(currentCol);
+
+    if (currentColIndex < 0) {
+      return;
+    }
+
+    var toCols = getNavigableCols(toRow);
+    // Focus the first focusable element inside the <td> or <th>
+    focusCell(toCols[currentColIndex]);
+    return true;
+  }
+
+  function moveToExtreme(direction) {
+    var currentRow = getRowWithFocus();
+    if (!currentRow) {
+      return;
+    }
+    var currentCol = getColWithFocus(currentRow);
+    if (currentCol) {
+      moveToExtremeCol(direction, currentRow);
+    } else {
+      // Move to first/last row
+      moveToExtremeRow(direction);
+    }
+  }
+
+  function moveByCol(direction) {
+    var currentRow = getRowWithFocus();
+    if (!currentRow) {
+      return;
+    }
+    var cols = getNavigableCols(currentRow);
+    var numCols = cols.length;
+    var currentCol = getColWithFocus(currentRow);
+    var currentColIndex = cols.indexOf(currentCol);
+    // First right arrow moves to first column
+    var newColIndex =
+      currentCol || direction < 0 ? currentColIndex + direction : 0;
+    // Moving past beginning focuses row
+    if (doAllowRowFocus && newColIndex < 0) {
+      focus(currentRow);
+      return;
+    }
+    newColIndex = restrictIndex(newColIndex, numCols);
+    focusCell(cols[newColIndex]);
+  }
+
+  function moveToExtremeCol(direction, currentRow) {
+    // Move to first/last col
+    var cols = getNavigableCols(currentRow);
+    var desiredColIndex = direction < 0 ? 0 : cols.length - 1;
+    focusCell(cols[desiredColIndex]);
+  }
+
+  function moveToExtremeRow(direction) {
+    var rows = getAllNavigableRows();
+    var newRow = rows[direction > 0 ? rows.length - 1 : 0];
+    if (!focusSameColInDifferentRow(getRowWithFocus(), newRow)) {
+      focus(newRow);
+    }
+  }
+
+  function doPrimaryAction() {
+    var currentRow = getRowWithFocus();
+    if (!currentRow) {
+      return;
+    }
+
+    // If first col has focused, toggle expand/collapse
+    toggleExpanded(currentRow);
+  }
+
+  function toggleExpanded(row) {
+    var cols = getNavigableCols(row);
+    var currentCol = getColWithFocus(row);
+    if (currentCol === cols[0] && isExpandable(row)) {
+      changeExpanded(!isExpanded(row), row);
+    }
+  }
+
+  function changeExpanded(doExpand, row) {
+    var currentRow = row || getRowWithFocus();
+    if (!currentRow) {
+      return;
+    }
+    var currentLevel = getLevel(currentRow);
+    var rows = getAllRows();
+    var currentRowIndex = rows.indexOf(currentRow);
+    var didChange;
+    var doExpandLevel = [];
+    doExpandLevel[currentLevel + 1] = doExpand;
+
+    while (++currentRowIndex < rows.length) {
+      var nextRow = rows[currentRowIndex];
+      var rowLevel = getLevel(nextRow);
+      if (rowLevel <= currentLevel) {
+        break; // Next row is not a level down from current row
+      }
+      // Only expand the next level if this level is expanded
+      // and previous level is expanded
+      doExpandLevel[rowLevel + 1] =
+        doExpandLevel[rowLevel] && isExpanded(nextRow);
+      var willHideRow = !doExpandLevel[rowLevel];
+      var isRowHidden = nextRow.classList.contains('hidden');
+
+      if (willHideRow !== isRowHidden) {
+        if (willHideRow) {
+          nextRow.classList.add('hidden');
+        } else {
+          nextRow.classList.remove('hidden');
+        }
+        didChange = true;
+      }
+    }
+    if (didChange) {
+      setAriaExpanded(currentRow, doExpand);
+      return true;
+    }
+  }
+
+  // Mirror aria-expanded from the row to the first cell in that row
+  // (TBD is this a good idea? How else will screen reader user hear
+  // that the cell represents the opportunity to collapse/expand rows?)
+  function moveAriaExpandedToFirstCell(row) {
+    var expandedValue = row.getAttribute('aria-expanded');
+    var firstCell = getNavigableCols(row)[0];
+    if (expandedValue) {
+      firstCell.setAttribute('aria-expanded', expandedValue);
+      row.removeAttribute('aria-expanded');
+    }
+  }
+
+  function getAriaExpandedElem(row) {
+    return doAllowRowFocus ? row : getNavigableCols(row)[0];
+  }
+
+  function setAriaExpanded(row, doExpand) {
+    var elem = getAriaExpandedElem(row);
+    elem.setAttribute('aria-expanded', doExpand);
+  }
+
+  function isExpandable(row) {
+    var elem = getAriaExpandedElem(row);
+    return elem.hasAttribute('aria-expanded');
+  }
+
+  function isExpanded(row) {
+    var elem = getAriaExpandedElem(row);
+    return elem.getAttribute('aria-expanded') === 'true';
+  }
+
+  function onKeyDown(event) {
+    var ENTER = 13;
+    var UP = 38;
+    var DOWN = 40;
+    var LEFT = 37;
+    var RIGHT = 39;
+    var HOME = 36;
+    var END = 35;
+    var CTRL_HOME = -HOME;
+    var CTRL_END = -END;
+
+    var numModifiersPressed =
+      event.ctrlKey + event.altKey + event.shiftKey + event.metaKey;
+
+    var key = event.keyCode;
+
+    if (numModifiersPressed === 1 && event.ctrlKey) {
+      key = -key; // Treat as negative key value when ctrl pressed
+    } else if (numModifiersPressed) {
+      return;
+    }
+
+    switch (key) {
+      case DOWN:
+        moveByRow(1);
+        break;
+      case UP:
+        moveByRow(-1);
+        break;
+      case LEFT:
+        if (isEditableFocused()) {
+          return; // Leave key for editable area
+        }
+        if (isRowFocused()) {
+          changeExpanded(false) || moveByRow(-1, true);
+        } else {
+          moveByCol(-1);
+        }
+        break;
+      case RIGHT:
+        if (isEditableFocused()) {
+          return; // Leave key for editable area
+        }
+
+        // If row: try to expand
+        // If col or can't expand, move column to right
+        if (!isRowFocused() || !changeExpanded(true)) {
+          moveByCol(1);
+        }
+        break;
+      case CTRL_HOME:
+        moveToExtremeRow(-1);
+        break;
+      case HOME:
+        if (isEditableFocused()) {
+          return; // Leave key for editable area
+        }
+        moveToExtreme(-1);
+        break;
+      case CTRL_END:
+        moveToExtremeRow(1);
+        break;
+      case END:
+        if (isEditableFocused()) {
+          return; // Leave key for editable area
+        }
+        moveToExtreme(1);
+        break;
+      case ENTER:
+        doPrimaryAction();
+        break;
+      default:
+        return;
+    }
+
+    if (key === ENTER && (
+      (document.activeElement.tagName.toLowerCase() === 'a' && document.activeElement.hasAttribute('href')) ||
+      document.activeElement.tagName.toLowerCase() === 'button'
+    )) {
+      return;
+    }
+    // Important: don't use key for anything else, such as scrolling
+    event.preventDefault();
+  }
+
+  function onClick(iconElem) {
+    const row = iconElem.closest('tr');
+    if (!row) return;
+    if (!isExpandable(row)) return;
+
+    changeExpanded(!isExpanded(row), row);
+  }
+  // Double click on row toggles expansion
+  function onDoubleClick(event) {
+    var row = getContainingRow(event.target);
+    if (row) {
+      if (isExpandable(row)) {
+        changeExpanded(!isExpanded(row), row);
+      }
+      event.preventDefault();
+    }
+  }
+
+  initAttributes();
+  treegridElem.addEventListener('keydown', onKeyDown);
+  treegridElem.querySelectorAll('.c-treegrid__icon').forEach(icon => icon.addEventListener('click', () => onClick(icon)));
+  treegridElem.addEventListener('dblclick', e => {
+    if (!e.target.closest('.c-treegrid__icon')) {
+      onDoubleClick(e);
+    }
+  });
+
+  // Polyfill for focusin necessary for Firefox < 52
+  window.addEventListener(
+    window.onfocusin ? 'focusin' : 'focus',
+    onFocusIn,
+    true
+  );
 }