Visual editor: Handling wlxml classes in transformations

[fnpeditor.git] / modules / visualEditor.transformations.js

diff --git a/modules/visualEditor.transformations.js b/modules/visualEditor.transformations.js
index cfb5877..49f9aa3 100644 (file)
--- a/modules/visualEditor.transformations.js
+++ b/modules/visualEditor.transformations.js
@@ -21,15 +21,24 @@ if(typeof module !== 'undefined' && module.exports) {
                 console.log('running ' + tagName);
                 toret.find(tagName).replaceWith(function() {
                     var suffix = tagName !== 'div'  ? tagName : 'block';
-                    return $('<div></div>').addClass('rng rng-' + suffix).append($(this).contents());
+                    var toret = $('<div></div>').attr('wlxml-tag', suffix);
+                    var currentTag = $(this);
+                    if(currentTag.attr('class'))
+                        toret.attr('wlxml-class', currentTag.attr('class'));
+                    toret.append(currentTag.contents());
+                    return toret;
                 });
             });
             
             toInline.forEach(function(tagName) {
                 tagName = tagName.toLowerCase();
                 toret.find(tagName).replaceWith(function() {
-                    var node = this;
-                    return $('<span></span>').addClass('rng rng-' + tagName).append($(this).contents());
+                    var currentTag = $(this);
+                    var toret = $('<span></span>').attr('wlxml-tag', tagName);
+                    if(currentTag.attr('class'))
+                        toret.attr('wlxml-class', currentTag.attr('class'));
+                    toret.append(currentTag.contents());
+                    return toret;
                 });
             });
             return toret.children();
@@ -59,7 +68,7 @@ if(typeof module !== 'undefined' && module.exports) {
             
             toret.find('div, span').replaceWith(function() {
                 var div = $(this);
-                var tagName = div.attr('class').split('rng-')[1];
+                var tagName = div.attr('wlxml-tag');
                 return $('<'+tagName+'>').append(div.contents());
             });