79 lines
2.7 KiB
JavaScript
79 lines
2.7 KiB
JavaScript
// main/services/iosai.js
|
||
const { app } = require('electron')
|
||
const path = require('path')
|
||
const fs = require('fs')
|
||
const { exec } = require('child_process')
|
||
const axios = require('axios')
|
||
|
||
let iosAIProcess = null
|
||
let userData = { tenantId: null, userId: null, tokenValue: null } // 如果需要共享，请提供 setter
|
||
|
||
function resolveIOSAIPath() {
|
||
const candidates = [
|
||
path.join(path.dirname(process.execPath), 'iOSAI', 'IOSAI.exe'),
|
||
path.join(process.resourcesPath || '', 'iOSAI', 'IOSAI.exe'),
|
||
path.join(__dirname, '..', 'iOSAI', 'IOSAI.exe'),
|
||
path.join(process.cwd(), 'iOSAI', 'IOSAI.exe'),
|
||
]
|
||
for (const p of candidates) {
|
||
try { if (fs.existsSync(p)) return p } catch { }
|
||
}
|
||
return null
|
||
}
|
||
|
||
function startIOSAIExecutable() {
|
||
if (process.platform !== 'win32') {
|
||
console.warn('[IOSAI] 非 Windows，跳过')
|
||
return
|
||
}
|
||
const exePath = resolveIOSAIPath()
|
||
if (!exePath) return console.error('[IOSAI] 未找到 IOSAI.exe')
|
||
|
||
const exeDir = path.dirname(exePath)
|
||
const exeFile = path.basename(exePath)
|
||
const cmd = `start "" /D "${exeDir}" "${exeFile}"`
|
||
|
||
try {
|
||
exec(cmd, { cwd: exeDir, windowsHide: false }, (err) => {
|
||
if (err) console.error('[IOSAI] 启动失败：', err)
|
||
else console.log('[IOSAI] 启动命令已执行')
|
||
})
|
||
} catch (e) { console.error('[IOSAI] 启动异常：', e) }
|
||
}
|
||
|
||
async function killIOSAI() {
|
||
try {
|
||
if (userData.userId && userData.tenantId) {
|
||
await axios.post('https://crawlclient.api.yolozs.com/api/user/aiChat-logout',
|
||
{ userId: userData.userId, tenantId: userData.tenantId },
|
||
{ headers: { 'Content-Type': 'application/json', 'vvtoken': userData.tokenValue } }
|
||
).catch(e => console.log('[IOSAI] 登出失败', e?.message))
|
||
}
|
||
if (iosAIProcess?.pid) {
|
||
exec(`taskkill /PID ${iosAIProcess.pid} /T /F`, (err) => {
|
||
if (err) {
|
||
exec('taskkill /IM IOSAI.exe /F')
|
||
}
|
||
})
|
||
} else {
|
||
exec('taskkill /IM IOSAI.exe /F')
|
||
exec('taskkill /IM iproxy.exe /F')
|
||
exec('taskkill /IM tidevice.exe /F')
|
||
}
|
||
} catch { }
|
||
iosAIProcess = null
|
||
}
|
||
|
||
function isProcessRunningWin(exeName) {
|
||
return new Promise((resolve) => {
|
||
if (process.platform !== 'win32') return resolve(true)
|
||
const cmd = `tasklist /FI "IMAGENAME eq ${exeName}" /FO CSV /NH`
|
||
exec(cmd, { windowsHide: true }, (err, stdout) => {
|
||
if (err || !stdout) return resolve(false)
|
||
resolve(stdout.toLowerCase().includes(`"${exeName.toLowerCase()}"`))
|
||
})
|
||
})
|
||
}
|
||
|
||
module.exports = { startIOSAIExecutable, killIOSAI, isProcessRunningWin }
|