@assistant-ui/react-mcp 0.0.0

package/dist/resources/McpManagerResource.js

@@ -0,0 +1,191 @@
+import { resource, tapState, tapResource, tapEffect, tapMemo, tapEffectEvent, tapRef, withKey, } from "@assistant-ui/tap";
+import { tapClientLookup, tapAssistantClientRef, attachTransformScopes, } from "@assistant-ui/store";
+import { ModelContext } from "@assistant-ui/core/store";
+import { McpServerResource } from "./McpServerResource.js";
+import { McpLocalStorage } from "./storage/McpLocalStorage.js";
+function defaultRedirectUri() {
+    if (typeof window === "undefined")
+        return "";
+    return `${window.location.origin}/mcp/callback`;
+}
+export const McpManagerResource = resource((props) => {
+    const connectors = props.connectors ?? [];
+    const autoConnect = props.autoConnect ?? true;
+    const redirectUri = props.oauthRedirectUri ?? defaultRedirectUri();
+    const storageElement = props.storage ?? McpLocalStorage();
+    const storage = tapResource(storageElement);
+    const [customServers, setCustomServers] = tapState([]);
+    const [isHydrated, setIsHydrated] = tapState(false);
+    const hydratedRef = tapRef(false);
+    const hydrate = tapEffectEvent(async (signal) => {
+        try {
+            const records = await storage.loadCustomServers();
+            if (signal.cancelled)
+                return;
+            // Merge rather than replace so any addCustomServer calls that
+            // happened before hydration resolved aren't silently overwritten.
+            // Persisted order wins; pre-hydration locals append.
+            setCustomServers((prev) => {
+                if (prev.length === 0)
+                    return records;
+                const persistedIds = new Set(records.map((r) => r.id));
+                return [...records, ...prev.filter((r) => !persistedIds.has(r.id))];
+            });
+        }
+        finally {
+            if (!signal.cancelled) {
+                hydratedRef.current = true;
+                setIsHydrated(true);
+            }
+        }
+    });
+    tapEffect(() => {
+        const signal = { cancelled: false };
+        void hydrate(signal);
+        return () => {
+            signal.cancelled = true;
+        };
+    }, []);
+    const persistCustomServers = tapEffectEvent(async (records) => {
+        if (!hydratedRef.current)
+            return;
+        await storage.saveCustomServers(records);
+    });
+    tapEffect(() => {
+        void persistCustomServers(customServers);
+    }, [customServers]);
+    const serverElements = tapMemo(() => {
+        const connectorElements = connectors.map((c) => withKey(c.id, McpServerResource({
+            id: c.id,
+            kind: "connector",
+            name: c.name,
+            url: c.url,
+            icon: c.icon,
+            auth: c.auth,
+            storage,
+            redirectUri,
+            autoConnect,
+            onRemove: async () => {
+                // connectors cannot be removed
+            },
+        })));
+        const customElements = customServers.map((s) => withKey(s.id, McpServerResource({
+            id: s.id,
+            kind: "custom",
+            name: s.name,
+            url: s.url,
+            auth: s.auth,
+            storage,
+            redirectUri,
+            autoConnect,
+            onRemove: async () => {
+                setCustomServers((prev) => prev.filter((x) => x.id !== s.id));
+            },
+        })));
+        return [...connectorElements, ...customElements];
+    }, [connectors, customServers, storage, redirectUri, autoConnect]);
+    const lookup = tapClientLookup(() => serverElements, [serverElements]);
+    const state = tapMemo(() => {
+        const all = lookup.state;
+        return {
+            servers: all,
+            connectors: all.filter((s) => s.kind === "connector"),
+            customServers: all.filter((s) => s.kind === "custom"),
+            isHydrated,
+        };
+    }, [lookup.state, isHydrated]);
+    // ─── Auto-register MCP tools as frontend tools in modelContext ─────
+    // Build the toolkit from connected servers; re-register when the visible
+    // tool surface changes. Tool names are prefixed with the server id to
+    // avoid collisions across connected servers — connector ids must not
+    // contain `__` (enforced by `defineConnector`); `addCustomServer`
+    // generates UUIDs that satisfy the constraint by construction.
+    const toolkit = tapMemo(() => {
+        const out = {};
+        for (const server of state.servers) {
+            if (server.connectionState !== "connected")
+                continue;
+            for (const tool of server.tools) {
+                const fullName = `${server.id}__${tool.name}`;
+                out[fullName] = {
+                    type: "frontend",
+                    ...(tool.description !== undefined
+                        ? { description: tool.description }
+                        : {}),
+                    parameters: tool.inputSchema,
+                    execute: (args) => lookup
+                        .get({ key: server.id })
+                        .callTool(tool.name, args),
+                };
+            }
+        }
+        return out;
+    }, [state, lookup]);
+    const clientRef = tapAssistantClientRef();
+    tapEffect(() => {
+        const client = clientRef.current;
+        if (!client)
+            return;
+        return client.modelContext().register({
+            getModelContext: () => ({ tools: toolkit }),
+        });
+    }, [toolkit, clientRef]);
+    const serverByKind = (kind, index) => {
+        const list = kind === "connector" ? state.connectors : state.customServers;
+        const entry = list[index];
+        if (!entry) {
+            throw new Error(`McpManagerResource: no ${kind} at index ${index} (length ${list.length})`);
+        }
+        return lookup.get({ key: entry.id });
+    };
+    return {
+        getState: () => state,
+        server: (query) => {
+            if ("id" in query)
+                return lookup.get({ key: query.id });
+            return serverByKind(query.kind, query.index);
+        },
+        connector: ({ index }) => serverByKind("connector", index),
+        customServer: ({ index }) => serverByKind("custom", index),
+        addCustomServer: async ({ name, url, auth }) => {
+            const record = {
+                id: typeof crypto !== "undefined" && "randomUUID" in crypto
+                    ? crypto.randomUUID()
+                    : `mcp-${Date.now()}-${Math.random().toString(36).slice(2)}`,
+                name,
+                url,
+                auth: auth,
+                createdAt: Date.now(),
+            };
+            setCustomServers((prev) => [...prev, record]);
+            return record.id;
+        },
+        removeServer: async (id) => {
+            // removeServer is custom-server only — connectors are app-defined
+            // and not user-removable. Refuse rather than silently no-op.
+            if (state.connectors.some((c) => c.id === id)) {
+                throw new Error(`Cannot remove connector "${id}" — connectors are app-defined and not removable. Use a custom server id instead.`);
+            }
+            // Delegate to McpServerResource.remove() which disconnects,
+            // clears auth state, and unregisters from customServers in one
+            // place. Fallback to manual cleanup if the lookup is empty
+            // (server already gone).
+            try {
+                await lookup.get({ key: id }).remove();
+            }
+            catch {
+                await storage.clearAuthState(id);
+                setCustomServers((prev) => prev.filter((s) => s.id !== id));
+            }
+        },
+    };
+});
+// Ensure modelContext exists as a sibling when the manager mounts. If an
+// ancestor (e.g. a chat runtime) already provides modelContext, this is a
+// no-op; otherwise it's auto-mounted alongside `mcp`.
+attachTransformScopes(McpManagerResource, (scopes, parent) => {
+    if (!scopes.modelContext && parent.modelContext.source === null) {
+        scopes.modelContext = ModelContext();
+    }
+});
+//# sourceMappingURL=McpManagerResource.js.map

package/dist/resources/McpManagerResource.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpManagerResource.js","sourceRoot":"","sources":["../../src/resources/McpManagerResource.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,QAAQ,EACR,WAAW,EACX,SAAS,EACT,OAAO,EACP,cAAc,EACd,MAAM,EACN,OAAO,GACR,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EACL,eAAe,EACf,qBAAqB,EACrB,qBAAqB,GAEtB,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAE,YAAY,EAAE,MAAM,0BAA0B,CAAC;AAExD,OAAO,EAAE,iBAAiB,EAAE,+BAA4B;AACxD,OAAO,EAAE,eAAe,EAAE,qCAAkC;AAkB5D,SAAS,kBAAkB;IACzB,IAAI,OAAO,MAAM,KAAK,WAAW;QAAE,OAAO,EAAE,CAAC;IAC7C,OAAO,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,eAAe,CAAC;AAClD,CAAC;AAED,MAAM,CAAC,MAAM,kBAAkB,GAAG,QAAQ,CACxC,CAAC,KAA8B,EAAuB,EAAE;IACtD,MAAM,UAAU,GAAG,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC;IAC1C,MAAM,WAAW,GAAG,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC;IAC9C,MAAM,WAAW,GAAG,KAAK,CAAC,gBAAgB,IAAI,kBAAkB,EAAE,CAAC;IAEnE,MAAM,cAAc,GAAG,KAAK,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC;IAC1D,MAAM,OAAO,GAAG,WAAW,CAAC,cAAc,CAAC,CAAC;IAE5C,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAChD,EAAE,CACH,CAAC;IACF,MAAM,CAAC,UAAU,EAAE,aAAa,CAAC,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IAEpD,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;IAElC,MAAM,OAAO,GAAG,cAAc,CAAC,KAAK,EAAE,MAA8B,EAAE,EAAE;QACtE,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,iBAAiB,EAAE,CAAC;YAClD,IAAI,MAAM,CAAC,SAAS;gBAAE,OAAO;YAC7B,8DAA8D;YAC9D,kEAAkE;YAClE,qDAAqD;YACrD,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE;gBACxB,IAAI,IAAI,CAAC,MAAM,KAAK,CAAC;oBAAE,OAAO,OAAO,CAAC;gBACtC,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;gBACvD,OAAO,CAAC,GAAG,OAAO,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YACtE,CAAC,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC;gBACtB,WAAW,CAAC,OAAO,GAAG,IAAI,CAAC;gBAC3B,aAAa,CAAC,IAAI,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,MAAM,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QACpC,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;QACrB,OAAO,GAAG,EAAE;YACV,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;QAC1B,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,oBAAoB,GAAG,cAAc,CACzC,KAAK,EAAE,OAAgC,EAAE,EAAE;QACzC,IAAI,CAAC,WAAW,CAAC,OAAO;YAAE,OAAO;QACjC,MAAM,OAAO,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC,CACF,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,KAAK,oBAAoB,CAAC,aAAa,CAAC,CAAC;IAC3C,CAAC,EAAE,CAAC,aAAa,CAAC,CAAC,CAAC;IAEpB,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,EAAE;QAClC,MAAM,iBAAiB,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC7C,OAAO,CACL,CAAC,CAAC,EAAE,EACJ,iBAAiB,CAAC;YAChB,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,GAAG,EAAE,CAAC,CAAC,GAAG;YACV,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,OAAO;YACP,WAAW;YACX,WAAW;YACX,QAAQ,EAAE,KAAK,IAAI,EAAE;gBACnB,+BAA+B;YACjC,CAAC;SACF,CAAC,CACH,CACF,CAAC;QACF,MAAM,cAAc,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAC7C,OAAO,CACL,CAAC,CAAC,EAAE,EACJ,iBAAiB,CAAC;YAChB,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,IAAI,EAAE,QAAQ;YACd,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,GAAG,EAAE,CAAC,CAAC,GAAG;YACV,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,OAAO;YACP,WAAW;YACX,WAAW;YACX,QAAQ,EAAE,KAAK,IAAI,EAAE;gBACnB,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAChE,CAAC;SACF,CAAC,CACH,CACF,CAAC;QACF,OAAO,CAAC,GAAG,iBAAiB,EAAE,GAAG,cAAc,CAAC,CAAC;IACnD,CAAC,EAAE,CAAC,UAAU,EAAE,aAAa,EAAE,OAAO,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC,CAAC;IAEnE,MAAM,MAAM,GAAG,eAAe,CAAC,GAAG,EAAE,CAAC,cAAc,EAAE,CAAC,cAAc,CAAC,CAAC,CAAC;IAEvE,MAAM,KAAK,GAAG,OAAO,CAAkB,GAAG,EAAE;QAC1C,MAAM,GAAG,GAAG,MAAM,CAAC,KAAK,CAAC;QACzB,OAAO;YACL,OAAO,EAAE,GAAG;YACZ,UAAU,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC;YACrD,aAAa,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,CAAC;YACrD,UAAU;SACX,CAAC;IACJ,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IAE/B,sEAAsE;IACtE,yEAAyE;IACzE,sEAAsE;IACtE,qEAAqE;IACrE,kEAAkE;IAClE,+DAA+D;IAC/D,MAAM,OAAO,GAAG,OAAO,CAAiC,GAAG,EAAE;QAC3D,MAAM,GAAG,GAAmC,EAAE,CAAC;QAC/C,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;YACnC,IAAI,MAAM,CAAC,eAAe,KAAK,WAAW;gBAAE,SAAS;YACrD,KAAK,MAAM,IAAI,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAChC,MAAM,QAAQ,GAAG,GAAG,MAAM,CAAC,EAAE,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;gBAC9C,GAAG,CAAC,QAAQ,CAAC,GAAG;oBACd,IAAI,EAAE,UAAU;oBAChB,GAAG,CAAC,IAAI,CAAC,WAAW,KAAK,SAAS;wBAChC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE;wBACnC,CAAC,CAAC,EAAE,CAAC;oBACP,UAAU,EAAE,IAAI,CAAC,WAAoB;oBACrC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAChB,MAAM;yBACH,GAAG,CAAC,EAAE,GAAG,EAAE,MAAM,CAAC,EAAE,EAAE,CAAC;yBACvB,QAAQ,CAAC,IAAI,CAAC,IAAI,EAAE,IAAe,CAAC;iBAC1C,CAAC;YACJ,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC;IAEpB,MAAM,SAAS,GAAG,qBAAqB,EAAE,CAAC;IAE1C,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC;QACjC,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,OAAO,MAAM,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC;YACpC,eAAe,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC;SAC5C,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC,CAAC;IAEzB,MAAM,YAAY,GAAG,CAAC,IAA4B,EAAE,KAAa,EAAE,EAAE;QACnE,MAAM,IAAI,GACR,IAAI,KAAK,WAAW,CAAC,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC;QAChE,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,MAAM,IAAI,KAAK,CACb,0BAA0B,IAAI,aAAa,KAAK,YAAY,IAAI,CAAC,MAAM,GAAG,CAC3E,CAAC;QACJ,CAAC;QACD,OAAO,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;IACvC,CAAC,CAAC;IAEF,OAAO;QACL,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK;QACrB,MAAM,EAAE,CAAC,KAAK,EAAE,EAAE;YAChB,IAAI,IAAI,IAAI,KAAK;gBAAE,OAAO,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;YACxD,OAAO,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QAC/C,CAAC;QACD,SAAS,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,YAAY,CAAC,WAAW,EAAE,KAAK,CAAC;QAC1D,YAAY,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,KAAK,CAAC;QAC1D,eAAe,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,EAAE,EAAE;YAC7C,MAAM,MAAM,GAA0B;gBACpC,EAAE,EACA,OAAO,MAAM,KAAK,WAAW,IAAI,YAAY,IAAI,MAAM;oBACrD,CAAC,CAAC,MAAM,CAAC,UAAU,EAAE;oBACrB,CAAC,CAAC,OAAO,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE;gBAChE,IAAI;gBACJ,GAAG;gBACH,IAAI,EAAE,IAAqB;gBAC3B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YACF,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,GAAG,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;YAC9C,OAAO,MAAM,CAAC,EAAE,CAAC;QACnB,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YACzB,kEAAkE;YAClE,6DAA6D;YAC7D,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC;gBAC9C,MAAM,IAAI,KAAK,CACb,4BAA4B,EAAE,mFAAmF,CAClH,CAAC;YACJ,CAAC;YACD,4DAA4D;YAC5D,+DAA+D;YAC/D,2DAA2D;YAC3D,yBAAyB;YACzB,IAAI,CAAC;gBACH,MAAM,MAAM,CAAC,GAAG,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC;YACzC,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,OAAO,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;gBACjC,gBAAgB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC,CACF,CAAC;AAEF,yEAAyE;AACzE,0EAA0E;AAC1E,sDAAsD;AACtD,qBAAqB,CAAC,kBAAkB,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE;IAC3D,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,MAAM,CAAC,YAAY,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;QAChE,MAAM,CAAC,YAAY,GAAG,YAAY,EAAE,CAAC;IACvC,CAAC;AACH,CAAC,CAAC,CAAC"}

package/dist/resources/McpServerResource.d.ts

@@ -0,0 +1,17 @@
+import type { ClientOutput } from "@assistant-ui/store";
+import type { MCPStorage } from "./storage/types.js";
+import type { MCPAuthConfig, MCPServerKind } from "../mcp-scope.js";
+export type McpServerResourceProps = {
+    id: string;
+    kind: MCPServerKind;
+    name: string;
+    url: string;
+    icon?: string | undefined;
+    auth: MCPAuthConfig;
+    storage: MCPStorage;
+    redirectUri: string;
+    autoConnect: boolean;
+    onRemove: () => Promise<void>;
+};
+export declare const McpServerResource: (props: McpServerResourceProps) => import("@assistant-ui/tap").ResourceElement<ClientOutput<"mcpServer">, McpServerResourceProps>;
+//# sourceMappingURL=McpServerResource.d.ts.map

package/dist/resources/McpServerResource.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpServerResource.d.ts","sourceRoot":"","sources":["../../src/resources/McpServerResource.ts"],"names":[],"mappings":"AAQA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAWxD,OAAO,KAAK,EAAE,UAAU,EAAE,2BAAwB;AAClD,OAAO,KAAK,EACV,aAAa,EAEb,aAAa,EAGd,wBAAqB;AAEtB,MAAM,MAAM,sBAAsB,GAAG;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,aAAa,CAAC;IACpB,IAAI,EAAE,MAAM,CAAC;IACb,GAAG,EAAE,MAAM,CAAC;IACZ,IAAI,CAAC,EAAE,MAAM,GAAG,SAAS,CAAC;IAC1B,IAAI,EAAE,aAAa,CAAC;IACpB,OAAO,EAAE,UAAU,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,WAAW,EAAE,OAAO,CAAC;IACrB,QAAQ,EAAE,MAAM,OAAO,CAAC,IAAI,CAAC,CAAC;CAC/B,CAAC;AAEF,eAAO,MAAM,iBAAiB,mIA0P7B,CAAC"}

package/dist/resources/McpServerResource.js

@@ -0,0 +1,239 @@
+import { resource, tapState, tapRef, tapEffect, tapMemo, tapEffectEvent, } from "@assistant-ui/tap";
+import { Client } from "@modelcontextprotocol/sdk/client/index.js";
+import { StreamableHTTPClientTransport, } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
+import { UnauthorizedError } from "@modelcontextprotocol/sdk/client/auth.js";
+import { createOAuthProvider } from "../auth/createOAuthProvider.js";
+import { buildHeaders } from "../auth/buildHeaders.js";
+import { assertValidServerId } from "../utils/serverId.js";
+export const McpServerResource = resource((props) => {
+    assertValidServerId(props.id);
+    const [connectionState, setConnectionState] = tapState("disconnected");
+    const [tools, setTools] = tapState([]);
+    const [lastError, setLastError] = tapState(null);
+    const [authorizationUrl, setAuthorizationUrl] = tapState(null);
+    const clientRef = tapRef(null);
+    const transportRef = tapRef(null);
+    const buildTransport = tapEffectEvent(async () => {
+        if (props.auth.type === "oauth") {
+            const authProvider = createOAuthProvider({
+                serverId: props.id,
+                config: props.auth,
+                storage: props.storage,
+                redirectUri: props.redirectUri,
+                onAuthorizationUrl: (url) => setAuthorizationUrl(url.toString()),
+            });
+            return new StreamableHTTPClientTransport(new URL(props.url), {
+                authProvider,
+            });
+        }
+        if (props.auth.type === "bearer") {
+            const persisted = await props.storage.loadAuthState(props.id);
+            const headers = buildHeaders(props.auth, persisted);
+            const transportOpts = {};
+            if (headers)
+                transportOpts.requestInit = { headers };
+            return new StreamableHTTPClientTransport(new URL(props.url), transportOpts);
+        }
+        return new StreamableHTTPClientTransport(new URL(props.url));
+    });
+    const finalizeConnect = tapEffectEvent(async (transport) => {
+        const client = new Client({
+            name: "assistant-ui-mcp",
+            version: "0.0.0",
+        });
+        // SDK's StreamableHTTPClientTransport.sessionId is `string | undefined`
+        // but Transport.sessionId is declared `string?` — under
+        // exactOptionalPropertyTypes the SDK's own classes don't satisfy its
+        // Transport interface. Cast to bridge the gap.
+        await client.connect(transport);
+        // Defer ref assignment until listTools() also succeeds — otherwise a
+        // post-connect failure leaves stale refs that `callTool()` would
+        // happily walk into, producing confusing SDK errors instead of
+        // "not connected".
+        const list = await client.listTools();
+        clientRef.current = client;
+        transportRef.current = transport;
+        setTools(list.tools.map((t) => {
+            const info = {
+                name: t.name,
+                inputSchema: t.inputSchema,
+            };
+            if (t.description !== undefined)
+                info.description = t.description;
+            return info;
+        }));
+        setConnectionState("connected");
+    });
+    const closeTransport = async () => {
+        const t = transportRef.current;
+        transportRef.current = null;
+        clientRef.current = null;
+        if (t) {
+            try {
+                await t.close();
+            }
+            catch {
+                // ignore close errors
+            }
+        }
+    };
+    const doConnect = tapEffectEvent(async () => {
+        // Close any prior transport/client so a re-connect doesn't leak.
+        await closeTransport();
+        setConnectionState("connecting");
+        setLastError(null);
+        setAuthorizationUrl(null);
+        // Clear tools so a reconnect (error → connect, or authRequired
+        // → connect) doesn't expose the previous attempt's tool list.
+        setTools([]);
+        let transport = null;
+        try {
+            transport = await buildTransport();
+            // Don't assign to transportRef until connect succeeds — otherwise a
+            // failed `listTools()` leaves an orphaned transport that future
+            // doConnect / doDisconnect calls treat as live.
+            await finalizeConnect(transport);
+        }
+        catch (err) {
+            if (err instanceof UnauthorizedError) {
+                // OAuth: keep the transport alive so completeAuth can call
+                // finishAuth on it. Closing it before storing would leave a
+                // closed transport on transportRef.
+                transportRef.current = transport;
+                setConnectionState("authRequired");
+            }
+            else {
+                if (transport) {
+                    try {
+                        await transport.close();
+                    }
+                    catch {
+                        // ignore close errors
+                    }
+                }
+                setLastError({
+                    message: err instanceof Error ? err.message : String(err),
+                });
+                setConnectionState("error");
+            }
+        }
+    });
+    const doDisconnect = tapEffectEvent(async () => {
+        setTools([]);
+        setAuthorizationUrl(null);
+        setConnectionState("disconnected");
+        await closeTransport();
+    });
+    const doCompleteAuth = tapEffectEvent(async (callbackUrl) => {
+        setConnectionState("authPending");
+        setLastError(null);
+        try {
+            const url = new URL(callbackUrl);
+            const code = url.searchParams.get("code");
+            if (!code)
+                throw new Error("missing authorization code in callback URL");
+            let transport = transportRef.current;
+            if (!transport) {
+                transport = await buildTransport();
+                transportRef.current = transport;
+            }
+            await transport.finishAuth(code);
+            setAuthorizationUrl(null);
+            await finalizeConnect(transport);
+        }
+        catch (err) {
+            await closeTransport();
+            setLastError({
+                message: err instanceof Error ? err.message : String(err),
+            });
+            setConnectionState("error");
+        }
+    });
+    const tryAutoConnect = tapEffectEvent(async (signal) => {
+        if (!props.autoConnect)
+            return;
+        if (props.auth.type === "none") {
+            void doConnect();
+            return;
+        }
+        // Static connector token short-circuits the storage read.
+        if (props.auth.type === "bearer" && props.auth.token) {
+            void doConnect();
+            return;
+        }
+        const persisted = await props.storage.loadAuthState(props.id);
+        if (signal.cancelled)
+            return;
+        if (props.auth.type === "oauth") {
+            if (!persisted?.tokens)
+                return;
+        }
+        else if (!persisted?.token) {
+            return;
+        }
+        void doConnect();
+    });
+    // Auto-connect on mount when usable auth exists.
+    tapEffect(() => {
+        const signal = { cancelled: false };
+        void tryAutoConnect(signal);
+        return () => {
+            signal.cancelled = true;
+            const t = transportRef.current;
+            transportRef.current = null;
+            clientRef.current = null;
+            if (t)
+                t.close().catch(() => { });
+        };
+    }, []);
+    const state = tapMemo(() => ({
+        id: props.id,
+        kind: props.kind,
+        name: props.name,
+        url: props.url,
+        icon: props.icon,
+        connectionState,
+        lastError,
+        tools,
+        authorizationUrl,
+    }), [
+        props.id,
+        props.kind,
+        props.name,
+        props.url,
+        props.icon,
+        connectionState,
+        lastError,
+        tools,
+        authorizationUrl,
+    ]);
+    return {
+        getState: () => state,
+        connect: doConnect,
+        disconnect: doDisconnect,
+        remove: async () => {
+            await doDisconnect();
+            await props.storage.clearAuthState(props.id);
+            await props.onRemove();
+        },
+        callTool: async (name, args) => {
+            const client = clientRef.current;
+            if (!client) {
+                throw new Error(`MCP server "${props.id}" is not connected`);
+            }
+            return await client.callTool({
+                name,
+                arguments: args,
+            });
+        },
+        readResource: async (uri) => {
+            const client = clientRef.current;
+            if (!client) {
+                throw new Error(`MCP server "${props.id}" is not connected`);
+            }
+            return await client.readResource({ uri });
+        },
+        completeAuth: doCompleteAuth,
+    };
+});
+//# sourceMappingURL=McpServerResource.js.map

package/dist/resources/McpServerResource.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpServerResource.js","sourceRoot":"","sources":["../../src/resources/McpServerResource.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,QAAQ,EACR,MAAM,EACN,SAAS,EACT,OAAO,EACP,cAAc,GACf,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EACL,6BAA6B,GAE9B,MAAM,oDAAoD,CAAC;AAC5D,OAAO,EAAE,iBAAiB,EAAE,MAAM,0CAA0C,CAAC;AAE7E,OAAO,EAAE,mBAAmB,EAAE,uCAAoC;AAClE,OAAO,EAAE,YAAY,EAAE,gCAA6B;AACpD,OAAO,EAAE,mBAAmB,EAAE,6BAA0B;AAuBxD,MAAM,CAAC,MAAM,iBAAiB,GAAG,QAAQ,CACvC,CAAC,KAA6B,EAA6B,EAAE;IAC3D,mBAAmB,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAC9B,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GACzC,QAAQ,CAAqB,cAAc,CAAC,CAAC;IAC/C,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,QAAQ,CAAgB,EAAE,CAAC,CAAC;IACtD,MAAM,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,QAAQ,CACxC,IAAI,CACL,CAAC;IACF,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,QAAQ,CACtD,IAAI,CACL,CAAC;IAEF,MAAM,SAAS,GAAG,MAAM,CAAgB,IAAI,CAAC,CAAC;IAC9C,MAAM,YAAY,GAAG,MAAM,CAAuC,IAAI,CAAC,CAAC;IAExE,MAAM,cAAc,GAAG,cAAc,CACnC,KAAK,IAA4C,EAAE;QACjD,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAChC,MAAM,YAAY,GAAG,mBAAmB,CAAC;gBACvC,QAAQ,EAAE,KAAK,CAAC,EAAE;gBAClB,MAAM,EAAE,KAAK,CAAC,IAAI;gBAClB,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,WAAW,EAAE,KAAK,CAAC,WAAW;gBAC9B,kBAAkB,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,mBAAmB,CAAC,GAAG,CAAC,QAAQ,EAAE,CAAC;aACjE,CAAC,CAAC;YACH,OAAO,IAAI,6BAA6B,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EAAE;gBAC3D,YAAY;aACb,CAAC,CAAC;QACL,CAAC;QACD,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,EAAE,CAAC;YACjC,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC9D,MAAM,OAAO,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;YACpD,MAAM,aAAa,GAAyC,EAAE,CAAC;YAC/D,IAAI,OAAO;gBAAE,aAAa,CAAC,WAAW,GAAG,EAAE,OAAO,EAAE,CAAC;YACrD,OAAO,IAAI,6BAA6B,CACtC,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,EAClB,aAAa,CACd,CAAC;QACJ,CAAC;QACD,OAAO,IAAI,6BAA6B,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;IAC/D,CAAC,CACF,CAAC;IAEF,MAAM,eAAe,GAAG,cAAc,CACpC,KAAK,EAAE,SAAwC,EAAE,EAAE;QACjD,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC;YACxB,IAAI,EAAE,kBAAkB;YACxB,OAAO,EAAE,OAAO;SACjB,CAAC,CAAC;QACH,wEAAwE;QACxE,wDAAwD;QACxD,qEAAqE;QACrE,+CAA+C;QAC/C,MAAM,MAAM,CAAC,OAAO,CAAC,SAAiC,CAAC,CAAC;QACxD,qEAAqE;QACrE,iEAAiE;QACjE,+DAA+D;QAC/D,mBAAmB;QACnB,MAAM,IAAI,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;QACtC,SAAS,CAAC,OAAO,GAAG,MAAM,CAAC;QAC3B,YAAY,CAAC,OAAO,GAAG,SAAS,CAAC;QACjC,QAAQ,CACN,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YACnB,MAAM,IAAI,GAAgB;gBACxB,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,WAAW,EAAE,CAAC,CAAC,WAAW;aAC3B,CAAC;YACF,IAAI,CAAC,CAAC,WAAW,KAAK,SAAS;gBAAE,IAAI,CAAC,WAAW,GAAG,CAAC,CAAC,WAAW,CAAC;YAClE,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CACH,CAAC;QACF,kBAAkB,CAAC,WAAW,CAAC,CAAC;IAClC,CAAC,CACF,CAAC;IAEF,MAAM,cAAc,GAAG,KAAK,IAAI,EAAE;QAChC,MAAM,CAAC,GAAG,YAAY,CAAC,OAAO,CAAC;QAC/B,YAAY,CAAC,OAAO,GAAG,IAAI,CAAC;QAC5B,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC;QACzB,IAAI,CAAC,EAAE,CAAC;YACN,IAAI,CAAC;gBACH,MAAM,CAAC,CAAC,KAAK,EAAE,CAAC;YAClB,CAAC;YAAC,MAAM,CAAC;gBACP,sBAAsB;YACxB,CAAC;QACH,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,SAAS,GAAG,cAAc,CAAC,KAAK,IAAI,EAAE;QAC1C,iEAAiE;QACjE,MAAM,cAAc,EAAE,CAAC;QACvB,kBAAkB,CAAC,YAAY,CAAC,CAAC;QACjC,YAAY,CAAC,IAAI,CAAC,CAAC;QACnB,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC1B,+DAA+D;QAC/D,8DAA8D;QAC9D,QAAQ,CAAC,EAAE,CAAC,CAAC;QACb,IAAI,SAAS,GAAyC,IAAI,CAAC;QAC3D,IAAI,CAAC;YACH,SAAS,GAAG,MAAM,cAAc,EAAE,CAAC;YACnC,oEAAoE;YACpE,gEAAgE;YAChE,gDAAgD;YAChD,MAAM,eAAe,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,GAAG,YAAY,iBAAiB,EAAE,CAAC;gBACrC,2DAA2D;gBAC3D,4DAA4D;gBAC5D,oCAAoC;gBACpC,YAAY,CAAC,OAAO,GAAG,SAAS,CAAC;gBACjC,kBAAkB,CAAC,cAAc,CAAC,CAAC;YACrC,CAAC;iBAAM,CAAC;gBACN,IAAI,SAAS,EAAE,CAAC;oBACd,IAAI,CAAC;wBACH,MAAM,SAAS,CAAC,KAAK,EAAE,CAAC;oBAC1B,CAAC;oBAAC,MAAM,CAAC;wBACP,sBAAsB;oBACxB,CAAC;gBACH,CAAC;gBACD,YAAY,CAAC;oBACX,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;iBAC1D,CAAC,CAAC;gBACH,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC9B,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,cAAc,CAAC,KAAK,IAAI,EAAE;QAC7C,QAAQ,CAAC,EAAE,CAAC,CAAC;QACb,mBAAmB,CAAC,IAAI,CAAC,CAAC;QAC1B,kBAAkB,CAAC,cAAc,CAAC,CAAC;QACnC,MAAM,cAAc,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,cAAc,CAAC,KAAK,EAAE,WAAmB,EAAE,EAAE;QAClE,kBAAkB,CAAC,aAAa,CAAC,CAAC;QAClC,YAAY,CAAC,IAAI,CAAC,CAAC;QACnB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,CAAC;YACjC,MAAM,IAAI,GAAG,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YAC1C,IAAI,CAAC,IAAI;gBACP,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;YAChE,IAAI,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC;YACrC,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,SAAS,GAAG,MAAM,cAAc,EAAE,CAAC;gBACnC,YAAY,CAAC,OAAO,GAAG,SAAS,CAAC;YACnC,CAAC;YACD,MAAM,SAAS,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;YACjC,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1B,MAAM,eAAe,CAAC,SAAS,CAAC,CAAC;QACnC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,cAAc,EAAE,CAAC;YACvB,YAAY,CAAC;gBACX,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC;aAC1D,CAAC,CAAC;YACH,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC,CAAC,CAAC;IAEH,MAAM,cAAc,GAAG,cAAc,CACnC,KAAK,EAAE,MAA8B,EAAE,EAAE;QACvC,IAAI,CAAC,KAAK,CAAC,WAAW;YAAE,OAAO;QAC/B,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC/B,KAAK,SAAS,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,0DAA0D;QAC1D,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;YACrD,KAAK,SAAS,EAAE,CAAC;YACjB,OAAO;QACT,CAAC;QACD,MAAM,SAAS,GAAG,MAAM,KAAK,CAAC,OAAO,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9D,IAAI,MAAM,CAAC,SAAS;YAAE,OAAO;QAC7B,IAAI,KAAK,CAAC,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAChC,IAAI,CAAC,SAAS,EAAE,MAAM;gBAAE,OAAO;QACjC,CAAC;aAAM,IAAI,CAAC,SAAS,EAAE,KAAK,EAAE,CAAC;YAC7B,OAAO;QACT,CAAC;QACD,KAAK,SAAS,EAAE,CAAC;IACnB,CAAC,CACF,CAAC;IAEF,iDAAiD;IACjD,SAAS,CAAC,GAAG,EAAE;QACb,MAAM,MAAM,GAAG,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QACpC,KAAK,cAAc,CAAC,MAAM,CAAC,CAAC;QAC5B,OAAO,GAAG,EAAE;YACV,MAAM,CAAC,SAAS,GAAG,IAAI,CAAC;YACxB,MAAM,CAAC,GAAG,YAAY,CAAC,OAAO,CAAC;YAC/B,YAAY,CAAC,OAAO,GAAG,IAAI,CAAC;YAC5B,SAAS,CAAC,OAAO,GAAG,IAAI,CAAC;YACzB,IAAI,CAAC;gBAAE,CAAC,CAAC,KAAK,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,GAAE,CAAC,CAAC,CAAC;QACnC,CAAC,CAAC;IACJ,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,KAAK,GAAG,OAAO,CACnB,GAAG,EAAE,CAAC,CAAC;QACL,EAAE,EAAE,KAAK,CAAC,EAAE;QACZ,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,GAAG,EAAE,KAAK,CAAC,GAAG;QACd,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,eAAe;QACf,SAAS;QACT,KAAK;QACL,gBAAgB;KACjB,CAAC,EACF;QACE,KAAK,CAAC,EAAE;QACR,KAAK,CAAC,IAAI;QACV,KAAK,CAAC,IAAI;QACV,KAAK,CAAC,GAAG;QACT,KAAK,CAAC,IAAI;QACV,eAAe;QACf,SAAS;QACT,KAAK;QACL,gBAAgB;KACjB,CACF,CAAC;IAEF,OAAO;QACL,QAAQ,EAAE,GAAG,EAAE,CAAC,KAAK;QACrB,OAAO,EAAE,SAAS;QAClB,UAAU,EAAE,YAAY;QACxB,MAAM,EAAE,KAAK,IAAI,EAAE;YACjB,MAAM,YAAY,EAAE,CAAC;YACrB,MAAM,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAC7C,MAAM,KAAK,CAAC,QAAQ,EAAE,CAAC;QACzB,CAAC;QACD,QAAQ,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;YAC7B,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC;YACjC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,CAAC,EAAE,oBAAoB,CAAC,CAAC;YAC/D,CAAC;YACD,OAAO,MAAM,MAAM,CAAC,QAAQ,CAAC;gBAC3B,IAAI;gBACJ,SAAS,EAAE,IAA2C;aACvD,CAAC,CAAC;QACL,CAAC;QACD,YAAY,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YAC1B,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC;YACjC,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,eAAe,KAAK,CAAC,EAAE,oBAAoB,CAAC,CAAC;YAC/D,CAAC;YACD,OAAO,MAAM,MAAM,CAAC,YAAY,CAAC,EAAE,GAAG,EAAE,CAAC,CAAC;QAC5C,CAAC;QACD,YAAY,EAAE,cAAc;KAC7B,CAAC;AACJ,CAAC,CACF,CAAC"}

package/dist/resources/storage/McpCustomStorage.d.ts

@@ -0,0 +1,3 @@
+import type { MCPStorage } from "./types.js";
+export declare const McpCustomStorage: (props: MCPStorage) => import("@assistant-ui/tap").ResourceElement<MCPStorage, MCPStorage>;
+//# sourceMappingURL=McpCustomStorage.d.ts.map

package/dist/resources/storage/McpCustomStorage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpCustomStorage.d.ts","sourceRoot":"","sources":["../../../src/resources/storage/McpCustomStorage.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,UAAU,EAAE,mBAAgB;AAE1C,eAAO,MAAM,gBAAgB,4FAE5B,CAAC"}

package/dist/resources/storage/McpCustomStorage.js

@@ -0,0 +1,3 @@
+import { resource } from "@assistant-ui/tap";
+export const McpCustomStorage = resource((impl) => impl);
+//# sourceMappingURL=McpCustomStorage.js.map

package/dist/resources/storage/McpCustomStorage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpCustomStorage.js","sourceRoot":"","sources":["../../../src/resources/storage/McpCustomStorage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAG7C,MAAM,CAAC,MAAM,gBAAgB,GAAG,QAAQ,CACtC,CAAC,IAAgB,EAAc,EAAE,CAAC,IAAI,CACvC,CAAC"}

package/dist/resources/storage/McpLocalStorage.d.ts

@@ -0,0 +1,9 @@
+import type { MCPStorage } from "./types.js";
+export type McpLocalStorageOptions = {
+    /** Namespace prefix for keys. Default "aui-mcp". */
+    keyPrefix?: string;
+    /** Override the underlying Storage. Defaults to globalThis.localStorage. */
+    storage?: Storage;
+};
+export declare const McpLocalStorage: () => import("@assistant-ui/tap").ResourceElement<MCPStorage, undefined>;
+//# sourceMappingURL=McpLocalStorage.d.ts.map

package/dist/resources/storage/McpLocalStorage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpLocalStorage.d.ts","sourceRoot":"","sources":["../../../src/resources/storage/McpLocalStorage.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,mBAAgB;AAE1C,MAAM,MAAM,sBAAsB,GAAG;IACnC,oDAAoD;IACpD,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,4EAA4E;IAC5E,OAAO,CAAC,EAAE,OAAO,CAAC;CACnB,CAAC;AAcF,eAAO,MAAM,eAAe,0EAoD3B,CAAC"}

package/dist/resources/storage/McpLocalStorage.js

@@ -0,0 +1,67 @@
+import { resource } from "@assistant-ui/tap";
+function resolveStorage(opts) {
+    if (opts.storage)
+        return opts.storage;
+    if (typeof globalThis !== "undefined" && "localStorage" in globalThis) {
+        try {
+            return globalThis.localStorage;
+        }
+        catch {
+            return null;
+        }
+    }
+    return null;
+}
+export const McpLocalStorage = resource((opts = {}) => {
+    const prefix = opts.keyPrefix ?? "aui-mcp";
+    const customServersKey = `${prefix}:custom-servers`;
+    const authKey = (id) => `${prefix}:auth:${id}`;
+    const storage = resolveStorage(opts);
+    const read = (key, fallback) => {
+        if (!storage)
+            return fallback;
+        try {
+            const raw = storage.getItem(key);
+            if (raw == null)
+                return fallback;
+            return JSON.parse(raw);
+        }
+        catch {
+            return fallback;
+        }
+    };
+    const write = (key, value) => {
+        if (!storage)
+            return;
+        try {
+            storage.setItem(key, JSON.stringify(value));
+        }
+        catch {
+            // quota or serialization failure — silently drop
+        }
+    };
+    const remove = (key) => {
+        if (!storage)
+            return;
+        try {
+            storage.removeItem(key);
+        }
+        catch {
+            // ignore
+        }
+    };
+    return {
+        loadCustomServers: async () => read(customServersKey, []),
+        saveCustomServers: async (records) => {
+            write(customServersKey, records);
+        },
+        loadAuthState: async (id) => read(authKey(id), null),
+        saveAuthState: async (id, state) => {
+            write(authKey(id), state);
+        },
+        clearAuthState: async (id) => {
+            remove(authKey(id));
+        },
+    };
+});
+//# sourceMappingURL=McpLocalStorage.js.map

package/dist/resources/storage/McpLocalStorage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpLocalStorage.js","sourceRoot":"","sources":["../../../src/resources/storage/McpLocalStorage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAY7C,SAAS,cAAc,CAAC,IAA4B;IAClD,IAAI,IAAI,CAAC,OAAO;QAAE,OAAO,IAAI,CAAC,OAAO,CAAC;IACtC,IAAI,OAAO,UAAU,KAAK,WAAW,IAAI,cAAc,IAAI,UAAU,EAAE,CAAC;QACtE,IAAI,CAAC;YACH,OAAQ,UAAwC,CAAC,YAAY,CAAC;QAChE,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,CAAC,MAAM,eAAe,GAAG,QAAQ,CACrC,CAAC,OAA+B,EAAE,EAAc,EAAE;IAChD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC;IAC3C,MAAM,gBAAgB,GAAG,GAAG,MAAM,iBAAiB,CAAC;IACpD,MAAM,OAAO,GAAG,CAAC,EAAU,EAAE,EAAE,CAAC,GAAG,MAAM,SAAS,EAAE,EAAE,CAAC;IACvD,MAAM,OAAO,GAAG,cAAc,CAAC,IAAI,CAAC,CAAC;IAErC,MAAM,IAAI,GAAG,CAAI,GAAW,EAAE,QAAW,EAAK,EAAE;QAC9C,IAAI,CAAC,OAAO;YAAE,OAAO,QAAQ,CAAC;QAC9B,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YACjC,IAAI,GAAG,IAAI,IAAI;gBAAE,OAAO,QAAQ,CAAC;YACjC,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAM,CAAC;QAC9B,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,KAAK,GAAG,CAAC,GAAW,EAAE,KAAc,EAAQ,EAAE;QAClD,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC;YACH,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC,CAAC;QAC9C,CAAC;QAAC,MAAM,CAAC;YACP,iDAAiD;QACnD,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,MAAM,GAAG,CAAC,GAAW,EAAQ,EAAE;QACnC,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC;YACH,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC,CAAC;IAEF,OAAO;QACL,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAC5B,IAAI,CAA0B,gBAAgB,EAAE,EAAE,CAAC;QACrD,iBAAiB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACnC,KAAK,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;QACnC,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAC1B,IAAI,CAA+B,OAAO,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC;QACvD,aAAa,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE;YACjC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,EAAE,KAAK,CAAC,CAAC;QAC5B,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC3B,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;QACtB,CAAC;KACF,CAAC;AACJ,CAAC,CACF,CAAC"}

package/dist/resources/storage/McpMemoryStorage.d.ts

@@ -0,0 +1,3 @@
+import type { MCPStorage } from "./types.js";
+export declare const McpMemoryStorage: () => import("@assistant-ui/tap").ResourceElement<MCPStorage, undefined>;
+//# sourceMappingURL=McpMemoryStorage.d.ts.map

package/dist/resources/storage/McpMemoryStorage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpMemoryStorage.d.ts","sourceRoot":"","sources":["../../../src/resources/storage/McpMemoryStorage.ts"],"names":[],"mappings":"AAGA,OAAO,KAAK,EAAE,UAAU,EAAE,mBAAgB;AAE1C,eAAO,MAAM,gBAAgB,0EAgB3B,CAAC"}

package/dist/resources/storage/McpMemoryStorage.js

@@ -0,0 +1,19 @@
+import { resource } from "@assistant-ui/tap";
+export const McpMemoryStorage = resource(() => {
+    let servers = [];
+    const auth = new Map();
+    return {
+        loadCustomServers: async () => [...servers],
+        saveCustomServers: async (records) => {
+            servers = [...records];
+        },
+        loadAuthState: async (id) => auth.get(id) ?? null,
+        saveAuthState: async (id, state) => {
+            auth.set(id, state);
+        },
+        clearAuthState: async (id) => {
+            auth.delete(id);
+        },
+    };
+});
+//# sourceMappingURL=McpMemoryStorage.js.map

package/dist/resources/storage/McpMemoryStorage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"McpMemoryStorage.js","sourceRoot":"","sources":["../../../src/resources/storage/McpMemoryStorage.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,mBAAmB,CAAC;AAK7C,MAAM,CAAC,MAAM,gBAAgB,GAAG,QAAQ,CAAC,GAAe,EAAE;IACxD,IAAI,OAAO,GAA4B,EAAE,CAAC;IAC1C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAiC,CAAC;IACtD,OAAO;QACL,iBAAiB,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,OAAO,CAAC;QAC3C,iBAAiB,EAAE,KAAK,EAAE,OAAO,EAAE,EAAE;YACnC,OAAO,GAAG,CAAC,GAAG,OAAO,CAAC,CAAC;QACzB,CAAC;QACD,aAAa,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,IAAI;QACjD,aAAa,EAAE,KAAK,EAAE,EAAE,EAAE,KAAK,EAAE,EAAE;YACjC,IAAI,CAAC,GAAG,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;QACtB,CAAC;QACD,cAAc,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC3B,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;QAClB,CAAC;KACF,CAAC;AACJ,CAAC,CAAC,CAAC"}

package/dist/resources/storage/types.d.ts

@@ -0,0 +1,12 @@
+import type { ResourceElement } from "@assistant-ui/tap";
+import type { MCPCustomServerRecord } from "../../mcp-scope.js";
+import type { MCPPersistedAuthState } from "../../auth/types.js";
+export type MCPStorage = {
+    loadCustomServers: () => Promise<MCPCustomServerRecord[]>;
+    saveCustomServers: (records: MCPCustomServerRecord[]) => Promise<void>;
+    loadAuthState: (serverId: string) => Promise<MCPPersistedAuthState | null>;
+    saveAuthState: (serverId: string, state: MCPPersistedAuthState) => Promise<void>;
+    clearAuthState: (serverId: string) => Promise<void>;
+};
+export type MCPStorageElement = ResourceElement<MCPStorage>;
+//# sourceMappingURL=types.d.ts.map

package/dist/resources/storage/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/resources/storage/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,KAAK,EAAE,qBAAqB,EAAE,2BAAwB;AAC7D,OAAO,KAAK,EAAE,qBAAqB,EAAE,4BAAyB;AAE9D,MAAM,MAAM,UAAU,GAAG;IACvB,iBAAiB,EAAE,MAAM,OAAO,CAAC,qBAAqB,EAAE,CAAC,CAAC;IAC1D,iBAAiB,EAAE,CAAC,OAAO,EAAE,qBAAqB,EAAE,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACvE,aAAa,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,qBAAqB,GAAG,IAAI,CAAC,CAAC;IAC3E,aAAa,EAAE,CACb,QAAQ,EAAE,MAAM,EAChB,KAAK,EAAE,qBAAqB,KACzB,OAAO,CAAC,IAAI,CAAC,CAAC;IACnB,cAAc,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACrD,CAAC;AAEF,MAAM,MAAM,iBAAiB,GAAG,eAAe,CAAC,UAAU,CAAC,CAAC"}

package/dist/resources/storage/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/resources/storage/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/resources/storage/types.ts"],"names":[],"mappings":""}

package/dist/utils/serverId.d.ts

@@ -0,0 +1,2 @@
+export declare function assertValidServerId(id: string): void;
+//# sourceMappingURL=serverId.d.ts.map