From 18823a22ae3b2315442b7efac7c3c31d299ea7e1 Mon Sep 17 00:00:00 2001
From: Thomas Steiner <tomac@google.com>
Date: Fri, 20 Jun 2025 16:28:59 +0200
Subject: [PATCH] Add Language Detector API support as a progressive
 enhancement

---
 app/components/publish/PublishWidget.vue | 41 ++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/app/components/publish/PublishWidget.vue b/app/components/publish/PublishWidget.vue
index 84f2ea6b..567953f5 100644
--- a/app/components/publish/PublishWidget.vue
+++ b/app/components/publish/PublishWidget.vue
@@ -1,6 +1,7 @@
 <script setup lang="ts">
 import type { DraftItem } from '#shared/types'
 import type { mastodon } from 'masto'
+import { PublishLanguagePicker } from '#components'
 import { EditorContent } from '@tiptap/vue-3'
 import stringLength from 'string-length'
 
@@ -236,6 +237,45 @@ function stopQuestionMarkPropagation(e: KeyboardEvent) {
   if (e.key === '?')
     e.stopImmediatePropagation()
 }
+
+const languageDetectorInGlobalThis = 'LanguageDetector' in globalThis
+let supportsLanguageDetector = languageDetectorInGlobalThis && await (globalThis as any).LanguageDetector.availability() === 'available'
+let languageDetector: { detect: (arg0: string) => any }
+// If the API is supported, but the model not loaded yet…
+if (languageDetectorInGlobalThis && !supportsLanguageDetector) {
+  // …trigger the model download
+  (globalThis as any).LanguageDetector.create().then((_languageDetector: { detect: (arg0: string) => any }) => {
+    supportsLanguageDetector = true
+    languageDetector = _languageDetector
+  })
+}
+
+function countLetters(text: string) {
+  const segmenter = new Intl.Segmenter('und', { granularity: 'grapheme' })
+  const letters = [...segmenter.segment(text)]
+  return letters.length
+}
+
+async function detectLanguage() {
+  if (!supportsLanguageDetector) {
+    return
+  }
+  if (!languageDetector) {
+    languageDetector = await (globalThis as any).LanguageDetector.create()
+  }
+  const text = htmlToText(editor.value?.getHTML() || '')
+  if (!text || countLetters(text) <= 5) {
+    draft.value.params.language = preferredLanguage.value
+    return
+  }
+  try {
+    const detectedLanguage = (await languageDetector.detect(text))[0].detectedLanguage
+    draft.value.params.language = detectedLanguage === 'und' ? preferredLanguage.value : detectedLanguage.substring(0, 2)
+  }
+  catch {
+    draft.value.params.language = preferredLanguage.value
+  }
+}
 </script>
 
 <template>
@@ -310,6 +350,7 @@ function stopQuestionMarkPropagation(e: KeyboardEvent) {
                 }"
                 @keydown="stopQuestionMarkPropagation"
                 @keydown.esc.prevent="editor?.commands.blur()"
+                @keyup="detectLanguage"
               />
             </div>