feat: Add check existence feature to bookmark and highlight service.

1 files changed, 17 insertions, 0 deletions

diff --git a/BackEnd/Timeline/Services/HighlightTimelineService.cs b/BackEnd/Timeline/Services/HighlightTimelineService.cs
index bf0aac91..557478c7 100644
--- a/BackEnd/Timeline/Services/HighlightTimelineService.cs
+++ b/BackEnd/Timeline/Services/HighlightTimelineService.cs
@@ -32,6 +32,15 @@ namespace Timeline.Services
         Task<List<long>> GetHighlightTimelines();

 

         /// <summary>

+        /// Check if a timeline is highlight timeline.

+        /// </summary>

+        /// <param name="timelineId">Timeline id.</param>

+        /// <param name="checkTimelineExistence">If true it will throw if timeline does not exist.</param>

+        /// <returns>True if timeline is highlight. Otherwise false.</returns>

+        /// <exception cref="TimelineNotExistException">Thrown when timeline does not exist and <paramref name="checkTimelineExistence"/> is true.</exception>

+        Task<bool> IsHighlightTimeline(long timelineId, bool checkTimelineExistence = true);

+

+        /// <summary>

         /// Add a timeline to highlight list.

         /// </summary>

         /// <param name="timelineId">The timeline id.</param>

@@ -172,5 +181,13 @@ namespace Timeline.Services
 

             await transaction.CommitAsync();

         }

+

+        public async Task<bool> IsHighlightTimeline(long timelineId, bool checkTimelineExistence = true)

+        {

+            if (checkTimelineExistence && !await _timelineService.CheckExistence(timelineId))

+                throw new TimelineNotExistException(timelineId);

+

+            return await _database.HighlightTimelines.AnyAsync(t => t.TimelineId == timelineId);

+        }

     }

 }