From df586fc68e6d39ce84dce6c65354ab7cbcf2cdb0 Mon Sep 17 00:00:00 2001
From: HugoCasa <hugo@casademont.ch>
Date: Tue, 30 Apr 2024 22:03:48 +0200
Subject: [PATCH] feat: import export worker group config (#3649)

---
 .../src/lib/components/WorkspaceGroup.svelte  | 18 +++++
 .../(root)/(logged)/workers/+page.svelte      | 67 ++++++++++++++++++-
 2 files changed, 84 insertions(+), 1 deletion(-)

diff --git a/frontend/src/lib/components/WorkspaceGroup.svelte b/frontend/src/lib/components/WorkspaceGroup.svelte
index 0bd23d6071..bf2a49ad5c 100644
--- a/frontend/src/lib/components/WorkspaceGroup.svelte
+++ b/frontend/src/lib/components/WorkspaceGroup.svelte
@@ -16,6 +16,7 @@
 	import Section from './Section.svelte'
 	import Label from './Label.svelte'
 	import AutoComplete from 'simple-svelte-autocomplete'
+	import YAML from 'yaml'
 
 	export let name: string
 	export let config:
@@ -817,6 +818,23 @@
 			</div>
 		</Button>
 
+		<Button
+			color="light"
+			size="xs"
+			on:click={() => {
+				loadNConfig()
+				navigator.clipboard.writeText(
+					YAML.stringify({
+						name,
+						...nconfig
+					})
+				)
+				sendUserToast('Worker config copied to clipboard as YAML')
+			}}
+		>
+			copy config
+		</Button>
+
 		{#if config}
 			<Button
 				color="light"
diff --git a/frontend/src/routes/(root)/(logged)/workers/+page.svelte b/frontend/src/routes/(root)/(logged)/workers/+page.svelte
index afd848f896..d6df077bd7 100644
--- a/frontend/src/routes/(root)/(logged)/workers/+page.svelte
+++ b/frontend/src/routes/(root)/(logged)/workers/+page.svelte
@@ -3,8 +3,11 @@
 	import CenteredPage from '$lib/components/CenteredPage.svelte'
 	import { Button, Popup, Skeleton } from '$lib/components/common'
 	import Badge from '$lib/components/common/badge/Badge.svelte'
+	import Drawer from '$lib/components/common/drawer/Drawer.svelte'
+	import DrawerContent from '$lib/components/common/drawer/DrawerContent.svelte'
 	import DefaultTags from '$lib/components/DefaultTags.svelte'
 	import PageHeader from '$lib/components/PageHeader.svelte'
+	import SimpleEditor from '$lib/components/SimpleEditor.svelte'
 	import Cell from '$lib/components/table/Cell.svelte'
 	import DataTable from '$lib/components/table/DataTable.svelte'
 	import Head from '$lib/components/table/Head.svelte'
@@ -16,6 +19,7 @@
 	import { displayDate, groupBy, truncate } from '$lib/utils'
 	import { AlertTriangle, Plus } from 'lucide-svelte'
 	import { onDestroy, onMount } from 'svelte'
+	import YAML from 'yaml'
 
 	let workers: WorkerPing[] | undefined = undefined
 	let workerGroups: Record<string, any> | undefined = undefined
@@ -104,8 +108,55 @@
 		await ConfigService.updateConfig({ name: 'worker__' + newConfigName, requestBody: {} })
 		loadWorkerGroups()
 	}
+
+	let importConfigDrawer: Drawer | undefined = undefined
+	let importConfigCode = ''
+
+	async function importConfig() {
+		const config = YAML.parse(importConfigCode)
+
+		if (!config || config.name == undefined) {
+			sendUserToast('Invalid worker config', true)
+			return
+		}
+
+		if (workerGroups?.hasOwnProperty(config.name)) {
+			sendUserToast('A worker config with this name already exists', true)
+			return
+		}
+
+		await ConfigService.updateConfig({
+			name: 'worker__' + config.name,
+			requestBody: { ...config, name: undefined }
+		})
+
+		importConfigDrawer?.toggleDrawer?.()
+
+		importConfigCode = ''
+
+		sendUserToast('Worker config successfully imported')
+
+		await loadWorkerGroups()
+	}
 </script>
 
+<Drawer bind:this={importConfigDrawer} size="800px">
+	<DrawerContent
+		title="Import config from YAML"
+		on:close={() => importConfigDrawer?.toggleDrawer?.()}
+	>
+		<SimpleEditor
+			bind:code={importConfigCode}
+			lang="yaml"
+			class="h-full"
+			fixedOverflowWidgets={false}
+		/>
+		<svelte:fragment slot="actions">
+			<Button size="sm" on:click={importConfig} disabled={!importConfigCode}>Import</Button>
+		</svelte:fragment>
+	</DrawerContent>
+</Drawer>
+
 <CenteredPage>
 	<PageHeader
 		title="Workers"
@@ -149,7 +200,21 @@
 					>
 						<svelte:fragment slot="button">
 							<div class="flex items-center">
-								<Button size="sm" startIcon={{ icon: Plus }} nonCaptureEvent>
+								<Button
+									size="sm"
+									startIcon={{ icon: Plus }}
+									nonCaptureEvent
+									dropdownItems={$enterpriseLicense
+										? [
+												{
+													label: 'Import config from YAML',
+													onClick: () => {
+														importConfigDrawer?.toggleDrawer?.()
+													}
+												}
+										  ]
+										: undefined}
+								>
 									New worker group config
 									<Tooltip light>
 										Worker Group configs are propagated to every workers in the worker group