X-Git-Url: https://git.mdrn.pl/fnpeditor.git/blobdiff_plain/6435513192b5be363719dcb1559c93b52944800d..02a91f42fa120c3f1d5f08100cf0f85bc571b5d1:/modules/documentCanvas/canvas/documentElement.js?ds=sidebyside

diff --git a/modules/documentCanvas/canvas/documentElement.js b/modules/documentCanvas/canvas/documentElement.js
index 50e680d..5e3983e 100644
--- a/modules/documentCanvas/canvas/documentElement.js
+++ b/modules/documentCanvas/canvas/documentElement.js
@@ -22,16 +22,27 @@ $.extend(DocumentElement.prototype, {
         var elementContent = this.$element.contents();
         elementContent.each(function(idx) {
             var element = documentElementFromHTMLElement(this);
-            if(
-                (toret.length === 0 && (element instanceof DocumentNodeElement)) ||
-                (toret.length > 0 && (toret[toret.length -1] instanceof DocumentNodeElement) && (element instanceof DocumentNodeElement))
-            )
-                toret.push(documentElementFromHTMLElement(document.createTextNode()));
+            if(idx === 0 && elementContent.length > 1 && elementContent[1].nodeType === Node.ELEMENT_NODE && (element instanceof DocumentTextElement) && $.trim($(this).text()) === '')
+                return true;
+            if(idx > 0 && element instanceof DocumentTextElement) {
+                if(toret[toret.length-1] instanceof DocumentNodeElement && $.trim($(this).text()) === '')
+                    return true;
+            }
             toret.push(element);
-            if((idx === elementContent.length - 1) && (element instanceof DocumentNodeElement))
-                toret.push(documentElementFromHTMLElement(document.createTextNode()));
         });
         return toret;
+    },
+    parent: function() {
+        return documentElementFromHTMLElement(this.$element.parent()[0]);
+    },
+
+    sameNode: function(other) {
+        return other && (typeof other === typeof this) && other.$element[0] === this.$element[0];
+    },
+
+    wrapWithNodeElement: function(wlxmlNode) {
+        this.$element.wrap($('<' + wlxmlNode.tag + ' class="' + wlxmlNode.klass + '"">')[0]);
+        return documentElementFromHTMLElement(this.$element.parent().get(0));
     }
 });
 
@@ -46,12 +57,32 @@ var DocumentTextElement = function(htmlElement) {
 DocumentNodeElement.prototype = new DocumentElement();
 DocumentTextElement.prototype = new DocumentElement();
 
+$.extend(DocumentNodeElement.prototype, {
+    append: function(params) {
+        var to_append = DocumentNodeElement.createDOM(params.tag, params.klass);
+        this.$element.append(to_append);
+        return documentElementFromHTMLElement(to_append);
+    },
+    after: function(params) {
+        var to_append = DocumentNodeElement.createDOM(params.tag, params.klass);
+        this.$element.after(to_append);
+        return documentElementFromHTMLElement(to_append);   
+    }
+});
+
+DocumentNodeElement.createDOM = function(tag, klass) {
+    var dom = $('<' + tag + '>');
+    if(klass)
+        dom.attr('class', klass);
+    return dom;
+};
+
 var documentElementFromHTMLElement = function(htmlElement) {
     if(htmlElement.nodeType === Node.ELEMENT_NODE)
         return new DocumentNodeElement(htmlElement);
     if(htmlElement.nodeType === Node.TEXT_NODE)
         return new DocumentTextElement(htmlElement);
-}
+};
 
 return {
     wrap: function(htmlElement) {