fix: don't rend the HTML tags

uetchy · Oct 4, 2022 · 5323ce0 · 5323ce0
1 parent fac1ccc
commit 5323ce0
Showing 1 changed file with 16 additions and 6 deletions.
diff --git a/PolyglotSafariExtension/Sources/content.ts b/PolyglotSafariExtension/Sources/content.ts
@@ -84,12 +84,9 @@ function translationHandler(message: ReceivedTranslation): void {
 
   const args = {
     sourceLanguage: message.sourceLanguage || null,
-    translation: message.translation.replace(/\n/g, "<br/>"),
-    transliteration: message.transliteration.replace(/\n/g, "<br/>"),
-    sourceTransliteration: message.sourceTransliteration.replace(
-      /\n/g,
-      "<br/>"
-    ),
+    translation: convertHTMLToDisplayText(message.translation),
+    transliteration: convertHTMLToDisplayText(message.transliteration),
+    sourceTransliteration: convertHTMLToDisplayText(message.sourceTransliteration),
     synonyms: message.synonyms
       ? message.synonyms.map((synonym) => ({
           pos: synonym.pos,
@@ -203,4 +200,17 @@ function getSelectedText(): string | undefined {
   }
 }
 
+function convertHTMLToDisplayText(htmlString: string): string {
+  let encodedString = htmlString.replace(/&/g, '&amp;');
+  encodedString = encodedString.replace(/</g, '&lt;');
+  encodedString = encodedString.replace(/>/g, '&gt;');
+  encodedString = encodedString.replace(/"/g, '&quot;');
+
+  return convertNewlineToHTML(encodedString)
+}
+
+function convertNewlineToHTML(text: string): string {
+  return text.replace(/\n/g, "<br/>")
+}
+
 init();