<script lang="ts">
	import {
		CancelablePromise,
		KafkaTriggerService,
		MqttTriggerService,
		NatsTriggerService,
		SqsTriggerService,
		PostgresTriggerService,
		WebsocketTriggerService,
		GcpTriggerService
	} from '$lib/gen'
	import { workspaceStore } from '$lib/stores'
	import { sendUserToast } from '$lib/toast'
	import Button from '../common/button/Button.svelte'

	export let kind: 'websocket' | 'nats' | 'kafka' | 'postgres' | 'sqs' | 'mqtt' | 'gcp'
	export let args: Record<string, any>
	export let noButton = false
	export let testLoading: boolean = false

	const kindToName: { [key: string]: string } = {
		websocket: 'WebSocket',
		nats: 'NATS server(s)',
		kafka: 'Kafka broker(s)',
		sqs: 'SQS',
		postgres: 'Postgres',
		mqtt: 'MQTT broker',
		gcp: 'Google Cloud Pub/Sub'
	}

	let promise: CancelablePromise<any> | null = null
	export async function testTriggerConnection() {
		if (testLoading) {
			promise?.cancel()
			return
		}

		testLoading = true
		try {
			if (kind === 'websocket') {
				promise = WebsocketTriggerService.testWebsocketConnection({
					workspace: $workspaceStore!,
					requestBody: args as any
				})
			} else if (kind === 'nats') {
				promise = NatsTriggerService.testNatsConnection({
					workspace: $workspaceStore!,
					requestBody: args as any
				})
			} else if (kind === 'kafka') {
				promise = KafkaTriggerService.testKafkaConnection({
					workspace: $workspaceStore!,
					requestBody: args as any
				})
			} else if (kind === 'mqtt') {
				promise = MqttTriggerService.testMqttConnection({
					workspace: $workspaceStore!,
					requestBody: args as any
				})
			} else if (kind === 'sqs') {
				promise = SqsTriggerService.testSqsConnection({
					workspace: $workspaceStore!,
					requestBody: args as any
				})
			} else if (kind === 'postgres') {
				promise = PostgresTriggerService.testPostgresConnection({
					workspace: $workspaceStore!,
					requestBody: args as any
				})
			} else if (kind === 'gcp') {
				promise = GcpTriggerService.testGcpConnection({
					workspace: $workspaceStore!,
					requestBody: args as any
				})
			}
			await promise
			sendUserToast(`Successfully connected to ${kindToName[kind]}`)
		} catch (err) {
			if (!promise?.isCancelled) {
				sendUserToast(`Error testing ${kindToName[kind]}: ${err?.body ?? 'Unknown error'}`, true)
			}
		} finally {
			testLoading = false
		}
	}
</script>

{#if !noButton}
	<div class="flex flex-row justify-end mt-1">
		<Button
			spacingSize="sm"
			size="xs"
			variant="default"
			on:click={testTriggerConnection}
			loading={testLoading}
			clickableWhileLoading
		>
			Test connection
		</Button>
	</div>
{/if}
