From 4d7bc5c651d1cb93e4e7bb32e61841f5c41ed6d0 Mon Sep 17 00:00:00 2001
From: Faton Ramadani <faton.ramadani14@gmail.com>
Date: Tue, 13 Aug 2024 14:45:59 +0200
Subject: [PATCH] feat(frontend): add support for triggers

---
 .../src/lib/components/graph/graphBuilder.ts  |  3 +-
 .../graph/renderers/edges/BaseEdge.svelte     | 35 ++++++++++++-------
 2 files changed, 25 insertions(+), 13 deletions(-)

diff --git a/frontend/src/lib/components/graph/graphBuilder.ts b/frontend/src/lib/components/graph/graphBuilder.ts
index 2a2704aa5c..5644f99314 100644
--- a/frontend/src/lib/components/graph/graphBuilder.ts
+++ b/frontend/src/lib/components/graph/graphBuilder.ts
@@ -88,7 +88,8 @@ export default function graphBuilder(
 				eventHandlers,
 				enableTrigger: sourceId === 'Input',
 				// If the index is -1, it means that the target module is not in the modules array, so we set it to the length of the array
-				index: index >= 0 ? index : mods?.length ?? 0
+				index: index >= 0 ? index : mods?.length ?? 0,
+				...extra
 			}
 		})
 	}
diff --git a/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte b/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
index c3dd2d3165..ea4e8bf4f7 100644
--- a/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
+++ b/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
@@ -7,6 +7,7 @@
 	import type { Writable } from 'svelte/store'
 	import type { GraphEventHandlers } from '../../graphBuilder'
 	import { getStraightLinePath } from '../utils'
+	import InsertTriggerButton from '$lib/components/flows/map/InsertTriggerButton.svelte'
 
 	export let sourceX: number
 	export let sourceY: number
@@ -23,6 +24,7 @@
 		eventHandlers: GraphEventHandlers
 		index: number
 		enableTrigger: boolean
+		disableAi: boolean
 	}
 
 	$: [edgePath, labelX, labelY] = getBezierPath({
@@ -52,25 +54,34 @@
 			style:transform="translate(-50%, -50%) translate({labelX}px,{labelY}px)"
 		>
 			<InsertModuleButton
+				disableAi={data.disableAi}
 				index={data.index ?? 0}
+				trigger={data.enableTrigger}
 				modules={data?.modules ?? []}
 				on:new={(e) => {
 					data?.eventHandlers.insert({ modules: data.modules, index: data.index, detail: e.detail })
 				}}
 			/>
 		</div>
-		<div
-			class="edgeButtonContainer nodrag nopan"
-			style:transform="translate(100%, -50%) translate({labelX}px,{labelY}px)"
-		>
-			<InsertModuleButton
-				index={data.index ?? 0}
-				modules={data?.modules ?? []}
-				on:new={(e) => {
-					data?.eventHandlers.insert({ modules: data.modules, index: data.index, detail: e.detail })
-				}}
-			/>
-		</div>
+		{#if data.enableTrigger}
+			<div
+				class="edgeButtonContainer nodrag nopan"
+				style:transform="translate(100%, -50%) translate({labelX}px,{labelY}px)"
+			>
+				<InsertTriggerButton
+					disableAi={data.disableAi}
+					on:new={(e) => {
+						data?.eventHandlers.insert({
+							modules: data.modules,
+							index: data.index,
+							detail: e.detail
+						})
+					}}
+					index={data?.index ?? 0}
+					modules={data?.modules ?? []}
+				/>
+			</div>
+		{/if}
 	{/if}
 
 	{#if data?.moving}