Update CommentModel.php

Author: Igor Chepurnoy

models/CommentModel.php

@@ -154,6 +154,24 @@ public function beforeSave($insert)
         }
     }
 
+    /**
+     * This method is called at the end of inserting or updating a record.
+     *
+     * @param bool $insert
+     * @param array $changedAttributes
+     */
+    public function afterSave($insert, $changedAttributes)
+    {
+        if (!$insert) {
+            // Mark all the nested comments as `deleted` after the comment was deleted
+            if (array_key_exists('status', $changedAttributes) && $this->status == CommentStatus::DELETED) {
+                self::updateAll(['status' => CommentStatus::DELETED], ['parentId' => $this->id]);
+            }
+        }
+        
+        parent::afterSave($insert, $changedAttributes);
+    }
+
     /**
      * Author relation
      *
@@ -223,21 +241,14 @@ protected static function buildTree(&$data, $rootID = 0)
 
     /**
      * Delete comment.
-     * All nested comments will also be deleted
      *
      * @return boolean whether comment was deleted or not
      */
     public function deleteComment()
     {
         $this->status = CommentStatus::DELETED;
 
-        if ($this->save(false, ['status', 'updatedBy', 'updatedAt'])) {
-            // Mark all the nested comments as deleted
-            self::updateAll(['status' => CommentStatus::DELETED], ['parentId' => $this->id]);
-            return true;
-        }
-
-        return false;
+        return $this->save(false, ['status', 'updatedBy', 'updatedAt']);
     }
 
     /**
@@ -356,4 +367,4 @@ public function getCommentsCount($onlyActiveComments = true)
 
         return $query->count();
     }
-}
+}