Replacing/removing comments and updating the thread view

Author: StanleyGoldman

src/GitHub.InlineReviews/Services/PullRequestSession.cs

@@ -173,6 +173,8 @@ await service.DeleteComment(
                 RepositoryOwner,
                 User,
                 number);
+
+            await RemoveComment(number);
         }
 
         /// <inheritdoc/>
@@ -186,6 +188,7 @@ public async Task<IPullRequestReviewCommentModel> EditComment(
                 number,
                 body);
 
+            await ReplaceComment(model);
             return model;
         }
 
@@ -311,6 +314,24 @@ async Task AddComment(IPullRequestReviewCommentModel comment)
             await Update(PullRequest);
         }
 
+        async Task ReplaceComment(IPullRequestReviewCommentModel comment)
+        {
+            PullRequest.ReviewComments = PullRequest.ReviewComments
+                .Select(model => model.Id == comment.Id ? comment: model)
+                .ToList();
+
+            await Update(PullRequest);
+        }
+
+        async Task RemoveComment(int commentId)
+        {
+            PullRequest.ReviewComments = PullRequest.ReviewComments
+                .Where(model => model.Id != commentId)
+                .ToList();
+
+            await Update(PullRequest);
+        }
+
         async Task AddReview(IPullRequestReviewModel review)
         {
             PullRequest.Reviews = PullRequest.Reviews