Merge pull request #41 from Wxw-Gu/codex/develop-xkey-4-1-13

Codex/develop xkey 4 1 13

src/renderer/src/components/DatabaseConnectionPage.tsx

@@ -137,6 +137,7 @@ export function DatabaseConnectionPage({
 }: DatabaseConnectionPageProps): React.ReactElement {
   const isMac = platform === 'darwin'
   const isIntelMac = isMac && environment?.architecture === 'x64'
+  const isAppleSiliconMac = isMac && !isIntelMac
   const defaultPath = isMac
     ? '~/Library/Containers/com.tencent.xinWeChat/Data/Library/Application Support/com.tencent.xinWeChat/'
     : 'C:\\Users\\...\\WeChat Files\\Msg'
@@ -189,18 +190,16 @@ export function DatabaseConnectionPage({
               <li>
                 <span>2</span>
                 <div>
-                  <strong>{isIntelMac ? '让微信停在登录页面' : '准备连接组件'}</strong>
+                  <strong>{isMac ? '让微信停在登录页面' : '准备连接组件'}</strong>
                   <small>
-                    {isIntelMac
-                      ? '启动微信，但暂时不要点击登录'
-                      : '页面会按当前系统给出对应步骤'}
+                    {isMac ? '启动微信，但暂时不要点击登录' : '页面会按当前系统给出对应步骤'}
                   </small>
                 </div>
               </li>
               <li>
                 <span>3</span>
                 <div>
-                  <strong>{isIntelMac ? '获取密钥并验证连接' : '登录并验证连接'}</strong>
+                  <strong>{isMac ? '获取密钥并验证连接' : '登录并验证连接'}</strong>
                   <small>验证通过后进入主界面</small>
                 </div>
               </li>
@@ -248,11 +247,13 @@ export function DatabaseConnectionPage({
                           : [
                               '检查本机环境',
                               '让微信停在登录页面',
-                              isIntelMac ? '确认开始获取密钥' : '确认开始准备',
+                              isMac ? '确认开始获取密钥' : '确认开始准备',
                               isIntelMac
                                 ? '正在获取 Intel Mac 密钥'
-                                : `正在完成 ${isMac ? 'macOS' : 'Windows'} 授权`,
-                              isIntelMac
+                                : isAppleSiliconMac
+                                  ? '正在监听 macOS 登录密钥'
+                                  : '正在完成 Windows 授权',
+                              isMac
                                 ? '获取成功，验证数据库连接'
                                 : '获取成功, 现在可以重新登录微信了',
                               '验证数据库连接'
@@ -264,14 +265,14 @@ export function DatabaseConnectionPage({
                           : status ||
                             [
                               '确认下方检测结果；没有找到目录时可以手动选择。',
-                              isIntelMac
+                              isMac
                                 ? '请启动微信并停留在未登录界面，暂时不要点击登录，然后点击“我已准备好”。'
                                 : '请退出当前微信账号，让微信停留在登录页面，然后点击“我已准备好”。',
-                              isIntelMac ? '确认开始获取密钥。' : '开始后请按页面提示完成系统授权。',
-                              isIntelMac
+                              isMac ? '确认开始获取密钥。' : '开始后请按页面提示完成系统授权。',
+                              isMac
                                 ? '等页面提示可以登录后，立即回到微信点击登录。'
                                 : '正在准备连接组件，请不要关闭微信或 TraceMemo。',
-                              isIntelMac
+                              isMac
                                 ? '密钥已获取，请继续验证数据库。'
                                 : '请回到微信完成登录，登录成功后再回来验证。',
                               '正在验证密钥和本地数据库，请稍候。'
@@ -445,12 +446,12 @@ export function DatabaseConnectionPage({
                 )}
                 {guideStep === 3 && (
                   <Button className="database-login-primary" onClick={onAutoGetKey}>
-                    {isIntelMac ? '开始获取密钥' : '开始准备连接组件'}
+                    {isMac ? '开始获取密钥' : '开始准备连接组件'}
                   </Button>
                 )}
                 {guideStep === 4 && (
                   <Button className="database-login-primary" disabled>
-                    正在准备连接组件…
+                    {isMac ? '正在获取密钥…' : '正在准备连接组件…'}
                   </Button>
                 )}
                 {guideStep === 5 && (
@@ -491,7 +492,7 @@ export function DatabaseConnectionPage({
                     <>
                       {isIntelMac
                         ? 'Intel Mac 首次使用需要先准备连接环境，按页面提示完成即可。'
-                        : 'macOS 首次获取密钥需要关闭 SIP。'}{' '}
+                        : 'macOS Apple 芯片首次获取密钥需要关闭 SIP，并在监听期间点击微信登录。'}{' '}
                       <a href={isIntelMac ? GUIDE_URL : macKeyFaqUrl} target="_blank" rel="noreferrer">
                         查看说明
                       </a>

src/renderer/src/features/settings/database-key/DatabaseKeyAutoDetect.tsx

@@ -2,7 +2,14 @@ import type { DatabaseKeyState } from './types'
 import { runtimePlatform } from '../../../utils/runtime-environment'
 import { Button } from '../../../components/ui'
 
-const PHASES = ['查找微信进程', '识别微信版本', '扫描候选密钥', '验证数据库', '获取完成']
+const DEFAULT_PHASES = ['查找微信进程', '识别微信版本', '扫描候选密钥', '验证数据库', '获取完成']
+const MAC_LOGIN_PHASES = [
+  '查找微信进程',
+  '等待管理员授权',
+  '监听登录密钥',
+  '验证数据库',
+  '获取完成'
+]
 
 export function DatabaseKeyAutoDetect({
   state,
@@ -17,8 +24,12 @@ export function DatabaseKeyAutoDetect({
 }): React.ReactElement {
   const environment = state.environment
   const platform = environment?.platform || runtimePlatform
+  const isMac = platform === 'darwin'
   const isIntelMac = platform === 'darwin' && environment?.architecture === 'x64'
-  if (platform !== 'win32' && !isIntelMac) {
+  const isAppleSiliconMac = isMac && !isIntelMac
+  const supported = environment?.autoDetectSupported ?? (platform === 'win32' || isMac)
+  const phases = isAppleSiliconMac ? MAC_LOGIN_PHASES : DEFAULT_PHASES
+  if (!supported) {
     return (
       <section className="settings-card database-key-auto database-key-auto-manual">
         <div>
@@ -32,11 +43,15 @@ export function DatabaseKeyAutoDetect({
     <section className="settings-card database-key-auto">
       <div className="database-key-auto-heading">
         <div>
-          <strong>{isIntelMac ? 'Intel Mac 自动获取' : 'Windows 自动获取'}</strong>
+          <strong>
+            {isIntelMac ? 'Intel Mac 自动获取' : isMac ? 'macOS 自动获取' : 'Windows 自动获取'}
+          </strong>
           <p>
             {isIntelMac
               ? '请让微信停在未登录界面，按页面提示操作即可。'
-              : 'TraceMemo 可在微信桌面端正在运行时，通过本机内存扫描尝试获取数据库密钥。'}
+              : isAppleSiliconMac
+                ? '执行时会请求管理员授权；授权后在微信登录界面点击“登录”即可，已有登录凭据时通常不需要扫码。监听最长两分钟，结束后会明确显示结果。'
+                : 'TraceMemo 可在微信桌面端正在运行时，通过本机内存扫描尝试获取数据库密钥。'}
           </p>
         </div>
         <Button variant="outline" onClick={onDetect} disabled={disabled}>
@@ -50,9 +65,7 @@ export function DatabaseKeyAutoDetect({
         <li className={environment?.accountIdentified ? 'ok' : ''}>
           当前账号：{environment?.accountIdentified ? '已识别' : '尚未识别'}
         </li>
-        <li className={platform === 'win32' || isIntelMac ? 'ok' : ''}>
-          当前平台：{platform === 'win32' || isIntelMac ? '支持' : '不支持'}
-        </li>
+        <li className={supported ? 'ok' : ''}>当前平台：{supported ? '支持' : '不支持'}</li>
         {isIntelMac && (
           <>
             <li className={environment?.pythonAvailable && environment?.fridaAvailable ? 'ok' : ''}>
@@ -66,7 +79,7 @@ export function DatabaseKeyAutoDetect({
       </ul>
       {state.status === 'auto-detecting' && (
         <ol className="database-key-phases">
-          {PHASES.map((phase, index) => (
+          {phases.map((phase, index) => (
             <li key={phase} className={state.autoPhase >= index + 1 ? 'active' : ''}>
               {phase}
             </li>
@@ -80,7 +93,9 @@ export function DatabaseKeyAutoDetect({
           <p>
             {isIntelMac
               ? '请让微信回到未登录界面，重新开始后按页面提示操作。'
-              : '请保持微信正在运行，登录目标账号并打开几个聊天窗口后重试。'}
+              : isAppleSiliconMac
+                ? '请先让微信停留在登录界面，点击自动获取并完成管理员授权，然后点击微信“登录”。'
+                : '请保持微信正在运行，登录目标账号并打开几个聊天窗口后重试。'}
           </p>
           <Button
             variant="link"

src/renderer/src/features/settings/database-key/useDatabaseKeyController.ts

@@ -155,15 +155,22 @@ export function useDatabaseKeyController({
 
   const autoDetectKey = useCallback(async (): Promise<void> => {
     dispatch({ type: 'AUTO_START' })
-    await refreshEnvironment()
-    const result = await window.api.autoGetDbKey(selfInfo?.accountRoot || '', { save: false })
-    if (!result.success || !result.key) {
-      dispatch({ type: 'AUTO_ERROR', error: result.error || '暂未找到有效密钥' })
-      return
+    try {
+      await refreshEnvironment()
+      const result = await window.api.autoGetDbKey(selfInfo?.accountRoot || '', { save: false })
+      if (!result.success || !result.key) {
+        dispatch({ type: 'AUTO_ERROR', error: result.error || '暂未找到有效密钥' })
+        return
+      }
+      onDbKeyChange(result.key)
+      dispatch({ type: 'AUTO_SUCCESS' })
+      await runValidation(result.key)
+    } catch {
+      dispatch({
+        type: 'AUTO_ERROR',
+        error: '自动获取密钥未完成，请确认微信仍在运行后重试。'
+      })
     }
-    onDbKeyChange(result.key)
-    dispatch({ type: 'AUTO_SUCCESS' })
-    await runValidation(result.key)
   }, [onDbKeyChange, refreshEnvironment, runValidation, selfInfo?.accountRoot])
 
   const clearSavedKey = useCallback(async (): Promise<void> => {

src/renderer/src/features/settings/database-key/utils.ts

@@ -8,8 +8,8 @@ export const isDatabaseKeyFormatValid = (value: string): boolean =>
 export function mapAutoDetectPhase(message: string): number {
   if (/完成|成功|已获取/.test(message)) return 5
   if (/验证|校验/.test(message)) return 4
-  if (/扫描|候选|获取/.test(message)) return 3
-  if (/版本|组件|窗口/.test(message)) return 2
+  if (/登录|监听|扫描|候选|获取/.test(message)) return 3
+  if (/授权|版本|组件|窗口/.test(message)) return 2
   return 1
 }