diff --git a/src/attributes.js b/src/attributes.js index 93e04336..c4c75009 100644 --- a/src/attributes.js +++ b/src/attributes.js @@ -31,8 +31,64 @@ jQuery.fn.extend({ return this; }, + addClass: function( value ) { + if ( value && typeof value === "string" ) { + var classNames = (value || "").split(/\s+/); + + for ( var i = 0, l = this.length; i < l; i++ ) { + var elem = this[i]; + + if ( elem.nodeType === 1 ) { + if ( !elem.className ) { + elem.className = value; + } else { + var className = " " + elem.className + " "; + for ( var c = 0, cl = classNames.length; c < cl; c++ ) { + if ( className.indexOf( " " + classNames[c] + " " ) < 0 ) { + elem.className += " " + classNames[c]; + } + } + } + } + } + } + + return this; + }, + + removeClass: function( value ) { + if ( (value && typeof value === "string") || value === undefined ) { + var classNames = (value || "").split(/\s+/); + + for ( var i = 0, l = this.length; i < l; i++ ) { + var elem = this[i]; + + if ( elem.nodeType === 1 && elem.className ) { + if ( value ) { + var className = " " + elem.className + " "; + for ( var c = 0, cl = classNames.length; c < cl; c++ ) { + className = className.replace(" " + classNames[c] + " ", " "); + } + elem.className = className.substring(1, className.length - 1); + } else { + elem.className = ""; + } + } + } + } + + return this; + }, + hasClass: function( selector ) { - return !!selector && this.is( "." + selector ); + var className = " " + selector + " "; + for ( var i = 0, l = this.length; i < l; i++ ) { + if ( (" " + this[i].className + " ").indexOf( className ) > -1 ) { + return true; + } + } + + return false; }, val: function( value ) { @@ -125,14 +181,6 @@ jQuery.each({ this.removeAttribute( name ); }, - addClass: function( classNames ) { - jQuery.className.add( this, classNames ); - }, - - removeClass: function( classNames ) { - jQuery.className.remove( this, classNames ); - }, - toggleClass: function( classNames, state ) { var type = typeof classNames; if ( type === "string" ) { @@ -141,8 +189,8 @@ jQuery.each({ classNames = classNames.split( /\s+/ ); while ( (className = classNames[ i++ ]) ) { // check each className given, space seperated list - state = isBool ? state : !jQuery.className.has( this, className ); - jQuery.className[ state ? "add" : "remove" ]( this, className ); + state = isBool ? state : !jQuery(this).hasClass( className ); + jQuery(this)[ state ? "addClass" : "removeClass" ]( className ); } } else if ( type === "undefined" || type === "boolean" ) { if ( this.className ) { @@ -160,31 +208,6 @@ jQuery.each({ }); jQuery.extend({ - className: { - // internal only, use addClass("class") - add: function( elem, classNames ) { - jQuery.each((classNames || "").split(/\s+/), function(i, className){ - if ( elem.nodeType == 1 && !jQuery.className.has( elem.className, className ) ) - elem.className += (elem.className ? " " : "") + className; - }); - }, - - // internal only, use removeClass("class") - remove: function( elem, classNames ) { - if (elem.nodeType == 1) - elem.className = classNames !== undefined ? - jQuery.grep(elem.className.split(/\s+/), function(className){ - return !jQuery.className.has( classNames, className ); - }).join(" ") : - ""; - }, - - // internal only, use hasClass("class") - has: function( elem, className ) { - return elem && jQuery.inArray( className, (elem.className || elem).toString().split(/\s+/) ) > -1; - } - }, - attr: function( elem, name, value ) { // don't set attributes on text and comment nodes if (!elem || elem.nodeType == 3 || elem.nodeType == 8) diff --git a/test/unit/attributes.js b/test/unit/attributes.js index 7887790b..e9880df2 100644 --- a/test/unit/attributes.js +++ b/test/unit/attributes.js @@ -272,6 +272,7 @@ test("removeClass(String) - simple", function() { ok( !$divs.is('.test'), "Remove Class" ); reset(); + $divs = jQuery('div'); $divs.addClass("test").addClass("foo").addClass("bar"); $divs.removeClass("test").removeClass("bar").removeClass("foo"); @@ -279,6 +280,7 @@ test("removeClass(String) - simple", function() { ok( !$divs.is('.test,.bar,.foo'), "Remove multiple classes" ); reset(); + $divs = jQuery('div'); // Make sure that a null value doesn't cause problems $divs.eq(0).addClass("test").removeClass(null); @@ -350,19 +352,24 @@ test("removeAttr(String", function() { equals( jQuery('#mark').removeAttr("class")[0].className, "", "remove class" ); }); -test("jQuery.className", function() { +test("addClass, removeClass, hasClass", function() { expect(6); - var x = jQuery("

Hi

")[0]; - var c = jQuery.className; - c.add(x, "hi"); + + var jq = jQuery("

Hi

"), x = jq[0]; + + jq.addClass("hi"); equals( x.className, "hi", "Check single added class" ); - c.add(x, "foo bar"); + + jq.addClass("foo bar"); equals( x.className, "hi foo bar", "Check more added classes" ); - c.remove(x); + + jq.removeClass(); equals( x.className, "", "Remove all classes" ); - c.add(x, "hi foo bar"); - c.remove(x, "foo"); + + jq.addClass("hi foo bar"); + jq.removeClass("foo"); equals( x.className, "hi bar", "Check removal of one class" ); - ok( c.has(x, "hi"), "Check has1" ); - ok( c.has(x, "bar"), "Check has2" ); + + ok( jq.hasClass("hi"), "Check has1" ); + ok( jq.hasClass("bar"), "Check has2" ); });