@civilization/dsh-workspace-files 0.3.0 → 0.3.2

package/src/files.ts

@@ -17,16 +17,37 @@ export interface WorkspaceFilesResult {
   totalDirs: number
 }
 
+// 常见大型构建产物、虚拟环境、开发缓存与三方依赖黑名单目录
 const DEFAULT_IGNORED_DIRS = new Set([
   '.git',
   '.svn',
   '.hg',
   'node_modules',
+  '.pnpm-store',
   '.turbo',
   '.next',
+  '.nuxt',
+  '.output',
   'dist',
   'build',
-  '.pnpm-store',
+  'out',
+  '.cache',
+  '.idea',
+  '.vscode',
+  '.husky',
+  'venv',
+  '.venv',
+  'env',
+  '__pycache__',
+  'target',      // Rust / Cargo / Maven
+  '.gradle',
+  '.cargo',
+  'vendor',      // PHP / Go
+  'coverage',
+  'tmp',
+  'temp',
+  'logs',
+  '.dsh',
 ])
 
 function isWithin(parent: string, child: string): boolean {
@@ -34,6 +55,24 @@ function isWithin(parent: string, child: string): boolean {
   return !rel.startsWith('..') && !rel.includes(':')
 }
 
+// 快速解析工作区根目录的 .gitignore，避免深入被 git 忽略的庞大目录
+async function loadGitIgnoreNames(rootDir: string): Promise<Set<string>> {
+  const names = new Set<string>()
+  try {
+    const gitignorePath = join(rootDir, '.gitignore')
+    const content = await fs.readFile(gitignorePath, 'utf8')
+    for (const rawLine of content.split(/\r?\n/)) {
+      const line = rawLine.trim()
+      if (!line || line.startsWith('#') || line.startsWith('!')) continue
+      const clean = line.replace(/^\/+/, '').replace(/\/+$/, '')
+      if (clean && !clean.includes('*') && !clean.includes('?') && !clean.includes('/')) {
+        names.add(clean)
+      }
+    }
+  } catch {}
+  return names
+}
+
 export async function scanWorkspaceTree(
   cwd: string,
   options: { maxDepth?: number; showHidden?: boolean; showIgnored?: boolean } = {},
@@ -43,11 +82,15 @@ export async function scanWorkspaceTree(
   const showHidden = options.showHidden ?? false
   const showIgnored = options.showIgnored ?? false
 
+  const gitIgnoredNames = showIgnored ? new Set<string>() : await loadGitIgnoreNames(root)
+
   let totalFiles = 0
   let totalDirs = 0
+  let totalNodes = 0
+  const MAX_NODES_SAFETY_LIMIT = 30000 // 防爆保护阈值
 
   async function walk(dirPath: string, currentDepth: number): Promise<FileItem[]> {
-    if (currentDepth > maxDepth) return []
+    if (currentDepth > maxDepth || totalNodes >= MAX_NODES_SAFETY_LIMIT) return []
 
     let dirents: import('node:fs').Dirent[]
     try {
@@ -56,26 +99,23 @@ export async function scanWorkspaceTree(
       return []
     }
 
-    // Sort: directories first, then files alphabetically
-    dirents.sort((a, b) => {
-      if (a.isDirectory() && !b.isDirectory()) return -1
-      if (!a.isDirectory() && b.isDirectory()) return 1
-      return a.name.localeCompare(b.name, undefined, { numeric: true, sensitivity: 'base' })
-    })
-
-    const items: FileItem[] = []
+    const dirTasks: Array<Promise<FileItem | null>> = []
+    const fileItems: FileItem[] = []
 
     for (const ent of dirents) {
+      if (totalNodes >= MAX_NODES_SAFETY_LIMIT) break
+
       const name = ent.name
 
-      // Hidden file filter (starts with .)
+      // 隐藏文件/文件夹过滤（以 . 开头）
       if (!showHidden && name.startsWith('.') && name !== '.gitignore' && name !== '.env') {
         if (!showIgnored) continue
       }
 
-      // Ignored directory filter
-      if (!showIgnored && ent.isDirectory() && DEFAULT_IGNORED_DIRS.has(name)) {
-        continue
+      // 忽略目录及 .gitignore 规则过滤
+      if (!showIgnored) {
+        if (ent.isDirectory() && DEFAULT_IGNORED_DIRS.has(name)) continue
+        if (gitIgnoredNames.has(name)) continue
       }
 
       const fullPath = join(dirPath, name)
@@ -85,34 +125,35 @@ export async function scanWorkspaceTree(
 
       if (ent.isDirectory()) {
         totalDirs += 1
-        const children = await walk(fullPath, currentDepth + 1)
-        items.push({
-          name,
-          path: relPath,
-          isDirectory: true,
-          children,
-        })
+        totalNodes += 1
+        dirTasks.push(
+          walk(fullPath, currentDepth + 1).then(children => ({
+            name,
+            path: relPath,
+            isDirectory: true,
+            children,
+          })),
+        )
       } else {
         totalFiles += 1
-        let size: number | undefined
-        let mtime: number | undefined
-        try {
-          const st = await fs.stat(fullPath)
-          size = st.size
-          mtime = st.mtimeMs
-        } catch {}
-
-        items.push({
+        totalNodes += 1
+        fileItems.push({
           name,
           path: relPath,
           isDirectory: false,
-          size,
-          mtime,
         })
       }
     }
 
-    return items
+    // 并行处理子目录
+    const dirResults = await Promise.all(dirTasks)
+    const validDirs = dirResults.filter((d): d is FileItem => d !== null)
+
+    // 目录优先，按自然字母排序
+    validDirs.sort((a, b) => a.name.localeCompare(b.name, undefined, { numeric: true, sensitivity: 'base' }))
+    fileItems.sort((a, b) => a.name.localeCompare(b.name, undefined, { numeric: true, sensitivity: 'base' }))
+
+    return [...validDirs, ...fileItems]
   }
 
   const items = await walk(root, 1)

package/src/index.ts

@@ -1,3 +1,5 @@
+import { spawn } from 'node:child_process'
+import { isAbsolute, relative, resolve, sep } from 'node:path'
 import type { Context } from '@deepseek-ai/cordis'
 import Schema from '@deepseek-ai/schemastery'
 import { scanWorkspaceTree } from './files.js'
@@ -77,6 +79,30 @@ function writeJson(res: any, status: number, body: any): void {
   res.end(data)
 }
 
+/**
+ * 拒绝工作区之外的目标：files.open 的 path 来自请求体，必须是工作区内的相对/绝对路径，
+ * 否则模型或页面脚本可以借宿主进程去打开任意本地路径。
+ */
+function assertInsideWorkspace(cwd: string, fullPath: string): void {
+  const root = resolve(cwd)
+  const target = resolve(fullPath)
+  const rel = relative(root, target)
+  const escaped = rel === '..' || rel.startsWith(`..${sep}`) || isAbsolute(rel)
+  if (escaped) throw new Error('Path must stay inside the workspace.')
+}
+
+/**
+ * 启动系统文件管理器/默认应用。spawn 在命令不存在时会 emit 'error'，
+ * 没有监听器就会变成未捕获异常拖垮宿主进程，因此这里必须兜住。
+ */
+function launch(command: string, args: string[]): void {
+  const child = spawn(command, args, { detached: true, stdio: 'ignore' })
+  child.on('error', (err: any) => {
+    console.warn(`[dsh-workspace-files] failed to launch ${command}:`, err?.message || String(err))
+  })
+  child.unref()
+}
+
 function resolveCwd(ctx: Context, body: Record<string, any>): string {
   if (body.cwd?.trim()) return body.cwd.trim()
   if (body.root?.trim()) return body.root.trim()
@@ -92,31 +118,22 @@ function resolveCwd(ctx: Context, body: Record<string, any>): string {
 
 export const SETTINGS_NS = 'workspace-files'
 
+interface CacheEntry {
+  timestamp: number
+  result: any
+}
+const scanCache = new Map<string, CacheEntry>()
+const CACHE_TTL_MS = 20_000 // 20 秒热缓存
+
 export function apply(ctx: Context, config: Config = {}): void {
   const liveConfig = {
     maxDepth: typeof config?.maxDepth === 'number' && config.maxDepth > 0 ? config.maxDepth : 16,
     showHidden: Boolean(config?.showHidden),
   }
 
-  // 关键：向 Host 的 settings 服务注册 settings namespace，使设置面板的前端控制器能够枚举到此插件卡片
-  if ((ctx as any).inject) {
-    (ctx as any).inject(['settings'], (scopedCtx: any) => {
-      try {
-        const scope = scopedCtx.settings?.register(SETTINGS_NS, Config, {
-          base: liveConfig,
-        })
-        const applyChange = () => {
-          const current = scope?.get?.()
-          if (typeof current?.maxDepth === 'number') liveConfig.maxDepth = current.maxDepth
-          if (typeof current?.showHidden === 'boolean') liveConfig.showHidden = current.showHidden
-        }
-        applyChange()
-        scope?.watch?.(applyChange)
-      } catch (err) {
-        console.warn('[dsh-workspace-files] failed to register settings scope:', err)
-      }
-    })
-  }
+  // 设置表单由 DSH 从本插件导出的 Config schema 自动生成（Host 侧 settings.describe()），
+  // 无需也不存在插件侧的 settings.register()：DSH 的 settings 服务只提供
+  // describe/update/replace/mutate/configure，namespace 就是 profile 里的 entry id。
 
   const webServer = (ctx as any).webServer
   if (!webServer) return
@@ -147,6 +164,47 @@ export function apply(ctx: Context, config: Config = {}): void {
       return
     }
 
+    if (method === 'files.open') {
+      try {
+        const body = await readJsonBody(req)
+        const cwd = resolveCwd(ctx, body)
+        const rawPath = String(body.path || '')
+        if (!rawPath) {
+          writeJson(res, 400, { ok: false, error: { message: 'Missing path parameter' } })
+          return
+        }
+
+        const fullPath = isAbsolute(rawPath) ? rawPath : resolve(cwd, rawPath)
+        assertInsideWorkspace(cwd, fullPath)
+        const action = body.action || 'reveal'
+        const isDirectory = Boolean(body.isDirectory)
+
+        if (process.platform === 'win32') {
+          const winPath = fullPath.replace(/\//g, '\\')
+          // explorer.exe 自己解析路径，不再经过 cmd.exe 的字符串拼接，避免 & | ^ 等注入
+          const args = action === 'reveal' && !isDirectory ? [`/select,${winPath}`] : [winPath]
+          launch('explorer.exe', args)
+        } else if (process.platform === 'darwin') {
+          if (action === 'reveal') {
+            launch('open', [isDirectory ? fullPath : '-R', fullPath])
+          } else {
+            launch('open', [fullPath])
+          }
+        } else {
+          if (action === 'reveal') {
+            launch('xdg-open', [isDirectory ? fullPath : resolve(fullPath, '..')])
+          } else {
+            launch('xdg-open', [fullPath])
+          }
+        }
+
+        writeJson(res, 200, { ok: true })
+      } catch (err: any) {
+        writeJson(res, 500, { ok: false, error: { message: err?.message || String(err) } })
+      }
+      return
+    }
+
     if (method !== 'files.tree') {
       writeJson(res, 404, { ok: false, error: { message: `Unknown method: ${method}` } })
       return
@@ -155,11 +213,29 @@ export function apply(ctx: Context, config: Config = {}): void {
     try {
       const body = await readJsonBody(req)
       const cwd = resolveCwd(ctx, body)
+      const showHidden = typeof body.showHidden === 'boolean' ? body.showHidden : liveConfig.showHidden
+      const showIgnored = Boolean(body.showIgnored)
+      const maxDepth = typeof body.maxDepth === 'number' && body.maxDepth > 0 ? body.maxDepth : liveConfig.maxDepth
+      const forceRefresh = Boolean(body.forceRefresh)
+
+      const cacheKey = `${cwd}::${showHidden}::${showIgnored}::${maxDepth}`
+      const now = Date.now()
+
+      if (!forceRefresh) {
+        const cached = scanCache.get(cacheKey)
+        if (cached && now - cached.timestamp < CACHE_TTL_MS) {
+          writeJson(res, 200, { ok: true, value: cached.result, cached: true })
+          return
+        }
+      }
+
       const result = await scanWorkspaceTree(cwd, {
-        showHidden: typeof body.showHidden === 'boolean' ? body.showHidden : liveConfig.showHidden,
-        showIgnored: Boolean(body.showIgnored),
-        maxDepth: typeof body.maxDepth === 'number' && body.maxDepth > 0 ? body.maxDepth : liveConfig.maxDepth,
+        showHidden,
+        showIgnored,
+        maxDepth,
       })
+
+      scanCache.set(cacheKey, { timestamp: now, result })
       writeJson(res, 200, { ok: true, value: result })
     } catch (err: any) {
       writeJson(res, 500, { ok: false, error: { message: err?.message || String(err) } })