perf(tui): session-scoped caching to eliminate progressive slowdown (#59)

* feat(tui): add TuiCache class for session-scoped caching

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* feat(tui): make getTuiContext() use session cache

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* perf(tui): wire session cache into wizard loop and invalidate after writes

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* perf(tui): cache marketplace plugin lists in browse/install flows

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* fix(tui): remove non-null assertion to satisfy biome lint

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* fix(test): update workspace status e2e test for user workspace fallback

The test expected exit code 1 for workspace status in a non-workspace dir,
but since the user workspace fallback was added, the command now succeeds
with empty data. Also added HOME mock for environment isolation.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

* perf(tui): cache listMarketplaces() calls and remove dead registry code

Replace unused MarketplaceRegistry cache with MarketplaceEntry[] cache
that matches what listMarketplaces() returns. Wire cached marketplace
list into runInstallPlugin, runBrowseMarketplaces, and getTuiContext.

Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>

---------

Co-authored-by: Claude Opus 4.5 <noreply@anthropic.com>

Author: christso

src/cli/tui/actions/plugins.ts

@@ -8,9 +8,41 @@ import {
   addMarketplace,
   removeMarketplace,
   updateMarketplace,
+  type MarketplaceEntry,
+  type MarketplacePluginsResult,
 } from '../../../core/marketplace.js';
 import { getWorkspaceStatus } from '../../../core/status.js';
 import type { TuiContext } from '../context.js';
+import type { TuiCache } from '../cache.js';
+
+/**
+ * Get marketplace list, using cache when available.
+ */
+async function getCachedMarketplaces(
+  cache?: TuiCache,
+): Promise<MarketplaceEntry[]> {
+  const cached = cache?.getMarketplaces();
+  if (cached) return cached;
+
+  const result = await listMarketplaces();
+  cache?.setMarketplaces(result);
+  return result;
+}
+
+/**
+ * Get marketplace plugins, using cache when available.
+ */
+async function getCachedMarketplacePlugins(
+  name: string,
+  cache?: TuiCache,
+): Promise<MarketplacePluginsResult> {
+  const cached = cache?.getMarketplacePlugins(name);
+  if (cached) return cached;
+
+  const result = await listMarketplacePlugins(name);
+  cache?.setMarketplacePlugins(name, result);
+  return result;
+}
 
 /**
  * Shared helper: determine scope, install a plugin, sync, and show success.
@@ -19,6 +51,7 @@ import type { TuiContext } from '../context.js';
 async function installSelectedPlugin(
   pluginRef: string,
   context: TuiContext,
+  cache?: TuiCache,
 ): Promise<boolean> {
   // Determine scope
   let scope: 'project' | 'user' = 'user';
@@ -69,6 +102,7 @@ async function installSelectedPlugin(
     syncS.stop('Sync complete');
   }
 
+  cache?.invalidate();
   p.note(`Installed: ${pluginRef}`, 'Success');
   return true;
 }
@@ -77,10 +111,10 @@ async function installSelectedPlugin(
  * Plugin installation flow.
  * Lists marketplace plugins, lets user pick one, installs it, and auto-syncs.
  */
-export async function runInstallPlugin(context: TuiContext): Promise<void> {
+export async function runInstallPlugin(context: TuiContext, cache?: TuiCache): Promise<void> {
   try {
     // Get available marketplaces
-    const marketplaces = await listMarketplaces();
+    const marketplaces = await getCachedMarketplaces(cache);
 
     if (marketplaces.length === 0) {
       p.note(
@@ -93,7 +127,7 @@ export async function runInstallPlugin(context: TuiContext): Promise<void> {
     // Collect plugins from all marketplaces
     const allPlugins: Array<{ label: string; value: string }> = [];
     for (const marketplace of marketplaces) {
-      const result = await listMarketplacePlugins(marketplace.name);
+      const result = await getCachedMarketplacePlugins(marketplace.name, cache);
       for (const plugin of result.plugins) {
         const label = plugin.description
           ? `${plugin.name} - ${plugin.description}`
@@ -119,7 +153,7 @@ export async function runInstallPlugin(context: TuiContext): Promise<void> {
       return;
     }
 
-    await installSelectedPlugin(selected, context);
+    await installSelectedPlugin(selected, context, cache);
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
     p.note(message, 'Error');
@@ -130,7 +164,7 @@ export async function runInstallPlugin(context: TuiContext): Promise<void> {
  * Plugin management (uninstall) flow.
  * Lists installed plugins, lets user pick which to remove, and auto-syncs.
  */
-export async function runManagePlugins(context: TuiContext): Promise<void> {
+export async function runManagePlugins(context: TuiContext, cache?: TuiCache): Promise<void> {
   try {
     const status = await getWorkspaceStatus(context.workspacePath ?? undefined);
 
@@ -202,6 +236,7 @@ export async function runManagePlugins(context: TuiContext): Promise<void> {
       await syncUserWorkspace();
     }
     syncS.stop('Sync complete');
+    cache?.invalidate();
 
     p.note(results.join('\n'), 'Removed');
   } catch (error) {
@@ -216,10 +251,11 @@ export async function runManagePlugins(context: TuiContext): Promise<void> {
  */
 export async function runBrowseMarketplaces(
   context: TuiContext,
+  cache?: TuiCache,
 ): Promise<void> {
   try {
     while (true) {
-      const marketplaces = await listMarketplaces();
+      const marketplaces = await getCachedMarketplaces(cache);
 
       const options: Array<{ label: string; value: string }> = [
         { label: '+ Add marketplace', value: '__add__' },
@@ -258,13 +294,15 @@ export async function runBrowseMarketplaces(
 
         if (!result.success) {
           p.note(result.error ?? 'Unknown error', 'Error');
+        } else {
+          cache?.invalidate();
         }
 
         continue;
       }
 
       // User selected a marketplace — show detail screen
-      await runMarketplaceDetail(selected, context);
+      await runMarketplaceDetail(selected, context, cache);
     }
   } catch (error) {
     const message = error instanceof Error ? error.message : String(error);
@@ -279,6 +317,7 @@ export async function runBrowseMarketplaces(
 async function runMarketplaceDetail(
   marketplaceName: string,
   context: TuiContext,
+  cache?: TuiCache,
 ): Promise<void> {
   while (true) {
     const action = await p.select({
@@ -297,7 +336,7 @@ async function runMarketplaceDetail(
 
     if (action === 'browse') {
       try {
-        const result = await listMarketplacePlugins(marketplaceName);
+        const result = await getCachedMarketplacePlugins(marketplaceName, cache);
 
         if (result.plugins.length === 0) {
           p.note('No plugins found in this marketplace.', 'Plugins');
@@ -323,7 +362,7 @@ async function runMarketplaceDetail(
         }
 
         const pluginRef = `${selectedPlugin}@${marketplaceName}`;
-        await installSelectedPlugin(pluginRef, context);
+        await installSelectedPlugin(pluginRef, context, cache);
       } catch (error) {
         const message = error instanceof Error ? error.message : String(error);
         p.note(message, 'Error');
@@ -344,6 +383,7 @@ async function runMarketplaceDetail(
           )
           .join('\n');
         s.stop('Update complete');
+        cache?.invalidate();
         p.note(summary || 'Marketplace updated.', 'Update');
       } catch (error) {
         const message = error instanceof Error ? error.message : String(error);
@@ -376,6 +416,8 @@ async function runMarketplaceDetail(
           p.note(result.error ?? 'Unknown error', 'Error');
           continue;
         }
+
+        cache?.invalidate();
       } catch (error) {
         const message = error instanceof Error ? error.message : String(error);
         p.note(message, 'Error');

src/cli/tui/cache.ts

@@ -0,0 +1,55 @@
+import type { MarketplaceEntry, MarketplacePluginsResult } from '../../core/marketplace.js';
+import type { TuiContext } from './context.js';
+
+/**
+ * Session-scoped cache for the TUI wizard.
+ * Holds expensive-to-compute data (registry, plugin lists, context) in memory.
+ * Call invalidate() after any write operation (install, remove, sync, etc.).
+ */
+export class TuiCache {
+  private marketplaces: MarketplaceEntry[] | undefined;
+  private context: TuiContext | undefined;
+  private marketplacePlugins: Map<string, MarketplacePluginsResult> = new Map();
+
+  /** Get cached marketplace list (undefined if not cached) */
+  getMarketplaces(): MarketplaceEntry[] | undefined {
+    return this.marketplaces;
+  }
+
+  /** Store marketplace list in cache */
+  setMarketplaces(marketplaces: MarketplaceEntry[]): void {
+    this.marketplaces = marketplaces;
+  }
+
+  /** Check if context is cached */
+  hasCachedContext(): boolean {
+    return this.context !== undefined;
+  }
+
+  /** Get cached context (undefined if not cached) */
+  getContext(): TuiContext | undefined {
+    return this.context;
+  }
+
+  /** Store context in cache */
+  setContext(context: TuiContext): void {
+    this.context = context;
+  }
+
+  /** Get cached marketplace plugins for a specific marketplace */
+  getMarketplacePlugins(name: string): MarketplacePluginsResult | undefined {
+    return this.marketplacePlugins.get(name);
+  }
+
+  /** Store marketplace plugins in cache */
+  setMarketplacePlugins(name: string, result: MarketplacePluginsResult): void {
+    this.marketplacePlugins.set(name, result);
+  }
+
+  /** Clear all cached data. Call after any write operation. */
+  invalidate(): void {
+    this.marketplaces = undefined;
+    this.context = undefined;
+    this.marketplacePlugins.clear();
+  }
+}

src/cli/tui/context.ts

@@ -5,6 +5,7 @@ import { getWorkspaceStatus } from '../../core/status.js';
 import { loadSyncState } from '../../core/sync-state.js';
 import { getUserWorkspaceConfig } from '../../core/user-workspace.js';
 import { listMarketplaces } from '../../core/marketplace.js';
+import type { TuiCache } from './cache.js';
 
 /**
  * Workspace context detected at TUI startup and after each action.
@@ -26,7 +27,13 @@ export interface TuiContext {
  */
 export async function getTuiContext(
   cwd: string = process.cwd(),
+  cache?: TuiCache,
 ): Promise<TuiContext> {
+  // Return cached context if available
+  const cachedContext = cache?.getContext();
+  if (cachedContext) {
+    return cachedContext;
+  }
   const configPath = join(cwd, CONFIG_DIR, WORKSPACE_CONFIG_FILE);
   const hasWorkspace = existsSync(configPath);
 
@@ -67,13 +74,19 @@ export async function getTuiContext(
   // Marketplace count
   let marketplaceCount = 0;
   try {
-    const marketplaces = await listMarketplaces();
-    marketplaceCount = marketplaces.length;
+    const cachedMarketplaces = cache?.getMarketplaces();
+    if (cachedMarketplaces) {
+      marketplaceCount = cachedMarketplaces.length;
+    } else {
+      const marketplaces = await listMarketplaces();
+      cache?.setMarketplaces(marketplaces);
+      marketplaceCount = marketplaces.length;
+    }
   } catch {
     // Marketplace listing failed -- degrade gracefully
   }
 
-  return {
+  const context: TuiContext = {
     hasWorkspace,
     workspacePath: hasWorkspace ? cwd : null,
     projectPluginCount,
@@ -82,6 +95,9 @@ export async function getTuiContext(
     hasUserConfig,
     marketplaceCount,
   };
+
+  cache?.setContext(context);
+  return context;
 }
 
 /**

src/cli/tui/wizard.ts

@@ -2,6 +2,7 @@ import * as p from '@clack/prompts';
 import chalk from 'chalk';
 import { relative } from 'node:path';
 import packageJson from '../../../package.json';
+import { TuiCache } from './cache.js';
 import { getTuiContext, type TuiContext } from './context.js';
 import { runInit } from './actions/init.js';
 import { runSync } from './actions/sync.js';
@@ -91,7 +92,8 @@ function buildSummary(context: TuiContext): string {
 export async function runWizard(): Promise<void> {
   p.intro(`${chalk.cyan('allagents')} v${packageJson.version}`);
 
-  let context = await getTuiContext();
+  const cache = new TuiCache();
+  let context = await getTuiContext(process.cwd(), cache);
 
   // biome-ignore lint/correctness/noConstantCondition: intentional wizard loop
   while (true) {
@@ -110,31 +112,34 @@ export async function runWizard(): Promise<void> {
     switch (action) {
       case 'init':
         await runInit();
+        cache.invalidate();
         break;
       case 'sync':
         await runSync(context);
+        cache.invalidate();
         break;
       case 'status':
         await runStatus(context);
         break;
       case 'install':
-        await runInstallPlugin(context);
+        await runInstallPlugin(context, cache);
         break;
       case 'manage':
-        await runManagePlugins(context);
+        await runManagePlugins(context, cache);
         break;
       case 'marketplace':
-        await runBrowseMarketplaces(context);
+        await runBrowseMarketplaces(context, cache);
         break;
       case 'update':
         await runUpdate();
+        cache.invalidate();
         break;
       case 'exit':
         p.outro('Bye');
         return;
     }
 
-    // Refresh context after each action
-    context = await getTuiContext();
+    // Refresh context after each action (cache makes this cheap when nothing changed)
+    context = await getTuiContext(process.cwd(), cache);
   }
 }

tests/e2e/cli-json-output.test.ts

@@ -91,7 +91,7 @@ describe('CLI --json error cases', () => {
   });
 
   it('workspace status --json in non-workspace dir falls back to user workspace', async () => {
-    const { stdout, exitCode } = await runCli(['workspace', 'status', '--json']);
+    const { stdout, exitCode } = await runCli(['workspace', 'status', '--json'], { HOME: mockHome });
     expect(exitCode).toBe(0);
     const json = parseJson(stdout);
     expect(json.success).toBe(true);