# scripts/step2_translate_ipa.py

import os
import json
from pathlib import Path
from google.cloud import translate_v2 as translate
from notion_client import Client
from notion_client.errors import APIResponseError
import eng_to_ipa as e2i
import logging 

def _update_single_notion_page(client, page_id: str, zh_text: str, ipa_text: str):
    """
    將翻譯和 IPA 結果更新回指定的 Notion 頁面。
    
    Args:
        client: 已初始化的 Notion Client。
        page_id (str): 要更新的 Notion 頁面 ID。
        zh_text (str): 中文翻譯文字。
        ipa_text (str): IPA 音標文字。
    """
    try:
        # --- ✨ 核心修改處：建立符合 Notion API 規範的 properties 物件 ---
        properties_payload = {
            # 假設您在 Notion 中的欄位名稱就是 "zh"
            # 如果不是，請修改 "zh" 為您實際的欄位名稱，例如 "中文翻譯"
            "zh": {
                "rich_text": [
                    {
                        "type": "text",
                        "text": {
                            "content": zh_text
                        }
                    }
                ]
            },
            # 假設您在 Notion 中的欄位名稱就是 "ipa"
            # 如果不是，請修改 "ipa" 為您實際的欄位名稱，例如 "IPA"
            "ipa": {
                "rich_text": [
                    {
                        "type": "text",
                        "text": {
                            "content": ipa_text
                        }
                    }
                ]
            }
        }

        logging.info(f"正在更新 Notion 頁面 ID: {page_id}")
        # 發送更新請求
        client.pages.update(page_id=page_id, properties=properties_payload)
        logging.info(f"✅ 成功將結果寫回 Notion 頁面 {page_id}")

    except Exception as e:
        # 捕捉並印出詳細錯誤，這對於除錯至關重要
        logging.error(f"❌ 更新 Notion 頁面 {page_id} 失敗: {e}")


def run_step2_translate_ipa(
    project_path: Path,
    google_creds_path: str,
    notion_api_key: str = None,
    notion_database_id: str = None
):
    """
    讀取專案中的 data.json，為其添加中文翻譯和 IPA 音標，
    並將結果寫回 data.json，同時可選擇性地更新回 Notion。
    Args:
        project_path (Path): 專案資料夾的路徑。
        google_creds_path (str): Google Cloud 認證 JSON 檔案的路徑。
        notion_api_key (str, optional): Notion API Key. Defaults to None.
        notion_database_id (str, optional): Notion Database ID. Defaults to None.
    Returns:
        tuple[bool, str]: (操作是否成功, 附帶的訊息)。
    """
    data_file = project_path / "data.json"
    if not data_file.exists():
        return False, f"錯誤：在專案 '{project_path.name}' 中找不到 data.json 檔案。"

    # 1. 初始化 API 客戶端
    try:
        os.environ["GOOGLE_APPLICATION_CREDENTIALS"] = google_creds_path
        translate_client = translate.Client()
    except Exception as e:
        return False, f"初始化 Google Translate 客戶端失敗: {e}"

    notion_client = None
    if notion_api_key and notion_database_id:
        try:
            notion_client = Client(auth=notion_api_key)
        except Exception as e:
            return False, f"初始化 Notion 客戶端失敗: {e}"

    # 2. 讀取並處理資料
    try:
        with data_file.open("r", encoding="utf-8") as f:
            data = json.load(f)
    except Exception as e:
        return False, f"讀取或解析 data.json 失敗: {e}"

    # --- ✨ 核心修改處：移除迴圈，直接處理 data 物件 ---
    
    # 使用 .get() 安全地獲取英文原文，避免 KeyError [7][14]
    # 'en' 是您在 data.json 中定義的鍵
    english_text = data.get("en")

    if not english_text or not isinstance(english_text, str):
        return False, "在 data.json 中找不到 'en' 鍵或其內容不是文字。"

    # --- 翻譯成中文 ---
    # 將英文原文按行切分，並過濾掉空行
    english_lines = [line.strip() for line in english_text.strip().split('\n') if line.strip()]
    translated_lines = []

    # 迴圈遍歷每一行英文，單獨進行翻譯
    for line in english_lines:
        try:
            result = translate_client.translate(line, target_language="zh-TW")
            translated_lines.append(result.get("translatedText", ""))
        except Exception as e:
            print(f"翻譯 '{line}' 時出錯: {e}")
            translated_lines.append("") # 如果單行出錯，則添加空字串
    # 將翻譯好的各行重新用換行符組合起來
    translated_text = '\n'.join(translated_lines)

    # --- 生成 IPA 音標 ---
    try:
        english_lines = [line.strip() for line in english_text.strip().split('\n') if line.strip()]
        ipa_lines = [e2i.convert(line) for line in english_lines]
        ipa_text = '\n'.join(ipa_lines)
    except Exception as e:
        print(f"獲取 '{english_text}' 的 IPA 時出錯: {e}")
        ipa_text = ""

    # 將結果存回主 data 字典中
    data["zh"] = translated_text
    data["ipa"] = ipa_text
    
    print(f"已處理: '{english_text[:30]}...' -> '{translated_text[:30]}...' | '{ipa_text[:30]}...'")

    # --- 如果提供了 Notion 憑證，則更新回 Notion ---
    page_id = data.get("id")
    if notion_client and page_id:
        _update_single_notion_page(notion_client, page_id, translated_text, ipa_text)

    # 3. 將更新後的單一物件寫回 JSON 檔案
    try:
        with data_file.open("w", encoding="utf-8") as f:
            json.dump(data, f, ensure_ascii=False, indent=2)
    except Exception as e:
        return False, f"寫回更新後的 data.json 失敗: {e}"

    return True, f"✅ 專案 '{project_path.name}' 的翻譯與 IPA 已成功處理！"