feat(scanner): enhance project scanning with JetBrains and VSCode recent projects detection

Author: MidnightCrowing

electron/ipc.ts

@@ -32,9 +32,9 @@ ipcMain.handle('project:import', () => projectSvc.importProjects())
 ipcMain.handle('project:export', () => projectSvc.exportProjects())
 
 // scanner
-ipcMain.handle('scanner:scan', (_e, payload) => scanner.scan(payload))
 ipcMain.handle('scanner:start', (e, payload) => scanner.start(e, payload))
 ipcMain.handle('scanner:stop', (_e, sessionId) => scanner.stop(sessionId))
+ipcMain.handle('scanner:detect-jb-config-root-path', () => scanner.detectJetBrainsConfigRootPath())
 ipcMain.handle('scanner:detect-vsc-state-db-path', () => scanner.detectVscodeStateDbPath())
 
 // system

electron/main.ts

@@ -5,7 +5,7 @@ import { fileURLToPath } from 'node:url'
 
 import { app, BrowserWindow, globalShortcut, Menu } from 'electron'
 
-// @ts-expect-error -- 知道要改tsconfig但不想改
+// @ts-expect-error -- ignore
 const __filename = fileURLToPath(import.meta.url)
 const __dirname = path.dirname(__filename)
 

electron/preload.ts

@@ -19,7 +19,7 @@ interface ProjectAnalyzeProgress {
   stage: ProjectAnalyzeStage
 }
 
-interface ScanStartPayload { roots: string[], existingPaths: string[] }
+interface ScanStartPayload { roots: string[], existingPaths: string[], projectPaths?: string[] }
 interface ScannerItemEvent { sessionId: number, item: any }
 interface ScannerDoneEvent { sessionId: number }
 interface ScannerErrorEvent { sessionId: number, error: string }
@@ -87,11 +87,7 @@ const api = {
   // ===== update =====
   checkUpdate: () => ipcRenderer.invoke('update:check'),
 
-  // ===== scanner (batch) =====
-  scanProjects: (payload: ScanStartPayload) =>
-    ipcRenderer.invoke('scanner:scan', payload),
-
-  // ===== scanner (stream) =====
+  // ===== scanner =====
   startProjectScan: (payload: ScanStartPayload) =>
     ipcRenderer.invoke('scanner:start', payload),
 
@@ -107,6 +103,9 @@ const api = {
   onScannerError: (cb: (data: ScannerErrorEvent) => void): Unlisten =>
     onIpc('scanner:error', cb),
 
+  detectJetBrainsConfigRootPath: () =>
+    ipcRenderer.invoke('scanner:detect-jb-config-root-path'),
+
   detectVscodeStateDbPath: () =>
     ipcRenderer.invoke('scanner:detect-vsc-state-db-path'),
 

electron/services/projectService.ts

@@ -14,7 +14,7 @@ import type { LinguistResult } from '../utils/linguist'
 export async function analyzeProject(event: IpcMainInvokeEvent, folderPath: string): Promise<LinguistResult | { error: string }> {
   return await new Promise((resolve) => {
     try {
-      // @ts-expect-error -- 知道要改tsconfig但不想改
+      // @ts-expect-error -- ignore
       const workerUrl = new URL('../workers/linguistAnalyze.worker.js', import.meta.url)
       const worker = new Worker(workerUrl, { type: 'module', workerData: { folderPath } } as any)
 

electron/services/scannerService.ts

@@ -6,56 +6,26 @@ import { Worker } from 'node:worker_threads'
 import type { IpcMainInvokeEvent } from 'electron'
 import { app } from 'electron'
 
-interface ScanPayload {
+export interface ScanPayload {
+  // file system roots scanning
+  rootsEnabled: boolean
   roots: string[]
-  existingPaths: string[]
-}
-interface ScanResultItem {
-  path: string
-  name: string
-  mainLang?: string
-  mainLangColor?: `#${string}`
-  langGroup?: Array<{ text: string, color: `#${string}`, percentage: number }>
-  error?: string
-}
 
-export async function scan(payload: ScanPayload): Promise<ScanResultItem[]> {
-  return new Promise((resolve, reject) => {
-    try {
-      // @ts-expect-error -- 知道要改tsconfig但不想改
-      const workerUrl = new URL('../workers/projectScanner.worker.js', import.meta.url)
-      const worker = new Worker(workerUrl, { type: 'module', workerData: payload } as any)
-
-      const items: ScanResultItem[] = []
-      const cleanup = () => worker.removeAllListeners()
-
-      worker.on('message', (msg: any) => {
-        if (msg?.type === 'item' && msg.item) {
-          items.push(msg.item as ScanResultItem)
-        }
-        else if (msg?.type === 'done') {
-          cleanup()
-          resolve(items)
-        }
-        else if (msg?.type === 'error') {
-          cleanup()
-          reject(new Error(msg.error || 'scan error'))
-        }
-      })
-
-      worker.once('error', (err) => {
-        cleanup()
-        reject(err)
-      })
-      worker.once('exit', (code) => {
-        if (code !== 0)
-          reject(new Error(`scanner worker exited with code ${code}`))
-      })
-    }
-    catch (e) {
-      reject(e)
-    }
-  })
+  // IDE scanning toggles
+  ideEnabled: boolean
+  jetbrains: {
+    enabled: boolean
+    configRootPath: string
+  }
+  vscode: {
+    enabled: boolean
+    stateDbPath: string
+  }
+  visualStudio: {
+    enabled: boolean
+  }
+
+  existingPaths: string[]
 }
 
 // 流式扫描：创建会话，逐条回推
@@ -64,7 +34,7 @@ let nextSessionId = 1
 
 export function start(event: IpcMainInvokeEvent, payload: ScanPayload) {
   const sessionId = nextSessionId++
-  // @ts-expect-error -- 知道要改tsconfig但不想改
+  // @ts-expect-error -- ignore
   const workerUrl = new URL('../workers/projectScanner.worker.js', import.meta.url)
   const worker = new Worker(workerUrl, { type: 'module', workerData: payload } as any)
 
@@ -121,6 +91,48 @@ export function stop(sessionId: number) {
   return { stopped: false }
 }
 
+/**
+ * 自动检测 JetBrains 配置根目录
+ * 返回例如：
+ *  - Windows: C:\\Users\\<user>\\AppData\\Roaming\\JetBrains
+ *  - macOS:   ~/Library/Application Support/JetBrains
+ *  - Linux:   ~/.config/JetBrains 或 ~/.JetBrains（优先 .config/JetBrains）
+ */
+export function detectJetBrainsConfigRootPath() {
+  const candidates: string[] = []
+
+  switch (process.platform) {
+    case 'win32': {
+      // Roaming AppData
+      candidates.push(path.join(app.getPath('appData'), 'JetBrains'))
+      break
+    }
+    case 'darwin': {
+      candidates.push(path.join(app.getPath('home'), 'Library', 'Application Support', 'JetBrains'))
+      break
+    }
+    default: {
+      // linux
+      const home = app.getPath('home')
+      candidates.push(path.join(home, '.config', 'JetBrains'))
+      candidates.push(path.join(home, '.JetBrains')) // 一些老版本
+      break
+    }
+  }
+
+  for (const p of candidates) {
+    try {
+      if (fs.existsSync(p) && fs.statSync(p).isDirectory())
+        return p
+    }
+    catch {
+      // ignore
+    }
+  }
+
+  return null
+}
+
 /**
  * 自动检测 VSCode state.vscdb 路径
  * 全平台通用

electron/utils/recent/jetbrainsRecent.ts

@@ -0,0 +1,126 @@
+import fs from 'node:fs'
+import os from 'node:os'
+import path from 'node:path'
+import process from 'node:process'
+
+import { XMLParser } from 'fast-xml-parser'
+
+import { uniqExistingDirs } from './shared'
+
+export interface JetbrainsRecentEntry {
+  path: string
+  ide: string // 例如 "PyCharm", "IntelliJIdea"
+}
+
+const ideNameMap = {
+  'IntelliJIdea': 'intellij-idea',
+  'PyCharm': 'pycharm',
+  'PhpStorm': 'phpstorm',
+  'GoLand': 'goLand',
+  'Rider': 'rider',
+  'CLion': 'clion',
+  'RustRover': 'rust-rover',
+  'WebStorm': 'webstorm',
+  'RubyMine': 'rubymine',
+  'Aqua': 'aqua',
+  'Fleet': 'fleet',
+  'AndroidStudio': 'android-studio',
+} as const
+
+/**
+ * 获取 JetBrains 所有 IDE 的最近项目列表
+ * @param configRoot JetBrains 配置根目录（例如 C:\Users\xxx\AppData\Roaming\JetBrains）
+ */
+function getAllJetbrainsRecentProjects(
+  configRoot: string,
+): JetbrainsRecentEntry[] {
+  if (!fs.existsSync(configRoot)) {
+    throw new Error(`JetBrains config root not found: ${configRoot}`)
+  }
+
+  const parser = new XMLParser({
+    ignoreAttributes: false,
+    attributeNamePrefix: '',
+  })
+
+  const userHome = os.homedir()
+  const normalizePath = (p: string): string =>
+    p.replace(/\$USER_HOME\$/g, userHome).replace(/\\/g, '/')
+
+  const entries: JetbrainsRecentEntry[] = []
+
+  // 遍历 JetBrains 目录下的所有子目录
+  for (const subdir of fs.readdirSync(configRoot)) {
+    const fullDir = path.join(configRoot, subdir)
+    if (!fs.statSync(fullDir).isDirectory())
+      continue
+
+    const xmlPath = path.join(fullDir, 'options', 'recentProjects.xml')
+    if (!fs.existsSync(xmlPath))
+      continue
+
+    const xmlContent = fs.readFileSync(xmlPath, 'utf-8')
+    const json = parser.parse(xmlContent)
+
+    const ideNameRaw = subdir.replace(/\d.*$/, '') // 去掉版本号，比如 "PyCharm2025.1" -> "PyCharm"
+    const ideName = ideNameMap[ideNameRaw] ?? null
+
+    const options = json?.application?.component?.option
+    if (!options)
+      continue
+
+    // 可能是数组，也可能是单个对象
+    const optionList = Array.isArray(options) ? options : [options]
+
+    const additionalInfo = optionList.find(opt => opt.name === 'additionalInfo')
+    if (!additionalInfo?.map?.entry)
+      continue
+
+    const projectEntries = additionalInfo.map.entry
+
+    if (Array.isArray(projectEntries)) {
+      for (const e of projectEntries) {
+        if (e.key)
+          entries.push({ path: normalizePath(e.key), ide: ideName })
+      }
+    }
+    else if (projectEntries?.key) {
+      entries.push({ path: normalizePath(projectEntries.key), ide: ideName })
+    }
+  }
+
+  // 去重：按 path 唯一
+  const seen = new Set<string>()
+  const uniqueEntries: JetbrainsRecentEntry[] = []
+  for (const e of entries) {
+    if (!seen.has(e.path)) {
+      seen.add(e.path)
+      uniqueEntries.push(e)
+    }
+  }
+
+  return uniqueEntries
+}
+
+export function collectFromJetbrains(configRoot: string | null): JetbrainsRecentEntry[] {
+  if (!configRoot) {
+    return []
+  }
+  let entries: JetbrainsRecentEntry[] = []
+  try {
+    entries = getAllJetbrainsRecentProjects(configRoot)
+  }
+  catch {
+    return []
+  }
+  const candidates: JetbrainsRecentEntry[] = []
+  for (const e of entries) {
+    const p = e.path
+    // Skip placeholders and JetBrains light-edit/scratches etc.
+    if (!p || /\$[A-Z_]+\$/.test(p) || /light-edit|scratches/i.test(p))
+      continue
+    const norm = process.platform === 'win32' ? p.replace(/\//g, '\\') : p
+    candidates.push({ path: norm, ide: e.ide })
+  }
+  return uniqExistingDirs(candidates)
+}

electron/utils/recent/shared.ts

@@ -0,0 +1,37 @@
+import fs from 'node:fs'
+import path from 'node:path'
+
+export function uniqExistingDirs(
+  paths: Iterable<{ path: string, ide: string | null }>,
+): { path: string, ide: string | null }[] {
+  const seen = new Set<string>()
+  const out: { path: string, ide: string | null }[] = []
+  for (const item of paths) {
+    const p = item.path
+    if (!p)
+      continue
+    let fp = p
+    // Normalize windows backslashes, remove trailing slashes
+    fp = path.normalize(fp)
+
+    try {
+      if (fs.existsSync(fp) && fs.statSync(fp).isDirectory()) {
+        // 统一大小写来判断唯一性（比如统一转小写）
+        const key = fp.toLowerCase()
+
+        if (!seen.has(key)) {
+          seen.add(key)
+
+          // 输出时修正盘符大写（只改第一个字母）
+          if (/^[a-z]:/.test(fp)) {
+            fp = fp.charAt(0).toUpperCase() + fp.slice(1)
+          }
+
+          out.push({ path: fp, ide: item.ide })
+        }
+      }
+    }
+    catch {}
+  }
+  return out
+}