undo/redo for colors

1 files changed, 7 insertions, 6 deletions

diff --git a/public/assets/javascripts/rectangles/util/undostack.js b/public/assets/javascripts/rectangles/util/undostack.js
index b93c79e..959e3d1 100644
--- a/public/assets/javascripts/rectangles/util/undostack.js
+++ b/public/assets/javascripts/rectangles/util/undostack.js
@@ -31,16 +31,17 @@
     this.types[ action.type ].redo(action.redo)
     return this.pointer < this.stack.length-1
   }
-  UndoStack.prototype.register = function(actionType){
-  	if (actionType.length) {
-  		actionType.forEach(this.registerOne.bind(this))
+  UndoStack.prototype.register = function(actions){
+  	if (actions.length) {
+  		actions.forEach(this.registerOne.bind(this))
   	}
   	else {
-  		this.registerOne(actionType)
+  		this.registerOne(actions)
 		}
   }
-  UndoStack.prototype.registerOne = function(actionType){
-		this.types[ actionType.type ] = actionType
+  UndoStack.prototype.registerOne = function(action){
+		if (! action.redo) { action.redo = action.undo }
+		this.types[ action.type ] = action
 	}  
   if ('window' in this) {
     window.UndoStack = new UndoStack