From 6b64b74f6c5c1cfd291ed3f544f714d322a9511e Mon Sep 17 00:00:00 2001
From: Chenlei Hu <huchenlei@proton.me>
Date: Wed, 5 Feb 2025 10:51:32 -0500
Subject: [PATCH] [Desktop] Use fallback mirrors for China users (#2424)

---
 .../install/MirrorsConfiguration.vue          | 12 ++++-
 src/components/install/mirror/MirrorItem.vue  |  6 ++-
 src/utils/networkUtil.ts                      | 50 +++++++++++++++++++
 3 files changed, 66 insertions(+), 2 deletions(-)

diff --git a/src/components/install/MirrorsConfiguration.vue b/src/components/install/MirrorsConfiguration.vue
index cce64e60d..27b936105 100644
--- a/src/components/install/MirrorsConfiguration.vue
+++ b/src/components/install/MirrorsConfiguration.vue
@@ -41,11 +41,12 @@ import {
 } from '@comfyorg/comfyui-electron-types'
 import Divider from 'primevue/divider'
 import Panel from 'primevue/panel'
-import { ModelRef, computed, ref } from 'vue'
+import { ModelRef, computed, onMounted, ref } from 'vue'
 
 import MirrorItem from '@/components/install/mirror/MirrorItem.vue'
 import { PYPI_MIRROR, PYTHON_MIRROR, UVMirror } from '@/constants/uvMirrors'
 import { t } from '@/i18n'
+import { isInChina } from '@/utils/networkUtil'
 import { ValidationState, mergeValidationStates } from '@/utils/validationUtil'
 
 const showMirrorInputs = ref(false)
@@ -101,4 +102,13 @@ const validationStateTooltip = computed(() => {
       return t('install.settings.checkingMirrors')
   }
 })
+
+onMounted(async () => {
+  // Check if user is in China and set fallback mirrors directly
+  if (await isInChina()) {
+    for (const [item, modelValue] of mirrors.value) {
+      modelValue.value = item.fallbackMirror
+    }
+  }
+})
 </script>
diff --git a/src/components/install/mirror/MirrorItem.vue b/src/components/install/mirror/MirrorItem.vue
index 2aa134a83..e607c9197 100644
--- a/src/components/install/mirror/MirrorItem.vue
+++ b/src/components/install/mirror/MirrorItem.vue
@@ -22,6 +22,7 @@
 <script setup lang="ts">
 import { computed, onMounted, ref, watch } from 'vue'
 
+import UrlInput from '@/components/common/UrlInput.vue'
 import { UVMirror } from '@/constants/uvMirrors'
 import { normalizeI18nKey } from '@/utils/formatUtil'
 import { checkMirrorReachable } from '@/utils/networkUtil'
@@ -43,7 +44,10 @@ const normalizedSettingId = computed(() => {
 })
 
 onMounted(() => {
-  modelValue.value = item.mirror
+  // Set mirror value if not already set
+  if (!modelValue.value) {
+    modelValue.value = item.mirror
+  }
 })
 
 watch(validationState, (newState) => {
diff --git a/src/utils/networkUtil.ts b/src/utils/networkUtil.ts
index a4a93d4f3..bc29d81e5 100644
--- a/src/utils/networkUtil.ts
+++ b/src/utils/networkUtil.ts
@@ -24,3 +24,53 @@ export const checkMirrorReachable = async (mirror: string) => {
     isValidUrl(mirror) && (await electronAPI().NetWork.canAccessUrl(mirror))
   )
 }
+
+/**
+ * Checks if the user is likely in mainland China by:
+ * 1. Checking navigator.language
+ * 2. Testing connectivity to commonly blocked services
+ * 3. Testing latency to China-specific domains
+ */
+export async function isInChina(): Promise<boolean> {
+  // Quick check based on language/locale
+  const isChineseLocale = navigator.language.toLowerCase().startsWith('zh-cn')
+
+  try {
+    // Test connectivity to Google - commonly blocked in China
+    const googleTest = await Promise.race([
+      fetch('https://www.google.com', {
+        mode: 'no-cors',
+        cache: 'no-cache'
+      }),
+      new Promise((_, reject) => setTimeout(() => reject(), 2000))
+    ])
+
+    // If Google is accessible, user is likely not in China
+    if (googleTest) {
+      return false
+    }
+  } catch {
+    // Google is not accessible - potential indicator of being in China
+    if (isChineseLocale) {
+      return true
+    }
+
+    // Additional check - test latency to a reliable Chinese domain
+    try {
+      const start = performance.now()
+      await fetch('https://www.baidu.com', {
+        mode: 'no-cors',
+        cache: 'no-cache'
+      })
+      const latency = performance.now() - start
+
+      // If Baidu responds quickly (<150ms), user is likely in China
+      return latency < 150
+    } catch {
+      // If both tests fail, default to locale check
+      return isChineseLocale
+    }
+  }
+
+  return false
+}