#!/usr/bin/env node
// Copy new translation keys from zh-CN to all other locales

import { readFileSync, writeFileSync } from 'fs';
import { resolve, dirname } from 'path';
import { fileURLToPath } from 'url';

const __dirname = dirname(fileURLToPath(import.meta.url));
const localesDir = resolve(__dirname, '../lib/i18n/locales');

// Read zh-CN content to extract new keys
const zhCNContent = readFileSync(resolve(localesDir, 'zh-CN.ts'), 'utf-8');

// New keys to add (manually extracted from zh-CN.ts)
const newKeysSection = `
  // ========== Play Page (PlayCanvas.tsx & play/page.tsx) ==========
  play: {
    loading: {
      firstFrame: "正 · 在 · 绘 · 制 · 第 · 一 · 幕",
      transitioning: "AI · 正 · 在 · 描 · 画 · 下 · 一 · 幕",
      visionThinking: "AI · 正 · 在 · 想 · 你 · 看 · 到 · 了 · 什 · 么",
      loadingFirst: "正 · 在 · 唤 · 起 · 第 · 一 · 幕",
      awakening: "载入中",
    },

    freeform: {
      placeholder: "输入你想说的或想做的...",
      title: "自由输入",
      ariaLabel: "自由输入",
    },

    choiceDisabled: "分享剧情未包含这条分支",

    tooltips: {
      openSettings: "打开设置",
      openHistory: "剧情回溯",
      fullscreen: "全屏 (F)",
      enterFullscreen: "进入全屏",
      exportGallery: "导出本局为可交互图集链接（含配音；只会保留最近两次的可交互图集链接）",
      exportGalleryLabel: "导出可交互图集",
      shareStory: "导出本局为可继续游玩的剧情 .infiplot（含配音）",
      shareStoryLabel: "分享当前剧情",
      mute: "静音",
      unmute: "取消静音",
      closeNudge: "关闭提示",
      silenceNudge: "效果不满意/经常没声音？填入自己的 API Key 试试",
      back: "返回",
    },

    imageAlt: "Generated scene",

    counter: {
      scene: "第 · {n} · 幕",
      beat: "{n} · 拍",
      middle: "·",
    },

    buttons: {
      fullscreen: "F · 键 · 全 · 屏",
      exportGallery: "导 · 出 · 图 · 集",
      shareStory: "分 · 享 · 剧 · 情",
      muted: "静 · 音",
      sound: "有 · 声",
    },

    error: {
      title: "出 · 了 · 点 · 状 · 况",
      back: "返 · 回",
    },

    previousStep: "上 · 一 · 步 ·",

    settingsFooter: "保存后配音 Key 会立即生效，用你自己的额度合成当前这一幕的配音。",

    shareErrors: {
      notFound: "没有找到要载入的剧情文件。",
      invalid: "剧情分享文件没有可载入的剧情。",
      noImage: "剧情分享文件缺少第一幕图片。",
      noNextImage: "剧情分享文件缺少下一幕图片。",
      noMemory: "剧情分享文件缺少初始剧情记忆，无法载入。",
      packFailed: "剧情分享打包失败",
    },
  },
`;

// Find the line where to insert (before '  language: {' or at end)
function addKeysToFile(content, locale) {
  // Check if file already has play section
  if (content.includes('play: {')) {
    console.log(`${locale} already has play section, skipping`);
    return null;
  }

  // Find position to insert (before the last '  language:' or before '}')
  const langIndex = content.lastIndexOf('  language:');
  if (langIndex > 0) {
    return content.slice(0, langIndex) + newKeysSection + content.slice(langIndex);
  }

  // If no language: found, find the end of the object
  const lastBrace = content.lastIndexOf('}');
  if (lastBrace > 0) {
    return content.slice(0, lastBrace) + ',' + newKeysSection + '\n}' + content.slice(lastBrace + 1);
  }

  return null;
}

// Target locales
const targetLocales = [
  'zh-TW', 'zh-HK', 'ja', 'ko', 'es', 'fr', 'de', 'pt-BR', 'pt', 'ru',
  'it', 'vi', 'th', 'id', 'tr', 'pl', 'nl', 'uk', 'hi', 'cs'
];

let successCount = 0;
for (const locale of targetLocales) {
  try {
    const filePath = resolve(localesDir, `${locale}.ts`);
    const content = readFileSync(filePath, 'utf-8');
    const newContent = addKeysToFile(content, locale);

    if (newContent) {
      writeFileSync(filePath, newContent);
      console.log(`✓ Updated ${locale}.ts`);
      successCount++;
    }
  } catch (e) {
    console.error(`✗ Error updating ${locale}:`, e.message);
  }
}

console.log(`\nDone! Updated ${successCount} locale files`);
console.log('Note: New keys are in Chinese. Run translation script to translate them.');