<script lang="ts">
	import { ICON_CLASS_DEFAULT } from '$lib/constants/css-classes';
	import { ChevronDown, ChevronRight } from '@lucide/svelte';
	import { Checkbox } from '$lib/components/ui/checkbox';
	import * as Collapsible from '$lib/components/ui/collapsible';
	import { TruncatedText, McpServerIdentity } from '$lib/components/app';
	import { toolsStore } from '$lib/stores/tools.svelte';
	import { permissionsStore } from '$lib/stores/permissions.svelte';
	import { mcpStore } from '$lib/stores/mcp.svelte';
	import { getBuiltinToolUi } from '$lib/constants/built-in-tools';
	import { ToolSource } from '$lib/enums/tools.enums';
	import { SvelteSet } from 'svelte/reactivity';

	let expandedGroups = new SvelteSet<string>();
	let groups = $derived(toolsStore.toolGroups);

	function toggleExpanded(key: string) {
		if (expandedGroups.has(key)) {
			expandedGroups.delete(key);
		} else {
			expandedGroups.add(key);
		}
	}
</script>

{#if groups.length === 0}
	<div class="py-8 text-center text-sm text-muted-foreground">No tools available</div>
{:else}
	<div class="space-y-2">
		{#each groups as group (group.key)}
			{@const isExpanded = expandedGroups.has(group.key)}
			<Collapsible.Root open={isExpanded} onOpenChange={() => toggleExpanded(group.key)}>
				<Collapsible.Trigger
					class="flex w-full items-center gap-2 rounded-lg px-3 py-2 text-sm hover:bg-muted/50"
				>
					{#if isExpanded}
						<ChevronDown class="h-3.5 w-3.5 shrink-0" />
					{:else}
						<ChevronRight class="h-3.5 w-3.5 shrink-0" />
					{/if}

					{@const faviconUrl = group.serverId ? mcpStore.getServerFavicon(group.serverId) : null}

					<span class="inline-flex min-w-0 items-center gap-1.5 font-medium">
						{#if group.source === 'mcp'}
							<McpServerIdentity
								iconClass={ICON_CLASS_DEFAULT}
								iconRounded="rounded-sm"
								showVersion={false}
								displayName={group.label}
								{faviconUrl}
							/>
						{:else}
							<TruncatedText text={group.label} class="font-medium" />
						{/if}
					</span>

					<span class="ml-auto shrink-0 text-xs text-muted-foreground">
						{group.tools.length} tool{group.tools.length !== 1 ? 's' : ''}
					</span>
				</Collapsible.Trigger>

				<Collapsible.Content>
					<div class="ml-4 border-l border-border/50 pl-2">
						<!-- Header row -->
						<div class="flex items-center gap-2 px-2 py-1 text-xs text-muted-foreground">
							<span class="min-w-0 flex-1">Tool</span>
							<span class="w-16 shrink-0 text-center">Enabled</span>
							<span class="w-20 shrink-0 text-center">Always allow</span>
						</div>

						{#each group.tools as entry (entry.key)}
							{@const toolName = entry.definition.function.name}
							{@const builtinUi =
								entry.source === ToolSource.BUILTIN || entry.source === ToolSource.FRONTEND
									? getBuiltinToolUi(toolName)
									: null}
							{@const displayLabel = builtinUi?.label ?? toolName}
							{@const IconComponent = builtinUi?.icon ?? null}
							{@const isEnabled = toolsStore.isToolEnabled(entry.key)}
							{@const permissionKey = entry.key}
							{@const isAlwaysAllowed = permissionsStore.hasTool(permissionKey)}

							<div class="flex items-center gap-2 rounded px-2 py-1.5 text-sm hover:bg-muted/50">
								<span class="flex min-w-0 flex-1 items-center gap-1.5">
									{#if IconComponent}
										<IconComponent class={ICON_CLASS_DEFAULT} />
									{/if}
									<TruncatedText text={displayLabel} class="min-w-0" showTooltip={true} />
								</span>

								<div class="flex w-16 shrink-0 justify-center">
									<Checkbox
										checked={isEnabled}
										onCheckedChange={() => toolsStore.toggleTool(entry.key)}
										class={ICON_CLASS_DEFAULT}
									/>
								</div>

								<div class="flex w-20 shrink-0 justify-center">
									<Checkbox
										checked={isAlwaysAllowed}
										onCheckedChange={() => {
											if (isAlwaysAllowed) {
												permissionsStore.revokeTool(permissionKey);
											} else {
												permissionsStore.allowTool(permissionKey);
											}
										}}
										class={ICON_CLASS_DEFAULT}
									/>
								</div>
							</div>
						{/each}
					</div>
				</Collapsible.Content>
			</Collapsible.Root>
		{/each}
	</div>
{/if}
