From 8586d58b8498156d19c2e71805f5085357da2f32 Mon Sep 17 00:00:00 2001
From: Daniel Roe <daniel@roe.dev>
Date: Fri, 25 Nov 2022 00:14:16 +0000
Subject: [PATCH] feat: allow translation of statuses (#76)

---
 components/status/StatusActions.vue | 18 +++++++++++
 components/status/StatusBody.vue    |  3 +-
 composables/translate.ts            | 47 +++++++++++++++++++++++++++++
 nuxt.config.ts                      |  3 ++
 4 files changed, 70 insertions(+), 1 deletion(-)
 create mode 100644 composables/translate.ts

diff --git a/components/status/StatusActions.vue b/components/status/StatusActions.vue
index b454748c..9746e9ec 100644
--- a/components/status/StatusActions.vue
+++ b/components/status/StatusActions.vue
@@ -22,6 +22,7 @@ const isLoading = $ref({
   favourited: false,
   bookmarked: false,
   pinned: false,
+  translation: false,
 })
 async function toggleStatusAction(action: 'reblogged' | 'favourited' | 'bookmarked' | 'pinned', newStatus: Promise<Status>) {
   // Optimistic update
@@ -59,6 +60,13 @@ const togglePin = async () => toggleStatusAction(
   masto.statuses[status.pinned ? 'unpin' : 'pin'](status.id),
 )
 
+const { toggle: _toggleTranslation, translation, enabled: isTranslationEnabled } = useTranslation(_status)
+const toggleTranslation = async () => {
+  isLoading.translation = true
+  await _toggleTranslation()
+  isLoading.translation = false
+}
+
 const copyLink = async () => {
   await clipboard.copy(`${location.origin}${getStatusPath(status)}`)
 }
@@ -160,6 +168,16 @@ function mention() {
       />
     </CommonTooltip>
 
+    <CommonTooltip v-if="isTranslationEnabled" placement="bottom" content="Translate">
+      <StatusActionButton
+        color="text-pink" hover="text-pink" group-hover="bg-pink/10"
+        icon="i-ri:translate"
+        :active="translation.visible"
+        :disabled="isLoading.translation"
+        @click="toggleTranslation()"
+      />
+    </CommonTooltip>
+
     <CommonDropdown placement="bottom">
       <CommonTooltip placement="bottom" content="More">
         <button flex gap-1 items-center rounded op50 hover="op100 text-purple" group>
diff --git a/components/status/StatusBody.vue b/components/status/StatusBody.vue
index 02ec35e2..af219de0 100644
--- a/components/status/StatusBody.vue
+++ b/components/status/StatusBody.vue
@@ -4,10 +4,11 @@ import type { Status } from 'masto'
 const { status } = defineProps<{
   status: Status
 }>()
+const { translation } = useTranslation(status)
 </script>
 
 <template>
   <div class="status-body">
-    <ContentRich :content="status.content" :emojis="status.emojis" />
+    <ContentRich :content="translation.visible ? translation.text : status.content" :emojis="status.emojis" />
   </div>
 </template>
diff --git a/composables/translate.ts b/composables/translate.ts
new file mode 100644
index 00000000..e4260a90
--- /dev/null
+++ b/composables/translate.ts
@@ -0,0 +1,47 @@
+import type { Status } from 'masto'
+
+export interface TranslationResponse {
+  translatedText: string
+  detectedLanguage: {
+    confidence: number
+    language: string
+  }
+}
+
+const config = useRuntimeConfig()
+
+export async function translateText(text: string) {
+  const { translatedText } = await $fetch<TranslationResponse>(config.public.translateApi, {
+    method: 'POST',
+    body: {
+      q: text,
+      source: 'auto',
+      target: navigator.language.replace(/-.*$/, ''),
+      format: 'html',
+      api_key: '',
+    },
+  })
+  return translatedText
+}
+
+const translations = new WeakMap<Status, { visible: boolean; text: string }>()
+
+export function useTranslation(status: Status) {
+  if (!translations.has(status))
+    translations.set(status, reactive({ visible: false, text: '' }))
+
+  const translation = translations.get(status)!
+
+  async function toggle() {
+    if (!translation.text)
+      translation.text = await translateText(status.content)
+
+    translation.visible = !translation.visible
+  }
+
+  return {
+    enabled: !!config.public.translateApi,
+    toggle,
+    translation,
+  }
+}
diff --git a/nuxt.config.ts b/nuxt.config.ts
index 2c2da80b..24b87123 100644
--- a/nuxt.config.ts
+++ b/nuxt.config.ts
@@ -40,5 +40,8 @@ export default defineNuxtConfig({
       namespaceId: '',
       apiToken: '',
     },
+    public: {
+      translateApi: '',
+    },
   },
 })