Widget: Process "classes" option during _setOption()

Author: Gabriel Schulhof

ui/widget.js

@@ -357,6 +357,7 @@ $.Widget.prototype = {
 
 		return this;
 	},
+
 	_setOptions: function( options ) {
 		var key;
 
@@ -366,7 +367,26 @@ $.Widget.prototype = {
 
 		return this;
 	},
+
+	_elementsFromClassKey: function( classKey ) {
+		if ( this.options.classes[ classKey ] ) {
+			return this.element;
+		} else {
+			return $();
+		}
+	},
+
 	_setOption: function( key, value ) {
+		if ( key === "classes" ) {
+			for ( var classKey in value ) {
+				if ( value[ classKey ] !== this.options.classes[ classKey ] ) {
+					this._elementsFromClassKey( classKey )
+						.removeClass( this._classesFromObject( classKey, this.options.classes ) )
+						.addClass( this._classesFromObject( classKey, value ) );
+				}
+			}
+		}
+
 		this.options[ key ] = value;
 
 		if ( key === "disabled" ) {
@@ -383,19 +403,23 @@ $.Widget.prototype = {
 		return this;
 	},
 
-	_classes: function( key ) {
+	_classesFromObject: function( key, classes ) {
 		var out = [],
 			parts = key.split( " " ),
 			i = parts.length;
 
 		while ( i-- ) {
 			out.push( parts[ i ] );
-			out.push( this.options.classes[ parts[ i ] ] );
+			out.push( classes[ parts[ i ] ] );
 		}
 
 		return out.join( " " );
 	},
 
+	_classes: function( key ) {
+		return this._classesFromObject( key, this.options.classes );
+	},
+
 	enable: function() {
 		return this._setOptions({ disabled: false });
 	},