smartxml: refactoring

diff --git a/src/smartxml/smartxml.js b/src/smartxml/smartxml.js
index e484070..05583c5 100644 (file)
--- a/src/smartxml/smartxml.js
+++ b/src/smartxml/smartxml.js
@@ -91,13 +91,12 @@ $.extend(DocumentNode.prototype, {
     }),
 
     wrapWith: function(node) {
     }),
 
     wrapWith: function(node) {

-        node = node instanceof ElementNode ? node : this.document.createElementNode(node);
-
+        var insertion = this.getNodeInsertion(node);

         if(this.parent()) {
         if(this.parent()) {

-            this.before(node);
+            this.before(insertion.ofNode);

         }
         }

-        node.append(this);
-        return node;
+        insertion.ofNode.append(this);
+        return insertion.ofNode;

     },
 
     /**
     },
 
     /**