inkch/txtdot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix(server): title in text output

Browse Source
This commit is contained in:
Artemy 2024-05-14 13:11:59 +03:00
parent dba42c64b4
commit 1a48066684
2 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
packages/server/src/distributor.ts
View File

@ -75,13 +75,17 @@ export class Distributor {
const purify = DOMPurify(dom); const purify = DOMPurify(dom);
const content = purify.sanitize(dom.document.toString()); const content = purify.sanitize(dom.document.toString());
const title = output.title || dom.document.title;
const lang = output.lang || dom.document.documentElement.lang;
const textContent =
html2text(output, dom.document, title) ||
'Text output cannot be generated.';
return { return {
content, content,
textContent: textContent,
html2text(output, dom.document) || 'Text output cannot be generated.', title,
title: output.title || dom.document.title, lang,
lang: output.lang || dom.document.documentElement.lang,
}; };
} }

11
packages/server/src/utils/html2text.ts
View File

@ -1,9 +1,14 @@
import { EngineOutput } from '@txtdot/sdk/dist/types/handler'; import { EngineOutput } from '@txtdot/sdk/dist/types/handler';
import config from '../config'; import config from '../config';
export function html2text(output: EngineOutput, doc: Document) { function setTitle(body: string | null, title: string) {
if (!body) return null;
return `${title.toUpperCase()}\n${'='.repeat(title.length)}\n\n${body}`;
}
export function html2text(output: EngineOutput, doc: Document, title: string) {
if (output.textContent) return output.textContent; if (output.textContent) return output.textContent;
else if (config.plugin.html2text) else if (config.plugin.html2text)
return config.plugin.html2text(output.content); return setTitle(config.plugin.html2text(output.content), title);
else return doc.documentElement.textContent; else return setTitle(doc.documentElement.textContent, title);
} }