Smart Filters & Dashboard Customization (#2282)

Co-authored-by: Robbie Davis <robbie@therobbiedavis.com>

API/Controllers/AccountController.cs

@@ -8,6 +8,7 @@ using API.Data;
 using API.Data.Repositories;
 using API.DTOs;
 using API.DTOs.Account;
+using API.DTOs.Dashboard;
 using API.DTOs.Email;
 using API.Entities;
 using API.Entities.Enums;
@@ -1035,4 +1036,123 @@ public class AccountController : BaseApiController
         return Ok(origin + "/" + baseUrl + "api/opds/" + user!.ApiKey);
 
     }
+
+    /// <summary>
+    /// Returns the layout of the user's dashboard
+    /// </summary>
+    /// <returns></returns>
+    [HttpGet("dashboard")]
+    public async Task<ActionResult<IEnumerable<DashboardStreamDto>>> GetDashboardLayout(bool visibleOnly = true)
+    {
+        var streams = await _unitOfWork.UserRepository.GetDashboardStreams(User.GetUserId(), visibleOnly);
+        return Ok(streams);
+    }
+
+    /// <summary>
+    /// Creates a Dashboard Stream from a SmartFilter and adds it to the user's dashboard as visible
+    /// </summary>
+    /// <param name="smartFilterId"></param>
+    /// <returns></returns>
+    [HttpPost("add-dashboard-stream")]
+    public async Task<ActionResult<DashboardStreamDto>> AddDashboard([FromQuery] int smartFilterId)
+    {
+        var user = await _unitOfWork.UserRepository.GetUserByIdAsync(User.GetUserId(), AppUserIncludes.DashboardStreams);
+        if (user == null) return Unauthorized();
+
+        var smartFilter = await _unitOfWork.AppUserSmartFilterRepository.GetById(smartFilterId);
+        if (smartFilter == null) return NoContent();
+
+        var stream = user?.DashboardStreams.FirstOrDefault(d => d.SmartFilter?.Id == smartFilterId);
+        if (stream != null) return BadRequest("There is an existing stream with this Filter");
+
+        var maxOrder = user!.DashboardStreams.Max(d => d.Order);
+        var createdStream = new AppUserDashboardStream()
+        {
+            Name = smartFilter.Name,
+            IsProvided = false,
+            StreamType = DashboardStreamType.SmartFilter,
+            Visible = true,
+            Order = maxOrder + 1,
+            SmartFilter = smartFilter
+        };
+
+        user.DashboardStreams.Add(createdStream);
+        _unitOfWork.UserRepository.Update(user);
+        await _unitOfWork.CommitAsync();
+
+        var ret = new DashboardStreamDto()
+        {
+            Name = createdStream.Name,
+            IsProvided = createdStream.IsProvided,
+            Visible = createdStream.Visible,
+            Order = createdStream.Order,
+            SmartFilterEncoded = smartFilter.Filter,
+            StreamType = createdStream.StreamType
+        };
+
+
+        await _eventHub.SendMessageToAsync(MessageFactory.DashboardUpdate, MessageFactory.DashboardUpdateEvent(user.Id),
+            User.GetUserId());
+        return Ok(ret);
+    }
+
+    /// <summary>
+    /// Updates the visibility of a dashboard stream
+    /// </summary>
+    /// <param name="dto"></param>
+    /// <returns></returns>
+    [HttpPost("update-dashboard-stream")]
+    public async Task<ActionResult> UpdateDashboardStream(DashboardStreamDto dto)
+    {
+        var stream = await _unitOfWork.UserRepository.GetDashboardStream(dto.Id);
+        if (stream == null) return BadRequest();
+        stream.Visible = dto.Visible;
+
+        _unitOfWork.UserRepository.Update(stream);
+        await _unitOfWork.CommitAsync();
+        var userId = User.GetUserId();
+        await _eventHub.SendMessageToAsync(MessageFactory.DashboardUpdate, MessageFactory.DashboardUpdateEvent(userId),
+            userId);
+        return Ok();
+    }
+
+    /// <summary>
+    /// Updates the position of a dashboard stream
+    /// </summary>
+    /// <param name="dto"></param>
+    /// <returns></returns>
+    [HttpPost("update-dashboard-position")]
+    public async Task<ActionResult> UpdateDashboardStreamPosition(UpdateDashboardStreamPositionDto dto)
+    {
+        var user = await _unitOfWork.UserRepository.GetUserByIdAsync(User.GetUserId(),
+            AppUserIncludes.DashboardStreams);
+        var stream = user?.DashboardStreams.FirstOrDefault(d => d.Id == dto.DashboardStreamId);
+        if (stream == null) return BadRequest();
+        if (stream.Order == dto.ToPosition) return Ok();
+
+        var list = user!.DashboardStreams.ToList();
+        ReorderItems(list, stream.Id, dto.ToPosition);
+        user.DashboardStreams = list;
+
+        _unitOfWork.UserRepository.Update(user);
+        await _unitOfWork.CommitAsync();
+        await _eventHub.SendMessageToAsync(MessageFactory.DashboardUpdate, MessageFactory.DashboardUpdateEvent(user.Id),
+            user.Id);
+        return Ok();
+    }
+
+    private static void ReorderItems(List<AppUserDashboardStream> items, int itemId, int toPosition)
+    {
+        var item = items.Find(r => r.Id == itemId);
+        if (item != null)
+        {
+            items.Remove(item);
+            items.Insert(toPosition, item);
+        }
+
+        for (var i = 0; i < items.Count; i++)
+        {
+            items[i].Order = i;
+        }
+    }
 }