elk/composables/content.ts

210 lines
5.9 KiB
TypeScript
Raw Normal View History

2022-11-21 07:14:07 +00:00
import type { Emoji } from 'masto'
import type { Node } from 'ultrahtml'
import { TEXT_NODE, parse, render, walkSync } from 'ultrahtml'
import type { VNode } from 'vue'
2022-11-24 03:42:03 +00:00
import { Fragment, h, isVNode } from 'vue'
2022-11-20 21:21:53 +00:00
import { RouterLink } from 'vue-router'
2022-11-24 03:42:03 +00:00
import ContentCode from '~/components/content/ContentCode.vue'
2022-11-30 07:08:10 +00:00
import AccountHoverWrapper from '~/components/account/AccountHoverWrapper.vue'
2022-11-20 21:21:53 +00:00
function handleMention(el: Node) {
2022-11-20 21:21:53 +00:00
// Redirect mentions to the user page
if (el.name === 'a' && el.attributes.class?.includes('mention')) {
const href = el.attributes.href
2022-11-20 21:21:53 +00:00
if (href) {
const matchUser = href.match(UserLinkRE)
2022-11-20 21:21:53 +00:00
if (matchUser) {
const [, server, username] = matchUser
2022-11-30 07:08:10 +00:00
const handle = `@${username}@${server.replace(/(.+\.)(.+\..+)/, '$2')}`
el.attributes.href = `/${server}/@${username}`
2022-11-30 07:08:10 +00:00
return h(AccountHoverWrapper, { handle, class: 'inline-block' }, () => nodeToVNode(el))
2022-11-20 21:21:53 +00:00
}
const matchTag = href.match(TagLinkRE)
2022-11-20 21:21:53 +00:00
if (matchTag) {
const [, , name] = matchTag
el.attributes.href = `/${currentServer.value}/tags/${name}`
2022-11-20 21:21:53 +00:00
}
}
}
2022-11-24 03:42:03 +00:00
return undefined
}
function handleCodeBlock(el: Node) {
if (el.name === 'pre' && el.children[0]?.name === 'code') {
const codeEl = el.children[0] as Node
const classes = codeEl.attributes.class as string
const lang = classes?.split(/\s/g).find(i => i.startsWith('language-'))?.replace('language-', '')
const code = codeEl.children[0] ? treeToText(codeEl.children[0]) : ''
return h(ContentCode, { lang, code: encodeURIComponent(code) })
2022-11-24 03:42:03 +00:00
}
}
function handleNode(el: Node) {
return handleCodeBlock(el) || handleMention(el) || el
2022-11-20 21:21:53 +00:00
}
2022-12-05 00:30:20 +00:00
const decoder = document.createElement('textarea')
function decode(text: string) {
decoder.innerHTML = text
return decoder.value
}
/**
* Parse raw HTML form Mastodon server to AST,
* with interop of custom emojis and inline Markdown syntax
*/
export function parseMastodonHTML(html: string, customEmojis: Record<string, Emoji> = {}) {
let processed = html
// custom emojis
2022-11-24 03:42:03 +00:00
.replace(/:([\w-]+?):/g, (_, name) => {
const emoji = customEmojis[name]
if (emoji)
2022-11-30 06:50:47 +00:00
return `<img src="${emoji.url}" alt=":${name}:" class="custom-emoji" data-emoji-id="${name}" />`
2022-11-24 03:42:03 +00:00
return `:${name}:`
})
// handle code blocks
.replace(/>(```|~~~)(\w*)([\s\S]+?)\1/g, (_1, _2, lang, raw) => {
const code = htmlToText(raw)
const classes = lang ? ` class="language-${lang}"` : ''
return `><pre><code${classes}>${code}</code></pre>`
})
2022-11-24 03:42:03 +00:00
walkSync(parse(processed), (node) => {
if (node.type !== TEXT_NODE)
return
const replacements = [
[/\*\*\*(.*?)\*\*\*/g, '<b><em>$1</em></b>'],
[/\*\*(.*?)\*\*/g, '<b>$1</b>'],
[/\*(.*?)\*/g, '<em>$1</em>'],
[/~~(.*?)~~/g, '<del>$1</del>'],
[/`([^`]+?)`/g, '<code>$1</code>'],
2022-12-05 00:30:20 +00:00
[/&[^;]+;/g, (val: string) => decode(val)],
] as any
for (const [re, replacement] of replacements) {
for (const match of node.value.matchAll(re)) {
if (node.loc) {
const start = match.index! + node.loc[0].start
const end = start + match[0].length + node.loc[0].start
processed = processed.slice(0, start) + match[0].replace(re, replacement) + processed.slice(end)
}
else {
processed = processed.replace(match[0], match[0].replace(re, replacement))
}
}
}
})
return parse(processed)
}
export async function convertMastodonHTML(html: string, customEmojis: Record<string, Emoji> = {}) {
const tree = parseMastodonHTML(html, customEmojis)
return await render(tree)
}
/**
* Raw HTML to VNodes
*/
export function contentToVNode(
content: string,
customEmojis: Record<string, Emoji> = {},
): VNode {
const tree = parseMastodonHTML(content, customEmojis)
return h(Fragment, (tree.children as Node[]).map(n => treeToVNode(n)))
2022-11-20 21:21:53 +00:00
}
2022-11-30 07:08:10 +00:00
function nodeToVNode(node: Node): VNode | string | null {
if (node.type === TEXT_NODE)
return node.value
if ('children' in node) {
if (node.name === 'a' && (node.attributes.href?.startsWith('/') || node.attributes.href?.startsWith('.'))) {
node.attributes.to = node.attributes.href
delete node.attributes.href
delete node.attributes.target
2022-11-20 21:21:53 +00:00
return h(
RouterLink as any,
node.attributes,
() => node.children.map(treeToVNode),
2022-11-20 21:21:53 +00:00
)
}
return h(
node.name,
node.attributes,
node.children.map(treeToVNode),
2022-11-20 21:21:53 +00:00
)
}
return null
}
2022-11-24 03:42:03 +00:00
2022-11-30 07:08:10 +00:00
function treeToVNode(
input: Node,
): VNode | string | null {
if (input.type === TEXT_NODE)
2022-11-30 07:08:10 +00:00
return input.value as string
if ('children' in input) {
2022-11-30 07:08:10 +00:00
const node = handleNode(input)
if (node == null)
return null
if (isVNode(node))
return node
return nodeToVNode(node)
}
return null
}
2022-11-25 16:17:15 +00:00
export function htmlToText(html: string) {
const tree = parse(html)
return (tree.children as Node[]).map(n => treeToText(n)).join('').trim()
2022-11-24 03:42:03 +00:00
}
2022-11-30 06:50:47 +00:00
export function treeToText(input: Node): string {
2022-11-24 03:42:03 +00:00
let pre = ''
2022-11-25 16:17:15 +00:00
let body = ''
let post = ''
2022-11-24 03:42:03 +00:00
if (input.type === TEXT_NODE)
2022-11-24 03:42:03 +00:00
return input.value
if (input.name === 'br')
2022-11-24 03:42:03 +00:00
return '\n'
if (['p', 'pre'].includes(input.name))
2022-11-24 03:42:03 +00:00
pre = '\n'
if (input.name === 'code') {
if (input.parent?.name === 'pre') {
const lang = input.attributes.class?.replace('language-', '')
2022-11-25 19:21:53 +00:00
2022-11-30 04:50:29 +00:00
pre = `\`\`\`${lang || ''}\n`
post = '\n```'
}
else {
pre = '`'
post = '`'
}
}
else if (input.name === 'b' || input.name === 'strong') {
2022-11-30 04:50:29 +00:00
pre = '**'
post = '**'
}
else if (input.name === 'i' || input.name === 'em') {
2022-11-30 04:50:29 +00:00
pre = '*'
post = '*'
2022-11-25 16:17:15 +00:00
}
else if (input.name === 'del') {
2022-11-30 06:50:47 +00:00
pre = '~~'
post = '~~'
}
2022-11-25 16:17:15 +00:00
if ('children' in input)
body = (input.children as Node[]).map(n => treeToText(n)).join('')
2022-11-24 03:42:03 +00:00
if (input.name === 'img' && input.attributes.class?.includes('custom-emoji'))
return `:${input.attributes['data-emoji-id']}:`
2022-11-30 06:50:47 +00:00
2022-11-25 16:17:15 +00:00
return pre + body + post
2022-11-24 03:42:03 +00:00
}