const fs = require('fs')
const path = require('path')

const rootDir = process.cwd()
const docsReadmeDir = path.join(rootDir, 'docs-readme')
const rootReadmePath = path.join(rootDir, 'README.md')

const locales = [
  { code: 'zh-CN', name: '简体中文' },
  { code: 'zh-TW', name: '繁體中文' },
  { code: 'en-US', name: 'English' },
  { code: 'ja-JP', name: '日本語' },
  { code: 'es-ES', name: 'Español' },
  { code: 'fr-FR', name: 'Français' },
  { code: 'tlh', name: 'Klingon' },
  { code: 'ko-KR', name: '한국어' },
  { code: 'ar-SA', name: 'العربية' },
  { code: 'tr-TR', name: 'Türkçe' },
  { code: 'vi-VN', name: 'Tiếng_Việt' },
  { code: 'de-DE', name: 'Deutsch' },
  { code: 'bn-BD', name: 'বাংলা' }
]

function generateNavBlock(isRoot) {
  const links = locales.map((locale) => {
    const href = isRoot
      ? `docs-readme/${locale.code}/README.md`
      : `../${locale.code}/README.md`
    return `  <a href="${href}"><img alt="${locale.name}" src="https://img.shields.io/badge/${locale.name}-d9d9d9"></a>`
  })

  return `<p align="center">\n${links.join('\n')}\n</p>`
}

function updateFile(filePath, isRoot) {
  if (!fs.existsSync(filePath)) {
    console.log(`File not found: ${filePath}`)
    return
  }

  let content = fs.readFileSync(filePath, 'utf8')
  const newNavBlock = generateNavBlock(isRoot)

  // Regex to match the existing nav block. It starts with <p align="center"> and contains the badges.
  // The badges in previous versions had slightly different names/links, but they all were inside <p align="center"> ... </p>
  // We look for the block containing "简体中文" or "English" to be sure it's the language nav.

  const regex =
    /<p align="center">\s*<a href=".*README.*\.md"><img alt=".*".*>\s*<\/a>[\s\S]*?<\/p>/

  if (regex.test(content)) {
    content = content.replace(regex, newNavBlock)
    fs.writeFileSync(filePath, content, 'utf8')
    console.log(`Updated ${filePath}`)
  } else {
    console.log(`Nav block not found in ${filePath}`)
    // Fallback: try to find where it should be (after the main title block?)
    // In previous steps, we saw it's after the main title block div end or inside it.
    // Let's try to match a simpler pattern if the complex one fails, or manual inspection might be needed.
    // Given I just wrote them, they should match.
  }
}

// Update root README
updateFile(rootReadmePath, true)

// Update docs-readme files
locales.forEach((locale) => {
  const filePath = path.join(docsReadmeDir, locale.code, 'README.md')
  updateFile(filePath, false)
})