wxid可以自己选择

This commit is contained in:
xuncha
2026-01-31 18:11:55 +08:00
parent 014f57f152
commit bfcd154a25
7 changed files with 177 additions and 16 deletions

View File

@@ -674,6 +674,10 @@ function registerIpcHandlers() {
return dbPathService.scanWxids(rootPath) return dbPathService.scanWxids(rootPath)
}) })
ipcMain.handle('dbpath:scanWxidCandidates', async (_, rootPath: string) => {
return dbPathService.scanWxidCandidates(rootPath)
})
ipcMain.handle('dbpath:getDefault', async () => { ipcMain.handle('dbpath:getDefault', async () => {
return dbPathService.getDefaultPath() return dbPathService.getDefaultPath()
}) })

View File

@@ -71,6 +71,7 @@ contextBridge.exposeInMainWorld('electronAPI', {
dbPath: { dbPath: {
autoDetect: () => ipcRenderer.invoke('dbpath:autoDetect'), autoDetect: () => ipcRenderer.invoke('dbpath:autoDetect'),
scanWxids: (rootPath: string) => ipcRenderer.invoke('dbpath:scanWxids', rootPath), scanWxids: (rootPath: string) => ipcRenderer.invoke('dbpath:scanWxids', rootPath),
scanWxidCandidates: (rootPath: string) => ipcRenderer.invoke('dbpath:scanWxidCandidates', rootPath),
getDefault: () => ipcRenderer.invoke('dbpath:getDefault') getDefault: () => ipcRenderer.invoke('dbpath:getDefault')
}, },

View File

@@ -118,6 +118,48 @@ export class DbPathService {
} }
} }
/**
* 扫描目录名候选(仅包含下划线的文件夹,排除 all_users
*/
scanWxidCandidates(rootPath: string): WxidInfo[] {
const wxids: WxidInfo[] = []
try {
if (existsSync(rootPath)) {
const entries = readdirSync(rootPath)
for (const entry of entries) {
const entryPath = join(rootPath, entry)
let stat: ReturnType<typeof statSync>
try {
stat = statSync(entryPath)
} catch {
continue
}
if (!stat.isDirectory()) continue
const lower = entry.toLowerCase()
if (lower === 'all_users') continue
if (!entry.includes('_')) continue
wxids.push({ wxid: entry, modifiedTime: stat.mtimeMs })
}
}
if (wxids.length === 0) {
const rootName = basename(rootPath)
if (rootName.includes('_') && rootName.toLowerCase() !== 'all_users') {
const rootStat = statSync(rootPath)
wxids.push({ wxid: rootName, modifiedTime: rootStat.mtimeMs })
}
}
} catch { }
return wxids.sort((a, b) => {
if (b.modifiedTime !== a.modifiedTime) return b.modifiedTime - a.modifiedTime
return a.wxid.localeCompare(b.wxid)
})
}
/** /**
* 扫描 wxid 列表 * 扫描 wxid 列表
*/ */

View File

@@ -451,7 +451,7 @@ export class KeyService {
return false return false
} }
private async closeWeChatWindows(timeoutMs = 8000): Promise<boolean> { private async closeWeChatWindows(): Promise<boolean> {
if (!this.ensureUser32()) return false if (!this.ensureUser32()) return false
let requested = false let requested = false
@@ -473,13 +473,15 @@ export class KeyService {
this.EnumWindows(enumWindowsCallback, 0) this.EnumWindows(enumWindowsCallback, 0)
this.koffi.unregister(enumWindowsCallback) this.koffi.unregister(enumWindowsCallback)
if (!requested) return true return requested
return await this.waitForWeChatExit(timeoutMs)
} }
private async killWeChatProcesses(): Promise<boolean> { private async killWeChatProcesses(): Promise<boolean> {
const gracefulOk = await this.closeWeChatWindows(8000) const requested = await this.closeWeChatWindows()
if (gracefulOk) return true if (requested) {
const gracefulOk = await this.waitForWeChatExit(1500)
if (gracefulOk) return true
}
try { try {
await execFileAsync('taskkill', ['/F', '/T', '/IM', 'Weixin.exe']) await execFileAsync('taskkill', ['/F', '/T', '/IM', 'Weixin.exe'])
@@ -491,6 +493,7 @@ export class KeyService {
return await this.waitForWeChatExit(5000) return await this.waitForWeChatExit(5000)
} }
// --- Window Detection --- // --- Window Detection ---
private getWindowTitle(hWnd: any): string { private getWindowTitle(hWnd: any): string {

View File

@@ -435,6 +435,58 @@
} }
} }
.wxid-select {
position: relative;
}
.wxid-dropdown {
position: absolute;
top: calc(100% + 6px);
left: 0;
right: 0;
background: var(--bg-primary);
border: 1px solid var(--border-color);
border-radius: 12px;
padding: 6px;
max-height: 220px;
overflow: auto;
z-index: 20;
box-shadow: 0 12px 24px rgba(0, 0, 0, 0.08);
}
.wxid-option {
width: 100%;
border: none;
background: transparent;
display: flex;
align-items: center;
justify-content: space-between;
gap: 12px;
padding: 10px 12px;
border-radius: 8px;
cursor: pointer;
color: var(--text-primary);
font-size: 13px;
&:hover {
background: var(--bg-hover);
}
&.active {
background: var(--primary-light);
}
}
.wxid-name {
font-weight: 600;
}
.wxid-time {
color: var(--text-tertiary);
font-size: 12px;
white-space: nowrap;
}
.field-with-toggle { .field-with-toggle {
position: relative; position: relative;
} }

View File

@@ -1,4 +1,4 @@
import { useState, useEffect } from 'react' import { useState, useEffect, useRef } from 'react'
import { useNavigate } from 'react-router-dom' import { useNavigate } from 'react-router-dom'
import { useAppStore } from '../stores/appStore' import { useAppStore } from '../stores/appStore'
import { dialog } from '../services/ipc' import { dialog } from '../services/ipc'
@@ -35,6 +35,8 @@ function WelcomePage({ standalone = false }: WelcomePageProps) {
const [cachePath, setCachePath] = useState('') const [cachePath, setCachePath] = useState('')
const [wxid, setWxid] = useState('') const [wxid, setWxid] = useState('')
const [wxidOptions, setWxidOptions] = useState<Array<{ wxid: string; modifiedTime: number }>>([]) const [wxidOptions, setWxidOptions] = useState<Array<{ wxid: string; modifiedTime: number }>>([])
const [showWxidSelect, setShowWxidSelect] = useState(false)
const wxidSelectRef = useRef<HTMLDivElement>(null)
const [error, setError] = useState('') const [error, setError] = useState('')
const [isConnecting, setIsConnecting] = useState(false) const [isConnecting, setIsConnecting] = useState(false)
const [isDetectingPath, setIsDetectingPath] = useState(false) const [isDetectingPath, setIsDetectingPath] = useState(false)
@@ -127,8 +129,22 @@ function WelcomePage({ standalone = false }: WelcomePageProps) {
useEffect(() => { useEffect(() => {
setWxidOptions([]) setWxidOptions([])
setWxid('') setWxid('')
setShowWxidSelect(false)
}, [dbPath]) }, [dbPath])
useEffect(() => {
const handleClickOutside = (event: MouseEvent) => {
if (!showWxidSelect) return
const target = event.target as Node
if (wxidSelectRef.current && !wxidSelectRef.current.contains(target)) {
setShowWxidSelect(false)
}
}
document.addEventListener('mousedown', handleClickOutside)
return () => document.removeEventListener('mousedown', handleClickOutside)
}, [showWxidSelect])
const currentStep = steps[stepIndex] const currentStep = steps[stepIndex]
const rootClassName = `welcome-page${isClosing ? ' is-closing' : ''}${standalone ? ' is-standalone' : ''}` const rootClassName = `welcome-page${isClosing ? ' is-closing' : ''}${standalone ? ' is-standalone' : ''}`
const showWindowControls = standalone const showWindowControls = standalone
@@ -217,6 +233,28 @@ function WelcomePage({ standalone = false }: WelcomePageProps) {
} }
} }
const handleScanWxidCandidates = async () => {
if (!dbPath) {
setError('请先选择数据库目录')
return
}
if (isScanningWxid) return
setIsScanningWxid(true)
setError('')
try {
const wxids = await window.electronAPI.dbPath.scanWxidCandidates(dbPath)
setWxidOptions(wxids)
setShowWxidSelect(true)
if (!wxids.length) {
setError('未检测到可用的账号目录,请检查路径')
}
} catch (e) {
setError(`扫描失败: ${e}`)
} finally {
setIsScanningWxid(false)
}
}
const handleAutoGetDbKey = async () => { const handleAutoGetDbKey = async () => {
if (isFetchingDbKey) return if (isFetchingDbKey) return
setIsFetchingDbKey(true) setIsFetchingDbKey(true)
@@ -556,14 +594,35 @@ function WelcomePage({ standalone = false }: WelcomePageProps) {
{currentStep.id === 'key' && ( {currentStep.id === 'key' && (
<div className="form-group"> <div className="form-group">
<label className="field-label"> (Wxid)</label> <label className="field-label"> (Wxid)</label>
<input <div className="wxid-select" ref={wxidSelectRef}>
type="text" <input
className="field-input" type="text"
placeholder="等待获取..." className="field-input"
value={wxid} placeholder="点击选择..."
readOnly value={wxid}
onChange={(e) => setWxid(e.target.value)} readOnly
/> onClick={handleScanWxidCandidates}
onChange={(e) => setWxid(e.target.value)}
/>
{showWxidSelect && wxidOptions.length > 0 && (
<div className="wxid-dropdown">
{wxidOptions.map((opt) => (
<button
key={opt.wxid}
type="button"
className={`wxid-option ${opt.wxid === wxid ? 'active' : ''}`}
onClick={() => {
setWxid(opt.wxid)
setShowWxidSelect(false)
}}
>
<span className="wxid-name">{opt.wxid}</span>
<span className="wxid-time">{formatModifiedTime(opt.modifiedTime)}</span>
</button>
))}
</div>
)}
</div>
<label className="field-label mt-4"></label> <label className="field-label mt-4"></label>
<div className="field-with-toggle"> <div className="field-with-toggle">
@@ -733,4 +792,3 @@ function WelcomePage({ standalone = false }: WelcomePageProps) {
} }
export default WelcomePage export default WelcomePage

View File

@@ -42,6 +42,7 @@ export interface ElectronAPI {
dbPath: { dbPath: {
autoDetect: () => Promise<{ success: boolean; path?: string; error?: string }> autoDetect: () => Promise<{ success: boolean; path?: string; error?: string }>
scanWxids: (rootPath: string) => Promise<WxidInfo[]> scanWxids: (rootPath: string) => Promise<WxidInfo[]>
scanWxidCandidates: (rootPath: string) => Promise<WxidInfo[]>
getDefault: () => Promise<string> getDefault: () => Promise<string>
} }
wcdb: { wcdb: {