117 lines
No EOL
2.3 KiB
C#
117 lines
No EOL
2.3 KiB
C#
using AngleSharp.Dom;
|
|
using AngleSharp.Html.Dom;
|
|
using Iceshrimp.Backend.Core.Database.Tables;
|
|
|
|
namespace Iceshrimp.Backend.Core.Helpers.LibMfm.Parsing;
|
|
|
|
internal class HtmlParser(IEnumerable<Note.MentionedUser> mentions)
|
|
{
|
|
internal string? ParseNode(INode node)
|
|
{
|
|
if (node.NodeType is NodeType.Text)
|
|
return node.TextContent;
|
|
if (node.NodeType is NodeType.Comment or NodeType.Document)
|
|
return null;
|
|
|
|
switch (node.NodeName)
|
|
{
|
|
case "BR":
|
|
{
|
|
return "\n";
|
|
}
|
|
case "A":
|
|
{
|
|
if (node is HtmlElement el)
|
|
{
|
|
var href = el.GetAttribute("href");
|
|
if (href == null) return $"<plain>{el.TextContent}</plain>";
|
|
|
|
if (el.ClassList.Contains("u-url") && el.ClassList.Contains("mention"))
|
|
{
|
|
var mention = mentions.FirstOrDefault(p => p.Uri == href || p.Url == href);
|
|
return mention != null
|
|
? $"@{mention.Username}@{mention.Host}"
|
|
: $"<plain>{el.TextContent}</plain>";
|
|
}
|
|
|
|
if (el.TextContent == href && (href.StartsWith("http://") || href.StartsWith("https://")))
|
|
return href;
|
|
|
|
return $"[{el.TextContent}]({href})";
|
|
}
|
|
|
|
return node.TextContent;
|
|
}
|
|
case "H1":
|
|
{
|
|
return $"【{ParseChildren(node)}】\n";
|
|
}
|
|
case "B":
|
|
case "STRONG":
|
|
{
|
|
return $"**{ParseChildren(node)}**";
|
|
}
|
|
case "SMALL":
|
|
{
|
|
return $"<small>{ParseChildren(node)}</small>";
|
|
}
|
|
case "S":
|
|
case "DEL":
|
|
{
|
|
return $"~~{ParseChildren(node)}~~";
|
|
}
|
|
case "I":
|
|
case "EM":
|
|
{
|
|
return $"*{ParseChildren(node)}*";
|
|
}
|
|
case "PRE":
|
|
{
|
|
return node.ChildNodes is [{ NodeName: "CODE" }]
|
|
? $"\n```\n{string.Join(null, node.ChildNodes[0].TextContent)}\n```\n"
|
|
: ParseChildren(node);
|
|
}
|
|
case "CODE":
|
|
{
|
|
return $"`{ParseChildren(node)}`";
|
|
}
|
|
case "BLOCKQUOTE":
|
|
{
|
|
return node.TextContent.Length > 0
|
|
? $"\n> {string.Join("\n> ", node.TextContent.Split("\n"))}"
|
|
: null;
|
|
}
|
|
|
|
case "P":
|
|
case "H2":
|
|
case "H3":
|
|
case "H4":
|
|
case "H5":
|
|
case "H6":
|
|
{
|
|
return $"\n\n{ParseChildren(node)}";
|
|
}
|
|
|
|
case "DIV":
|
|
case "HEADER":
|
|
case "FOOTER":
|
|
case "ARTICLE":
|
|
case "LI":
|
|
case "DT":
|
|
case "DD":
|
|
{
|
|
return $"\n{ParseChildren(node)}";
|
|
}
|
|
|
|
default:
|
|
{
|
|
return ParseChildren(node);
|
|
}
|
|
}
|
|
}
|
|
|
|
private string ParseChildren(INode node)
|
|
{
|
|
return string.Join(null, node.ChildNodes.Select(ParseNode));
|
|
}
|
|
} |