From c3369742b6e68714d0a7df46a99a0798eb2d6940 Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Tue, 11 Aug 2020 01:48:57 +0800
Subject: Handle post deletion on user deletion correctly.

---
 Timeline/Services/TimelineService.cs | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

(limited to 'Timeline/Services/TimelineService.cs')

diff --git a/Timeline/Services/TimelineService.cs b/Timeline/Services/TimelineService.cs
index ad410d15..283938fb 100644
--- a/Timeline/Services/TimelineService.cs
+++ b/Timeline/Services/TimelineService.cs
@@ -219,6 +219,12 @@ namespace Timeline.Services
         /// </remarks>
         Task DeletePost(string timelineName, long postId);
 
+        /// <summary>
+        /// Delete all posts of the given user. Used when delete a user.
+        /// </summary>
+        /// <param name="userId">The id of the user.</param>
+        Task DeleteAllPostsOfUser(long userId);
+
         /// <summary>
         /// Change member of timeline.
         /// </summary>
@@ -776,6 +782,35 @@ namespace Timeline.Services
             }
         }
 
+        public async Task DeleteAllPostsOfUser(long userId)
+        {
+            var posts = await _database.TimelinePosts.Where(p => p.AuthorId == userId).ToListAsync();
+
+            var now = _clock.GetCurrentTime();
+
+            var dataTags = new List<string>();
+
+            foreach (var post in posts)
+            {
+                if (post.Content != null)
+                {
+                    if (post.ContentType == TimelinePostContentTypes.Image)
+                    {
+                        dataTags.Add(post.Content);
+                    }
+                    post.Content = null;
+                }
+                post.LastUpdated = now;
+            }
+
+            await _database.SaveChangesAsync();
+
+            foreach (var dataTag in dataTags)
+            {
+                await _dataManager.FreeEntry(dataTag);
+            }
+        }
+
         public async Task ChangeProperty(string timelineName, TimelineChangePropertyRequest newProperties)
         {
             if (timelineName == null)
-- 
cgit v1.2.3