Add selectionMatchFunction

addon/-private/collapse-tree.js

@@ -44,16 +44,25 @@ export const TableRowMeta = EmberObject.extend({
     },
   }),
 
-  isSelected: computed('_tree.selection.[]', '_parentMeta.isSelected', function() {
-    let rowValue = get(this, '_rowValue');
-    let selection = get(this, '_tree.selection');
+  // eslint-disable-next-line ember/use-brace-expansion
+  isSelected: computed(
+    '_tree.{selection.[],selectionMatchFunction}',
+    '_parentMeta.isSelected',
+    function() {
+      let rowValue = get(this, '_rowValue');
+      let selection = get(this, '_tree.selection');
+      let selectionMatchFunction = get(this, '_tree.selectionMatchFunction');
 
-    if (isArray(selection)) {
-      return this.get('isGroupSelected');
-    }
+      if (isArray(selection)) {
+        return this.get('isGroupSelected');
+      }
 
-    return selection === rowValue || get(this, '_parentMeta.isSelected');
-  }),
+      if (selectionMatchFunction) {
+        return selectionMatchFunction(selection, rowValue);
+      }
+      return selection === rowValue || get(this, '_parentMeta.isSelected');
+    }
+  ),
 
   isGroupSelected: computed('_tree.selection.[]', '_parentMeta.isSelected', function() {
     let rowValue = get(this, '_rowValue');

addon/components/ember-tbody/component.js

@@ -94,6 +94,14 @@ export default Component.extend({
   */
   selection: null,
 
+  /**
+    A function that will override how selection is compared to row value.
+
+    @argument selectionMatchFunction
+    @type function?
+  */
+  selectionMatchFunction: null,
+
   /**
     An action that is called when the row selection of the table changes.
     Will be called with either an array or individual row, depending on the
@@ -282,6 +290,7 @@ export default Component.extend({
     'enableCollapse',
     'enableTree',
     'selection',
+    'selectionMatchFunction',
     'selectingChildrenSelectsParent',
     'onSelect',
 
@@ -294,6 +303,7 @@ export default Component.extend({
       this.collapseTree.set('enableCollapse', this.get('enableCollapse'));
       this.collapseTree.set('enableTree', this.get('enableTree'));
       this.collapseTree.set('selection', this.get('selection'));
+      this.collapseTree.set('selectionMatchFunction', this.get('selectionMatchFunction'));
       this.collapseTree.set(
         'selectingChildrenSelectsParent',
         this.get('selectingChildrenSelectsParent')