@rooode/dsh-plugin-preview 0.1.2 → 0.1.4

package/lib/index.js

@@ -130,6 +130,146 @@ export class PreviewService extends Service {
     });
   }
 
+  readWorkspaceTree(targetDir, maxDepth = 4, includeAll = false) {
+    try {
+      const normalizedRoot = path.resolve(targetDir);
+      if (!fs.existsSync(normalizedRoot)) {
+        return { ok: false, error: { code: 'DIR_NOT_FOUND', message: '工作区目录不存在: ' + normalizedRoot } };
+      }
+      const stat = fs.statSync(normalizedRoot);
+      if (!stat.isDirectory()) {
+        return { ok: false, error: { code: 'NOT_A_DIRECTORY', message: '目标路径不是目录: ' + normalizedRoot } };
+      }
+      const ignoreDirs = new Set([
+        'node_modules', '.git', '.dsh', '.gemini', '.tempmediastorage', '.idea', '.vscode',
+        'dist', 'build', 'coverage', '.next', '.nuxt', '.output', '.turbo', '.cache',
+        '__pycache__', '.pytest_cache', 'target', 'vendor'
+      ]);
+      let totalFiles = 0;
+      let totalDirs = 0;
+
+      const buildTree = (dirPath, currentDepth) => {
+        if (currentDepth > maxDepth) return [];
+        let entries = [];
+        try {
+          entries = fs.readdirSync(dirPath, { withFileTypes: true });
+        } catch (e) {
+          return [];
+        }
+        const nodes = [];
+        for (const entry of entries) {
+          const entryName = entry.name;
+          const lowerName = entryName.toLowerCase();
+          if (!includeAll) {
+            if (ignoreDirs.has(lowerName)) continue;
+            if (entryName.startsWith('.') && entryName !== '.env' && entryName !== '.gitignore') continue;
+          }
+          const fullPath = path.join(dirPath, entryName);
+          const relPath = path.relative(normalizedRoot, fullPath).replace(/\\/g, '/');
+          try {
+            if (entry.isDirectory()) {
+              totalDirs++;
+              const children = buildTree(fullPath, currentDepth + 1);
+              nodes.push({
+                name: entryName,
+                path: fullPath,
+                relativePath: relPath,
+                isDir: true,
+                children,
+                childCount: children.length,
+              });
+            } else if (entry.isFile()) {
+              totalFiles++;
+              const ext = path.extname(entryName).toLowerCase();
+              let size = 0;
+              let mtime = 0;
+              try {
+                const s = fs.statSync(fullPath);
+                size = s.size;
+                mtime = s.mtimeMs;
+              } catch (e) {}
+              nodes.push({
+                name: entryName,
+                path: fullPath,
+                relativePath: relPath,
+                isDir: false,
+                extension: ext,
+                size,
+                mtime,
+              });
+            }
+          } catch (e) {}
+        }
+        nodes.sort((a, b) => {
+          if (a.isDir !== b.isDir) return a.isDir ? -1 : 1;
+          return a.name.localeCompare(b.name, undefined, { numeric: true, sensitivity: 'base' });
+        });
+        return nodes;
+      };
+
+      const tree = buildTree(normalizedRoot, 1);
+      const workspaceName = path.basename(normalizedRoot) || normalizedRoot;
+      return {
+        ok: true,
+        workspaceRoot: normalizedRoot,
+        workspaceName,
+        tree,
+        totalFiles,
+        totalDirs,
+      };
+    } catch (err) {
+      return { ok: false, error: { code: 'SCAN_ERROR', message: err.message || '扫描工作区文件树失败' } };
+    }
+  }
+
+  listDirectoryEntries(targetDir, includeAll = false) {
+    try {
+      const normalized = path.resolve(targetDir);
+      if (!fs.existsSync(normalized) || !fs.statSync(normalized).isDirectory()) {
+        return { ok: false, error: { code: 'DIR_NOT_FOUND', message: '目录不存在' } };
+      }
+      const ignoreDirs = new Set(['node_modules', '.git', '.dsh', '.gemini', '.tempmediastorage', '.idea', '.vscode']);
+      const rawEntries = fs.readdirSync(normalized, { withFileTypes: true });
+      const nodes = [];
+      for (const entry of rawEntries) {
+        const name = entry.name;
+        if (!includeAll) {
+          if (ignoreDirs.has(name.toLowerCase())) continue;
+          if (name.startsWith('.') && name !== '.env' && name !== '.gitignore') continue;
+        }
+        const fullPath = path.join(normalized, name);
+        const isDir = entry.isDirectory();
+        let size = 0;
+        let mtime = 0;
+        let ext = '';
+        if (!isDir) {
+          ext = path.extname(name).toLowerCase();
+          try {
+            const s = fs.statSync(fullPath);
+            size = s.size;
+            mtime = s.mtimeMs;
+          } catch (e) {}
+        }
+        nodes.push({
+          name,
+          path: fullPath,
+          relativePath: name,
+          isDir,
+          extension: ext,
+          size,
+          mtime,
+        });
+      }
+      nodes.sort((a, b) => {
+        if (a.isDir !== b.isDir) return a.isDir ? -1 : 1;
+        return a.name.localeCompare(b.name, undefined, { numeric: true, sensitivity: 'base' });
+      });
+      return { ok: true, entries: nodes };
+    } catch (err) {
+      return { ok: false, error: { code: 'LIST_ERROR', message: err.message || '读取目录失败' } };
+    }
+  }
+
   async handleHttpRequest(req, res) {
     this.setCorsHeaders(res);
     if (req.method === 'OPTIONS') {
@@ -144,7 +284,7 @@ export class PreviewService extends Service {
     try {
       if (pathname === '/api/preview/status') {
         res.statusCode = 200;
-        res.end(JSON.stringify({ ok: true, version: '0.1.0', service: 'preview' }));
+        res.end(JSON.stringify({ ok: true, version: '0.1.3', service: 'preview' }));
         return;
       }
       if (pathname === '/api/preview/read' && req.method === 'GET') {
@@ -159,6 +299,37 @@ export class PreviewService extends Service {
         res.end(JSON.stringify(result));
         return;
       }
+      if (pathname === '/api/preview/workspace-tree' && req.method === 'GET') {
+        const targetPath = url.searchParams.get('path') || process.cwd();
+        const depth = parseInt(url.searchParams.get('depth') || '4', 10);
+        const includeAll = url.searchParams.get('includeAll') === 'true';
+        const result = this.readWorkspaceTree(targetPath, isNaN(depth) ? 4 : depth, includeAll);
+        res.statusCode = result.ok ? 200 : 404;
+        res.end(JSON.stringify(result));
+        return;
+      }
+      if (pathname === '/api/preview/list-dir' && req.method === 'GET') {
+        const targetPath = url.searchParams.get('path') || process.cwd();
+        const includeAll = url.searchParams.get('includeAll') === 'true';
+        const result = this.listDirectoryEntries(targetPath, includeAll);
+        res.statusCode = result.ok ? 200 : 404;
+        res.end(JSON.stringify(result));
+        return;
+      }
+      if (pathname === '/api/preview/save' && req.method === 'POST') {
+        const body = await this.parseRequestBody(req);
+        if (!body.path || typeof body.content !== 'string') {
+          res.statusCode = 400;
+          res.end(JSON.stringify({ ok: false, error: { code: 'INVALID_PARAMS', message: '参数格式错误' } }));
+          return;
+        }
+        const normalized = path.resolve(body.path);
+        fs.writeFileSync(normalized, body.content, 'utf-8');
+        const stat = fs.statSync(normalized);
+        res.statusCode = 200;
+        res.end(JSON.stringify({ ok: true, size: stat.size, mtime: stat.mtimeMs }));
+        return;
+      }
       if (pathname === '/api/preview/reveal' && req.method === 'POST') {
         const body = await this.parseRequestBody(req);
         if (!body.path) {

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "@rooode/dsh-plugin-preview",
-  "version": "0.1.2",
-  "description": "DeepSeek Harness Markdown 文档右侧预览插件 (参考 WorkBuddy FileTabs 实现，支持多标签、GFM 渲染、TOC 大纲、源码分栏)",
+  "version": "0.1.4",
+  "description": "DeepSeek Harness Markdown 文档与工作空间文件浏览器右侧预览插件 (支持工作区文件树、JSON 交互结构树/格式化、YAML/JS/TS/Python 多语言语法高亮、代码符号大纲、自动换行与多标签 FileTabs)",
   "type": "module",
   "main": "lib/index.js",
   "types": "lib/index.d.ts",