windmill/frontend/src/lib/components/ExploreAssetButton.svelte

<script module lang="ts">
	export function assetCanBeExplored(
		asset: Asset,
		_resourceMetadata?: { resource_type?: string }
	): boolean {
		return (
			asset.kind === 'ducklake' ||
			asset.kind === 'datatable' ||
			asset.kind === 's3object' ||
			asset.kind === 'volume' ||
			(asset.kind === 'resource' && isDbType(_resourceMetadata?.resource_type))
		)
	}
</script>

<script lang="ts">
	import { isDbType } from '$lib/components/dbTypes'
	import { formatAsset, type Asset } from '$lib/components/assets/lib'
	import { Button, ButtonType } from '$lib/components/common'
	import S3FilePicker from '$lib/components/S3FilePicker.svelte'
	import { VolumeService } from '$lib/gen'
	import { globalDbManagerDrawer, userStore, workspaceStore } from '$lib/stores'
	import { isS3Uri } from '$lib/utils'
	import { Database, File, HardDriveIcon } from 'lucide-svelte'
	import DucklakeIcon from './icons/DucklakeIcon.svelte'

	const {
		asset,
		_resourceMetadata,
		s3FilePicker,
		onClick,
		class: className = '',
		noText = false,
		buttonVariant = 'default',
		btnClasses = '',
		disabled = false,
		workspace = undefined
	}: {
		asset: Asset
		_resourceMetadata?: { resource_type?: string }
		s3FilePicker?: S3FilePicker
		onClick?: () => void
		class?: string
		noText?: boolean
		buttonVariant?: ButtonType.Variant
		btnClasses?: string
		disabled?: boolean
		/** Workspace the explored asset lives in; defaults to the nav workspace. */
		workspace?: string
	} = $props()

	let dbManagerDrawer = $derived(globalDbManagerDrawer.val)
	let ws = $derived(workspace ?? $workspaceStore)
	const assetUri = $derived(formatAsset(asset))
</script>

<Button
	disabled={$userStore?.operator || disabled}
	unifiedSize={'md'}
	variant={buttonVariant}
	wrapperClasses={className}
	iconOnly={noText}
	{btnClasses}
	on:click={async () => {
		if (asset.kind === 'resource' && isDbType(_resourceMetadata?.resource_type)) {
			let [resourcePath, specificTable] = asset.path.split('?table=')
			dbManagerDrawer?.openDrawer(
				{
					type: 'database',
					resourceType: _resourceMetadata.resource_type,
					resourcePath,
					specificTable
				},
				ws
			)
		} else if (asset.kind === 's3object' && isS3Uri(assetUri)) {
			s3FilePicker?.open(assetUri)
		} else if (asset.kind === 'volume') {
			const storage = (await VolumeService.getVolumeStorage({ workspace: ws! })) ?? undefined
			s3FilePicker?.open({ s3: `volumes/${ws}/${asset.path}/`, storage })
		} else if (asset.kind === 'ducklake') {
			let ducklake = asset.path.split('/')[0]
			let specificTableSplit = asset.path.split('/')[1]?.split('.') as string[] | undefined
			let [specificSchema, specificTable] =
				specificTableSplit?.length === 2
					? [specificTableSplit[0], specificTableSplit[1]]
					: [undefined, specificTableSplit?.[0]]
			dbManagerDrawer?.openDrawer({ type: 'ducklake', ducklake, specificSchema, specificTable }, ws)
		} else if (asset.kind === 'datatable') {
			let datatable = asset.path.split('/')[0]
			let specificTableSplit = asset.path.split('/')[1]?.split('.') as string[] | undefined
			let [specificSchema, specificTable] =
				specificTableSplit?.length === 2
					? [specificTableSplit[0], specificTableSplit[1]]
					: [undefined, specificTableSplit?.[0]]
			dbManagerDrawer?.openDrawer(
				{
					type: 'database',
					resourceType: 'postgresql',
					resourcePath: `datatable://${datatable}`,
					specificTable,
					specificSchema
				},
				ws
			)
		}
		onClick?.()
	}}
	endIcon={asset.kind === 's3object'
		? { icon: File }
		: asset.kind === 'resource' || asset.kind === 'datatable'
			? { icon: Database }
			: asset.kind === 'ducklake'
				? { icon: DucklakeIcon }
				: asset.kind === 'volume'
					? { icon: HardDriveIcon }
					: undefined}
>
	{#if asset.kind === 's3object' || asset.kind === 'volume'}
		<span class:hidden={noText}>Explore</span>
	{:else if asset.kind === 'resource' || asset.kind === 'ducklake' || asset.kind === 'datatable'}
		<span class:hidden={noText}>Manage</span>
	{/if}
</Button>