X-Git-Url: https://git.mdrn.pl/fnpeditor.git/blobdiff_plain/2ec50c1931fa2337309067dd1b4ee0a3aeaaaa8f..66cc0acfd75edaf656c8ce482421674a8d0fa126:/src/smartxml/smartxml.js

diff --git a/src/smartxml/smartxml.js b/src/smartxml/smartxml.js
index 43e3041..248333e 100644
--- a/src/smartxml/smartxml.js
+++ b/src/smartxml/smartxml.js
@@ -5,21 +5,14 @@ define([
 'use strict';
 
 
-var TEXT_NODE = Node.TEXT_NODE, ELEMENT_NODE = Node.ELEMENT_NODE;
+var TEXT_NODE = Node.TEXT_NODE;
 
-var parseXML = function(xml) {
-    return $(xml)[0];
-};
 
-var Document = function(nativeNode) {
-    var $document = $(nativeNode);
-
-
-    Object.defineProperty(this, 'root', {get: function() { return new ElementNode($document[0]);}});
-};
-
-
-var DocumentNode = function(nativeNode) {
+var DocumentNode = function(nativeNode, document) {
+    if(!document) {
+        throw new Error('undefined document for a node');
+    }
+    this.document = document;
     this.nativeNode = nativeNode;
     this._$ = $(nativeNode);
 };
@@ -32,7 +25,7 @@ $.extend(DocumentNode.prototype, {
     },
 
     parent: function() {
-        return this.nativeNode.parentNode ? new ElementNode(this.nativeNode.parentNode) : null;
+        return this.nativeNode.parentNode ? this.document.createElementNode(this.nativeNode.parentNode) : null;
     },
 
     before: function(node) {
@@ -40,14 +33,15 @@ $.extend(DocumentNode.prototype, {
     },
 
     wrapWith: function(node) {
-        if(this.parent())
+        if(this.parent()) {
             this.before(node);
+        }
         node.append(this);
     },
 });
 
-var ElementNode = function(nativeNode) {
-    DocumentNode.apply(this, arguments);
+var ElementNode = function(nativeNode, document) {
+    DocumentNode.call(this, nativeNode, document);
 };
 
 $.extend(ElementNode.prototype, DocumentNode.prototype, {
@@ -58,12 +52,15 @@ $.extend(ElementNode.prototype, DocumentNode.prototype, {
     },
 
     contents: function() {
-        var toret = [];
+        var toret = [],
+            document = this.document;
         this._$.contents().each(function() {
-            if(this.nodeType === Node.ELEMENT_NODE)
-                toret.push(new ElementNode(this));
-            else if(this.nodeType === Node.TEXT_NODE)
-                toret.push(new TextNode(this));
+            if(this.nodeType === Node.ELEMENT_NODE) {
+                toret.push(document.createElementNode(this));
+            }
+            else if(this.nodeType === Node.TEXT_NODE) {
+                toret.push(document.createTextNode(this));
+            }
         });
         return toret;
     },
@@ -80,21 +77,31 @@ $.extend(ElementNode.prototype, DocumentNode.prototype, {
         this._$.attr(name, value);
     },
 
+    getAttrs: function() {
+        var toret = [];
+        for(var i = 0; i < this.nativeNode.attributes.length; i++) {
+            toret.push(this.nativeNode.attributes[i]);
+        }
+        return toret;
+    },
+
     append: function(documentNode) {
         this._$.append(documentNode.nativeNode);
     },
 
     unwrapContent: function() {
         var parent = this.parent();
-        if(!parent)
+        if(!parent) {
             return;
+        }
 
         var parentContents = parent.contents(),
             myContents = this.contents(),
             myIdx = parent.indexOf(this);
 
-        if(myContents.length === 0)
+        if(myContents.length === 0) {
             return this.detach();
+        }
 
         var moveLeftRange, moveRightRange, leftMerged;
 
@@ -126,12 +133,17 @@ $.extend(ElementNode.prototype, DocumentNode.prototype, {
             element1: parent.contents()[myIdx + (moveLeftRange ? -1 : 0)],
             element2: parent.contents()[myIdx + childrenLength-1 + (moveRightRange ? 1 : 0)]
         };
-    }
+    },
 
+    toXML: function() {
+        var wrapper = $('<div>');
+        wrapper.append(this._$);
+        return wrapper.html();
+    }
 });
 
-var TextNode = function(nativeNode) {
-    DocumentNode.apply(this, arguments);
+var TextNode = function(nativeNode, document) {
+    DocumentNode.call(this, nativeNode, document);
 };
 
 $.extend(TextNode.prototype, DocumentNode.prototype, {
@@ -151,14 +163,48 @@ $.extend(TextNode.prototype, DocumentNode.prototype, {
 });
 
 
+var parseXML = function(xml) {
+    return $(xml)[0];
+};
+
+var Document = function(xml) {
+    var $document = $(parseXML(xml));
+
+    var doc = this;
+    Object.defineProperty(this, 'root', {get: function() {
+        return doc.createElementNode($document[0]);
+    }});
+};
+$.extend(Document.prototype, {
+    ElementNodeFactory: ElementNode,
+    TextNodeFactory: TextNode,
+
+    createElementNode: function(nativeNode) {
+        return new this.ElementNodeFactory(nativeNode, this);
+    },
+
+    createTextNode: function(nativeNode) {
+        return new this.TextNodeFactory(nativeNode, this);
+    },
+
+    toXML: function() {
+        return this.root.toXML();
+    }
+});
+
+
 return {
     documentFromXML: function(xml) {
         return new Document(parseXML(xml));
     },
 
     elementNodeFromXML: function(xml) {
-        return new ElementNode(parseXML(xml));
-    }
+        return this.documentFromXML(xml).root;
+    },
+
+    Document: Document,
+    DocumentNode: DocumentNode,
+    ElementNode: ElementNode
 };
 
 });
\ No newline at end of file