monographs: add slug field which regenerates on update

Author: 01zulfi

Notesnook.API/Controllers/MonographsController.cs

@@ -18,27 +18,24 @@ You should have received a copy of the Affero GNU General Public License
 */
 
 using System;
-using System.Collections.Generic;
 using System.Linq;
 using System.Security.Claims;
 using System.Text.Json;
 using System.Threading.Tasks;
 using AngleSharp;
-using AngleSharp.Dom;
 using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Http;
 using Microsoft.AspNetCore.Mvc;
 using Microsoft.Extensions.Logging;
 using MongoDB.Bson;
 using MongoDB.Driver;
-using Notesnook.API.Authorization;
+using NanoidDotNet;
 using Notesnook.API.Models;
 using Notesnook.API.Services;
 using Streetwriters.Common;
 using Streetwriters.Common.Helpers;
 using Streetwriters.Common.Interfaces;
 using Streetwriters.Common.Messages;
-using Streetwriters.Data.Interfaces;
 using Streetwriters.Data.Repositories;
 
 namespace Notesnook.API.Controllers
@@ -95,6 +92,22 @@ private async Task<Monograph> FindMonographAsync(string itemId)
             return await result.FirstOrDefaultAsync();
         }
 
+        private async Task<Monograph> FindMonographBySlugAsync(string slug)
+        {
+            var result = await monographs.Collection.FindAsync(
+                Builders<Monograph>.Filter.Eq("Slug", slug),
+                new FindOptions<Monograph>
+                {
+                    Limit = 1
+                });
+            return await result.FirstOrDefaultAsync();
+        }
+
+        private static string GenerateSlug()
+        {
+            return Nanoid.Generate(size: 24);
+        }
+
         [HttpPost]
         public async Task<IActionResult> PublishAsync([FromQuery] string? deviceId, [FromBody] Monograph monograph)
         {
@@ -120,6 +133,7 @@ public async Task<IActionResult> PublishAsync([FromQuery] string? deviceId, [Fro
                 }
                 monograph.Deleted = false;
                 monograph.ViewCount = 0;
+                monograph.Slug = GenerateSlug();
                 await monographs.Collection.ReplaceOneAsync(
                     CreateMonographFilter(userId, monograph),
                     monograph,
@@ -131,7 +145,8 @@ await monographs.Collection.ReplaceOneAsync(
                 return Ok(new
                 {
                     id = monograph.ItemId,
-                    datePublished = monograph.DatePublished
+                    datePublished = monograph.DatePublished,
+                    slug = monograph.Slug
                 });
             }
             catch (Exception e)
@@ -164,6 +179,7 @@ public async Task<IActionResult> UpdateAsync([FromQuery] string? deviceId, [From
                     monograph.Content = null;
 
                 monograph.DatePublished = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
+                monograph.Slug = GenerateSlug();
                 var result = await monographs.Collection.UpdateOneAsync(
                     CreateMonographFilter(userId, monograph),
                     Builders<Monograph>.Update
@@ -172,6 +188,7 @@ public async Task<IActionResult> UpdateAsync([FromQuery] string? deviceId, [From
                     .Set(m => m.EncryptedContent, monograph.EncryptedContent)
                     .Set(m => m.SelfDestruct, monograph.SelfDestruct)
                     .Set(m => m.Title, monograph.Title)
+                    .Set(m => m.Slug, monograph.Slug)
                     .Set(m => m.Password, monograph.Password)
                 );
                 if (!result.IsAcknowledged) return BadRequest();
@@ -181,7 +198,8 @@ public async Task<IActionResult> UpdateAsync([FromQuery] string? deviceId, [From
                 return Ok(new
                 {
                     id = monograph.ItemId,
-                    datePublished = monograph.DatePublished
+                    datePublished = monograph.DatePublished,
+                    slug = monograph.Slug
                 });
             }
             catch (Exception e)
@@ -208,11 +226,25 @@ public async Task<IActionResult> GetUserMonographsAsync()
             return Ok(userMonographs.Select((m) => m.ItemId ?? m.Id));
         }
 
-        [HttpGet("{id}")]
+        [HttpGet("{slugOrId}")]
         [AllowAnonymous]
-        public async Task<IActionResult> GetMonographAsync([FromRoute] string id)
+        public async Task<IActionResult> GetMonographAsync([FromRoute] string slugOrId)
         {
-            var monograph = await FindMonographAsync(id);
+            var monograph = await FindMonographBySlugAsync(slugOrId);
+
+            if (monograph == null)
+            {
+                monograph = await FindMonographAsync(slugOrId);
+                if (!string.IsNullOrEmpty(monograph?.Slug))
+                {
+                    return NotFound(new
+                    {
+                        error = "invalid_id",
+                        error_description = $"No such monograph found."
+                    });
+                }
+            }
+
             if (monograph == null || monograph.Deleted)
             {
                 return NotFound(new

Notesnook.API/Hubs/SyncV2Hub.cs

@@ -282,6 +282,7 @@ private async Task<SyncV2Metadata> HandleRequestFetch(string deviceId, bool incl
                         Password = m.Password,
                         SelfDestruct = m.SelfDestruct,
                         Title = m.Title,
+                        Slug = m.Slug,
                         ItemId = m.ItemId ?? m.Id.ToString(),
                         ViewCount = m.ViewCount
                     }).ToListAsync();

Notesnook.API/Models/Monograph.cs

@@ -56,6 +56,10 @@ public Monograph()
         [JsonPropertyName("title")]
         public string? Title { get; set; }
 
+        [JsonPropertyName("slug")]
+        [MessagePack.Key("slug")]
+        public string? Slug { get; set; }
+
         [JsonPropertyName("userId")]
         public string? UserId { get; set; }
 

Notesnook.API/Models/MonographMetadata.cs

@@ -37,6 +37,9 @@ public required string ItemId
         [JsonPropertyName("title")]
         public string? Title { get; set; }
 
+        [JsonPropertyName("slug")]
+        public string? Slug { get; set; }
+
         [JsonPropertyName("selfDestruct")]
         public bool SelfDestruct { get; set; }