use axios

package.json

@@ -31,6 +31,7 @@
     "dayjs": "^1.11.13",
     "express": "^5.0.0",
     "webdav": "^5.7.1",
+    "ws": "^8.18.0",
     "yaml": "^2.5.1"
   },
   "devDependencies": {

pnpm-lock.yaml

@@ -35,6 +35,9 @@ importers:
       webdav:
         specifier: ^5.7.1
         version: 5.7.1
+      ws:
+        specifier: ^8.18.0
+        version: 8.18.0
       yaml:
         specifier: ^2.5.1
         version: 2.5.1
@@ -5264,6 +5267,18 @@ packages:
   wrappy@1.0.2:
     resolution: {integrity: sha512-l4Sp/DRseor9wL6EvV2+TuQn63dMkPjZ/sp9XkghTEbV9KlPS1xUsZ3u7/IQO4wxtcFB4bgpQPRcR3QCvezPcQ==}
 
+  ws@8.18.0:
+    resolution: {integrity: sha512-8VbfWfHLbbwu3+N6OKsOMpBdT4kXPDDB9cJk2bJ6mh9ucxdlnNvH1e+roYkKmN9Nxw2yjz7VzeO9oOz2zJ04Pw==}
+    engines: {node: '>=10.0.0'}
+    peerDependencies:
+      bufferutil: ^4.0.1
+      utf-8-validate: '>=5.0.2'
+    peerDependenciesMeta:
+      bufferutil:
+        optional: true
+      utf-8-validate:
+        optional: true
+
   xmlbuilder@15.1.1:
     resolution: {integrity: sha512-yMqGBqtXyeN1e3TGYvgNgDVZ3j84W4cwkOXQswghol6APgZWaff9lnbvN7MHYJOiXsvGPXtjTYJEiC9J2wv9Eg==}
     engines: {node: '>=8.0'}
@@ -11849,6 +11864,8 @@ snapshots:
 
   wrappy@1.0.2: {}
 
+  ws@8.18.0: {}
+
   xmlbuilder@15.1.1: {}
 
   y18n@5.0.8: {}

scripts/prepare.mjs

@@ -288,7 +288,7 @@ const resolveMonitor = async () => {
   if (fs.existsSync(targetPath)) {
     fs.rmSync(targetPath, { recursive: true })
   }
-  zip.extractAllTo(resDir, true)
+  zip.extractAllTo(targetPath, true)
 
   console.log(`[INFO]: TrafficMonitor finished`)
 }

src/main/core/mihomoApi.ts

@@ -1,116 +1,78 @@
-import net from 'net'
-import { getRuntimeConfig } from './factory'
+import axios, { AxiosInstance } from 'axios'
 import { getAppConfig, getControledMihomoConfig } from '../config'
 import { mainWindow } from '..'
+import WebSocket from 'ws'
 import { tray } from '../resolve/tray'
 import { calcTraffic } from '../utils/calc'
+import { getRuntimeConfig } from './factory'
 import { join } from 'path'
 import { mihomoWorkDir } from '../utils/dirs'
 
-type HttpMethod = 'GET' | 'POST' | 'PUT' | 'DELETE' | 'PATCH'
-
-let mihomoTrafficWs: net.Socket | null = null
+let axiosIns: AxiosInstance = null!
+let mihomoTrafficWs: WebSocket | null = null
 let trafficRetry = 10
-let mihomoMemoryWs: net.Socket | null = null
+let mihomoMemoryWs: WebSocket | null = null
 let memoryRetry = 10
-let mihomoLogsWs: net.Socket | null = null
+let mihomoLogsWs: WebSocket | null = null
 let logsRetry = 10
-let mihomoConnectionsWs: net.Socket | null = null
+let mihomoConnectionsWs: WebSocket | null = null
 let connectionsRetry = 10
 
-function trimJson(data: string): string {
-  if (data.trim().length === 0) return ''
-  const start = data.indexOf('{')
-  const end = data.lastIndexOf('}')
-  return data.slice(start, end + 1)
-}
-
-async function mihomoHttp<T>(method: HttpMethod, path: string, data?: object): Promise<T> {
+export const getAxios = async (force: boolean = false): Promise<AxiosInstance> => {
   const {
     'external-controller-pipe': mihomoPipe = '\\\\.\\pipe\\MihomoParty\\mihomo',
     'external-controller-unix': mihomoUnix = 'mihomo-party.sock'
   } = await getControledMihomoConfig()
-  return new Promise((resolve, reject) => {
-    const client = net.connect(
-      process.platform === 'win32' ? mihomoPipe : join(mihomoWorkDir(), mihomoUnix)
+  if (axiosIns && !force) return axiosIns
+
+  axiosIns = axios.create({
+    baseURL: `http://localhost`,
+    socketPath: process.platform === 'win32' ? mihomoPipe : join(mihomoWorkDir(), mihomoUnix),
+    timeout: 15000
+  })
+
+  axiosIns.interceptors.response.use(
+    (response) => {
+      return response.data
+    },
+    (error) => {
+      if (error.response && error.response.data) {
+        return Promise.reject(error.response.data)
+      }
+      return Promise.reject(error)
+    }
   )
-    const parseResult = (str: string): void => {
-      try {
-        const data = str.split('\r\n\r\n')[1]
-        const json = trimJson(data)
-        if (str.includes('HTTP/1.1 4') || str.includes('HTTP/1.1 5')) {
-          reject(json ? JSON.parse(json) : data)
-        } else {
-          resolve(json ? JSON.parse(json) : undefined)
-        }
-      } catch (e) {
-        reject(e)
-      } finally {
-        client.end()
-      }
-    }
-    let buffer = ''
-    client.on('data', function (res) {
-      if (res.toString().includes('Transfer-Encoding: chunked') || buffer !== '') {
-        buffer += res.toString()
-        if (buffer.endsWith('\r\n\r\n')) {
-          parseResult(buffer)
-        }
-      } else {
-        parseResult(res.toString())
-      }
-    })
-    client.on('error', function (error) {
-      reject(error)
-    })
-    if (data) {
-      const json = JSON.stringify(data)
-      client.write(
-        `${method} ${path} HTTP/1.1\r\nHost: mihomo-party\r\nContent-Type: application/json\r\nContent-Length: ${Buffer.from(json).length}\r\n\r\n${json}`
-      )
-    } else {
-      client.write(`${method} ${path} HTTP/1.1\r\nHost: mihomo-party\r\n\r\n`)
-    }
-  })
+  return axiosIns
 }
 
-async function mihomoWs(path: string): Promise<net.Socket> {
-  const {
-    'external-controller-pipe': mihomoPipe = '\\\\.\\pipe\\MihomoParty\\mihomo',
-    'external-controller-unix': mihomoUnix = 'mihomo-party.sock'
-  } = await getControledMihomoConfig()
-  const client = net.connect(
-    process.platform === 'win32' ? mihomoPipe : join(mihomoWorkDir(), mihomoUnix)
-  )
-  client.write(
-    `GET ${path} HTTP/1.1\r\nHost: mihomo-party\r\nConnection: Upgrade\r\nUpgrade: websocket\r\nSec-WebSocket-Version: 13\r\nSec-WebSocket-Key: xxxxxxxxxxxxxxxxxxxxxxxx\r\n\r\n`
-  )
-
-  return client
-}
-
-export const mihomoVersion = async (): Promise<IMihomoVersion> => {
-  return await mihomoHttp('GET', '/version')
+export async function mihomoVersion(): Promise<IMihomoVersion> {
+  const instance = await getAxios()
+  return await instance.get('/version')
 }
 
 export const patchMihomoConfig = async (patch: Partial<IMihomoConfig>): Promise<void> => {
-  return await mihomoHttp('PATCH', '/configs', patch)
+  const instance = await getAxios()
+  return await instance.patch('/configs', patch)
 }
 
 export const mihomoCloseConnection = async (id: string): Promise<void> => {
-  return await mihomoHttp('DELETE', `/connections/${id}`)
+  const instance = await getAxios()
+  return await instance.delete(`/connections/${encodeURIComponent(id)}`)
 }
 
 export const mihomoCloseAllConnections = async (): Promise<void> => {
-  return await mihomoHttp('DELETE', '/connections')
+  const instance = await getAxios()
+  return await instance.delete('/connections')
 }
 
 export const mihomoRules = async (): Promise<IMihomoRulesInfo> => {
-  return await mihomoHttp('GET', '/rules')
+  const instance = await getAxios()
+  return await instance.get('/rules')
 }
 
 export const mihomoProxies = async (): Promise<IMihomoProxies> => {
-  const proxies = (await mihomoHttp('GET', '/proxies')) as IMihomoProxies
+  const instance = await getAxios()
+  const proxies = (await instance.get('/proxies')) as IMihomoProxies
   if (!proxies.proxies['GLOBAL']) {
     throw new Error('GLOBAL proxy not found')
   }
@@ -141,199 +103,271 @@ export const mihomoGroups = async (): Promise<IMihomoMixedGroup[]> => {
 }
 
 export const mihomoProxyProviders = async (): Promise<IMihomoProxyProviders> => {
-  return await mihomoHttp('GET', '/providers/proxies')
+  const instance = await getAxios()
+  return await instance.get('/providers/proxies')
 }
 
 export const mihomoUpdateProxyProviders = async (name: string): Promise<void> => {
-  return await mihomoHttp('PUT', `/providers/proxies/${encodeURIComponent(name)}`)
+  const instance = await getAxios()
+  return await instance.put(`/providers/proxies/${encodeURIComponent(name)}`)
 }
 
 export const mihomoRuleProviders = async (): Promise<IMihomoRuleProviders> => {
-  return await mihomoHttp('GET', '/providers/rules')
+  const instance = await getAxios()
+  return await instance.get('/providers/rules')
 }
 
 export const mihomoUpdateRuleProviders = async (name: string): Promise<void> => {
-  return await mihomoHttp('PUT', `/providers/rules/${encodeURIComponent(name)}`)
+  const instance = await getAxios()
+  return await instance.put(`/providers/rules/${encodeURIComponent(name)}`)
 }
 
 export const mihomoChangeProxy = async (group: string, proxy: string): Promise<IMihomoProxy> => {
-  return await mihomoHttp('PUT', `/proxies/${encodeURIComponent(group)}`, { name: proxy })
+  const instance = await getAxios()
+  return await instance.put(`/proxies/${encodeURIComponent(group)}`, { name: proxy })
 }
 
-export const mihomoUnfixedProxy = async (group: string): Promise<void> => {
-  return await mihomoHttp('DELETE', `/proxies/${encodeURIComponent(group)}`)
+export const mihomoUnfixedProxy = async (group: string): Promise<IMihomoProxy> => {
+  const instance = await getAxios()
+  return await instance.delete(`/proxies/${encodeURIComponent(group)}`)
 }
 
 export const mihomoUpgradeGeo = async (): Promise<void> => {
-  return await mihomoHttp('POST', '/configs/geo')
+  const instance = await getAxios()
+  return await instance.post('/configs/geo')
 }
 
 export const mihomoProxyDelay = async (proxy: string, url?: string): Promise<IMihomoDelay> => {
   const appConfig = await getAppConfig()
   const { delayTestUrl, delayTestTimeout } = appConfig
-
-  return await mihomoHttp(
-    'GET',
-    `/proxies/${encodeURIComponent(proxy)}/delay?url=${encodeURIComponent(url || delayTestUrl || 'https://www.gstatic.com/generate_204')}&timeout=${delayTestTimeout || 5000}`
-  )
+  const instance = await getAxios()
+  return await instance.get(`/proxies/${encodeURIComponent(proxy)}/delay`, {
+    params: {
+      url: url || delayTestUrl || 'https://www.gstatic.com/generate_204',
+      timeout: delayTestTimeout || 5000
+    }
+  })
 }
 
 export const mihomoGroupDelay = async (group: string, url?: string): Promise<IMihomoGroupDelay> => {
   const appConfig = await getAppConfig()
   const { delayTestUrl, delayTestTimeout } = appConfig
-  return await mihomoHttp(
-    'GET',
-    `/group/${encodeURIComponent(group)}/delay?url=${encodeURIComponent(url || delayTestUrl || 'https://www.gstatic.com/generate_204')}&timeout=${delayTestTimeout || 5000}`
-  )
+  const instance = await getAxios()
+  return await instance.get(`/group/${encodeURIComponent(group)}/delay`, {
+    params: {
+      url: url || delayTestUrl || 'https://www.gstatic.com/generate_204',
+      timeout: delayTestTimeout || 5000
+    }
+  })
 }
 
 export const mihomoUpgrade = async (): Promise<void> => {
-  return await mihomoHttp('POST', '/upgrade')
+  const instance = await getAxios()
+  return await instance.post('/upgrade')
 }
 
 export const startMihomoTraffic = async (): Promise<void> => {
   await mihomoTraffic()
 }
 
-export const stopMihomoTraffic = async (): Promise<void> => {
+export const stopMihomoTraffic = (): void => {
   if (mihomoTrafficWs) {
-    mihomoTrafficWs.end()
+    mihomoTrafficWs.removeAllListeners()
+    if (mihomoTrafficWs.readyState === WebSocket.OPEN) {
+      mihomoTrafficWs.close()
+    }
     mihomoTrafficWs = null
   }
 }
 
 const mihomoTraffic = async (): Promise<void> => {
-  stopMihomoTraffic()
-  mihomoTrafficWs = await mihomoWs('/traffic')
-  mihomoTrafficWs.on('data', (data) => {
-    try {
-      const json = JSON.parse(trimJson(data.toString())) as IMihomoTrafficInfo
+  const {
+    'external-controller-pipe': mihomoPipe = '\\\\.\\pipe\\MihomoParty\\mihomo',
+    'external-controller-unix': mihomoUnix = 'mihomo-party.sock'
+  } = await getControledMihomoConfig()
+
+  mihomoTrafficWs = new WebSocket(
+    `ws+unix:${process.platform === 'win32' ? mihomoPipe : join(mihomoWorkDir(), mihomoUnix)}:/traffic`
+  )
+
+  mihomoTrafficWs.onmessage = async (e): Promise<void> => {
+    const data = e.data as string
+    const json = JSON.parse(data) as IMihomoTrafficInfo
     trafficRetry = 10
+    try {
       mainWindow?.webContents.send('mihomoTraffic', json)
+      if (process.platform !== 'linux') {
         tray?.setToolTip(
           '↑' +
             `${calcTraffic(json.up)}/s`.padStart(9) +
             '\n↓' +
             `${calcTraffic(json.down)}/s`.padStart(9)
         )
+      }
     } catch {
       // ignore
     }
-  })
-  mihomoTrafficWs.on('close', () => {
+  }
+
+  mihomoTrafficWs.onclose = (): void => {
     if (trafficRetry) {
       trafficRetry--
       mihomoTraffic()
     }
-  })
+  }
 
-  mihomoTrafficWs.on('error', (): void => {
-    stopMihomoTraffic()
-  })
+  mihomoTrafficWs.onerror = (): void => {
+    if (mihomoTrafficWs) {
+      mihomoTrafficWs.close()
+      mihomoTrafficWs = null
+    }
+  }
 }
 
 export const startMihomoMemory = async (): Promise<void> => {
   await mihomoMemory()
 }
 
-export const stopMihomoMemory = async (): Promise<void> => {
+export const stopMihomoMemory = (): void => {
   if (mihomoMemoryWs) {
-    mihomoMemoryWs.end()
+    mihomoMemoryWs.removeAllListeners()
+    if (mihomoMemoryWs.readyState === WebSocket.OPEN) {
+      mihomoMemoryWs.close()
+    }
     mihomoMemoryWs = null
   }
 }
 
 const mihomoMemory = async (): Promise<void> => {
-  stopMihomoMemory()
-  mihomoMemoryWs = await mihomoWs('/memory')
-  mihomoMemoryWs.on('data', (data) => {
-    try {
-      const json = JSON.parse(trimJson(data.toString())) as IMihomoMemoryInfo
+  const {
+    'external-controller-pipe': mihomoPipe = '\\\\.\\pipe\\MihomoParty\\mihomo',
+    'external-controller-unix': mihomoUnix = 'mihomo-party.sock'
+  } = await getControledMihomoConfig()
+
+  mihomoMemoryWs = new WebSocket(
+    `ws+unix:${process.platform === 'win32' ? mihomoPipe : join(mihomoWorkDir(), mihomoUnix)}:/memory`
+  )
+
+  mihomoMemoryWs.onmessage = (e): void => {
+    const data = e.data as string
     memoryRetry = 10
-      mainWindow?.webContents.send('mihomoMemory', json)
+    try {
+      mainWindow?.webContents.send('mihomoMemory', JSON.parse(data) as IMihomoMemoryInfo)
     } catch {
       // ignore
     }
-  })
-  mihomoMemoryWs.on('close', () => {
+  }
+
+  mihomoMemoryWs.onclose = (): void => {
     if (memoryRetry) {
       memoryRetry--
       mihomoMemory()
     }
-  })
+  }
 
-  mihomoMemoryWs.on('error', (): void => {
-    stopMihomoMemory()
-  })
+  mihomoMemoryWs.onerror = (): void => {
+    if (mihomoMemoryWs) {
+      mihomoMemoryWs.close()
+      mihomoMemoryWs = null
+    }
+  }
 }
 
 export const startMihomoLogs = async (): Promise<void> => {
   await mihomoLogs()
 }
 
-export const stopMihomoLogs = async (): Promise<void> => {
+export const stopMihomoLogs = (): void => {
   if (mihomoLogsWs) {
-    mihomoLogsWs.end()
+    mihomoLogsWs.removeAllListeners()
+    if (mihomoLogsWs.readyState === WebSocket.OPEN) {
+      mihomoLogsWs.close()
+    }
     mihomoLogsWs = null
   }
 }
 
 const mihomoLogs = async (): Promise<void> => {
-  stopMihomoLogs()
-  const { 'log-level': logLevel } = await getControledMihomoConfig()
-  mihomoLogsWs = await mihomoWs(`/logs?level=${logLevel}`)
-  mihomoLogsWs.on('data', (data) => {
-    try {
-      const json = JSON.parse(trimJson(data.toString())) as IMihomoLogInfo
+  const {
+    'external-controller-pipe': mihomoPipe = '\\\\.\\pipe\\MihomoParty\\mihomo',
+    'external-controller-unix': mihomoUnix = 'mihomo-party.sock',
+    'log-level': logLevel = 'info'
+  } = await getControledMihomoConfig()
+
+  mihomoLogsWs = new WebSocket(
+    `ws+unix:${process.platform === 'win32' ? mihomoPipe : join(mihomoWorkDir(), mihomoUnix)}:/logs?level=${logLevel}`
+  )
+
+  mihomoLogsWs.onmessage = (e): void => {
+    const data = e.data as string
     logsRetry = 10
-      mainWindow?.webContents.send('mihomoLogs', json)
+    try {
+      mainWindow?.webContents.send('mihomoLogs', JSON.parse(data) as IMihomoLogInfo)
     } catch {
       // ignore
     }
-  })
-  mihomoLogsWs.on('close', () => {
+  }
+
+  mihomoLogsWs.onclose = (): void => {
     if (logsRetry) {
       logsRetry--
       mihomoLogs()
     }
-  })
+  }
 
-  mihomoLogsWs.on('error', (): void => {
-    stopMihomoLogs()
-  })
+  mihomoLogsWs.onerror = (): void => {
+    if (mihomoLogsWs) {
+      mihomoLogsWs.close()
+      mihomoLogsWs = null
+    }
+  }
 }
 
 export const startMihomoConnections = async (): Promise<void> => {
   await mihomoConnections()
 }
 
-export const stopMihomoConnections = async (): Promise<void> => {
+export const stopMihomoConnections = (): void => {
   if (mihomoConnectionsWs) {
-    mihomoConnectionsWs.end()
+    mihomoConnectionsWs.removeAllListeners()
+    if (mihomoConnectionsWs.readyState === WebSocket.OPEN) {
+      mihomoConnectionsWs.close()
+    }
     mihomoConnectionsWs = null
   }
 }
 
 const mihomoConnections = async (): Promise<void> => {
-  stopMihomoConnections()
-  mihomoConnectionsWs = await mihomoWs('/connections')
-  mihomoConnectionsWs.on('data', (data) => {
-    try {
-      const json = JSON.parse(trimJson(data.toString())) as IMihomoConnectionsInfo
+  const {
+    'external-controller-pipe': mihomoPipe = '\\\\.\\pipe\\MihomoParty\\mihomo',
+    'external-controller-unix': mihomoUnix = 'mihomo-party.sock'
+  } = await getControledMihomoConfig()
+
+  mihomoConnectionsWs = new WebSocket(
+    `ws+unix:${process.platform === 'win32' ? mihomoPipe : join(mihomoWorkDir(), mihomoUnix)}:/connections`
+  )
+
+  mihomoConnectionsWs.onmessage = (e): void => {
+    const data = e.data as string
     connectionsRetry = 10
-      mainWindow?.webContents.send('mihomoConnections', json)
+    try {
+      mainWindow?.webContents.send('mihomoConnections', JSON.parse(data) as IMihomoConnectionsInfo)
     } catch {
       // ignore
     }
-  })
-  mihomoConnectionsWs.on('close', () => {
+  }
+
+  mihomoConnectionsWs.onclose = (): void => {
     if (connectionsRetry) {
       connectionsRetry--
       mihomoConnections()
     }
-  })
+  }
 
-  mihomoConnectionsWs.on('error', (): void => {
-    stopMihomoConnections()
-  })
+  mihomoConnectionsWs.onerror = (): void => {
+    if (mihomoConnectionsWs) {
+      mihomoConnectionsWs.close()
+      mihomoConnectionsWs = null
+    }
+  }
 }