Post notification when view is being dismissed instead of immediately.

Author: ScoutHarris

WordPress/Classes/ViewRelated/Reader/Comments/ReaderCommentsViewController.h

@@ -31,4 +31,7 @@ typedef NS_ENUM(NSUInteger, ReaderCommentsSource) {
 /// Navigates to the specified comment when the view appears
 @property (nonatomic, strong) NSNumber *navigateToCommentID;
 
+
+// Comment moderation support.
+@property (nonatomic, assign, readwrite) BOOL commentModified;
 @end

WordPress/Classes/ViewRelated/Reader/Comments/ReaderCommentsViewController.m

@@ -144,6 +144,7 @@ - (void)viewDidLoad
 {
     [super viewDidLoad];
     self.view.backgroundColor = [UIColor murielBasicBackground];
+    self.commentModified = NO;
 
     [self checkIfLoggedIn];
 
@@ -195,6 +196,11 @@ - (void)viewWillDisappear:(BOOL)animated
 {
     [super viewWillDisappear:animated];
     [self dismissNotice];
+    
+    if (self.commentModified) {
+        // Don't post the notification until the view is being dismissed to avoid purging cached comments prematurely.
+        [self postCommentModifiedNotification];
+    }
 
 #pragma clang diagnostic push
 #pragma clang diagnostic ignored "-Wunused-result"

WordPress/Classes/ViewRelated/Reader/Comments/ReaderCommentsViewController.swift

@@ -142,6 +142,12 @@ extension NSNotification.Name {
         WPAnalytics.trackReader(.readerArticleCommentsOpened, properties: properties)
     }
 
+    // MARK: - Notification
+
+    @objc func postCommentModifiedNotification() {
+        NotificationCenter.default.post(name: .ReaderCommentModifiedNotification, object: nil)
+    }
+
 }
 
 // MARK: - Popover Presentation Delegate
@@ -223,7 +229,7 @@ private extension ReaderCommentsViewController {
             CommentAnalytics.trackCommentEdited(comment: comment)
 
             self?.commentService.uploadComment(comment, success: {
-                NotificationCenter.default.post(name: .ReaderCommentModifiedNotification, object: nil)
+                self?.commentModified = true
 
                 // update the thread again in case the approval status changed.
                 tableView.reloadRows(at: [indexPath], with: .automatic)
@@ -239,7 +245,7 @@ private extension ReaderCommentsViewController {
 
     func moderateComment(_ comment: Comment, status: CommentStatusType, handler: WPTableViewHandler) {
         let successBlock: (String) -> Void = { [weak self] noticeText in
-            NotificationCenter.default.post(name: .ReaderCommentModifiedNotification, object: nil)
+            self?.commentModified = true
 
             // Adjust the ReaderPost's comment count.
             if let post = self?.post, let commentCount = post.commentCount?.intValue {