84 lines
1.9 KiB
TypeScript
84 lines
1.9 KiB
TypeScript
import { parseFragment, DefaultTreeDocumentFragment } from 'parse5';
|
|
import { URL } from 'url';
|
|
import { urlRegex } from './prelude';
|
|
|
|
export function fromHtml(html: string): string {
|
|
if (html == null) return null;
|
|
|
|
const dom = parseFragment(html) as DefaultTreeDocumentFragment;
|
|
|
|
let text = '';
|
|
|
|
for (const n of dom.childNodes) {
|
|
analyze(n);
|
|
}
|
|
|
|
return text.trim();
|
|
|
|
function getText(node: any): string {
|
|
if (node.nodeName == '#text') return node.value;
|
|
|
|
if (node.childNodes) {
|
|
return node.childNodes.map((n: any) => getText(n)).join('');
|
|
}
|
|
|
|
return '';
|
|
}
|
|
|
|
function analyze(node: any) {
|
|
switch (node.nodeName) {
|
|
case '#text':
|
|
text += node.value;
|
|
break;
|
|
|
|
case 'br':
|
|
text += '\n';
|
|
break;
|
|
|
|
case 'a':
|
|
const txt = getText(node);
|
|
const rel = node.attrs.find((x: any) => x.name == 'rel');
|
|
const href = node.attrs.find((x: any) => x.name == 'href');
|
|
const isHashtag = rel && rel.value.match('tag') !== null;
|
|
|
|
// ハッシュタグ / hrefがない / txtがURL
|
|
if (isHashtag || !href || href.value == txt) {
|
|
text += isHashtag || txt.match(urlRegex) ? txt : `<${txt}>`;
|
|
// メンション
|
|
} else if (txt.startsWith('@') && !(rel && rel.value.match(/^me /))) {
|
|
const part = txt.split('@');
|
|
|
|
if (part.length == 2) {
|
|
//#region ホスト名部分が省略されているので復元する
|
|
const acct = `${txt}@${(new URL(href.value)).hostname}`;
|
|
text += acct;
|
|
//#endregion
|
|
} else if (part.length == 3) {
|
|
text += txt;
|
|
}
|
|
// その他
|
|
} else {
|
|
text += `[${txt}](${href.value})`;
|
|
}
|
|
break;
|
|
|
|
case 'p':
|
|
text += '\n\n';
|
|
if (node.childNodes) {
|
|
for (const n of node.childNodes) {
|
|
analyze(n);
|
|
}
|
|
}
|
|
break;
|
|
|
|
default:
|
|
if (node.childNodes) {
|
|
for (const n of node.childNodes) {
|
|
analyze(n);
|
|
}
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
}
|