copilot-usage-studio 0.2.0 → 0.2.1

package/lib/local-runtime.mjs

@@ -1,9 +1,18 @@
-import { spawn } from 'node:child_process';
-import { createReadStream, existsSync, readFileSync, statSync } from 'node:fs';
+import { fork, spawn } from 'node:child_process';
+import {
+  appendFileSync,
+  createReadStream,
+  existsSync,
+  mkdirSync,
+  readFileSync,
+  statSync,
+  writeFileSync,
+} from 'node:fs';
 import { createServer } from 'node:http';
 import { dirname, extname, join, normalize, resolve, sep } from 'node:path';
+import { fileURLToPath } from 'node:url';
 
-import { scanVsCodeSessions, writeSessionData } from './scanner-api.mjs';
+import { writeSessionData } from './scanner-api.mjs';
 
 const contentTypes = new Map([
   ['.css', 'text/css; charset=utf-8'],
@@ -19,6 +28,8 @@ const contentTypes = new Map([
   ['.woff2', 'font/woff2'],
 ]);
 
+const scannerWorkerPath = fileURLToPath(new URL('./scanner-worker.mjs', import.meta.url));
+
 export function createLocalRuntime(options = {}) {
   const host = options.host ?? '127.0.0.1';
   const port = Number(options.port ?? 4312);
@@ -28,11 +39,17 @@ export function createLocalRuntime(options = {}) {
     : resolve(options.seedDataFile ?? 'public/data/sessions.json');
   const staticDir = resolve(options.staticDir ?? 'dist/copilot-usage-studio/browser');
   const scanOptions = options.scanOptions ?? {};
-  const scanner = options.scanner ?? scanVsCodeSessions;
+  const scanner = options.scanner ?? scanInChildProcess;
   const writer = options.writer ?? writeSessionData;
   const openPath = options.openPath ?? openLocalPath;
   const logger = options.logger ?? console;
   const backendOnly = options.backendOnly === true;
+  const firstDataWaitMs = Math.max(0, Number(options.firstDataWaitMs ?? 5000));
+  const logFile = options.logFile === null ? '' : resolve(options.logFile ?? join(dirname(dataFile), 'runtime.log'));
+  const logEntries = [];
+  const progressHistory = [];
+  const workspaceProgress = new Map();
+  let scanProgress = null;
 
   let sessionData = readCachedSessionData(dataFile, logger)
     ?? (seedDataFile ? readCachedSessionData(seedDataFile, logger) : null);
@@ -42,6 +59,10 @@ export function createLocalRuntime(options = {}) {
   let lastScanCompletedAt = '';
   let lastScanDurationMs = 0;
   let activeScan = null;
+  let activeScanController = null;
+  let activeScanId = 0;
+  let activeScanMode = '';
+  let scanSequence = 0;
 
   function status() {
     return {
@@ -55,55 +76,154 @@ export function createLocalRuntime(options = {}) {
       lastScanCompletedAt,
       lastScanDurationMs,
       lastError,
+      activeScanId,
+      activeScanMode,
+      scanProgress,
+      progressHistory: progressHistory.slice(-40),
+      workspaceProgress: [...workspaceProgress.values()],
+      logFile,
+      recentLogs: logEntries.slice(-12),
     };
   }
 
-  function refresh(reason = 'manual') {
+  function diagnosticsReport() {
+    return {
+      status: status(),
+      dataFile,
+      staticDir,
+      seedDataFile,
+      logFile,
+      scanOptions: summarizeScanOptions(scanOptions),
+      progressHistory,
+      workspaceProgress: [...workspaceProgress.values()],
+      logs: logEntries,
+    };
+  }
+
+  function refresh(reason = 'manual', refreshOptions = {}) {
     if (activeScan) {
+      emitLog('log', `Scan already running; reusing active ${reason} request.`);
       return activeScan;
     }
 
+    const scanMode = normalizeScanMode(refreshOptions.mode ?? 'quick');
+    const effectiveScanOptions = scanOptionsForMode(scanMode, scanOptions);
     const started = Date.now();
+    const scanId = ++scanSequence;
+    activeScanId = scanId;
+    activeScanMode = scanMode;
+    activeScanController = new AbortController();
     lastScanStartedAt = new Date(started).toISOString();
     lastError = '';
     phase = 'scanning';
+    progressHistory.length = 0;
+    workspaceProgress.clear();
+    scanProgress = rememberProgress({
+      stage: 'starting',
+      message: `Starting scan #${scanId} (${reason}).`,
+    }, scanId, reason, lastScanStartedAt);
+    emitLog('log', `Starting local data refresh #${scanId} (${reason}).`);
 
     activeScan = Promise.resolve()
-      .then(() => scanner(scanOptions))
+      .then(() =>
+        scanner({
+          ...effectiveScanOptions,
+          signal: activeScanController.signal,
+          onProgress: (progress) => {
+            scanProgress = rememberProgress(progress, scanId, reason);
+            if (progress?.message) {
+              emitLog('log', progress.message);
+            }
+          },
+        }),
+      )
       .then((nextSessionData) => {
-        writer(nextSessionData, dataFile);
-        sessionData = nextSessionData;
+        const finalSessionData = sessionData
+          ? scanMode === 'customizations'
+            ? mergeCustomizationScan(sessionData, nextSessionData)
+            : preserveCustomizationSnapshot(sessionData, nextSessionData)
+          : nextSessionData;
+        writer(finalSessionData, dataFile);
+        sessionData = finalSessionData;
         phase = 'ready';
         lastScanCompletedAt = new Date().toISOString();
         lastScanDurationMs = Date.now() - started;
-        logger.log(
-          `Local data refresh (${reason}) imported ${nextSessionData.sessions.length} sessions in ${lastScanDurationMs}ms.`,
+        scanProgress = rememberProgress({
+          stage: 'complete',
+          message: scanMode === 'customizations'
+            ? `Customization usage check complete.`
+            : `Scan #${scanId} imported ${finalSessionData.sessions.length} sessions.`,
+          sessions: finalSessionData.sessions.length,
+        }, scanId, reason, lastScanCompletedAt);
+        emitLog(
+          'log',
+          scanMode === 'customizations'
+            ? `Customization usage check #${scanId} (${reason}) completed in ${lastScanDurationMs}ms.`
+            : `Local data refresh #${scanId} (${reason}) imported ${finalSessionData.sessions.length} sessions in ${lastScanDurationMs}ms.`,
         );
-        return nextSessionData;
+        return finalSessionData;
       })
       .catch((error) => {
-        lastError = error instanceof Error ? error.message : String(error);
-        phase = sessionData ? 'ready' : 'error';
+        const message = error instanceof Error ? error.message : String(error);
+        const stopped = isScanStoppedError(error);
+        lastError = stopped ? 'Scan stopped by user.' : message;
+        phase = stopped ? (sessionData ? 'ready' : 'stopped') : (sessionData ? 'ready' : 'error');
         lastScanCompletedAt = new Date().toISOString();
         lastScanDurationMs = Date.now() - started;
-        logger.error(`Local data refresh failed: ${lastError}`);
+        scanProgress = rememberProgress({
+          stage: stopped ? 'stopped' : 'failed',
+          message: stopped ? 'Scan stopped. Existing data was kept.' : lastError,
+        }, scanId, reason, lastScanCompletedAt);
+        emitLog(stopped ? 'warn' : 'error', `Local data refresh #${scanId} (${reason}) ${stopped ? 'stopped' : 'failed'}: ${lastError}`);
+        if (!stopped && error instanceof Error && error.stack) {
+          emitLog('error', error.stack);
+        }
         throw error;
       })
       .finally(() => {
         activeScan = null;
+        activeScanController = null;
+        activeScanId = 0;
+        activeScanMode = '';
       });
 
     return activeScan;
   }
 
+  function cancelScan() {
+    if (!activeScan || !activeScanController) {
+      return false;
+    }
+    activeScanController.abort();
+    emitLog('warn', 'Local data refresh was stopped by the user.');
+    return true;
+  }
+
   const server = createServer(async (request, response) => {
     try {
       const url = new URL(request.url ?? '/', `http://${request.headers.host ?? `${host}:${port}`}`);
 
+      if (request.method === 'OPTIONS') {
+        response.writeHead(204, corsHeaders());
+        response.end();
+        return;
+      }
+
       if (request.method === 'GET' && url.pathname === '/api/status') {
         return sendJson(response, 200, status());
       }
 
+      if (request.method === 'GET' && url.pathname === '/api/diagnostics') {
+        return sendJson(response, 200, diagnosticsReport());
+      }
+
+      if (request.method === 'GET' && url.pathname === '/api/logs') {
+        return sendJson(response, 200, {
+          logFile,
+          entries: logEntries,
+        });
+      }
+
       if (request.method === 'GET' && url.pathname === '/api/sessions') {
         await waitForFirstData();
         return sessionData
@@ -120,13 +240,18 @@ export function createLocalRuntime(options = {}) {
 
       if (request.method === 'POST' && url.pathname === '/api/scan') {
         try {
-          const nextSessionData = await refresh('manual');
+          const body = await readJsonBody(request);
+          const nextSessionData = await refresh('manual', body);
           return sendJson(response, 200, { sessionData: nextSessionData, status: status() });
         } catch {
           return sendJson(response, 500, { error: lastError, status: status() });
         }
       }
 
+      if (request.method === 'POST' && url.pathname === '/api/scan/cancel') {
+        return sendJson(response, 200, { canceled: cancelScan(), status: status() });
+      }
+
       const memoryAction = url.pathname.match(/^\/api\/memories\/([a-f0-9]{24})\/open$/);
       if (request.method === 'POST' && memoryAction) {
         const memory = sessionData?.memories?.find((candidate) => candidate.id === memoryAction[1]);
@@ -146,7 +271,7 @@ export function createLocalRuntime(options = {}) {
 
       return serveStaticFile(request, response, staticDir, url.pathname);
     } catch (error) {
-      logger.error(error);
+      emitLog('error', error instanceof Error ? (error.stack ?? error.message) : String(error));
       return sendJson(response, 500, { error: 'Local runtime request failed.' });
     }
   });
@@ -161,31 +286,332 @@ export function createLocalRuntime(options = {}) {
       server.listen(port, host, () => {
         server.off('error', rejectListen);
         const address = server.address();
-        logger.log(
+        emitLog(
+          'log',
           backendOnly
             ? `Copilot Usage Studio backend API: http://${host}:${address.port}/ (used by the dev server)`
             : `Copilot Usage Studio local app: http://${host}:${address.port}/`,
         );
+        if (logFile) {
+          emitLog('log', `Runtime log file: ${logFile}`);
+        }
         resolveListen(address);
 
         if (options.scanOnStart !== false) {
-          void refresh('startup').catch(() => {});
+          void refresh('startup', { mode: options.startupScanMode ?? 'quick' }).catch(() => {});
         }
       });
     }),
     refresh,
+    cancelScan,
     status,
+    diagnostics: diagnosticsReport,
   };
 
   async function waitForFirstData() {
     if (!sessionData && activeScan) {
       try {
-        await activeScan;
+        await waitForActiveScanOrTimeout(activeScan, firstDataWaitMs);
       } catch {
         // The response below reports the retained empty/error state.
       }
     }
   }
+
+  function emitLog(level, message) {
+    const entry = {
+      at: new Date().toISOString(),
+      level,
+      message: String(message),
+    };
+    logEntries.push(entry);
+    if (logEntries.length > 200) {
+      logEntries.splice(0, logEntries.length - 200);
+    }
+
+    if (logFile) {
+      try {
+        mkdirSync(dirname(logFile), { recursive: true });
+        if (!existsSync(logFile)) {
+          writeFileSync(logFile, '', 'utf8');
+        }
+        appendFileSync(logFile, `${entry.at} ${entry.level.toUpperCase()} ${entry.message}\n`, 'utf8');
+      } catch {
+        // Logging must never prevent the local runtime from starting.
+      }
+    }
+
+    const sink = level === 'error' ? logger.error : level === 'warn' ? logger.warn : logger.log;
+    if (typeof sink === 'function') {
+      sink.call(logger, entry.message);
+    }
+  }
+
+  function rememberProgress(progress = {}, scanId, reason, updatedAt = new Date().toISOString()) {
+    const event = {
+      ...progress,
+      scanId,
+      reason,
+      updatedAt,
+    };
+    progressHistory.push(event);
+    if (progressHistory.length > 200) {
+      progressHistory.splice(0, progressHistory.length - 200);
+    }
+    if (event.workspace || event.workspaceDir) {
+      const key = event.workspaceDir || event.workspace;
+      const existing = workspaceProgress.get(key) ?? {};
+      workspaceProgress.set(key, {
+        ...existing,
+        workspace: event.workspace ?? existing.workspace ?? '',
+        workspaceDir: event.workspaceDir ?? existing.workspaceDir ?? '',
+        workspaceIndex: event.workspaceIndex ?? existing.workspaceIndex ?? null,
+        workspaceTotal: event.workspaceTotal ?? existing.workspaceTotal ?? null,
+        lastStage: event.stage ?? existing.lastStage ?? '',
+        message: event.message ?? existing.message ?? '',
+        elapsedMs: event.elapsedMs ?? existing.elapsedMs ?? 0,
+        updatedAt,
+        debugLogFolders: event.debugLogFolders ?? existing.debugLogFolders ?? null,
+        chatSnapshots: event.chatSnapshots ?? existing.chatSnapshots ?? null,
+        hasMemoryRoot: event.hasMemoryRoot ?? existing.hasMemoryRoot ?? null,
+        customizationInventory: event.customizationInventory ?? existing.customizationInventory ?? null,
+        total: event.total ?? existing.total ?? null,
+        index: event.index ?? existing.index ?? null,
+        sessions: event.sessions ?? existing.sessions ?? null,
+        memories: event.memories ?? existing.memories ?? null,
+        customizations: event.customizations ?? existing.customizations ?? null,
+        completed: event.stage === 'workspace-complete' || existing.completed === true,
+      });
+    }
+    return event;
+  }
+}
+
+function summarizeScanOptions(scanOptions = {}) {
+  return {
+    roots: Array.isArray(scanOptions.roots) ? scanOptions.roots.map(String) : [],
+    customizationWorkspaceFolders: Array.isArray(scanOptions.customizationWorkspaceFolders)
+      ? scanOptions.customizationWorkspaceFolders.map(String)
+      : [],
+    includeCustomizations: scanOptions.includeCustomizations !== false,
+    sqlite: scanOptions.sqlite !== false,
+  };
+}
+
+function normalizeScanMode(mode) {
+  return ['quick', 'full', 'customizations'].includes(String(mode)) ? String(mode) : 'quick';
+}
+
+function scanOptionsForMode(mode, baseScanOptions = {}) {
+  if (mode === 'full' || mode === 'customizations') {
+    const { customizationWorkspaceFolders, ...scanOptions } = baseScanOptions;
+    if (mode === 'customizations' && Array.isArray(customizationWorkspaceFolders)) {
+      return {
+        ...scanOptions,
+        includeCustomizations: true,
+        requireWorkspaceFolders: true,
+        workspaceFolders: customizationWorkspaceFolders,
+        customizationEvidence: {
+          maxSessions: 0,
+          maxModelCalls: 0,
+          maxElapsedMs: 0,
+          maxPartChars: 250_000,
+          ...(scanOptions.customizationEvidence ?? {}),
+        },
+      };
+    }
+
+    return {
+      ...scanOptions,
+      includeCustomizations: true,
+      customizationEvidence: {
+        maxSessions: 60,
+        maxModelCalls: 500,
+        maxElapsedMs: 90_000,
+        maxPartChars: 250_000,
+        ...(scanOptions.customizationEvidence ?? {}),
+      },
+    };
+  }
+
+  return {
+    ...baseScanOptions,
+    includeCustomizations: false,
+  };
+}
+
+function mergeCustomizationScan(previousData, currentWorkspaceData) {
+  const scannedWorkspaces = new Set(
+    (currentWorkspaceData.ingestion?.workspaceScans ?? [])
+      .map((scan) => scan.workspace)
+      .filter(Boolean),
+  );
+  const sessionsById = new Map((previousData.sessions ?? []).map((session) => [session.id, session]));
+  for (const session of currentWorkspaceData.sessions ?? []) {
+    sessionsById.set(session.id, session);
+  }
+  const memoriesById = new Map((previousData.memories ?? []).map((memory) => [memory.id, memory]));
+  for (const memory of currentWorkspaceData.memories ?? []) {
+    memoriesById.set(memory.id, memory);
+  }
+  const retainedCustomizations = (previousData.customizations ?? []).filter(
+    (customization) => !scannedWorkspaces.has(customization.workspace),
+  );
+  const customizationsById = new Map(retainedCustomizations.map((customization) => [customization.id, customization]));
+  for (const customization of currentWorkspaceData.customizations ?? []) {
+    customizationsById.set(customization.id, customization);
+  }
+  const sessions = [...sessionsById.values()].sort((a, b) => b.startedAt.localeCompare(a.startedAt));
+  const memories = [...memoriesById.values()].sort((a, b) => b.modifiedAt.localeCompare(a.modifiedAt));
+  const customizations = [...customizationsById.values()].sort((a, b) =>
+    b.modifiedAt.localeCompare(a.modifiedAt),
+  );
+
+  return {
+    ...previousData,
+    generatedAt: currentWorkspaceData.generatedAt,
+    ingestion: {
+      ...previousData.ingestion,
+      ...currentWorkspaceData.ingestion,
+      importedSessions: sessions.length,
+      importedCustomizations: customizations.length,
+    },
+    sessions,
+    memories,
+    customizations,
+  };
+}
+
+function preserveCustomizationSnapshot(previousData, nextData) {
+  const hasPreviousCustomizations = Array.isArray(previousData.customizations) && previousData.customizations.length > 0;
+  const nextHasCustomizations = Array.isArray(nextData.customizations) && nextData.customizations.length > 0;
+  if (!hasPreviousCustomizations || nextHasCustomizations) {
+    return nextData;
+  }
+
+  return {
+    ...nextData,
+    ingestion: {
+      ...nextData.ingestion,
+      scannedCustomizationLocations:
+        nextData.ingestion?.scannedCustomizationLocations ??
+        previousData.ingestion?.scannedCustomizationLocations,
+      importedCustomizations:
+        nextData.ingestion?.importedCustomizations ??
+        previousData.ingestion?.importedCustomizations,
+      customizationEvidenceScannedSessions:
+        nextData.ingestion?.customizationEvidenceScannedSessions ??
+        previousData.ingestion?.customizationEvidenceScannedSessions,
+      customizationEvidenceModelCalls:
+        nextData.ingestion?.customizationEvidenceModelCalls ??
+        previousData.ingestion?.customizationEvidenceModelCalls,
+      customizationEvidenceTextParts:
+        nextData.ingestion?.customizationEvidenceTextParts ??
+        previousData.ingestion?.customizationEvidenceTextParts,
+      customizationEvidenceMatchedCustomizations:
+        nextData.ingestion?.customizationEvidenceMatchedCustomizations ??
+        previousData.ingestion?.customizationEvidenceMatchedCustomizations,
+    },
+    customizations: previousData.customizations,
+  };
+}
+
+function waitForActiveScanOrTimeout(activeScan, timeoutMs) {
+  if (!timeoutMs) {
+    return Promise.resolve();
+  }
+
+  return Promise.race([
+    activeScan,
+    new Promise((resolveTimeout) => setTimeout(resolveTimeout, timeoutMs)),
+  ]);
+}
+
+function isScanStoppedError(error) {
+  return /scan stopped by user/i.test(error instanceof Error ? error.message : String(error));
+}
+
+function scanInChildProcess(options = {}) {
+  const onProgress = typeof options.onProgress === 'function' ? options.onProgress : () => {};
+  const signal = options.signal;
+  const scanOptions = { ...options };
+  delete scanOptions.onProgress;
+  delete scanOptions.signal;
+
+  return new Promise((resolveScan, rejectScan) => {
+    const child = fork(scannerWorkerPath, {
+      stdio: ['ignore', 'ignore', 'pipe', 'ipc'],
+      windowsHide: true,
+    });
+    let settled = false;
+    let stderr = '';
+
+    const abort = () => {
+      if (settled) {
+        return;
+      }
+      settled = true;
+      child.kill();
+      rejectScan(new Error('Scan stopped by user.'));
+    };
+
+    if (signal?.aborted) {
+      abort();
+      return;
+    }
+    signal?.addEventListener?.('abort', abort, { once: true });
+
+    child.stderr?.on('data', (chunk) => {
+      stderr = `${stderr}${chunk.toString()}`.slice(-8000);
+    });
+
+    child.on('message', (message) => {
+      if (!message || typeof message !== 'object') {
+        return;
+      }
+      if (message.type === 'progress') {
+        onProgress(message.event ?? {});
+        return;
+      }
+      if (message.type === 'result') {
+        settled = true;
+        signal?.removeEventListener?.('abort', abort);
+        resolveScan(message.sessionData);
+        child.kill();
+        return;
+      }
+      if (message.type === 'error') {
+        settled = true;
+        signal?.removeEventListener?.('abort', abort);
+        const error = new Error(message.error?.message ?? 'Scanner worker failed.');
+        error.stack = message.error?.stack || error.stack;
+        rejectScan(error);
+        child.kill();
+      }
+    });
+
+    child.once('error', (error) => {
+      if (!settled) {
+        settled = true;
+        signal?.removeEventListener?.('abort', abort);
+        rejectScan(error);
+      }
+    });
+
+    child.once('exit', (code, exitSignal) => {
+      if (!settled) {
+        settled = true;
+        options.signal?.removeEventListener?.('abort', abort);
+        rejectScan(
+          new Error(
+            `Scanner worker exited before returning data (${exitSignal ?? `code ${code}`}).${stderr ? `\n${stderr}` : ''}`,
+          ),
+        );
+      }
+    });
+
+    child.send({ type: 'scan', options: scanOptions });
+  });
 }
 
 function readJsonBody(request) {
@@ -259,6 +685,7 @@ function readCachedSessionData(dataFile, logger) {
 function sendJson(response, statusCode, body) {
   const payload = JSON.stringify(body);
   response.writeHead(statusCode, {
+    ...corsHeaders(),
     'cache-control': 'no-store',
     'content-length': Buffer.byteLength(payload),
     'content-type': 'application/json; charset=utf-8',
@@ -266,6 +693,15 @@ function sendJson(response, statusCode, body) {
   response.end(payload);
 }
 
+function corsHeaders() {
+  return {
+    'access-control-allow-headers': 'content-type',
+    'access-control-allow-methods': 'GET,POST,OPTIONS',
+    'access-control-allow-origin': '*',
+    'access-control-max-age': '600',
+  };
+}
+
 function serveStaticFile(request, response, staticDir, pathname) {
   if (!existsSync(staticDir)) {
     return sendJson(response, 404, {

package/lib/scanner-worker.mjs

@@ -0,0 +1,25 @@
+import { scanVsCodeSessions } from './scanner-api.mjs';
+
+process.on('message', async (message) => {
+  if (!message || typeof message !== 'object' || message.type !== 'scan') {
+    return;
+  }
+
+  try {
+    const sessionData = await scanVsCodeSessions({
+      ...(message.options ?? {}),
+      onProgress: (event) => {
+        process.send?.({ type: 'progress', event });
+      },
+    });
+    process.send?.({ type: 'result', sessionData });
+  } catch (error) {
+    process.send?.({
+      type: 'error',
+      error: {
+        message: error instanceof Error ? error.message : String(error),
+        stack: error instanceof Error ? error.stack : '',
+      },
+    });
+  }
+});

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "copilot-usage-studio",
-  "version": "0.2.0",
+  "version": "0.2.1",
   "description": "Local VS Code GitHub Copilot usage, cost, and session insights.",
   "type": "module",
   "bin": {
@@ -17,6 +17,12 @@
     "docs/scanner-api.md",
     "lib/",
     "scripts/pricing-utils.mjs",
+    "scripts/scanner-customization-evidence.mjs",
+    "scripts/scanner-customization-inventory.mjs",
+    "scripts/scanner-memory.mjs",
+    "scripts/scanner-session-parser.mjs",
+    "scripts/scanner-traversal.mjs",
+    "scripts/scanner-workspace.mjs",
     "scripts/scan-vscode-sessions.mjs"
   ],
   "scripts": {
@@ -38,7 +44,14 @@
     "verify:data": "node scripts/verify-session-data.mjs",
     "schema:audit": "node scripts/audit-vscode-schema.mjs",
     "schema:accept": "node scripts/audit-vscode-schema.mjs --accept",
+    "vscode:typecheck": "tsc -p vscode-extension/tsconfig.json --noEmit",
+    "vscode:compile": "npm run vscode:typecheck && node scripts/build-vscode-extension.mjs --extension-only",
+    "vscode:build": "npm run build && npm run vscode:typecheck && node scripts/build-vscode-extension.mjs",
+    "vscode:verify": "npm run vscode:build && node scripts/verify-vscode-extension.mjs",
+    "vscode:vsix:dry-run": "npm run vscode:verify && cd vscode-extension && vsce ls --tree",
+    "vscode:package": "npm run vscode:verify && node scripts/package-vscode-extension.mjs",
     "verify:package": "node scripts/verify-package.mjs",
+    "release:notes": "node scripts/generate-release-notes.mjs",
     "release:check": "npm run test:scripts && npm test -- --watch=false && npm run build && npm run verify:package",
     "prepack": "npm run build"
   },
@@ -77,6 +90,9 @@
     "@angular/forms": "^21.2.0",
     "@angular/platform-browser": "^21.2.0",
     "@angular/router": "^21.2.0",
+    "@types/node": "^26.0.0",
+    "@types/vscode": "^1.125.0",
+    "@vscode/vsce": "^3.9.2",
     "jsdom": "^28.0.0",
     "prettier": "^3.8.1",
     "rxjs": "~7.8.0",

package/scripts/pricing-utils.mjs

@@ -13,10 +13,15 @@ export function normalizeModel(model, pricing) {
     .trim();
   const key = modelKey(raw);
   const knownModels = Object.keys(pricing);
+  const knownAliases = knownModels.flatMap((name) =>
+    (pricing[name].aliases ?? []).map((alias) => ({ name, aliasKey: modelKey(alias) })),
+  );
 
   return (
     knownModels.find((name) => modelKey(name) === key) ??
+    knownAliases.find(({ aliasKey }) => aliasKey === key)?.name ??
     knownModels.find((name) => key.includes(modelKey(name))) ??
+    knownAliases.find(({ aliasKey }) => key.includes(aliasKey))?.name ??
     (raw || 'Unknown model')
   );
 }