Save a reference to the snapshot on the immutable value

This allows us to later get a firebase reference to it and to get it's priority, among other nice things.
FirebaseExtended · Jan 28, 2015 · 391dad3 · 391dad3
1 parent 009c993
commit 391dad3
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/reactfire.js b/src/reactfire.js
@@ -133,7 +133,9 @@ var ReactFireMixin = {
       else if (typeof(snapshot) === "object") {
         out = out.withMutations(function(map) {
           snapshot.forEach(function(child) {
-            map.set(child.key(), Immutable.fromJS(child.val()));
+            var immutableChild = Immutable.fromJS(child.val());
+            immutableChild.snapshot = child;
+            map.set(child.key(), immutableChild);
           });
         });
       }
@@ -151,7 +153,9 @@ var ReactFireMixin = {
       else if (typeof(snapshot) === "object") {
         out = out.withMutations(function(list) {
           snapshot.forEach(function(child) {
-            list.push(Immutable.fromJS(child.val()));
+            var immutableChild = Immutable.fromJS(child.val());
+            immutableChild.snapshot = child;
+            list.push(immutableChild);
           });
         });
       }