From 4efc40fbb0214fba65f6ff95782e6ead3b20bc71 Mon Sep 17 00:00:00 2001
From: Faton Ramadani <faton.ramadani14@gmail.com>
Date: Mon, 26 Aug 2024 16:36:22 +0200
Subject: [PATCH] fix(frontend): fix inserting Ws scripts from the search menu
 (#4290)

* fix(frontend): fix inserting Ws scripts from the AI gen

* fix(frontend): fix inserting Ws scripts from the search menu
---
 .../src/lib/components/copilot/StepGen.svelte    |  3 ++-
 .../flows/map/InsertTriggerButton.svelte         |  2 +-
 .../graph/renderers/edges/BaseEdge.svelte        | 16 ++++++++++++++++
 3 files changed, 19 insertions(+), 2 deletions(-)

diff --git a/frontend/src/lib/components/copilot/StepGen.svelte b/frontend/src/lib/components/copilot/StepGen.svelte
index ff009b9e69..8b2aa3605b 100644
--- a/frontend/src/lib/components/copilot/StepGen.svelte
+++ b/frontend/src/lib/components/copilot/StepGen.svelte
@@ -36,7 +36,8 @@
 	async function loadScripts(): Promise<void> {
 		const loadedScripts = await ScriptService.listScripts({
 			workspace: $workspaceStore!,
-			perPage: 300
+			perPage: 300,
+			kinds: trigger ? 'trigger' : 'script'
 		})
 
 		scripts = loadedScripts
diff --git a/frontend/src/lib/components/flows/map/InsertTriggerButton.svelte b/frontend/src/lib/components/flows/map/InsertTriggerButton.svelte
index 88b3593f2f..03d819b2c5 100644
--- a/frontend/src/lib/components/flows/map/InsertTriggerButton.svelte
+++ b/frontend/src/lib/components/flows/map/InsertTriggerButton.svelte
@@ -32,7 +32,7 @@
 		<Zap size={14} />
 	</button>
 	{#if !disableAi}
-		<StepGen {index} bind:funcDesc bind:open {close} {modules} trigger />
+		<StepGen {index} bind:funcDesc bind:open {close} {modules} trigger on:insert />
 	{/if}
 	{#if funcDesc.length === 0}
 		<div class="font-mono divide-y text-xs w-full text-secondary whitespace-nowrap">
diff --git a/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte b/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
index 39bac15583..b29e481fb9 100644
--- a/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
+++ b/frontend/src/lib/components/graph/renderers/edges/BaseEdge.svelte
@@ -65,6 +65,14 @@
 				on:new={(e) => {
 					data?.eventHandlers.insert({ modules: data.modules, index: data.index, detail: e.detail })
 				}}
+				on:insert={(e) => {
+					data?.eventHandlers.insert({
+						modules: data.modules,
+						index: data.index,
+						script: e.detail,
+						detail: 'script'
+					})
+				}}
 				bind:open={menuOpen}
 			/>
 		</div>
@@ -82,6 +90,14 @@
 							detail: e.detail
 						})
 					}}
+					on:insert={(e) => {
+						data?.eventHandlers.insert({
+							modules: data.modules,
+							index: data.index,
+							script: e.detail,
+							detail: 'script'
+						})
+					}}
 					index={data?.index ?? 0}
 					modules={data?.modules ?? []}
 				/>