Merge pull request #797 from wqoq/fix-edit-classname

Add/remove className options when PolyVerticesEdit is enabled/disabled
Leaflet · Oct 20, 2017 · 7659f8d · 7659f8d
2 parents febb0e0 + 35c6b57
commit 7659f8d
Showing 1 changed file with 28 additions and 0 deletions.
diff --git a/src/edit/handler/Edit.Poly.js b/src/edit/handler/Edit.Poly.js
@@ -127,6 +127,7 @@ L.Edit.PolyVerticesEdit = L.Handler.extend({
 	// Add listener hooks to this handler.
 	addHooks: function () {
 		var poly = this._poly;
+		var path = poly._path;
 
 		if (!(poly instanceof L.Polygon)) {
 			poly.options.fill = false;
@@ -135,6 +136,19 @@ L.Edit.PolyVerticesEdit = L.Handler.extend({
 			}
 		}
 
+		if (path) {
+			if (poly.options.editing.className) {
+				if (poly.options.original.className) {
+					poly.options.original.className.split(' ').forEach(function(className) {
+						L.DomUtil.removeClass(path, className);
+					});
+				}
+				poly.options.editing.className.split(' ').forEach(function(className) {
+					L.DomUtil.addClass(path, className);
+				});
+			}
+		}
+
 		poly.setStyle(poly.options.editing);
 
 		if (this._poly._map) {
@@ -152,6 +166,20 @@ L.Edit.PolyVerticesEdit = L.Handler.extend({
 	// Remove listener hooks from this handler.
 	removeHooks: function () {
 		var poly = this._poly;
+		var path = poly._path;
+
+		if (path) {
+			if (poly.options.editing.className) {
+				poly.options.editing.className.split(' ').forEach(function(className) {
+					L.DomUtil.removeClass(path, className);
+				});
+				if (poly.options.original.className) {
+					poly.options.original.className.split(' ').forEach(function(className) {
+						L.DomUtil.addClass(path, className);
+					});
+				}
+			}
+		}
 
 		poly.setStyle(poly.options.original);