type-challenges/scripts/translate.ts

import path from 'node:path'
import fs from 'fs-extra'
import { translate } from '@vitalets/google-translate-api'
import { QUIZ_ROOT, loadQuizByNo, loadQuizzes } from './loader'
import { resolveFilePath } from './utils/resolve'
import type { Quiz } from './types'
import type { SupportedLocale } from './locales'
import { t } from './locales'

export async function translateQuizByNo(no: number, from: SupportedLocale, to: SupportedLocale) {
  const quiz = await loadQuizByNo(no)
  if (!quiz)
    throw new Error(`Quiz #${no} not founded`)
  return await translateQuiz(quiz, from, to)
}

export async function translateQuiz(quiz: Quiz, from: SupportedLocale, to: SupportedLocale) {
  let translatedReadme = await translateMarkdown(quiz.readme[from], from, to)

  if (!translatedReadme)
    throw new Error(`Quiz #${quiz.no} empty translation`)

  translatedReadme = `> ${t(to, 'readme.google-translated')}\n\n${translatedReadme.trim()}`

  const readmePath = resolveFilePath(path.join(QUIZ_ROOT, quiz.path), 'README', 'md', to)
  await fs.writeFile(readmePath, translatedReadme, 'utf-8')

  console.log(`Translated [${quiz.no}] ${from} → ${to} | saved to ${readmePath}`)
}

export async function translateMarkdown(code: string, from: SupportedLocale, to: SupportedLocale) {
  // to replace the code blocks intro a placeholder then feed it into translator
  // then replace back for the results
  const codeBlocks: string[] = []

  const source = code
    .replace(/```[\s\S\n]+?```/g, (v) => {
      const placeholder = `__${codeBlocks.length}__`
      codeBlocks.push(v)
      return placeholder
    })
    .replace(/`[\s\S\n]+?`/g, (v) => {
      const placeholder = `__${codeBlocks.length}__`
      codeBlocks.push(v)
      return placeholder
    })

  const { text } = await translate(source, {
    from,
    to,
  })

  if (!text)
    return

  const result = text.replace(/__\s*?(\d+?)\s*?__/g, (_, i) => codeBlocks[+i])

  return result
}

export async function translateAllQuizzes(from: SupportedLocale, to: SupportedLocale) {
  const quizzes = await loadQuizzes()

  for (const quiz of quizzes) {
    if (quiz.readme[to] || !quiz.readme[from]) {
      console.log(`Skipped #${quiz.no}`)
      continue
    }

    console.log(`Translating #${quiz.no} to ${to}`)
    await translateQuiz(quiz, from, to)
  }
}