Merge pull request #203 from joeybaker/patch-1

EditToolbar now sets it's state correctly
Leaflet · Oct 13, 2013 · 5ac5672 · 5ac5672
2 parents d7c4103 + 2923321
commit 5ac5672
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/edit/EditToolbar.js b/src/edit/EditToolbar.js
@@ -123,13 +123,14 @@ L.EditToolbar = L.Toolbar.extend({
 
 	_checkDisabled: function () {
 		var featureGroup = this.options.featureGroup,
-			hasLayers = featureGroup.getLayers().length === 0,
+			hasLayers = featureGroup.getLayers().length !== 0,
 			button;
 
 		if (this.options.edit) {
 			button = this._modes[L.EditToolbar.Edit.TYPE].button;
 
-			L.DomUtil.toggleClass(button, 'leaflet-disabled');
+			if (hasLayers) { L.DomUtil.removeClass(button, 'leaflet-disabled'); }
+			else { L.DomUtil.addClass(button, 'leaflet-disabled'); }
 
 			button.setAttribute(
 				'title',
@@ -142,7 +143,8 @@ L.EditToolbar = L.Toolbar.extend({
 		if (this.options.remove) {
 			button = this._modes[L.EditToolbar.Delete.TYPE].button;
 
-			L.DomUtil.toggleClass(button, 'leaflet-disabled');
+			if (hasLayers) { L.DomUtil.removeClass(button, 'leaflet-disabled'); }
+			else { L.DomUtil.addClass(button, 'leaflet-disabled'); }
 
 			button.setAttribute(
 				'title',
@@ -164,4 +166,4 @@ if (!L.DomUtil.toggleClass) {
 			}
 		}
 	});
-}
+}