taiji-pad-v0/contexts/language-context.tsx

"use client"

import type React from "react"
import { createContext, useContext, useState, useEffect } from "react"

type Language = "zh" | "en"

interface LanguageContextType {
  language: Language
  setLanguage: (lang: Language) => void
  t: (zh: string, en: string) => string
}

const LanguageContext = createContext<LanguageContextType | undefined>(undefined)

export function LanguageProvider({ children }: { children: React.ReactNode }) {
  // 始终从 "zh" 开始，确保服务器端和客户端初始状态一致
  const [language, setLanguage] = useState<Language>("zh")

  useEffect(() => {
    // 只在客户端挂载后读取 localStorage
    const savedLang = localStorage.getItem("language") as Language
    if (savedLang && (savedLang === "zh" || savedLang === "en")) {
      setLanguage(savedLang)
    }
  }, [])

  const handleSetLanguage = (lang: Language) => {
    setLanguage(lang)
    localStorage.setItem("language", lang)
  }

  const t = (zh: string, en: string) => {
    return language === "zh" ? zh : en
  }

  return (
    <LanguageContext.Provider value={{ language, setLanguage: handleSetLanguage, t }}>
      {children}
    </LanguageContext.Provider>
  )
}

export function useLanguage() {
  const context = useContext(LanguageContext)
  if (!context) {
    throw new Error("useLanguage must be used within a LanguageProvider")
  }
  return context
}