X-Git-Url: https://git.mdrn.pl/fnpeditor.git/blobdiff_plain/48c904bb773b7fcc8e9249de731d03a17760f727..0f5cbf957bc0bbdcff93bdb33c227f78041e332b:/src/smartxml/smartxml.js

diff --git a/src/smartxml/smartxml.js b/src/smartxml/smartxml.js
index 0c76cd2..9215287 100644
--- a/src/smartxml/smartxml.js
+++ b/src/smartxml/smartxml.js
@@ -1,60 +1,932 @@
 define([
-    'libs/jquery'
-], function($) {
+    'libs/jquery',
+    'libs/underscore',
+    'libs/backbone',
+    'smartxml/events',
+    'smartxml/transformations'
+], function($, _, Backbone, events, transformations) {
     
 'use strict';
+/* globals Node */
 
+var TEXT_NODE = Node.TEXT_NODE;
 
-var parseXML = function(xml) {
-    return $(xml)[0];
-}
 
-var Document = function(nativeNode) {
-    var $document = $(nativeNode);
+var INSERTION = function(implementation) {
+    var toret = function(node) {
+        var insertion = this.getNodeInsertion(node),
+            nodeWasContained = this.document.containsNode(insertion.ofNode),
+            nodeParent;
+        if(!(this.document.containsNode(this))) {
+            nodeParent = insertion.ofNode.parent();
+        }
+        implementation.call(this, insertion.ofNode.nativeNode);
+        this.triggerChangeEvent(insertion.insertsNew ? 'nodeAdded' : 'nodeMoved', {node: insertion.ofNode}, nodeParent, nodeWasContained);
+        return insertion.ofNode;
+    };
+    return toret;
+};
+
+var DocumentNode = function(nativeNode, document) {
+    if(!document) {
+        throw new Error('undefined document for a node');
+    }
+    this.document = document;
+    this._setNativeNode(nativeNode);
+
+};
+
+$.extend(DocumentNode.prototype, {
+
+    transform: function(name, args) {
+        var Transformation = this.transformations.get(name),
+            transformation;
+        if(Transformation) {
+            transformation = new Transformation(this.document, this, args);
+        }
+        return this.document.transform(transformation);
+    },
+
+    _setNativeNode: function(nativeNode) {
+        this.nativeNode = nativeNode;
+        this._$ = $(nativeNode);
+    },
+
+    clone: function() {
+        var clone = this._$.clone(true, true);
+        // clone.find('*').addBack().each(function() {
+        //     var n = $(this);
+        //     if(n.data('canvasElement')) {
+        //         n.data('canvasElement', $.extend(true, {}, n.data('canvasElement')));
+        //         n.data('canvasElement').$element = n.data('canvasElement').$element.clone(true, true);
+        //     }
+        // });
+        return this.document.createDocumentNode(clone[0]);
+    },
+
+    getPath: function(ancestor) {
+        var nodePath = [this].concat(this.parents()),
+            toret, idx;
+        ancestor = ancestor || this.document.root;
+
+        nodePath.some(function(node, i) {
+            if(node.sameNode(ancestor)) {
+                idx = i;
+                return true;
+            }
+        });
+
+        if(idx !== 'undefined') {
+            nodePath = nodePath.slice(0, idx);
+        }
+        toret = nodePath.map(function(node) {return node.getIndex(); });
+        toret.reverse();
+        return toret;
+    },
+
+    isRoot: function() {
+        return this.document.root.sameNode(this);
+    },
+
+    detach: function() {
+        var parent = this.parent();
+        this._$.detach();
+        this.triggerChangeEvent('nodeDetached', {parent: parent});
+        return this;
+    },
+
+    replaceWith: function(node) {
+        var toret;
+        if(this.isRoot()) {
+            return this.document.replaceRoot(node);
+        }
+        toret = this.after(node);
+        this.detach();
+        return toret;
+    },
+
+    sameNode: function(otherNode) {
+        return !!(otherNode) && this.nativeNode === otherNode.nativeNode;
+    },
 
+    parent: function() {
+        var parentNode = this.nativeNode.parentNode;
+        if(parentNode && parentNode.nodeType === Node.ELEMENT_NODE) {
+            return this.document.createDocumentNode(parentNode);
+        }
+        return null;
+    },
+
+    parents: function() {
+        var parent = this.parent(),
+            parents = parent ? parent.parents() : [];
+        if(parent) {
+            parents.unshift(parent);
+        }
+        return parents;
+    },
+
+    prev: function() {
+        var myIdx = this.getIndex();
+        return myIdx > 0 ? this.parent().contents()[myIdx-1] : null;
+    },
+
+    next: function() {
+        if(this.isRoot()) {
+            return null;
+        }
+        var myIdx = this.getIndex(),
+            parentContents = this.parent().contents();
+        return myIdx < parentContents.length - 1 ? parentContents[myIdx+1] : null;
+    },
+
+    isSurroundedByTextElements: function() {
+        var prev = this.prev(),
+            next = this.next();
+        return prev && (prev.nodeType === Node.TEXT_NODE) && next && (next.nodeType === Node.TEXT_NODE);
+    },
 
-    Object.defineProperty(this, 'root', {get: function() { return new ElementNode($document[0])}}); 
-}
+    after: INSERTION(function(nativeNode) {
+        return this._$.after(nativeNode);
+    }),
 
+    before: INSERTION(function(nativeNode) {
+        return this._$.before(nativeNode);
+    }),
 
-var ElementNode = function(nativeNode) {
-    this.nativeNode = nativeNode;
-    this._$ = $(nativeNode);
+    wrapWith: function(node) {
+        var insertion = this.getNodeInsertion(node);
+        if(this.parent()) {
+            this.before(insertion.ofNode);
+        }
+        insertion.ofNode.append(this);
+        return insertion.ofNode;
+    },
+
+    /**
+    * Removes parent of a node if node has no siblings.
+    */
+    unwrap: function() {
+        if(this.isRoot()) {
+            return;
+        }
+        var parent = this.parent(),
+            grandParent;
+        if(parent.contents().length === 1) {
+            grandParent = parent.parent();
+            parent.unwrapContent();
+            return grandParent;
+        }
+    },
+
+    triggerChangeEvent: function(type, metaData, origParent, nodeWasContained) {
+        var node = (metaData && metaData.node) ? metaData.node : this,
+            event = new events.ChangeEvent(type, $.extend({node: node}, metaData || {}));
+        if(type === 'nodeDetached' || this.document.containsNode(event.meta.node)) {
+            this.document.trigger('change', event);
+        }
+        if((type === 'nodeAdded' || type === 'nodeMoved') && !this.document.containsNode(this) && nodeWasContained) {
+             event = new events.ChangeEvent('nodeDetached', {node: node, parent: origParent});
+             this.document.trigger('change', event);
+        }
+    },
+    
+    getNodeInsertion: function(node) {
+        return this.document.getNodeInsertion(node);
+    },
+
+    getIndex: function() {
+        if(this.isRoot()) {
+            return 0;
+        }
+        return this.parent().indexOf(this);
+    }
+});
+
+var ElementNode = function(nativeNode, document) {
+    DocumentNode.call(this, nativeNode, document);
 };
+ElementNode.prototype = Object.create(DocumentNode.prototype);
 
 $.extend(ElementNode.prototype, {
+    nodeType: Node.ELEMENT_NODE,
+
+    detach: function() {
+        var next;
+        if(this.parent() && this.isSurroundedByTextElements()) {
+            next = this.next();
+            this.prev().appendText(next.getText());
+            next.detach();
+        }
+        return DocumentNode.prototype.detach.call(this);
+    },
+
+    setData: function(key, value) {
+        if(value !== undefined) {
+            this._$.data(key, value);
+        } else {
+            this._$.removeData(_.keys(this._$.data()));
+            this._$.data(key);
+        }
+    },
+
+    getData: function(key) {
+        if(key) {
+            return this._$.data(key);
+        }
+        return this._$.data();
+    },
+
     getTagName: function() {
         return this.nativeNode.tagName.toLowerCase();
     },
 
-    append: function(documentNode) {
-        this._$.append(documentNode.nativeNode);
+    contents: function(selector) {
+        var toret = [],
+            document = this.document;
+        if(selector) {
+            this._$.children(selector).each(function() {
+                toret.push(document.createDocumentNode(this));
+            });
+        } else {
+            this._$.contents().each(function() {
+                toret.push(document.createDocumentNode(this));
+            });
+        }
+        return toret;
     },
 
-    contents: function() {
-        var toret = [];
-        this._$.contents().each(function() {
-            if(this.nodeType === Node.ELEMENT_NODE)
-                toret.push(new ElementNode(this));
+    indexOf: function(node) {
+        return this._$.contents().index(node._$);
+    },
+
+    setTag: function(tagName) {
+        var node = this.document.createDocumentNode({tagName: tagName}),
+            oldTagName = this.getTagName(),
+            myContents = this._$.contents();
+
+        this.getAttrs().forEach(function(attribute) {
+            node.setAttr(attribute.name, attribute.value, true);
         });
+        node.setData(this.getData());
+
+        if(this.sameNode(this.document.root)) {
+            defineDocumentProperties(this.document, node._$);
+        }
+        this._$.replaceWith(node._$);
+        this._setNativeNode(node._$[0]);
+        this._$.append(myContents);
+        this.triggerChangeEvent('nodeTagChange', {oldTagName: oldTagName, newTagName: this.getTagName()});
+    },
+
+    getAttr: function(name) {
+        return this._$.attr(name);
+    },
+
+    setAttr: function(name, value, silent) {
+        var oldVal = this.getAttr(name);
+        this._$.attr(name, value);
+        if(!silent) {
+            this.triggerChangeEvent('nodeAttrChange', {attr: name, oldVal: oldVal, newVal: value});
+        }
+    },
+
+    getAttrs: function() {
+        var toret = [];
+        for(var i = 0; i < this.nativeNode.attributes.length; i++) {
+            toret.push(this.nativeNode.attributes[i]);
+        }
         return toret;
     },
 
+    append: INSERTION(function(nativeNode) {
+        this._$.append(nativeNode);
+    }),
 
-    sameNode: function(otherNode) {
-        return this.nativeNode === otherNode.nativeNode;
+    prepend: INSERTION(function(nativeNode) {
+        this._$.prepend(nativeNode);
+    }),
+
+    insertAtIndex: function(nativeNode, index) {
+        var contents = this.contents();
+        if(index < contents.length) {
+            return contents[index].before(nativeNode);
+        } else if(index === contents.length) {
+            return this.append(nativeNode);
+        }
+    },
+
+    unwrapContent: function() {
+        var parent = this.parent();
+        if(!parent) {
+            return;
+        }
+
+        var myContents = this.contents(),
+            myIdx = parent.indexOf(this);
+
+
+        if(myContents.length === 0) {
+            return this.detach();
+        }
+
+        var prev = this.prev(),
+            next = this.next(),
+            moveLeftRange, moveRightRange, leftMerged;
+
+        if(prev && (prev.nodeType === TEXT_NODE) && (myContents[0].nodeType === TEXT_NODE)) {
+            prev.appendText(myContents[0].getText());
+            myContents[0].detach();
+            moveLeftRange = true;
+            leftMerged = true;
+        } else {
+            leftMerged = false;
+        }
+
+        if(!(leftMerged && myContents.length === 1)) {
+            var lastContents = _.last(myContents);
+            if(next && (next.nodeType === TEXT_NODE) && (lastContents.nodeType === TEXT_NODE)) {
+                next.prependText(lastContents.getText());
+                lastContents.detach();
+                moveRightRange = true;
+            }
+        }
+
+        var childrenLength = this.contents().length;
+        this.contents().forEach(function(child) {
+            this.before(child);
+        }.bind(this));
+
+        this.detach();
+
+        return {
+            element1: parent.contents()[myIdx + (moveLeftRange ? -1 : 0)],
+            element2: parent.contents()[myIdx + childrenLength-1 + (moveRightRange ? 1 : 0)]
+        };
+    },
+
+    wrapText: function(params) {
+        return this.document._wrapText(_.extend({inside: this}, params));
+    },
+
+    toXML: function() {
+        var wrapper = $('<div>');
+        wrapper.append(this._getXMLDOMToDump());
+        return wrapper.html();
+    },
+    
+    _getXMLDOMToDump: function() {
+        return this._$;
     }
+});
+
+// trans
+
+// todo - split+append
+
+// ElementNode.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'smartxml.setAttr',
+//     impl: function(args) {
+//         this.setAttr(args.name, args.value);
+//     },
+//     getChangeRoot: function() {
+//         return this.context;
+//     }
+// }));
+
+// ElementNode.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'smartxml.setAttr2',
+//     impl: function(args) {
+//         this.prevAttr = this.getAttr(args.name);
+//         this.setAttr(args.name, args.value);
+//     },
+//     undo: function(args) {
+//         this.setAttr(args.name, this.prevAttr);
+//     }
+// }));
+
+// DocumentNode.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'smartxml.wrapWith',
+//     getChangeRoot: function() {
+//         return this.context.parent();
+//     },
+//     impl: function(args) {
+//         return this.wrapWith(args);
+//     }
+// }));
+
+// DocumentNode.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'smartxml.wrapText',
+//     getChangeRoot: function() {
+//         return this.context;
+//     },
+//     impl: function(args) {
+//         return this.wrapText(args);
+//     }
+// }));
+
+// DocumentNode.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'smartxml.detach',
+//     getChangeRoot: function() {
+//         return this.context.parent();
+//     },
+//     impl: function(args) {
+//         return this.detach();
+//     }
+// }));
+
+///
+
+var TextNode = function(nativeNode, document) {
+    DocumentNode.call(this, nativeNode, document);
+};
+TextNode.prototype = Object.create(DocumentNode.prototype);
 
+$.extend(TextNode.prototype, {
+    nodeType: Node.TEXT_NODE,
+
+    getText: function() {
+        return this.nativeNode.data;
+    },
+
+    setText: function(text) {
+        //console.log('smartxml: ' + text);
+        this.nativeNode.data = text;
+        this.triggerTextChangeEvent();
+    },
+
+    appendText: function(text) {
+        this.nativeNode.data = this.nativeNode.data + text;
+        this.triggerTextChangeEvent();
+    },
+
+    prependText: function(text) {
+        this.nativeNode.data = text + this.nativeNode.data;
+        this.triggerTextChangeEvent();
+    },
+
+    wrapWith: function(desc) {
+        if(typeof desc.start === 'number' && typeof desc.end === 'number') {
+            return this.document._wrapText({
+                inside: this.parent(),
+                textNodeIdx: this.parent().indexOf(this),
+                offsetStart: Math.min(desc.start, desc.end),
+                offsetEnd: Math.max(desc.start, desc.end),
+                _with: {tagName: desc.tagName, attrs: desc.attrs}
+            });
+        } else {
+            return DocumentNode.prototype.wrapWith.call(this, desc);
+        }
+    },
+
+    split: function(params) {
+        var parentElement = this.parent(),
+            passed = false,
+            succeedingChildren = [],
+            prefix = this.getText().substr(0, params.offset),
+            suffix = this.getText().substr(params.offset);
+
+        parentElement.contents().forEach(function(child) {
+            if(passed) {
+                succeedingChildren.push(child);
+            }
+            if(child.sameNode(this)) {
+                passed = true;
+            }
+        }.bind(this));
+
+        if(prefix.length > 0) {
+            this.setText(prefix);
+        }
+        else {
+            this.detach();
+        }
+
+        var attrs = {};
+        parentElement.getAttrs().forEach(function(attr) {attrs[attr.name] = attr.value; });
+        var newElement = this.document.createDocumentNode({tagName: parentElement.getTagName(), attrs: attrs});
+        parentElement.after(newElement);
+
+        if(suffix.length > 0) {
+            newElement.append({text: suffix});
+        }
+        succeedingChildren.forEach(function(child) {
+            newElement.append(child);
+        });
+
+        return {first: parentElement, second: newElement};
+    },
+
+    triggerTextChangeEvent: function() {
+        var event = new events.ChangeEvent('nodeTextChange', {node: this});
+        this.document.trigger('change', event);
+    }
+});
+
+
+// TextNode.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'rng.breakContent',
+//     // impl: function(args) {
+//     //     var node = this.context,
+//     //         newNodes, emptyNode, emptyText;
+//     //     newNodes = node.transform('smartxml.split', {offset: args.offset});
+//     //     [newNodes.first, newNodes.second].some(function(newNode) {
+//     //         if(!(newNode.contents().length)) {
+//     //             newNode.transform('smartxml.append', {text: ''});
+//     //             return true; // break
+//     //         }
+//     //     });
+//     //     return _.extend(newNodes, {emptyText: emptyText});
+//     // },
+//     impl: function(args) {
+//         var node = this,
+//             newNodes, emptyNode, emptyText;
+//         newNodes = node.split({offset: args.offset});
+//         [newNodes.first, newNodes.second].some(function(newNode) {
+//             if(!(newNode.contents().length)) {
+//                 newNode.append({text: ''});
+//                 return true; // break
+//             }
+//         });
+//         return _.extend(newNodes, {emptyText: emptyText});
+//     },
+//     getChangeRoot: function() {
+//         return this.context.parent().parent();
+//     },
+//     isAllowed: function(args) {
+//         var parent = this.parent();
+//         return !!(parent && parent.parent());
+//     }
+// }));
+
+
+// ElementNode.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'smartxml.setText',
+//     impl: function(args) {
+//         this.setText(args.text);
+//     },
+//     getChangeRoot: function() {
+//         return this.context;
+//     }
+// }));
+
+
+var parseXML = function(xml) {
+    return $($.trim(xml))[0];
+};
+
+var Document = function(xml) {
+    this.loadXML(xml);
+    this.undoStack = [];
+    this.redoStack = [];
+    this._transformationLevel = 0;
+    this.transformations = new transformations.TransformationStorage();
+    
+    this._nodeMethods = {};
+    this._nodeTransformations = new transformations.TransformationStorage();
+};
+
+$.extend(Document.prototype, Backbone.Events, {
+    ElementNodeFactory: ElementNode,
+    TextNodeFactory: TextNode,
+
+    createDocumentNode: function(from) {
+        if(!(from instanceof Node)) {
+            if(from.text !== undefined) {
+                /* globals document */
+                from = document.createTextNode(from.text);
+            } else {
+                var node = $('<' + from.tagName + '>');
+
+                _.keys(from.attrs || {}).forEach(function(key) {
+                    node.attr(key, from.attrs[key]);
+                });
+
+                from = node[0];
+            }
+        }
+        var Factory;
+        if(from.nodeType === Node.TEXT_NODE) {
+            Factory = this.TextNodeFactory;
+        } else if(from.nodeType === Node.ELEMENT_NODE) {
+            Factory = this.ElementNodeFactory;
+        }
+        var toret = new Factory(from, this);
+        _.extend(toret, this._nodeMethods);
+        toret.transformations = this._nodeTransformations;
+        return toret;
+    },
+
+    loadXML: function(xml, options) {
+        options = options || {};
+        defineDocumentProperties(this, $(parseXML(xml)));
+        if(!options.silent) {
+            this.trigger('contentSet');
+        }
+    },
+
+    toXML: function() {
+        return this.root.toXML();
+    },
+
+    containsNode: function(node) {
+        return this.root && (node.nativeNode === this.root.nativeNode || node._$.parents().index(this.root._$) !== -1);
+    },
+
+    wrapNodes: function(params) {
+        if(!(params.node1.parent().sameNode(params.node2.parent()))) {
+            throw new Error('Wrapping non-sibling nodes not supported.');
+        }
+
+        var parent = params.node1.parent(),
+            parentContents = parent.contents(),
+            wrapper = this.createDocumentNode({
+                tagName: params._with.tagName,
+                attrs: params._with.attrs}),
+            idx1 = parent.indexOf(params.node1),
+            idx2 = parent.indexOf(params.node2);
+
+        if(idx1 > idx2) {
+            var tmp = idx1;
+            idx1 = idx2;
+            idx2 = tmp;
+        }
+
+        var insertingMethod, insertingTarget;
+        if(idx1 === 0) {
+            insertingMethod = 'prepend';
+            insertingTarget = parent;
+        } else {
+            insertingMethod = 'after';
+            insertingTarget = parentContents[idx1-1];
+        }
+
+        for(var i = idx1; i <= idx2; i++) {
+            wrapper.append(parentContents[i].detach());
+        }
+
+        insertingTarget[insertingMethod](wrapper);
+        return wrapper;
+    },
+
+    getSiblingParents: function(params) {
+        var parents1 = [params.node1].concat(params.node1.parents()).reverse(),
+            parents2 = [params.node2].concat(params.node2.parents()).reverse(),
+            noSiblingParents = null;
+
+        if(parents1.length === 0 || parents2.length === 0 || !(parents1[0].sameNode(parents2[0]))) {
+            return noSiblingParents;
+        }
+
+        var i;
+        for(i = 0; i < Math.min(parents1.length, parents2.length); i++) {
+            if(parents1[i].sameNode(parents2[i])) {
+                continue;
+            }
+            break;
+        }
+        return {node1: parents1[i], node2: parents2[i]};
+    },
+
+    _wrapText: function(params) {
+        params = _.extend({textNodeIdx: 0}, params);
+        if(typeof params.textNodeIdx === 'number') {
+            params.textNodeIdx = [params.textNodeIdx];
+        }
+        
+        var contentsInside = params.inside.contents(),
+            idx1 = Math.min.apply(Math, params.textNodeIdx),
+            idx2 = Math.max.apply(Math, params.textNodeIdx),
+            textNode1 = contentsInside[idx1],
+            textNode2 = contentsInside[idx2],
+            sameNode = textNode1.sameNode(textNode2),
+            prefixOutside = textNode1.getText().substr(0, params.offsetStart),
+            prefixInside = textNode1.getText().substr(params.offsetStart),
+            suffixInside = textNode2.getText().substr(0, params.offsetEnd),
+            suffixOutside = textNode2.getText().substr(params.offsetEnd)
+        ;
+
+        if(!(textNode1.parent().sameNode(textNode2.parent()))) {
+            throw new Error('Wrapping text in non-sibling text nodes not supported.');
+        }
+        
+        var wrapperElement = this.createDocumentNode({tagName: params._with.tagName, attrs: params._with.attrs});
+        textNode1.after(wrapperElement);
+        textNode1.detach();
+        
+        if(prefixOutside.length > 0) {
+            wrapperElement.before({text:prefixOutside});
+        }
+        if(sameNode) {
+            var core = textNode1.getText().substr(params.offsetStart, params.offsetEnd - params.offsetStart);
+            wrapperElement.append({text: core});
+        } else {
+            textNode2.detach();
+            if(prefixInside.length > 0) {
+                wrapperElement.append({text: prefixInside});
+            }
+            for(var i = idx1 + 1; i < idx2; i++) {
+                wrapperElement.append(contentsInside[i]);
+            }
+            if(suffixInside.length > 0) {
+                wrapperElement.append({text: suffixInside});
+            }
+        }
+        if(suffixOutside.length > 0) {
+            wrapperElement.after({text: suffixOutside});
+        }
+        return wrapperElement;
+    },
+
+    trigger: function() {
+        //console.log('trigger: ' + arguments[0] + (arguments[1] ? ', ' + arguments[1].type : ''));
+        Backbone.Events.trigger.apply(this, arguments);
+    },
+
+    getNodeInsertion: function(node) {
+        var insertion = {};
+        if(node instanceof DocumentNode) {
+            insertion.ofNode = node;
+            insertion.insertsNew = !this.containsNode(node);
+        } else {
+          insertion.ofNode = this.createDocumentNode(node);
+          insertion.insertsNew = true;
+        }
+        return insertion;
+    },
+
+    replaceRoot: function(node) {
+        var insertion = this.getNodeInsertion(node);
+        this.root.detach();
+        defineDocumentProperties(this, insertion.ofNode._$);
+        insertion.ofNode.triggerChangeEvent('nodeAdded');
+        return insertion.ofNode;
+    },
+
+    registerMethod: function(methodName, method) {
+        this[methodName] = method;
+    },
+
+    registerTransformation: function(Transformation) {
+        return this.transformations.register(Transformation);
+    },
+
+    registerNodeMethod: function(methodName, method) {
+        this._nodeMethods[methodName] = method;
+    },
+
+    registerNodeTransformation: function(Transformation) {
+        this._nodeTransformations.register(Transformation);
+    },
+
+    registerExtension: function(extension) {
+        //debugger;
+        var doc = this,
+            existingPropertyNames = _.values(this);
+
+        var getTrans = function(desc, methodName) {
+            if(typeof desc === 'function') {
+                desc = {impl: desc};
+            }
+            if(!desc.impl) {
+                throw new Error('Got transformation description without implementation.')
+            }
+            desc.name = desc.name || methodName;
+            return desc;
+        };
+
+        [
+            {source: extension.document, target: doc},
+            {source: extension.documentNode, target: [doc.ElementNodeFactory.prototype, doc.TextNodeFactory.prototype]},
+
+        ].forEach(function(mapping) {
+            if(mapping.source) {
+                if(mapping.source.methods) {
+                    existingPropertyNames = _.values(mapping.target)
+                    _.pairs(mapping.source.methods).forEach(function(pair) {
+                        var methodName = pair[0],
+                            method = pair[1],
+                            targets = _.isArray(mapping.target) ? mapping.target : [mapping.target];
+                        if(_.contains(existingPropertyNames, methodName)) {
+                            throw new Error('Cannot extend {target} with method name {methodName}. Name already exists.'
+                                .replace('{target}', mapping.target)
+                                .replace('{methodName}', methodName)
+                            );
+                        }
+                        targets.forEach(function(target) {
+                            if(target === doc) {
+                                target.registerMethod(methodName, method);
+                            } else {
+                                doc.registerNodeMethod(methodName, method);
+                            }
+
+                        });
+                    });
+                }
+
+                if(mapping.source.transformations) {
+                    _.pairs(mapping.source.transformations).forEach(function(pair) {
+                        var transformation = getTrans(pair[1], pair[0]),
+                            targets = _.isArray(mapping.target) ? mapping.target : [mapping.target];
+                        targets.forEach(function(target) {
+                            if(target === doc) {
+                                target.registerTransformation(transformations.createContextTransformation(transformation));    
+                            } else {
+                                doc.registerNodeTransformation(transformations.createContextTransformation(transformation));
+                            }
+
+                            
+                        });
+                    });
+                }
+            }
+        });
+    },
+
+    transform: function(transformation, args) {
+        //console.log('transform');
+        var Transformation, toret;
+        if(typeof transformation === 'string') {
+            Transformation = this.transformations.get(transformation);
+            if(Transformation) {
+                transformation = new Transformation(this, this, args);
+            }
+        } 
+        if(transformation) {
+            this._transformationLevel++;
+            toret = transformation.run();
+            if(this._transformationLevel === 1) {
+                this.undoStack.push(transformation);
+            }
+            this._transformationLevel--;
+            //console.log('clearing redo stack');
+            this.redoStack = [];
+            return toret;
+        } else {
+            throw new Error('Transformation ' + transformation + ' doesn\'t exist!');
+        }
+    },
+    undo: function() {
+        var transformation = this.undoStack.pop();
+        if(transformation) {
+            transformation.undo();
+            this.redoStack.push(transformation);
+        }
+    },
+    redo: function() {
+        var transformation = this.redoStack.pop();
+        if(transformation) {
+            transformation.run();
+            this.undoStack.push(transformation);
+        }
+    },
+
+    getNodeByPath: function(path) {
+        var toret = this.root;
+        path.forEach(function(idx) {
+            toret = toret.contents()[idx];
+        });
+        return toret;
+    }
 });
 
+var defineDocumentProperties = function(doc, $document) {
+    Object.defineProperty(doc, 'root', {get: function() {
+        return doc.createDocumentNode($document[0]);
+    }, configurable: true});
+    Object.defineProperty(doc, 'dom', {get: function() {
+        return $document[0];
+    }, configurable: true});
+};
+
+// Document.prototype.transformations.register(transformations.createContextTransformation({
+//     name: 'smartxml.wrapNodes',
+//     // init: function() {
+
+//     // },
+//     // getChangeRoot: function() {
+//     //     return this.context;
+//     // },
+//     impl: function(args) {
+//         this.wrapNodes(args);
+//     },
+
+// }));
+
+
 return {
     documentFromXML: function(xml) {
-        return new Document(parseXML(xml));
+        return new Document(xml);
     },
 
     elementNodeFromXML: function(xml) {
-        return new ElementNode(parseXML(xml));
-    }
+        return this.documentFromXML(xml).root;
+    },
+
+    Document: Document,
+    DocumentNode: DocumentNode,
+    ElementNode: ElementNode,
+    TextNode: TextNode
 };
 
 });
\ No newline at end of file