1 files changed, 19 insertions, 2 deletions
diff --git a/public/assets/javascripts/rectangles/engine/map/ui_editor.js b/public/assets/javascripts/rectangles/engine/map/ui_editor.js
index 016a8ad..4b5f784 100644
--- a/public/assets/javascripts/rectangles/engine/map/ui_editor.js
+++ b/public/assets/javascripts/rectangles/engine/map/ui_editor.js
@@ -30,6 +30,7 @@ Map.UI.Editor = function(map){
 	//
 	
 	function down (e, cursor){
+	  var room
 		cursor.x.div(map.dimensions.a).add(0.5).mul(map.dimensions.a / map.zoom).add(map.center.a)
 		cursor.y.div(map.dimensions.b).sub(0.5).mul(map.dimensions.b / map.zoom).sub(map.center.b)
 
@@ -52,8 +53,17 @@ Map.UI.Editor = function(map){
 
 		if (intersects.length && base.permissions.destroy) {
 			base.mouse.down = false
-			Rooms.remove(intersects[0])
-			app.tube("builder-destroy-room", intersects[0])
+			
+			room = intersects[0]
+
+      UndoStack.push({
+        type: "destroy-room",
+        prev: room.clone(),
+        next: { id: room.id },
+      })
+
+			Rooms.remove(room)
+			app.tube("builder-destroy-room", room)
 			return
 		}
 		else if (intersects.length) {
@@ -151,6 +161,13 @@ Map.UI.Editor = function(map){
 				cursor.x.abs().quantize(1)
 				cursor.y.abs().quantize(1)
 				var room = Rooms.add_with_rect( cursor )
+
+        UndoStack.push({
+          type: "create-room",
+          prev: { id: room.id },
+          next: room.clone()
+        })
+        
 				app.tube("builder-pick-room", room)
 			}
 		}