Add support for non-image attachments

pull/4/head
Alexey Golub 7 years ago
parent a209252259
commit d17fef6721

@ -8,14 +8,14 @@
public string FileName { get; }
public long ContentLength { get; }
public bool IsImage { get; }
public Attachment(string id, string url, string fileName, long contentLength)
public Attachment(string id, string url, string fileName, bool isImage)
{
Id = id;
Url = url;
FileName = fileName;
ContentLength = contentLength;
IsImage = isImage;
}
}
}

@ -37,9 +37,9 @@ namespace DiscordChatExporter.Services
string attachmentId = attachmentJson.Value<string>("id");
string attachmentUrl = attachmentJson.Value<string>("url");
string attachmentFileName = attachmentJson.Value<string>("filename");
long attachmentContentLength = attachmentJson.Value<long>("size");
bool attachmentIsImage = attachmentJson["width"] != null;
var attachment = new Attachment(attachmentId, attachmentUrl, attachmentFileName, attachmentContentLength);
var attachment = new Attachment(attachmentId, attachmentUrl, attachmentFileName, attachmentIsImage);
attachments.Add(attachment);
}

@ -164,11 +164,22 @@ namespace DiscordChatExporter.Services
// Attachments
foreach (var attachment in message.Attachments)
{
messageBodyHtml.AppendChild(
HtmlNode.CreateNode("<div class=\"msg-attachment\">" +
$"<a href=\"{attachment.Url}\">" +
$"<img class=\"msg-attachment\" src=\"{attachment.Url}\" />" +
"</a></div>"));
if (attachment.IsImage)
{
messageBodyHtml.AppendChild(
HtmlNode.CreateNode("<div class=\"msg-attachment\">" +
$"<a href=\"{attachment.Url}\">" +
$"<img class=\"msg-attachment\" src=\"{attachment.Url}\" />" +
"</a></div>"));
}
else
{
messageBodyHtml.AppendChild(
HtmlNode.CreateNode("<div class=\"msg-attachment\">" +
$"<a href=\"{attachment.Url}\">" +
$"Attachment: {attachment.FileName}" +
"</a></div>"));
}
}
}
}

Loading…
Cancel
Save