<script lang="ts">
import { FolderOpen, Plus, Loader2, Braces } from '@lucide/svelte';
import { toast } from 'svelte-sonner';
import * as Dialog from '$lib/components/ui/dialog';
import { Button } from '$lib/components/ui/button';
import { mcpStore } from '$lib/stores/mcp.svelte';
import { conversationsStore } from '$lib/stores/conversations.svelte';
import {
mcpResources,
mcpTotalResourceCount,
mcpResourceStore
} from '$lib/stores/mcp-resources.svelte';
import {
McpResourceBrowser,
McpResourcePreview,
McpResourceTemplateForm
} from '$lib/components/app';
import { getResourceDisplayName } from '$lib/utils';
import type { MCPResourceInfo, MCPResourceContent, MCPResourceTemplateInfo } from '$lib/types';
import { SvelteSet } from 'svelte/reactivity';
interface Props {
open?: boolean;
onOpenChange?: (open: boolean) => void;
onAttach?: (resource: MCPResourceInfo) => void;
preSelectedUri?: string;
}
let { open = $bindable(false), onOpenChange, onAttach, preSelectedUri }: Props = $props();
let selectedResources = new SvelteSet<string>();
let lastSelectedUri = $state<string | null>(null);
let isAttaching = $state(false);
let selectedTemplate = $state<MCPResourceTemplateInfo | null>(null);
let templatePreviewUri = $state<string | null>(null);
let templatePreviewContent = $state<MCPResourceContent[] | null>(null);
let templatePreviewLoading = $state(false);
let templatePreviewError = $state<string | null>(null);
const totalCount = $derived(mcpTotalResourceCount());
$effect(() => {
if (open) {
loadResources();
if (preSelectedUri) {
selectedResources.clear();
selectedResources.add(preSelectedUri);
lastSelectedUri = preSelectedUri;
}
}
});
async function loadResources() {
const perChatOverrides = conversationsStore.getAllMcpServerOverrides();
const initialized = await mcpStore.ensureInitialized(perChatOverrides);
if (initialized) {
await mcpStore.fetchAllResources();
}
}
function handleOpenChange(newOpen: boolean) {
open = newOpen;
onOpenChange?.(newOpen);
if (!newOpen) {
selectedResources.clear();
lastSelectedUri = null;
clearTemplateState();
}
}
function clearTemplateState() {
selectedTemplate = null;
templatePreviewUri = null;
templatePreviewContent = null;
templatePreviewLoading = false;
templatePreviewError = null;
}
function handleTemplateSelect(template: MCPResourceTemplateInfo) {
selectedResources.clear();
lastSelectedUri = null;
if (
selectedTemplate?.uriTemplate === template.uriTemplate &&
selectedTemplate?.serverName === template.serverName
) {
clearTemplateState();
return;
}
selectedTemplate = template;
templatePreviewUri = null;
templatePreviewContent = null;
templatePreviewLoading = false;
templatePreviewError = null;
}
async function handleTemplateResolve(uri: string, serverName: string) {
templatePreviewUri = uri;
templatePreviewContent = null;
templatePreviewLoading = true;
templatePreviewError = null;
try {
const content = await mcpStore.readResourceByUri(serverName, uri);
if (content) {
templatePreviewContent = content;
} else {
templatePreviewError = 'Failed to read resource';
}
} catch (error) {
templatePreviewError = error instanceof Error ? error.message : 'Unknown error';
} finally {
templatePreviewLoading = false;
}
}
function handleTemplateCancelForm() {
clearTemplateState();
}
async function handleAttachTemplateResource() {
if (!templatePreviewUri || !selectedTemplate || !templatePreviewContent) return;
isAttaching = true;
try {
const knownResource = mcpResourceStore.findResourceByUri(templatePreviewUri);
if (knownResource) {
if (!mcpResourceStore.isAttached(knownResource.uri)) {
await mcpStore.attachResource(knownResource.uri);
}
toast.success(`Resource attached: ${knownResource.title || knownResource.name}`);
} else {
if (mcpResourceStore.isAttached(templatePreviewUri)) {
toast.info('Resource already attached');
handleOpenChange(false);
return;
}
const resourceInfo: MCPResourceInfo = {
uri: templatePreviewUri,
name: templatePreviewUri.split('/').pop() || templatePreviewUri,
serverName: selectedTemplate.serverName
};
const attachment = mcpResourceStore.addAttachment(resourceInfo);
mcpResourceStore.updateAttachmentContent(attachment.id, templatePreviewContent);
toast.success(`Resource attached: ${resourceInfo.name}`);
}
handleOpenChange(false);
} catch (error) {
console.error('Failed to attach template resource:', error);
} finally {
isAttaching = false;
}
}
function handleResourceSelect(resource: MCPResourceInfo, shiftKey: boolean = false) {
clearTemplateState();
if (shiftKey && lastSelectedUri) {
const allResources = getAllResourcesFlatInTreeOrder();
const lastIndex = allResources.findIndex((r) => r.uri === lastSelectedUri);
const currentIndex = allResources.findIndex((r) => r.uri === resource.uri);
if (lastIndex !== -1 && currentIndex !== -1) {
const start = Math.min(lastIndex, currentIndex);
const end = Math.max(lastIndex, currentIndex);
for (let i = start; i <= end; i++) {
selectedResources.add(allResources[i].uri);
}
}
} else {
selectedResources.clear();
selectedResources.add(resource.uri);
lastSelectedUri = resource.uri;
}
}
function handleResourceToggle(resource: MCPResourceInfo, checked: boolean) {
clearTemplateState();
if (checked) {
selectedResources.add(resource.uri);
} else {
selectedResources.delete(resource.uri);
}
lastSelectedUri = resource.uri;
}
function getAllResourcesFlatInTreeOrder(): MCPResourceInfo[] {
const allResources: MCPResourceInfo[] = [];
const resourcesMap = mcpResources();
for (const [serverName, serverRes] of resourcesMap.entries()) {
for (const resource of serverRes.resources) {
allResources.push({ ...resource, serverName });
}
}
return allResources.sort((a, b) => {
const aName = getResourceDisplayName(a);
const bName = getResourceDisplayName(b);
return aName.localeCompare(bName);
});
}
async function handleAttach() {
if (selectedResources.size === 0) return;
isAttaching = true;
try {
const allResources = getAllResourcesFlatInTreeOrder();
const resourcesToAttach = allResources.filter((r) => selectedResources.has(r.uri));
for (const resource of resourcesToAttach) {
await mcpStore.attachResource(resource.uri);
onAttach?.(resource);
}
const count = resourcesToAttach.length;
toast.success(
count === 1
? `Resource attached: ${resourcesToAttach[0].name}`
: `${count} resources attached`
);
handleOpenChange(false);
} catch (error) {
console.error('Failed to attach resources:', error);
} finally {
isAttaching = false;
}
}
const selectedTemplateUri = $derived(selectedTemplate?.uriTemplate ?? null);
const hasTemplateResult = $derived(
!!selectedTemplate && !!templatePreviewContent && !!templatePreviewUri
);
</script>
<Dialog.Root {open} onOpenChange={handleOpenChange}>
<Dialog.Content class="max-h-[80vh] !max-w-4xl overflow-hidden p-0">
<Dialog.Header class="border-b border-border/30 px-6 py-4">
<Dialog.Title class="flex items-center gap-2">
<FolderOpen class="h-5 w-5" />
<span>MCP Resources</span>
{#if totalCount > 0}
<span class="text-sm font-normal text-muted-foreground">({totalCount})</span>
{/if}
</Dialog.Title>
<Dialog.Description>
Browse and attach resources from connected MCP servers to your chat context.
</Dialog.Description>
</Dialog.Header>
<div class="flex h-[500px] min-w-0">
<div class="w-72 shrink-0 overflow-y-auto border-r border-border/30 p-4">
<McpResourceBrowser
onSelect={handleResourceSelect}
onToggle={handleResourceToggle}
onTemplateSelect={handleTemplateSelect}
selectedUris={selectedResources}
{selectedTemplateUri}
expandToUri={preSelectedUri}
/>
</div>
<div class="min-w-0 flex-1 overflow-auto p-4">
{#if selectedTemplate && !templatePreviewContent}
<div class="flex h-full flex-col">
<div class="mb-3 flex items-center gap-2">
<Braces class="h-4 w-4 text-muted-foreground" />
<span class="text-sm font-medium">
{selectedTemplate.title || selectedTemplate.name}
</span>
</div>
{#if selectedTemplate.description}
<p class="mb-4 text-xs text-muted-foreground">
{selectedTemplate.description}
</p>
{/if}
<div class="mb-4 rounded-md border border-border/50 bg-muted/30 px-3 py-2">
<p class="font-mono text-xs break-all text-muted-foreground">
{selectedTemplate.uriTemplate}
</p>
</div>
{#if templatePreviewLoading}
<div class="flex flex-1 items-center justify-center">
<Loader2 class="h-6 w-6 animate-spin text-muted-foreground" />
</div>
{:else if templatePreviewError}
<div class="flex flex-1 flex-col items-center justify-center gap-2 text-red-500">
<span class="text-sm">{templatePreviewError}</span>
<Button
size="sm"
variant="outline"
onclick={() => {
templatePreviewError = null;
}}
>
Try again
</Button>
</div>
{:else}
<McpResourceTemplateForm
template={selectedTemplate}
onResolve={handleTemplateResolve}
onCancel={handleTemplateCancelForm}
/>
{/if}
</div>
{:else if hasTemplateResult}
<!-- Template resolved: show preview -->
<McpResourcePreview
resource={{
uri: templatePreviewUri ?? '',
name: templatePreviewUri?.split('/').pop() || (templatePreviewUri ?? ''),
serverName: selectedTemplate?.serverName || ''
}}
preloadedContent={templatePreviewContent}
/>
{:else if selectedResources.size === 1}
{@const allResources = getAllResourcesFlatInTreeOrder()}
{@const selectedResource = allResources.find((r) => selectedResources.has(r.uri))}
<McpResourcePreview resource={selectedResource ?? null} />
{:else if selectedResources.size > 1}
<div class="flex flex-col gap-10">
{#each getAllResourcesFlatInTreeOrder() as resource (resource.uri)}
{#if selectedResources.has(resource.uri)}
<McpResourcePreview {resource} />
{/if}
{/each}
</div>
{:else}
<div class="flex h-full items-center justify-center text-sm text-muted-foreground">
Select a resource to preview
</div>
{/if}
</div>
</div>
<Dialog.Footer class="border-t border-border/30 px-6 py-4">
<Button variant="outline" onclick={() => handleOpenChange(false)}>Cancel</Button>
{#if hasTemplateResult}
<Button onclick={handleAttachTemplateResource} disabled={isAttaching}>
{#if isAttaching}
<Loader2 class="mr-2 h-4 w-4 animate-spin" />
{:else}
<Plus class="mr-2 h-4 w-4" />
{/if}
Attach Resource
</Button>
{:else}
<Button onclick={handleAttach} disabled={selectedResources.size === 0 || isAttaching}>
{#if isAttaching}
<Loader2 class="mr-2 h-4 w-4 animate-spin" />
{:else}
<Plus class="mr-2 h-4 w-4" />
{/if}
Attach {selectedResources.size > 0 ? `(${selectedResources.size})` : 'Resource'}
</Button>
{/if}
</Dialog.Footer>
</Dialog.Content>
</Dialog.Root>