AngularJS Select All directive

select-all.js

/**
 * Directive to instantly enable/disable multiple checkboxes based on a master checkbox.
 * Changing slave checkboxes will update the master checkbox accordingly, including the indeterminate state.
 *
 * Usage example:
 *
 * <label><input type="checkbox" select-all="theProperties"> Select all</label>
 * <div ng-repeat="property in properties">
 *   <label><input type="checkbox" rel="theProperties" ng-model="property.checked"> {{ property.label }}</label>
 * </div>
 *
 * Note that the 'rel' attribute on slave checkboxes should match the 'select-all' attribute.
 */
.directive('selectAll', function($timeout, $parse) {
  return {
    restrict: 'A',
    link: function (scope, masterElement, attrs) {
      var slaveName = attrs.selectAll;
      var slaveSelector = ':checkbox[rel="' + slaveName + '"]';

      masterElement.on('click', function() {
        angular.element(slaveSelector).each(function(i, elem) {
          var localScope = angular.element(elem).scope();
          var model = $parse(angular.element(elem).attr('ng-model'));
          model.assign(localScope, masterElement.prop('checked'));
          localScope.$apply();
        });
      });

      $timeout(function() {
        var slaveElements = angular.element(slaveSelector);
        var setMasterState = function() {
          var checkedSlaves = slaveElements.filter(function(i, elem) {
            return angular.element(elem).prop('checked');
          });
          var isChecked = (checkedSlaves.length === slaveElements.length);
          var isIndeterminate = (checkedSlaves.length > 0 && checkedSlaves.length < slaveElements.length);
          masterElement.prop('checked', isChecked);
          masterElement.prop('indeterminate', isIndeterminate);
        };
        setMasterState();
        slaveElements.on('click', setMasterState);
      });
    }
  };
})

Awesome work!
Unfortunately there's an issue when using checkboxes that are wrapped in another directive.
The indeterminate feature is not working, and toggling the checkboxes doesn't update the select-all checkbox state.
See a demo here: https://plnkr.co/edit/AekVeiDndoOLE1LLCXoz?p=preview
Setting the $timeout to 300 or more seems to take care of the problem, but changing priority for the directives doesn't.

Any fix?

is there something for angular2?