web-fusion/src/stores/countryStore.js

import { defineStore } from 'pinia'
import { ref } from 'vue'
import { ElMessage, ElMessageBox } from 'element-plus'
import { getCountryName } from '@/utils/countryUtil'

export const useCountryStore = defineStore('country', () => {
  // 状态
  const countryData = ref('') // 中文国家名
  const countryDataEN = ref('') // 英文国家名
  const isLoading = ref(false) // 是否正在获取
  const hasInitialized = ref(false) // 是否已初始化
  const lastFetchTime = ref(null) // 上次获取时间

  /**
   * 获取 IP 国家信息
   * @param {Function} t - 国际化函数
   * @param {boolean} showDialog - 获取失败时是否显示弹窗（默认 true，只在第一次获取时显示）
   */
  const fetchCountryInfo = async (t, showDialog = true) => {
    if (isLoading.value) return

    isLoading.value = true
    try {
      const response = await fetch('https://ipapi.co/json/')
      if (!response.ok) {
        throw new Error('请求失败')
      }
      const data = await response.json()

      countryDataEN.value = data.country_name
      countryData.value = getCountryName(data.country)
      lastFetchTime.value = Date.now()
      hasInitialized.value = true

      return { success: true }
    } catch (error) {
      console.error('获取国家信息失败:', error)

      // 只在允许显示弹窗且未初始化时才显示
      if (showDialog && !hasInitialized.value) {
        showCountryInputDialog(t)
      }

      return { success: false, error }
    } finally {
      isLoading.value = false
    }
  }

  /**
   * 刷新国家信息
   * @param {Function} t - 国际化函数
   */
  const refreshCountry = async (t) => {
    if (isLoading.value) return

    try {
      const result = await fetchCountryInfo(t, false) // 刷新时不自动弹窗

      if (result.success) {
        ElMessage.success(t('workbenchesSetup.refreshSuccess') || t('hostsList.refreshSuccess') || '刷新成功')
      } else {
        // 刷新失败时，给用户选择是否手动输入
        ElMessageBox.confirm(
          t('workbenchesSetup.refreshFailed') || t('hostsList.refreshFailed') || '刷新失败，是否手动输入国家？',
          t('workbenchesSetup.enterCountryTitle') || t('hostsList.enterCountryTitle') || '提示',
          {
            confirmButtonText: t('workbenchesSetup.confirm') || t('hostsList.confirm') || '手动输入',
            cancelButtonText: t('workbenchesSetup.cancel') || t('hostsList.cancel') || '取消',
            type: 'warning',
          }
        ).then(() => {
          showCountryInputDialog(t)
        }).catch(() => {
          // 用户取消
        })
      }
    } catch (error) {
      ElMessage.error(t('workbenchesSetup.refreshFailed') || t('hostsList.refreshFailed') || '刷新失败')
    }
  }

  /**
   * 显示手动输入国家的弹窗
   * @param {Function} t - 国际化函数
   */
  const showCountryInputDialog = (t) => {
    ElMessageBox.prompt(
      t('workbenchesSetup.enterCountryPrompt') || t('hostsList.enterCountryPrompt') || '由于网络原因无法自动获取国家信息，请手动输入当前网络所在国家（中文名）',
      t('workbenchesSetup.enterCountryTitle') || t('hostsList.enterCountryTitle') || '获取国家失败',
      {
        confirmButtonText: t('workbenchesSetup.confirm') || t('hostsList.confirm') || '确定',
        cancelButtonText: t('workbenchesSetup.cancel') || t('hostsList.cancel') || '取消',
        inputPlaceholder: t('workbenchesSetup.countryPlaceholder') || t('hostsList.countryPlaceholder') || '例如：美国、日本、英国',
        inputValidator: (value) => {
          if (!value || value.trim() === '') {
            return t('workbenchesSetup.countryRequired') || t('hostsList.countryRequired') || '请输入国家名称'
          }
          return true
        }
      }
    ).then(({ value }) => {
      countryData.value = value.trim()
      countryDataEN.value = value.trim()
      hasInitialized.value = true
      lastFetchTime.value = Date.now()
      ElMessage.success(t('workbenchesSetup.countrySetSuccess') || t('hostsList.countrySetSuccess') || '国家设置成功')
    }).catch(() => {
      // 用户取消输入
      if (!hasInitialized.value) {
        countryData.value = t('workbenchesSetup.unknown') || t('hostsList.unknown') || '未知'
        countryDataEN.value = 'Unknown'
        hasInitialized.value = true
      }
    })
  }

  /**
   * 初始化国家信息（只在第一次调用时获取）
   * @param {Function} t - 国际化函数
   */
  const initCountryInfo = async (t) => {
    if (hasInitialized.value) {
      return // 已经初始化过，不再重复获取
    }
    await fetchCountryInfo(t, true)
  }

  /**
   * 手动设置国家信息
   * @param {string} countryName - 国家名称（中文）
   * @param {Function} t - 国际化函数（可选）
   */
  const setCountryManually = (countryName, t = null) => {
    if (!countryName || countryName.trim() === '') {
      return false
    }

    countryData.value = countryName.trim()
    countryDataEN.value = countryName.trim()
    hasInitialized.value = true
    lastFetchTime.value = Date.now()

    if (t) {
      ElMessage.success(t('workbenchesSetup.countrySetSuccess') || t('hostsList.countrySetSuccess') || '国家设置成功')
    }

    return true
  }

  /**
   * 显示编辑国家的弹窗
   * @param {Function} t - 国际化函数
   * @returns {Promise} 确认时 resolve 新的国家名称，取消时 reject
   */
  const showEditCountryDialog = (t) => {
    return ElMessageBox.prompt(
      t('workbenchesSetup.editCountryPrompt') || t('hostsList.editCountryPrompt') || '请输入国家名称（中文）',
      t('workbenchesSetup.editCountryTitle') || t('hostsList.editCountryTitle') || '编辑国家',
      {
        confirmButtonText: t('workbenchesSetup.confirm') || t('hostsList.confirm') || '确定',
        cancelButtonText: t('workbenchesSetup.cancel') || t('hostsList.cancel') || '取消',
        inputPlaceholder: t('workbenchesSetup.countryPlaceholder') || t('hostsList.countryPlaceholder') || '例如：美国、日本、英国',
        inputValue: countryData.value, // 预填充当前值
        inputValidator: (value) => {
          if (!value || value.trim() === '') {
            return t('workbenchesSetup.countryRequired') || t('hostsList.countryRequired') || '请输入国家名称'
          }
          return true
        }
      }
    ).then(({ value }) => {
      setCountryManually(value, t)
      return value.trim() // 返回新的国家名称
    })
  }

  return {
    // 状态
    countryData,
    countryDataEN,
    isLoading,
    hasInitialized,
    lastFetchTime,

    // 方法
    fetchCountryInfo,
    refreshCountry,
    showCountryInputDialog,
    initCountryInfo,
    setCountryManually,
    showEditCountryDialog,
  }
})