Fix disk usage display: use native du/PowerShell instead of recursive stat

Previous async Promise.all approach had two problems:
1. Created tens of thousands of concurrent stat() calls for large directories
   (venv has 100k+ files), overwhelming the OS I/O queue on Windows and
   returning 0 or hanging indefinitely
2. Math.round(bytes/1048576) silently showed '0 MB' for dirs < 0.5 MB

Fix:
- Linux/macOS: use 'du -sm <dir>' via exec() — fast, non-blocking, OS-native
- Windows: use PowerShell Get-ChildItem | Measure-Object — reliable, async
- For excludeDirs: compute total minus excluded subdirs separately
- Display: show '< 1 MB' for small dirs, 'not installed' for missing venv

Author: nodeeeeee

electron/main.js

@@ -6,7 +6,7 @@ const os     = require('os');
 const fs     = require('fs');
 const https  = require('https');
 const http   = require('http');
-const { spawnSync, spawn } = require('child_process');
+const { spawnSync, spawn, exec } = require('child_process');
 
 // ── Optional: node-pty for real tty on Unix (tqdm in-place refresh) ──────────
 let nodePty = null;
@@ -627,37 +627,44 @@ function discoverLectures(courseId) {
 }
 
 // ── Uninstall helpers ─────────────────────────────────────────────────────────
-async function getDirSizeBytes(dirPath) {
-  let bytes = 0;
-  try {
-    const entries = await fs.promises.readdir(dirPath, { withFileTypes: true });
-    await Promise.all(entries.map(async e => {
-      const full = path.join(dirPath, e.name);
-      if (e.isDirectory()) {
-        bytes += await getDirSizeBytes(full);
-      } else {
-        try { bytes += (await fs.promises.stat(full)).size; } catch {}
-      }
-    }));
-  } catch {}
-  return bytes;
+// Returns directory size in MB using native OS tools (non-blocking, fast).
+function getDirSizeMBNative(dirPath) {
+  return new Promise(resolve => {
+    if (!fs.existsSync(dirPath)) { resolve(0); return; }
+    if (process.platform === 'win32') {
+      // PowerShell: sum all file lengths under the directory
+      const esc = dirPath.replace(/'/g, "''");
+      const cmd = `powershell -NoProfile -NonInteractive -Command ` +
+        `"(Get-ChildItem -LiteralPath '${esc}' -Recurse -File ` +
+        `-ErrorAction SilentlyContinue | Measure-Object -Property Length -Sum).Sum"`;
+      exec(cmd, { timeout: 30000, windowsHide: true }, (_, stdout) => {
+        const bytes = parseInt((stdout || '').trim()) || 0;
+        resolve(Math.round(bytes / 1048576));
+      });
+    } else {
+      // Linux / macOS: du -sm gives megabytes directly
+      const esc = dirPath.replace(/\\/g, '\\\\').replace(/"/g, '\\"');
+      exec(`du -sm -- "${esc}"`, { timeout: 30000 }, (err, stdout) => {
+        if (err) { resolve(0); return; }
+        const mb = parseInt((stdout || '').trim().split(/\s+/)[0]) || 0;
+        resolve(mb);
+      });
+    }
+  });
 }
 
 async function getDirSizeMB(dirPath, excludeDirs = []) {
-  let bytes = 0;
-  try {
-    const entries = await fs.promises.readdir(dirPath, { withFileTypes: true });
-    await Promise.all(entries.map(async e => {
-      if (excludeDirs.includes(e.name)) return;
-      const full = path.join(dirPath, e.name);
-      if (e.isDirectory()) {
-        bytes += await getDirSizeBytes(full);
-      } else {
-        try { bytes += (await fs.promises.stat(full)).size; } catch {}
-      }
-    }));
-  } catch {}
-  return Math.round(bytes / (1024 * 1024));
+  if (!fs.existsSync(dirPath)) return 0;
+  const total = await getDirSizeMBNative(dirPath);
+  if (excludeDirs.length === 0) return total;
+  // Subtract excluded subdirs from the total
+  const excluded = await Promise.all(
+    excludeDirs
+      .map(d => path.join(dirPath, d))
+      .filter(p => fs.existsSync(p))
+      .map(getDirSizeMBNative)
+  );
+  return Math.max(0, total - excluded.reduce((a, b) => a + b, 0));
 }
 
 function rmRecursive(dirPath) {

electron/package.json

@@ -1,6 +1,6 @@
 {
   "name": "auto-note",
-  "version": "0.9.6",
+  "version": "0.9.7",
   "description": "AutoNote — lecture notes generator from Canvas recordings",
   "homepage": "https://github.com/nodeeeeee/Auto-Note",
   "author": {

electron/renderer/app.js

@@ -1303,9 +1303,14 @@ async function attachPageHandlers() {
       if (!el) return;
       try {
         const s = await window.api.getUninstallSizes();
-        const fmt = mb => mb >= 1024 ? `${(mb/1024).toFixed(1)} GB` : `${mb} MB`;
+        const fmt = mb => {
+          if (mb >= 1024) return `${(mb / 1024).toFixed(1)} GB`;
+          if (mb >= 1)    return `${mb} MB`;
+          return '< 1 MB';
+        };
+        const venvLabel = s.venv === 0 ? '<span style="color:var(--c-white-45)">not installed</span>' : `<b>${fmt(s.venv)}</b>`;
         let html = `<b>Disk usage (controlled by checkboxes below):</b><br>`;
-        html += `&nbsp;&nbsp;• ML Environment (~/.auto_note/venv): <b>${fmt(s.venv)}</b><br>`;
+        html += `&nbsp;&nbsp;• ML Environment (~/.auto_note/venv): ${venvLabel}<br>`;
         html += `&nbsp;&nbsp;• Settings &amp; config (~/.auto_note/): <b>${fmt(s.settings)}</b><br>`;
         if (s.content !== null) {
           html += `&nbsp;&nbsp;• Generated content (${s.outputDir}): <b>${fmt(s.content)}</b>`;