1 files changed, 128 insertions, 311 deletions

diff --git a/js/mediators/element-mediator.js b/js/mediators/element-mediator.js
index d62fa1f8..919aaec1 100755
--- a/js/mediators/element-mediator.js
+++ b/js/mediators/element-mediator.js
@@ -5,146 +5,97 @@ No rights, expressed or implied, whatsoever to this software are provided by Mot
 </copyright> */
 
 var Montage = require("montage/core/core").Montage,
-    NJComponent = require("js/lib/nj-base").NJComponent;
+    Component = require("montage/ui/component").Component;
 
 var ElementController = require("js/controllers/elements/element-controller").ElementController,
     Command     =   require("js/controllers/undo-controller").Command,
-    GroupCommand = require("js/controllers/undo-controller").GroupCommand,
     NJUtils = require("js/lib/NJUtils").NJUtils;
 
-exports.ElementMediator = Montage.create(NJComponent, {
+exports.ElementMediator = Montage.create(Component, {
 
-    deleteDelegate: {
-        value: null
-    },
 
-    deserializedFromTemplate: {
-        value: function () {
-            this.eventManager.addEventListener("elementAdding", this, false);
-            this.eventManager.addEventListener("deleting", this, false);
-        }
+    addDelegate: {
+        enumerable: false,
+        value: null
     },
 
-    // TODO use the specific controller to be able to subclass the functionality
-    handleElementAdding: {
-        value: function(event) {
-            this.addElement(event.detail.el, event.detail.data);
-        }
+    deleteDelegate: {
+        enumerable: false,
+        value: null
     },
 
-    handleDeleting: {
-        value: function(event) {
-            if(this.deleteDelegate && (typeof this.deleteDelegate.handleDelete === 'function')) {
-                this.deleteDelegate.handleDelete();
-            } else {
-                // Add the Undo/Redo
-                var els = [],
-                    len = this.application.ninja.selectedElements.length;
-
-                if(len) {
-                    for(var i = 0; i<len; i++) {
-                        els.push(this.application.ninja.selectedElements[i]);
-                    }
-                    
-                    for(i=0; i<len; i++) {
-                        this._removeElement(els[i]._element);
+    addElements: {
+        value: function(elements, rules, notify) {
+            if(Array.isArray(elements)) {
+                elements.forEach(function(element) {
+                    ElementController.addElement(element, rules);
+                    if(element.elementModel && element.elementModel.props3D) {
+                        element.elementModel.props3D.init(element, false);
                     }
-
-                    NJevent( "deleteSelection", els );
+                });
+            } else {
+                ElementController.addElement(elements, rules);
+                if(elements.elementModel && elements.elementModel.props3D) {
+                    elements.elementModel.props3D.init(elements, false);
                 }
             }
-        }
-    },
-
-    addElement: {
-        value: function(el, rules, noEvent) {
-            var command = Montage.create(Command, {
-                _el:            { value: el },
-                _rules:         { value: rules },
-                _noEvent:       { value: noEvent },
 
-                description: { value: "Adding Element"},
+            if(this.addDelegate && typeof (this.addDelegate['onAddElements']) === "function") {
+                this.addDelegate['onAddElements'].call(this.addDelegate, elements);
+            }
 
-                receiver: { value: this},
+            var undoLabel = "add element";
 
-                execute: {
-                    value: function() {
-                        this.receiver._addElement(this._el, this._rules, this._noEvent);
-                        return this._el;
-                    }
-                },
+            document.application.undoManager.add(undoLabel, this.removeElements, this, elements, notify);
 
-                unexecute: {
-                    value: function() {
-                        this.receiver._removeElement(this._el, this._rules, this._noEvent);
-                        return this._el;
-                    }
-                }
-            });
-
-            NJevent("sendToUndo", command);
-            command.execute();
-        }
-    },
+            this.application.ninja.documentController.activeDocument.needsSave = true;
 
-    _addElement: {
-        value: function(el, rules, noEvent) {
-            ElementController.addElement(el, rules);
-            var p3d = this.get3DProperties(el);
-            if(p3d) {
-                el.elementModel.controller["set3DProperties"](el, [p3d], 0, true);
-            }
-            if(!noEvent) {
-                this.application.ninja.documentController.activeDocument.needsSave = true;
-                NJevent("elementAdded", el);
+            if(notify || notify === undefined) {
+                NJevent("elementAdded", elements);
             }
         }
     },
 
-    deleteElements: {
-         value: function(items) {
-            // Add the Undo/Redo
-            var len, el;
+    removeElements: {
+        value: function(elements, notify /* Used for the add undo */) {
 
-            len = items.length;
+            if(this.deleteDelegate && (typeof this.deleteDelegate.handleDelete === 'function')) {
+                return this.deleteDelegate.handleDelete();
+                // this.handleDelete.call(deleteDelegate);
+            }
 
-            if(len) {
+            if(Array.isArray(elements)) {
+                elements = Array.prototype.slice.call(elements, 0);
+                elements.forEach(function(element) {
+                    ElementController.removeElement(element);
+                });
+            } else {
+                ElementController.removeElement(elements);
+            }
 
-                for(var i = len - 1; i >= 0; i--) {
-                    el = items[i]._element || items[i];
-                    this._removeElement(el);
-                }
+            var undoLabel = "add element";
 
-                NJevent( "deleteSelection", items );
-            }
-         }
-    },
+            document.application.undoManager.add(undoLabel, this.addElements, this, elements, null, notify);
 
-    _removeElement: {
-        value: function(el, rules) {
-            ElementController.removeElement(el, rules);
             this.application.ninja.documentController.activeDocument.needsSave = true;
-            NJevent("elementDeleted", el);
+
+            NJevent("elementsRemoved", elements);
         }
     },
 
     replaceElement: {
-        value: function(el, el2) {
-            el2.elementModel = el.elementModel;
-            this.application.ninja.currentDocument.documentRoot.replaceChild(el2, el);
-        }
-    },
+        value: function(newChild, oldChild, notify) {
 
-    getNJProperty: {
-        value: function(el, p) {
-            if(el.elementModel) {
-                if(el.elementModel.hasOwnProperty(p)) {
-                    return el.elementModel[p];
-                } else {
-                    console.log("Element Model does not have ", p);
-                }
-            } else {
-                console.log("Element has no Model -- Create one");
+            this.application.ninja.currentDocument.documentRoot.replaceChild(newChild, oldChild);
+
+            var undoLabel = "replace element";
+
+            document.application.undoManager.add(undoLabel, this.replaceElement, this, oldChild, newChild);
+
+            this.application.ninja.documentController.activeDocument.needsSave = true;
+
+            if(notify || notify === undefined) {
+                NJevent("elementReplaced", {type : "replaceElement", data: {"newChild": newChild, "oldChild": oldChild}});
             }
         }
     },
@@ -187,74 +138,25 @@ exports.ElementMediator = Montage.create(NJComponent, {
     },
 
     /**
-     Set a property change command for an element or array of elements
-     @param els: Array of elements. Can contain 1 or more elements
-     @param p: Property to set
-     @param value: Value to be set. This is an array of values corresponding to the array of elements
-     @param eventType: Change/Changing. Will be passed to the dispatched event
-     @param source: String for the source object making the call
-     @param currentValue *OPTIONAL*: current value array. If not found the current value is calculated
-     @param stageRedraw: *OPTIONAL*: True. If set to false the stage will not redraw the selection/outline
-     */
+    Set a property change command for an element or array of elements
+    @param element: Element
+    @param attribute: Attribute to set
+    @param value: Value to be set.
+    @param currentValue: current value
+    @param source: String for the source object making the call
+    */