support change data dir

Author: pompurin404

src/main/config/controledMihomo.ts

@@ -2,7 +2,7 @@ import { controledMihomoConfigPath } from '../utils/dirs'
 import { readFile, writeFile } from 'fs/promises'
 import yaml from 'yaml'
 import { getAxios, startMihomoMemory, startMihomoTraffic } from '../core/mihomoApi'
-import { generateProfile } from '../resolve/factory'
+import { generateProfile } from '../core/factory'
 import { getAppConfig } from './app'
 import { defaultControledMihomoConfig } from '../utils/template'
 

src/main/resolve/factory.ts src/main/core/factory.ts

@@ -8,6 +8,7 @@ import {
 import { mihomoWorkConfigPath } from '../utils/dirs'
 import yaml from 'yaml'
 import fs from 'fs'
+import { readFile } from 'fs/promises'
 
 export async function generateProfile(): Promise<void> {
   const { current } = await getProfileConfig()
@@ -59,3 +60,11 @@ function runOverrideScript(profile: IMihomoConfig, script: string): IMihomoConfi
     return profile
   }
 }
+
+export async function getRuntimeConfigStr(): Promise<string> {
+  return await readFile(mihomoWorkConfigPath(), 'utf8')
+}
+
+export async function getRuntimeConfig(): Promise<IMihomoConfig> {
+  return yaml.parse(await getRuntimeConfigStr())
+}

src/main/core/manager.ts

@@ -6,7 +6,7 @@ import {
   mihomoWorkConfigPath,
   mihomoWorkDir
 } from '../utils/dirs'
-import { generateProfile } from '../resolve/factory'
+import { generateProfile } from './factory'
 import { getAppConfig, patchAppConfig } from '../config'
 import { dialog, safeStorage } from 'electron'
 import { pauseWebsockets } from './mihomoApi'

src/main/core/mihomoApi.ts

@@ -2,7 +2,7 @@ import axios, { AxiosInstance } from 'axios'
 import { getAppConfig, getControledMihomoConfig } from '../config'
 import { mainWindow } from '..'
 import WebSocket from 'ws'
-import { tray } from './tray'
+import { tray } from '../resolve/tray'
 import { calcTraffic } from '../utils/calc'
 
 let axiosIns: AxiosInstance = null!

src/main/index.ts

@@ -2,14 +2,14 @@ import { electronApp, optimizer, is } from '@electron-toolkit/utils'
 import { registerIpcMainHandlers } from './utils/ipc'
 import windowStateKeeper from 'electron-window-state'
 import { app, shell, BrowserWindow, Menu, dialog, Notification } from 'electron'
+import { startMihomoMemory, stopMihomoMemory } from './core/mihomoApi'
+import { addProfileItem, getAppConfig } from './config'
 import { stopCore } from './core/manager'
-import { triggerSysProxy } from './resolve/sysproxy'
+import { triggerSysProxy } from './sys/sysproxy'
 import icon from '../../resources/icon.png?asset'
-import { createTray } from './core/tray'
-import { init } from './resolve/init'
-import { addProfileItem, getAppConfig } from './config'
+import { createTray } from './resolve/tray'
+import { init } from './utils/init'
 import { join } from 'path'
-import { startMihomoMemory, stopMihomoMemory } from './core/mihomoApi'
 
 export let mainWindow: BrowserWindow | null = null
 export let destroyTimer: NodeJS.Timeout | null = null

src/main/core/tray.ts src/main/resolve/tray.ts

@@ -7,11 +7,11 @@ import {
 import icoIcon from '../../../resources/icon.ico?asset'
 import pngIcon from '../../../resources/icon.png?asset'
 import templateIcon from '../../../resources/iconTemplate.png?asset'
-import { patchMihomoConfig } from './mihomoApi'
+import { patchMihomoConfig } from '../core/mihomoApi'
 import { mainWindow, showMainWindow } from '..'
 import { app, ipcMain, Menu, nativeImage, shell, Tray } from 'electron'
 import { dataDir, logDir, mihomoCoreDir, mihomoWorkDir } from '../utils/dirs'
-import { triggerSysProxy } from '../resolve/sysproxy'
+import { triggerSysProxy } from '../sys/sysproxy'
 
 export let tray: Tray | null = null
 
@@ -105,7 +105,7 @@ const buildContextMenu = async (): Promise<Menu> => {
         {
           type: 'normal',
           label: '应用目录',
-          click: (): Promise<string> => shell.openPath(dataDir)
+          click: (): Promise<string> => shell.openPath(dataDir())
         },
         {
           type: 'normal',

src/main/resolve/autoRun.ts src/main/sys/autoRun.ts

@@ -76,7 +76,7 @@ export async function checkAutoRun(): Promise<boolean> {
 export async function enableAutoRun(): Promise<void> {
   if (process.platform === 'win32') {
     const execPromise = promisify(exec)
-    const taskFilePath = path.join(dataDir, `${appName}.xml`)
+    const taskFilePath = path.join(dataDir(), `${appName}.xml`)
     await writeFile(taskFilePath, taskXml)
     await execPromise(`schtasks /create /tn "${appName}" /xml "${taskFilePath}" /f`)
   }

src/main/sys/misc.ts

@@ -0,0 +1,43 @@
+import { exec, execFile } from 'child_process'
+import { dialog } from 'electron'
+import { readFile } from 'fs/promises'
+import path from 'path'
+import { promisify } from 'util'
+import { exePath, mihomoCorePath, resourcesDir } from '../utils/dirs'
+
+export function getFilePath(ext: string[]): string[] | undefined {
+  return dialog.showOpenDialogSync({
+    title: '选择订阅文件',
+    filters: [{ name: `${ext} file`, extensions: ext }],
+    properties: ['openFile']
+  })
+}
+
+export async function readTextFile(filePath: string): Promise<string> {
+  return await readFile(filePath, 'utf8')
+}
+
+export async function openUWPTool(): Promise<void> {
+  const execFilePromise = promisify(execFile)
+  const uwpToolPath = path.join(resourcesDir(), 'files', 'enableLoopback.exe')
+  await execFilePromise(uwpToolPath)
+}
+
+export async function setupFirewall(): Promise<void> {
+  const execPromise = promisify(exec)
+  const removeCommand = `
+  Remove-NetFirewallRule -DisplayName "mihomo" -ErrorAction SilentlyContinue
+  Remove-NetFirewallRule -DisplayName "mihomo-alpha" -ErrorAction SilentlyContinue
+  Remove-NetFirewallRule -DisplayName "Mihomo Party" -ErrorAction SilentlyContinue
+  `
+  const createCommand = `
+  New-NetFirewallRule -DisplayName "mihomo" -Direction Inbound -Action Allow -Program "${mihomoCorePath('mihomo')}" -Enabled True -Profile Any -ErrorAction SilentlyContinue
+  New-NetFirewallRule -DisplayName "mihomo-alpha" -Direction Inbound -Action Allow -Program "${mihomoCorePath('mihomo-alpha')}" -Enabled True -Profile Any -ErrorAction SilentlyContinue
+  New-NetFirewallRule -DisplayName "Mihomo Party" -Direction Inbound -Action Allow -Program "${exePath()}" -Enabled True -Profile Any -ErrorAction SilentlyContinue
+  `
+
+  if (process.platform === 'win32') {
+    await execPromise(removeCommand, { shell: 'powershell' })
+    await execPromise(createCommand, { shell: 'powershell' })
+  }
+}

src/main/resolve/sysproxy.ts src/main/sys/sysproxy.ts

@@ -1,6 +1,6 @@
 import { triggerAutoProxy, triggerManualProxy } from '@mihomo-party/sysproxy'
 import { getAppConfig, getControledMihomoConfig } from '../config'
-import { pacPort } from './server'
+import { pacPort } from '../resolve/server'
 
 let defaultBypass: string[]
 

src/main/utils/dirs.ts

@@ -1,10 +1,37 @@
 import { is } from '@electron-toolkit/utils'
 import { app } from 'electron'
+import fs from 'fs'
+import { rm, writeFile } from 'fs/promises'
 import path from 'path'
 
-export const dataDir = app.getPath('userData')
 export const homeDir = app.getPath('home')
 
+export function isPortable(): boolean {
+  return fs.existsSync(path.join(exeDir(), 'PORTABLE'))
+}
+
+export async function setPortable(portable: boolean): Promise<void> {
+  if (portable) {
+    await writeFile(path.join(exeDir(), 'PORTABLE'), '')
+  } else {
+    await rm(path.join(exeDir(), 'PORTABLE'))
+  }
+  app.relaunch()
+  app.quit()
+}
+
+export function dataDir(): string {
+  if (isPortable()) {
+    return path.join(exeDir(), 'data')
+  } else {
+    return app.getPath('userData')
+  }
+}
+
+export function exeDir(): string {
+  return path.dirname(exePath())
+}
+
 export function exePath(): string {
   return app.getPath('exe')
 }
@@ -35,51 +62,51 @@ export function mihomoCorePath(core: string): string {
 }
 
 export function appConfigPath(): string {
-  return path.join(dataDir, 'config.yaml')
+  return path.join(dataDir(), 'config.yaml')
 }
 
 export function controledMihomoConfigPath(): string {
-  return path.join(dataDir, 'mihomo.yaml')
+  return path.join(dataDir(), 'mihomo.yaml')
 }
 
 export function profileConfigPath(): string {
-  return path.join(dataDir, 'profile.yaml')
+  return path.join(dataDir(), 'profile.yaml')
 }
 
 export function profilesDir(): string {
-  return path.join(dataDir, 'profiles')
+  return path.join(dataDir(), 'profiles')
 }
 
 export function profilePath(id: string): string {
   return path.join(profilesDir(), `${id}.yaml`)
 }
 
 export function overrideDir(): string {
-  return path.join(dataDir, 'override')
+  return path.join(dataDir(), 'override')
 }
 
 export function overrideConfigPath(): string {
-  return path.join(dataDir, 'override.yaml')
+  return path.join(dataDir(), 'override.yaml')
 }
 
 export function overridePath(id: string): string {
   return path.join(overrideDir(), `${id}.js`)
 }
 
 export function mihomoWorkDir(): string {
-  return path.join(dataDir, 'work')
+  return path.join(dataDir(), 'work')
 }
 
 export function mihomoTestDir(): string {
-  return path.join(dataDir, 'test')
+  return path.join(dataDir(), 'test')
 }
 
 export function mihomoWorkConfigPath(): string {
   return path.join(mihomoWorkDir(), 'config.yaml')
 }
 
 export function logDir(): string {
-  return path.join(dataDir, 'logs')
+  return path.join(dataDir(), 'logs')
 }
 
 export function logPath(): string {

src/main/resolve/init.ts src/main/utils/init.ts

@@ -11,29 +11,29 @@ import {
   profilePath,
   profilesDir,
   resourcesFilesDir
-} from '../utils/dirs'
+} from './dirs'
 import {
   defaultConfig,
   defaultControledMihomoConfig,
   defaultOverrideConfig,
   defaultProfile,
   defaultProfileConfig
-} from '../utils/template'
+} from './template'
 import yaml from 'yaml'
 import { mkdir, writeFile, copyFile } from 'fs/promises'
 import { existsSync } from 'fs'
 import path from 'path'
-import { startPacServer } from './server'
-import { triggerSysProxy } from './sysproxy'
+import { startPacServer } from '../resolve/server'
+import { triggerSysProxy } from '../sys/sysproxy'
 import { getAppConfig } from '../config'
 import { app } from 'electron'
 import { startCore } from '../core/manager'
 import { initProfileUpdater } from '../core/profileUpdater'
 import { startMihomoTraffic } from '../core/mihomoApi'
 
 async function initDirs(): Promise<void> {
-  if (!existsSync(dataDir)) {
-    await mkdir(dataDir)
+  if (!existsSync(dataDir())) {
+    await mkdir(dataDir())
   }
   if (!existsSync(profilesDir())) {
     await mkdir(profilesDir())

src/main/utils/ipc.ts

@@ -1,4 +1,4 @@
-import { app, dialog, ipcMain, safeStorage } from 'electron'
+import { app, ipcMain, safeStorage } from 'electron'
 import {
   mihomoChangeProxy,
   mihomoCloseAllConnections,
@@ -19,7 +19,7 @@ import {
   stopMihomoConnections,
   stopMihomoLogs
 } from '../core/mihomoApi'
-import { checkAutoRun, disableAutoRun, enableAutoRun } from '../resolve/autoRun'
+import { checkAutoRun, disableAutoRun, enableAutoRun } from '../sys/autoRun'
 import {
   getAppConfig,
   patchAppConfig,
@@ -45,14 +45,11 @@ import {
   updateOverrideItem
 } from '../config'
 import { isEncryptionAvailable, manualGrantCorePermition, restartCore } from '../core/manager'
-import { triggerSysProxy } from '../resolve/sysproxy'
+import { triggerSysProxy } from '../sys/sysproxy'
 import { checkUpdate } from '../resolve/autoUpdater'
-import { exePath, mihomoCorePath, mihomoWorkConfigPath, resourcesDir } from './dirs'
-import { exec, execFile } from 'child_process'
-import yaml from 'yaml'
-import path from 'path'
-import { promisify } from 'util'
-import { readFile } from 'fs/promises'
+import { getFilePath, openUWPTool, readTextFile, setupFirewall } from '../sys/misc'
+import { getRuntimeConfig, getRuntimeConfigStr } from '../core/factory'
+import { isPortable, setPortable } from './dirs'
 
 function ipcErrorWrapper<T>( // eslint-disable-next-line @typescript-eslint/no-explicit-any
   fn: (...args: any[]) => Promise<T> // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -138,50 +135,7 @@ export function registerIpcMainHandlers(): void {
   ipcMain.handle('platform', () => process.platform)
   ipcMain.handle('openUWPTool', ipcErrorWrapper(openUWPTool))
   ipcMain.handle('setupFirewall', ipcErrorWrapper(setupFirewall))
+  ipcMain.handle('setPortable', (_e, portable) => ipcErrorWrapper(setPortable)(portable))
+  ipcMain.handle('isPortable', isPortable)
   ipcMain.handle('quitApp', () => app.quit())
 }
-
-function getFilePath(ext: string[]): string[] | undefined {
-  return dialog.showOpenDialogSync({
-    title: '选择订阅文件',
-    filters: [{ name: `${ext} file`, extensions: ext }],
-    properties: ['openFile']
-  })
-}
-
-async function readTextFile(filePath: string): Promise<string> {
-  return await readFile(filePath, 'utf8')
-}
-
-async function getRuntimeConfigStr(): Promise<string> {
-  return readFile(mihomoWorkConfigPath(), 'utf8')
-}
-
-async function getRuntimeConfig(): Promise<IMihomoConfig> {
-  return yaml.parse(await getRuntimeConfigStr())
-}
-
-async function openUWPTool(): Promise<void> {
-  const execFilePromise = promisify(execFile)
-  const uwpToolPath = path.join(resourcesDir(), 'files', 'enableLoopback.exe')
-  await execFilePromise(uwpToolPath)
-}
-
-async function setupFirewall(): Promise<void> {
-  const execPromise = promisify(exec)
-  const removeCommand = `
-  Remove-NetFirewallRule -DisplayName "mihomo" -ErrorAction SilentlyContinue
-  Remove-NetFirewallRule -DisplayName "mihomo-alpha" -ErrorAction SilentlyContinue
-  Remove-NetFirewallRule -DisplayName "Mihomo Party" -ErrorAction SilentlyContinue
-  `
-  const createCommand = `
-  New-NetFirewallRule -DisplayName "mihomo" -Direction Inbound -Action Allow -Program "${mihomoCorePath('mihomo')}" -Enabled True -Profile Any -ErrorAction SilentlyContinue
-  New-NetFirewallRule -DisplayName "mihomo-alpha" -Direction Inbound -Action Allow -Program "${mihomoCorePath('mihomo-alpha')}" -Enabled True -Profile Any -ErrorAction SilentlyContinue
-  New-NetFirewallRule -DisplayName "Mihomo Party" -Direction Inbound -Action Allow -Program "${exePath()}" -Enabled True -Profile Any -ErrorAction SilentlyContinue
-  `
-
-  if (process.platform === 'win32') {
-    await execPromise(removeCommand, { shell: 'powershell' })
-    await execPromise(createCommand, { shell: 'powershell' })
-  }
-}