Fixed typo from commit 542099a278e79dce38e814e7e7b448a1b73df82f.

[jquery.git] / src / attributes.js

diff --git a/src/attributes.js b/src/attributes.js
index 4405b43..0a60b0d 100644 (file)
--- a/src/attributes.js
+++ b/src/attributes.js
@@ -4,6 +4,12 @@ jQuery.fn.extend({
        },
 
        addClass: function( value ) {
+               if(jQuery.isFunction(value)) {
+                       return this.each(function() {
+                               jQuery(this).addClass( value.call(this) );
+                       });
+               }
+
                if ( value && typeof value === "string" ) {
                        var classNames = (value || "").split(/\s+/);
 
@@ -29,6 +35,12 @@ jQuery.fn.extend({
        },
 
        removeClass: function( value ) {
+               if(jQuery.isFunction(value)) {
+                       return this.each(function() {
+                               jQuery(this).removeClass( value.call(this) );
+                       });
+               }
+
                if ( (value && typeof value === "string") || value === undefined ) {
                        var classNames = (value || "").split(/\s+/);
 
@@ -113,7 +125,7 @@ jQuery.fn.extend({
                // Typecast once if the value is a number
                if ( typeof value === "number" ) {
                        value += '';
-               }       
+               }
                var val = value;
 
                return this.each(function(){
@@ -122,10 +134,10 @@ jQuery.fn.extend({
                                // Typecast each time if the value is a Function and the appended
                                // value is therefore different each time.
                                if( typeof val === "number" ) {
-                                       val += ''; 
+                                       val += '';
                                }
                        }
-                       
+
                        if ( this.nodeType != 1 ) {
                                return;
                        }
@@ -151,8 +163,8 @@ jQuery.fn.extend({
 
 jQuery.each({
        removeAttr: function( name ) {
-               jQuery.attr( this, name, "" );
-               if (this.nodeType == 1) {
+               if ( this.nodeType === 1 ) {
+                       this[ jQuery.isXMLDoc( this ) ? name : jQuery.props[ name ] || name ] = null;
                        this.removeAttribute( name );
                }
        },
@@ -178,21 +190,41 @@ jQuery.each({
                }
        }
 }, function(name, fn){
-       jQuery.fn[ name ] = function(){
+       jQuery.fn[ name ] = function(val, state){
+               if( jQuery.isFunction( val ) ) {
+                       return this.each(function() { jQuery(this)[ name ]( val.call(this), state ); });
+               }
+
                return this.each( fn, arguments );
        };
 });
 
 jQuery.extend({
+       attrFn: {
+               val: true,
+               addClass: true,
+               css: true,
+               html: true,
+               text: true,
+               append: true,
+               prepend: true,
+               data: true,
+               width: true,
+               height: true,
+               offset: true,
+               bind: true
+       },
+               
        attr: function( elem, name, value ) {
                // don't set attributes on text and comment nodes
                if (!elem || elem.nodeType == 3 || elem.nodeType == 8) {
                        return undefined;
                }
-               if ( name in jQuery.fn && name !== "attr" ) {
+
+               if ( name in jQuery.attrFn ) {
                        return jQuery(elem)[name](value);
                }
-               
+
                var notxml = elem.nodeType !== 1 || !jQuery.isXMLDoc( elem ),
                        // Whether we are setting (or getting)
                        set = value !== undefined;