stacklok / toolhive-studio / 24717623940

70%

push

github

feat(playground): persist MCP App UI metadata across restarts (#2057)

* feat(db): add mcp_app_ui_metadata table migration

Introduces migration 004 with mcp_app_ui_metadata(tool_name PK, server_name, resource_uri, updated_at) and an index on server_name. Applies it in the in-memory test DB helper. Groundwork for persisting the Playground MCP App UI metadata cache across app restarts.

* feat(db): add reader and writer for mcp app ui metadata

Adds readAllMcpAppUiMetadata, replaceAllMcpAppUiMetadata (transactional DELETE + bulk INSERT so each call is atomic and prunes stale rows), and clearAllMcpAppUiMetadata. Covered by 5 new tests: empty read, round-trip, replace-without-accumulation, empty-replace clears the table, and clearAll.

* feat(playground): persist MCP App UI metadata cache across restarts

getCachedUiMetadata() now seeds the in-memory cache from SQLite on first access via ensureUiMetadataLoaded(), and createMcpTools() calls replaceAllMcpAppUiMetadata(cachedUiMetadata) at the end of each stream setup (even when empty, so servers that lost all UI tools have their stale rows pruned). Historical tool parts with _meta.ui now remount McpAppView on app restart without first requiring a new streaming session to repopulate the cache.

* fix(playground): preserve persisted MCP App UI metadata on discovery failure

When createMcpTools() used to reset the module-level cachedUiMetadata at
the top and unconditionally persist it at the bottom, a transient
fetchWorkloads() or getEnabledMcpTools() rejection would wipe the
previously persisted rows — defeating the whole point of the cache.
Build discovery into a local map and only swap + persist once the
required calls have resolved.

Made-with: Cursor

* fix(playground): retry MCP App UI metadata hydration after transient DB read failure

ensureUiMetadataLoaded() was latching uiMetadataLoaded=true before the
SQLite read, so a transient failure (e.g. database locked at startup)
would permanently disable re... (continued)

3844 of 6337 branches covered (60.66%)

42 of 43 new or added lines in 4 files covered. (97.67%)

5983 of 9023 relevant lines covered (66.31%)

119.52 hits per line