Write embed videos in JSON export

Closes Tyrrrz#1120
cooljeanius · Nov 8, 2023 · bf8ed97 · bf8ed97
1 parent 0d18c90
commit bf8ed97
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/DiscordChatExporter.Core/Exporting/JsonMessageWriter.cs b/DiscordChatExporter.Core/Exporting/JsonMessageWriter.cs
@@ -143,6 +143,31 @@ private async ValueTask WriteUserAsync(User user, CancellationToken cancellation
         await _writer.FlushAsync(cancellationToken);
     }
 
+    private async ValueTask WriteEmbedVideoAsync(
+        EmbedVideo embedVideo,
+        CancellationToken cancellationToken = default
+    )
+    {
+        _writer.WriteStartObject();
+
+        if (!string.IsNullOrWhiteSpace(embedVideo.Url))
+        {
+            _writer.WriteString(
+                "url",
+                await Context.ResolveAssetUrlAsync(
+                    embedVideo.ProxyUrl ?? embedVideo.Url,
+                    cancellationToken
+                )
+            );
+        }
+
+        _writer.WriteNumber("width", embedVideo.Width);
+        _writer.WriteNumber("height", embedVideo.Height);
+
+        _writer.WriteEndObject();
+        await _writer.FlushAsync(cancellationToken);
+    }
+
     private async ValueTask WriteEmbedFooterAsync(
         EmbedFooter embedFooter,
         CancellationToken cancellationToken = default
@@ -224,6 +249,12 @@ await FormatMarkdownAsync(embed.Description ?? "", cancellationToken)
             await WriteEmbedImageAsync(embed.Image, cancellationToken);
         }
 
+        if (embed.Video is not null)
+        {
+            _writer.WritePropertyName("video");
+            await WriteEmbedVideoAsync(embed.Video, cancellationToken);
+        }
+
         if (embed.Footer is not null)
         {
             _writer.WritePropertyName("footer");