From ccb4b8a49d88d135fc96268a6dda997407119718 Mon Sep 17 00:00:00 2001 From: Yanlong Wang Date: Wed, 13 Nov 2024 00:39:07 +0800 Subject: [PATCH] fix: potential invalid html --- backend/functions/src/services/jsdom.ts | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/backend/functions/src/services/jsdom.ts b/backend/functions/src/services/jsdom.ts index 7036e65..08ae9df 100644 --- a/backend/functions/src/services/jsdom.ts +++ b/backend/functions/src/services/jsdom.ts @@ -46,7 +46,10 @@ export class JSDomControl extends AsyncService { if (options?.withShadowDom && snapshot.shadowExpanded) { sourceHTML = snapshot.shadowExpanded; } - const jsdom = this.linkedom.parseHTML(sourceHTML); + let jsdom = this.linkedom.parseHTML(sourceHTML); + if (!jsdom.window.document.documentElement) { + jsdom = this.linkedom.parseHTML(`${sourceHTML}`); + } const allNodes: Node[] = []; jsdom.window.document.querySelectorAll('svg').forEach((x) => x.innerHTML = ''); if (options?.withIframe) {