Archive/DiscordChatExporter
1
0
Fork 0
mirror of https://github.com/Tyrrrz/DiscordChatExporter.git synced 2026-03-31 09:33:03 -06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Refactor

Browse source
This commit is contained in:
Tyrrrz 2026-02-27 17:39:36 +02:00
parent b660edfe78
commit 4674c517e3
1 changed files with 45 additions and 49 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

94
DiscordChatExporter.Core/Exporting/JsonMessageWriter.cs
View file

@ -119,6 +119,24 @@ internal class JsonMessageWriter(Stream stream, ExportContext context)
await _writer.FlushAsync(cancellationToken); await _writer.FlushAsync(cancellationToken);
} }
private async ValueTask WriteAttachmentAsync(
Attachment attachment,
CancellationToken cancellationToken = default
)
{
_writer.WriteStartObject();
_writer.WriteString("id", attachment.Id.ToString());
_writer.WriteString(
"url",
await Context.ResolveAssetUrlAsync(attachment.Url, cancellationToken)
);
_writer.WriteString("fileName", attachment.FileName);
_writer.WriteNumber("fileSizeBytes", attachment.FileSize.TotalBytes);
_writer.WriteEndObject();
}
private async ValueTask WriteEmbedAuthorAsync( private async ValueTask WriteEmbedAuthorAsync(
EmbedAuthor embedAuthor, EmbedAuthor embedAuthor,
CancellationToken cancellationToken = default CancellationToken cancellationToken = default
@ -334,6 +352,24 @@ internal class JsonMessageWriter(Stream stream, ExportContext context)
_writer.WriteEndObject(); _writer.WriteEndObject();
await _writer.FlushAsync(cancellationToken); await _writer.FlushAsync(cancellationToken);
} }
private async ValueTask WriteStickerAsync(
Sticker sticker,
CancellationToken cancellationToken = default
)
{
_writer.WriteStartObject();
_writer.WriteString("id", sticker.Id.ToString());
_writer.WriteString("name", sticker.Name);
_writer.WriteString("format", sticker.Format.ToString());
_writer.WriteString(
"sourceUrl",
await Context.ResolveAssetUrlAsync(sticker.SourceUrl, cancellationToken)
);
_writer.WriteEndObject();
}
public override async ValueTask WritePreambleAsync( public override async ValueTask WritePreambleAsync(
CancellationToken cancellationToken = default CancellationToken cancellationToken = default
@ -437,19 +473,7 @@ internal class JsonMessageWriter(Stream stream, ExportContext context)
_writer.WriteStartArray("attachments"); _writer.WriteStartArray("attachments");
foreach (var attachment in message.Attachments) foreach (var attachment in message.Attachments)
{ await WriteAttachmentAsync(attachment, cancellationToken);
_writer.WriteStartObject();
_writer.WriteString("id", attachment.Id.ToString());
_writer.WriteString(
"url",
await Context.ResolveAssetUrlAsync(attachment.Url, cancellationToken)
);
_writer.WriteString("fileName", attachment.FileName);
_writer.WriteNumber("fileSizeBytes", attachment.FileSize.TotalBytes);
_writer.WriteEndObject();
}
_writer.WriteEndArray(); _writer.WriteEndArray();
@ -465,19 +489,7 @@ internal class JsonMessageWriter(Stream stream, ExportContext context)
_writer.WriteStartArray("stickers"); _writer.WriteStartArray("stickers");
foreach (var sticker in message.Stickers) foreach (var sticker in message.Stickers)
{ await WriteStickerAsync(sticker, cancellationToken);
_writer.WriteStartObject();
_writer.WriteString("id", sticker.Id.ToString());
_writer.WriteString("name", sticker.Name);
_writer.WriteString("format", sticker.Format.ToString());
_writer.WriteString(
"sourceUrl",
await Context.ResolveAssetUrlAsync(sticker.SourceUrl, cancellationToken)
);
_writer.WriteEndObject();
}
_writer.WriteEndArray(); _writer.WriteEndArray();
@ -527,10 +539,10 @@ internal class JsonMessageWriter(Stream stream, ExportContext context)
if (message.Reference is not null) if (message.Reference is not null)
{ {
_writer.WriteStartObject("reference"); _writer.WriteStartObject("reference");
_writer.WriteString("type", message.Reference.Kind.ToString());
_writer.WriteString("messageId", message.Reference.MessageId?.ToString()); _writer.WriteString("messageId", message.Reference.MessageId?.ToString());
_writer.WriteString("channelId", message.Reference.ChannelId?.ToString()); _writer.WriteString("channelId", message.Reference.ChannelId?.ToString());
_writer.WriteString("guildId", message.Reference.GuildId?.ToString()); _writer.WriteString("guildId", message.Reference.GuildId?.ToString());
_writer.WriteString("type", message.Reference.Kind.ToString());
_writer.WriteEndObject(); _writer.WriteEndObject();
} }
@ -556,18 +568,10 @@ internal class JsonMessageWriter(Stream stream, ExportContext context)
// Forwarded attachments // Forwarded attachments
_writer.WriteStartArray("attachments"); _writer.WriteStartArray("attachments");
foreach (var attachment in message.ForwardedMessage.Attachments) foreach (var attachment in message.ForwardedMessage.Attachments)
{ await WriteAttachmentAsync(attachment, cancellationToken);
_writer.WriteStartObject();
_writer.WriteString("id", attachment.Id.ToString());
_writer.WriteString(
"url",
await Context.ResolveAssetUrlAsync(attachment.Url, cancellationToken)
);
_writer.WriteString("fileName", attachment.FileName);
_writer.WriteNumber("fileSizeBytes", attachment.FileSize.TotalBytes);
_writer.WriteEndObject();
}
_writer.WriteEndArray(); _writer.WriteEndArray();
// Forwarded embeds // Forwarded embeds
@ -578,18 +582,10 @@ internal class JsonMessageWriter(Stream stream, ExportContext context)
// Forwarded stickers // Forwarded stickers
_writer.WriteStartArray("stickers"); _writer.WriteStartArray("stickers");
foreach (var sticker in message.ForwardedMessage.Stickers) foreach (var sticker in message.ForwardedMessage.Stickers)
{ await WriteStickerAsync(sticker, cancellationToken);
_writer.WriteStartObject();
_writer.WriteString("id", sticker.Id.ToString());
_writer.WriteString("name", sticker.Name);
_writer.WriteString("format", sticker.Format.ToString());
_writer.WriteString(
"sourceUrl",
await Context.ResolveAssetUrlAsync(sticker.SourceUrl, cancellationToken)
);
_writer.WriteEndObject();
}
_writer.WriteEndArray(); _writer.WriteEndArray();
_writer.WriteEndObject(); _writer.WriteEndObject();