dendisuhubdy
diff --git a/‎database/scripts/main.js
Lines changed: 13 additions & 11 deletions b/‎database/scripts/main.js
Lines changed: 13 additions & 11 deletions
@@ -62,15 +62,17 @@ function writeNewPost(uid, username, title, body) {
 // [START post_stars_transaction]
 function toggleStar(postRef, uid) {
   postRef.transaction(function(post) {
-    if (post.stars && post.stars[uid]) {
-      post.starCount--;
-      post.stars[uid] = null;
-    } else {
-      post.starCount++;
-      if (!post.stars) {
-        post.stars = {};
+    if (post) {
+      if (post.stars && post.stars[uid]) {
+        post.starCount--;
+        post.stars[uid] = null;
+      } else {
+        post.starCount++;
+        if (!post.stars) {
+          post.stars = {};
+        }
+        post.stars[uid] = true;
       }
-      post.stars[uid] = true;
     }
     return post;
   });
@@ -80,7 +82,7 @@ function toggleStar(postRef, uid) {
 /**
  * Creates a post element.
  */
-function createPostElement(postId, title, text, author) {
+function createPostElement(postId, title, text, author, authorId) {
   var uid = firebase.auth().currentUser.uid;
 
   var html =
@@ -167,7 +169,7 @@ function createPostElement(postId, title, text, author) {
   // Bind starring action.
   var onStarClicked = function() {
     var globalPostRef = firebase.database().ref('/posts/' + postId);
-    var userPostRef = firebase.database().ref('/user-posts/' + uid + '/' + postId);
+    var userPostRef = firebase.database().ref('/user-posts/' + authorId + '/' + postId);
     toggleStar(globalPostRef, uid);
     toggleStar(userPostRef, uid);
   };
@@ -256,7 +258,7 @@ function startDatabaseQueries() {
     postsRef.on('child_added', function(data) {
       var containerElement = sectionElement.getElementsByClassName('posts-container')[0];
       containerElement.insertBefore(
-          createPostElement(data.key, data.val().title, data.val().body, data.val().author),
+          createPostElement(data.key, data.val().title, data.val().body, data.val().author, data.val().uid),
           containerElement.firstChild);
     });
   };