fix(jqLite) - toggleClass & removeClass reflect documentation

src/jqLite.js

@@ -381,15 +381,26 @@ function jqLiteHasClass(element, selector) {
       indexOf(" " + selector + " ") > -1);
 }
 
-function jqLiteRemoveClass(element, cssClasses) {
-  if (cssClasses && element.setAttribute) {
-    forEach(cssClasses.split(' '), function(cssClass) {
-      element.setAttribute('class', trim(
-          (" " + (element.getAttribute('class') || '') + " ")
-          .replace(/[\n\t]/g, " ")
-          .replace(" " + trim(cssClass) + " ", " "))
-      );
-    });
+function jqLiteRemoveClass(element, cssClassesOrFunc) {
+  if (element.setAttribute) {
+    if (cssClassesOrFunc) {
+      if (typeof cssClassesOrFunc === 'function') {
+        forEach(element.className.split(/\s+/), function(className, index) {
+          cssClassesOrFunc(index, className);
+        });
+      } else {
+        forEach(cssClassesOrFunc.split(' '), function(cssClass) {
+          element.setAttribute('class', trim(
+              (" " + (element.getAttribute('class') || '') + " ")
+              .replace(/[\n\t]/g, " ")
+              .replace(" " + trim(cssClass) + " ", " "))
+          );
+        });
+      }
+    } else {
+      // if no class names are specified in the parameter, all classes will be removed
+      element.setAttribute('class', '');
+    }
   }
 }
 
@@ -935,14 +946,35 @@ forEach({
   removeClass: jqLiteRemoveClass,
 
   toggleClass: function(element, selector, condition) {
+
     if (selector) {
-      forEach(selector.split(' '), function(className) {
-        var classCondition = condition;
-        if (isUndefined(classCondition)) {
-          classCondition = !jqLiteHasClass(element, className);
-        }
-        (classCondition ? jqLiteAddClass : jqLiteRemoveClass)(element, className);
-      });
+
+      // the class name to be toggled can be determined by passing in a function.
+      if (typeof selector === 'function') {
+
+        forEach(element.className.split(/\s+/), function(className, index) {
+          if (Array.isArray(condition) && condition.length >= index) {
+            selector(index, className, condition[index]);
+          } else {
+            selector(index, className);
+          }
+        });
+
+      // if has the class, then it is removed; if does not have the class, then it is added.
+      } else {
+
+        forEach(selector.split(' '), function(className) {
+          var classCondition = condition;
+          if (isUndefined(classCondition)) {
+            classCondition = !jqLiteHasClass(element, className);
+          }
+          (classCondition ? jqLiteAddClass : jqLiteRemoveClass)(element, className);
+        });
+      }
+
+    // if no arguments are passed all class names on the element will be toggled.
+    } else {
+      jqLiteRemoveClass(element);
     }
   },
 

test/jqLiteSpec.js

@@ -839,6 +839,65 @@ describe('jqLite', function() {
         expect(jqLite(b).hasClass('cde')).toBe(false);
       });
 
+      it('should allow function to be passed with conditions', function() {
+        var selector = jqLite([a, b]);
+        selector.addClass('abc');
+        selector.addClass('cde');
+        selector.addClass('fgh');
+        selector.addClass('ijk');
+
+        expect(selector.toggleClass(function(index, className, state) {
+          selector.toggleClass(className, state);
+        }, [true, true, false, false])).toBe(selector);
+
+        expect(jqLite(a).hasClass('abc')).toBe(true);
+        expect(jqLite(a).hasClass('cde')).toBe(true);
+        expect(jqLite(a).hasClass('fgh')).toBe(false);
+        expect(jqLite(a).hasClass('ijk')).toBe(false);
+
+        expect(jqLite(b).hasClass('abc')).toBe(true);
+        expect(jqLite(b).hasClass('cde')).toBe(true);
+        expect(jqLite(b).hasClass('fgh')).toBe(false);
+        expect(jqLite(b).hasClass('ijk')).toBe(false);
+      });
+
+      it('should allow function to be passed without condition', function() {
+        var selector = jqLite([a, b]);
+        selector.addClass('abc');
+        selector.addClass('cde');
+        selector.addClass('fgh');
+        selector.addClass('ijk');
+
+        expect(selector.toggleClass(function(index, className) {
+          if (className === 'abc' || className === 'cde') {
+            selector.removeClass(className);
+          }
+        })).toBe(selector);
+
+        expect(jqLite(a).hasClass('abc')).toBe(false);
+        expect(jqLite(a).hasClass('cde')).toBe(false);
+        expect(jqLite(a).hasClass('fgh')).toBe(true);
+        expect(jqLite(a).hasClass('ijk')).toBe(true);
+
+        expect(jqLite(b).hasClass('abc')).toBe(false);
+        expect(jqLite(b).hasClass('cde')).toBe(false);
+        expect(jqLite(b).hasClass('fgh')).toBe(true);
+        expect(jqLite(b).hasClass('ijk')).toBe(true);
+      });
+
+      it('should toggle all classes if no selector is passed', function() {
+        var selector = jqLite([a, b]);
+        selector.addClass('abc');
+        selector.addClass('cde');
+
+        expect(selector.toggleClass()).toBe(selector);
+
+        expect(jqLite(a).hasClass('abc')).toBe(false);
+        expect(jqLite(a).hasClass('cde')).toBe(false);
+        expect(jqLite(b).hasClass('abc')).toBe(false);
+        expect(jqLite(b).hasClass('cde')).toBe(false);
+      });
+
       it('should not break for null / undefined selectors', function() {
         var selector = jqLite([a, b]);
         expect(selector.toggleClass(null)).toBe(selector);
@@ -869,6 +928,32 @@ describe('jqLite', function() {
       });
 
 
+      it('should remove all classes if no class is specified', function() {
+        var element = jqLite('<div class="foo bar baz"></div>');
+
+        expect(element.removeClass()).toEqual(element);
+
+        expect(element.hasClass('foo')).toBe(false);
+        expect(element.hasClass('bar')).toBe(false);
+        expect(element.hasClass('baz')).toBe(false);
+      });
+
+
+      it('should allow a function as a paramater', function() {
+        var element = jqLite('<div class="foo bar baz"></div>');
+
+        expect(element.removeClass(function(index, className) {
+          if (className === 'foo' || className === 'bar') {
+            element.removeClass(className);
+          }
+        })).toEqual(element);
+
+        expect(element.hasClass('foo')).toBe(false);
+        expect(element.hasClass('bar')).toBe(false);
+        expect(element.hasClass('baz')).toBe(true);
+      });
+
+
       it('should remove multiple classes specified as one string', function() {
         var jqA = jqLite(a);