AngularJS directive ngClass-like with callbacks