From 1a44a615ad79054218d493f8bc4bf8563201002b Mon Sep 17 00:00:00 2001
From: crupest <crupest@outlook.com>
Date: Thu, 7 Jan 2021 20:48:47 +0800
Subject: feat: Add check existence feature to bookmark and highlight service.

---
 .../Timeline/Services/BookmarkTimelineService.cs   | 23 ++++++++++++++++++++++
 1 file changed, 23 insertions(+)

(limited to 'BackEnd/Timeline/Services/BookmarkTimelineService.cs')

diff --git a/BackEnd/Timeline/Services/BookmarkTimelineService.cs b/BackEnd/Timeline/Services/BookmarkTimelineService.cs
index 4c8bfdae..4930686e 100644
--- a/BackEnd/Timeline/Services/BookmarkTimelineService.cs
+++ b/BackEnd/Timeline/Services/BookmarkTimelineService.cs
@@ -33,6 +33,18 @@ namespace Timeline.Services
         /// <exception cref="UserNotExistException">Thrown when user does not exist.</exception>
         Task<List<long>> GetBookmarks(long userId);
 
+        /// <summary>
+        /// Check if a timeline is a bookmark.
+        /// </summary>
+        /// <param name="userId">The user id.</param>
+        /// <param name="timelineId">Timeline id.</param>
+        /// <param name="checkUserExistence">If true it will throw when user does not exist.</param>
+        /// <param name="checkTimelineExistence">If true it will throw when timeline does not exist.</param>
+        /// <returns>True if timeline is a bookmark. Otherwise false.</returns>
+        /// <exception cref="UserNotExistException">Throw if user does not exist and <paramref name="checkUserExistence"/> is true.</exception>
+        /// <exception cref="TimelineNotExistException">Thrown if timeline does not exist and <paramref name="checkTimelineExistence"/> is true.</exception>
+        Task<bool> IsBookmark(long userId, long timelineId, bool checkUserExistence = true, bool checkTimelineExistence = true);
+
         /// <summary>
         /// Add a bookmark to tail to a user.
         /// </summary>
@@ -110,6 +122,17 @@ namespace Timeline.Services
             return entities.Select(e => e.TimelineId).ToList();
         }
 
+        public async Task<bool> IsBookmark(long userId, long timelineId, bool checkUserExistence = true, bool checkTimelineExistence = true)
+        {
+            if (checkUserExistence && !await _userService.CheckUserExistence(userId))
+                throw new UserNotExistException(userId);
+
+            if (checkTimelineExistence && !await _timelineService.CheckExistence(timelineId))
+                throw new TimelineNotExistException(timelineId);
+
+            return await _database.BookmarkTimelines.AnyAsync(b => b.TimelineId == timelineId && b.UserId == userId);
+        }
+
         public async Task MoveBookmark(long userId, long timelineId, long newPosition)
         {
             if (!await _userService.CheckUserExistence(userId))
-- 
cgit v1.2.3