feat: add aihubmix oauth

src/renderer/src/i18n/locales/en-us.json

@@ -501,6 +501,7 @@
         "docs_check": "Check",
         "docs_more_details": "for more details",
         "get_api_key": "Get API Key",
+        "charge": "Charge",
         "no_models": "Please add models first before checking the API connection",
         "not_checked": "Not Checked",
         "remove_duplicate_keys": "Remove Duplicate Keys",
@@ -703,7 +704,8 @@
       "oauth_button": "Auth with {{provider}}",
       "get_key": "Get",
       "get_key_success": "API key automatically obtained successfully",
-      "login": "Login"
+      "login": "Login",
+      "error": "API key automatically obtained failed, please get it manually"
     }
   }
 }

src/renderer/src/i18n/locales/ja-jp.json

@@ -492,6 +492,7 @@
         "docs_check": "チェック",
         "docs_more_details": "詳細を確認",
         "get_api_key": "APIキーを取得",
+        "charge": "充電",
         "no_models": "API接続をチェックする前に、モデルを追加してください",
         "not_checked": "未チェック",
         "remove_duplicate_keys": "重複キーを削除",
@@ -683,7 +684,8 @@
       "oauth_button": "{{provider}}で認証",
       "get_key": "取得",
       "get_key_success": "APIキーの自動取得に成功しました",
-      "login": "認証"
+      "login": "認証",
+      "error": "APIキーの自動取得に失敗しました。手動で取得してください"
     }
   }
 }

src/renderer/src/i18n/locales/ru-ru.json

@@ -493,6 +493,7 @@
         "docs_check": "Проверить",
         "docs_more_details": "для получения дополнительной информации",
         "get_api_key": "Получить ключ API",
+        "charge": "Пополнить",
         "no_models": "Пожалуйста, добавьте модели перед проверкой соединения с API",
         "not_checked": "Не проверено",
         "remove_duplicate_keys": "Удалить дубликаты ключей",
@@ -695,7 +696,8 @@
       "oauth_button": "Авторизоваться с {{provider}}",
       "get_key": "Получить",
       "get_key_success": "Автоматический получение ключа API успешно",
-      "login": "Войти"
+      "login": "Войти",
+      "error": "Автоматический получение ключа API не удалось, пожалуйста, получите ключ вручную"
     }
   }
 }

src/renderer/src/i18n/locales/zh-cn.json

@@ -499,6 +499,7 @@
         "docs_check": "查看",
         "docs_more_details": "获取更多详情",
         "get_api_key": "点击这里获取密钥",
+        "charge": "充值",
         "no_models": "请先添加模型再检查 API 连接",
         "not_checked": "未检查",
         "remove_duplicate_keys": "移除重复密钥",
@@ -690,7 +691,8 @@
       "oauth_button": "使用{{provider}}登录",
       "get_key": "获取",
       "get_key_success": "自动获取密钥成功",
-      "login": "登录"
+      "login": "登录",
+      "error": "自动获取密钥失败，请手动获取"
     }
   }
 }

src/renderer/src/i18n/locales/zh-tw.json

@@ -498,6 +498,7 @@
         "docs_check": "檢查",
         "docs_more_details": "查看更多細節",
         "get_api_key": "點擊這裡獲取密鑰",
+        "charge": "充值",
         "no_models": "請先添加模型再檢查 API 連接",
         "not_checked": "未檢查",
         "remove_duplicate_keys": "移除重複密鑰",
@@ -689,7 +690,8 @@
       "oauth_button": "使用{{provider}}登入",
       "get_key": "獲取",
       "get_key_success": "自動獲取密鑰成功",
-      "login": "登入"
+      "login": "登入",
+      "error": "自動獲取密鑰失敗，請手動獲取"
     }
   }
 }

src/renderer/src/services/ProviderService.ts

@@ -16,6 +16,6 @@ export function getProviderName(id: string) {
 }
 
 export function isProviderSupportAuth(provider: Provider) {
-  const supportProviders = ['silicon']
+  const supportProviders = ['silicon', 'aihubmix']
   return supportProviders.includes(provider.id)
 }

src/renderer/src/utils/oauth.ts

@@ -1,5 +1,6 @@
 import { SILICON_CLIENT_ID } from '@renderer/config/constant'
 import { getLanguageCode } from '@renderer/i18n'
+import i18n from '@renderer/i18n'
 export const oauthWithSiliconFlow = async (setKey) => {
   const authUrl = `https://account.siliconflow.cn/oauth?client_id=${SILICON_CLIENT_ID}`
 
@@ -22,7 +23,7 @@ export const oauthWithSiliconFlow = async (setKey) => {
 }
 
 export const oauthWithAihubmix = async (setKey) => {
-  const authUrl = `https://aihubmix.com/login?cherry_studio_oauth=true&lang=${getLanguageCode()}&aff=SJyh`
+  const authUrl = ` https://aihubmix.com/oauth?client_id=cherry_studio_oauth&lang=${getLanguageCode()}&aff=SJyh`
 
   const popup = window.open(
     authUrl,
@@ -30,15 +31,25 @@ export const oauthWithAihubmix = async (setKey) => {
     'width=720,height=720,toolbar=no,location=no,status=no,menubar=no,scrollbars=yes,resizable=yes,alwaysOnTop=yes,alwaysRaised=yes'
   )
 
-  const messageHandler = (event) => {
+  const messageHandler = async (event) => {
     const data = event.data
 
     if (data && data.key === 'cherry_studio_oauth_callback') {
-      const apiKeys = data?.data?.apiKeys
-      if (apiKeys && apiKeys.length > 0) {
-        setKey(apiKeys[0].value)
+      const { iv, encryptedData } = data.data
+
+      try {
+        const secret = import.meta.env.RENDERER_VITE_AIHUBMIX_SECRET || ''
+        const decryptedData: any = await window.api.aes.decrypt(encryptedData, iv, secret)
+        const { api_keys } = JSON.parse(decryptedData)
+        if (api_keys && api_keys.length > 0) {
+          setKey(api_keys[0].value)
+          popup?.close()
+          window.removeEventListener('message', messageHandler)
+        }
+      } catch (error) {
+        console.error('[oauthWithAihubmix] error', error)
         popup?.close()
-        window.removeEventListener('message', messageHandler)
+        window.message.error(i18n.t('oauth.error'))
       }
     }
   }