From 91ab717c2fcd5beb68cd1476e3725ab07b21de90 Mon Sep 17 00:00:00 2001
From: Mohamed-Hacene <mohamedhacene.b@gmail.com>
Date: Fri, 20 Dec 2024 18:58:29 +0100
Subject: [PATCH] feat: add reference entity field for enterprise ebios

---
 backend/ebios_rm/serializers.py               |   1 +
 .../Forms/ModelForm/EbiosRmForm.svelte        | 186 ++++++++++++++++++
 frontend/messages/en.json                     |   4 +-
 frontend/messages/fr.json                     |   4 +-
 frontend/src/lib/utils/crud.ts                |   3 +-
 frontend/src/lib/utils/schemas.ts             |   3 +-
 .../workshop-1/ebios-rm-study/+page.svelte    |  10 +
 7 files changed, 207 insertions(+), 4 deletions(-)
 create mode 100644 enterprise/frontend/src/lib/components/Forms/ModelForm/EbiosRmForm.svelte

diff --git a/backend/ebios_rm/serializers.py b/backend/ebios_rm/serializers.py
index 908a63477..6a9457242 100644
--- a/backend/ebios_rm/serializers.py
+++ b/backend/ebios_rm/serializers.py
@@ -52,6 +52,7 @@ class EbiosRMStudyReadSerializer(BaseModelSerializer):
     str = serializers.CharField(source="__str__")
     project = FieldsRelatedField(["id", "folder"])
     folder = FieldsRelatedField()
+    reference_entity = FieldsRelatedField()
     risk_matrix = FieldsRelatedField()
     reference_entity = FieldsRelatedField()
     assets = FieldsRelatedField(many=True)
diff --git a/enterprise/frontend/src/lib/components/Forms/ModelForm/EbiosRmForm.svelte b/enterprise/frontend/src/lib/components/Forms/ModelForm/EbiosRmForm.svelte
new file mode 100644
index 000000000..17f64f811
--- /dev/null
+++ b/enterprise/frontend/src/lib/components/Forms/ModelForm/EbiosRmForm.svelte
@@ -0,0 +1,186 @@
+<script lang="ts">
+	import type { SuperValidated } from 'sveltekit-superforms';
+	import type { ModelInfo, CacheLock } from '$lib/utils/types';
+	import TextField from '$lib/components/Forms/TextField.svelte';
+	import AutocompleteSelect from '$lib/components/Forms/AutocompleteSelect.svelte';
+	import * as m from '$paraglide/messages.js';
+	import { getOptions } from '$lib/utils/crud';
+	import TextArea from '../TextArea.svelte';
+	import { page } from '$app/stores';
+
+	export let form: SuperValidated<any>;
+	export let model: ModelInfo;
+	export let cacheLocks: Record<string, CacheLock> = {};
+	export let formDataCache: Record<string, any> = {};
+	export let initialData: Record<string, any> = {};
+	export let context: string;
+
+	let activeActivity: string | null = null;
+
+	$page.url.searchParams.forEach((value, key) => {
+		if (key === 'activity' && value === 'one') {
+			activeActivity = 'one';
+		} else if (key === 'activity' && value === 'two') {
+			activeActivity = 'two';
+		}
+	});
+</script>
+
+{#if context != 'selectAudit'}
+	<TextField
+		{form}
+		field="name"
+		label={m.name()}
+		cacheLock={cacheLocks['name']}
+		bind:cachedValue={formDataCache['name']}
+		data-focusindex="0"
+	/>
+{/if}
+{#if context !== 'ebiosRmStudy' && context !== 'selectAudit'}
+	<TextField
+		{form}
+		field="version"
+		label={m.version()}
+		cacheLock={cacheLocks['version']}
+		bind:cachedValue={formDataCache['version']}
+	/>
+	<TextField
+		{form}
+		field="ref_id"
+		label={m.refId()}
+		cacheLock={cacheLocks['ref_id']}
+		bind:cachedValue={formDataCache['ref_id']}
+	/>
+	<AutocompleteSelect
+		{form}
+		options={getOptions({ objects: model.foreignKeys['folder'] })}
+		field="folder"
+		cacheLock={cacheLocks['folder']}
+		bind:cachedValue={formDataCache['folder']}
+		label={m.domain()}
+		hidden={initialData.folder}
+	/>
+	<AutocompleteSelect
+		{form}
+		options={getOptions({ objects: model.foreignKeys['reference_entity'] })}
+		field="reference_entity"
+		cacheLock={cacheLocks['reference_entity']}
+		bind:cachedValue={formDataCache['reference_entity']}
+		label={m.referenceEntity()}
+		hidden={initialData.reference_entity}
+	/>
+	<AutocompleteSelect
+		{form}
+		options={getOptions({ objects: model.foreignKeys['risk_matrix'] })}
+		field="risk_matrix"
+		cacheLock={cacheLocks['risk_matrix']}
+		bind:cachedValue={formDataCache['risk_matrix']}
+		label={m.riskMatrix()}
+		placeholder={m.ebiosRmRiskMatrix()}
+		helpText={m.ebiosRmMatrixHelpText()}
+	/>
+{:else if context === 'ebiosRmStudy'}
+	<div
+		class="relative p-2 space-y-2 rounded-md {activeActivity === 'one'
+			? 'border-2 border-primary-500'
+			: 'border-2 border-gray-300 border-dashed'}"
+	>
+		<p
+			class="absolute -top-3 bg-white font-bold {activeActivity === 'one'
+				? 'text-primary-500'
+				: 'text-gray-500'}"
+		>
+			{m.activityOne()}
+		</p>
+		<TextArea
+			{form}
+			field="description"
+			label={m.description()}
+			cacheLock={cacheLocks['description']}
+			bind:cachedValue={formDataCache['description']}
+			data-focusindex="1"
+		/>
+		<TextField
+			{form}
+			field="version"
+			label={m.version()}
+			cacheLock={cacheLocks['version']}
+			bind:cachedValue={formDataCache['version']}
+		/>
+		<TextField
+			{form}
+			field="ref_id"
+			label={m.refId()}
+			cacheLock={cacheLocks['ref_id']}
+			bind:cachedValue={formDataCache['ref_id']}
+		/>
+		<AutocompleteSelect
+		{form}
+			options={getOptions({ objects: model.foreignKeys['reference_entity'] })}
+			field="reference_entity"
+			cacheLock={cacheLocks['reference_entity']}
+			bind:cachedValue={formDataCache['reference_entity']}
+			label={m.referenceEntity()}
+			hidden={initialData.reference_entity}
+		/>
+		<AutocompleteSelect
+			multiple
+			{form}
+			options={getOptions({ objects: model.foreignKeys['authors'], label: 'email' })}
+			field="authors"
+			cacheLock={cacheLocks['authors']}
+			bind:cachedValue={formDataCache['authors']}
+			label={m.authors()}
+		/>
+		<AutocompleteSelect
+			multiple
+			{form}
+			options={getOptions({ objects: model.foreignKeys['reviewers'], label: 'email' })}
+			field="reviewers"
+			cacheLock={cacheLocks['reviewers']}
+			bind:cachedValue={formDataCache['reviewers']}
+			label={m.reviewers()}
+		/>
+	</div>
+	<div
+		class="relative p-2 space-y-2 rounded-md {activeActivity === 'two'
+			? 'border-2 border-primary-500'
+			: 'border-2 border-gray-300 border-dashed'}"
+	>
+		<p
+			class="absolute -top-3 bg-white font-bold {activeActivity === 'two'
+				? 'text-primary-500'
+				: 'text-gray-500'}"
+		>
+			{m.activityTwo()}
+		</p>
+		<AutocompleteSelect
+			multiple
+			{form}
+			options={getOptions({
+				objects: model.foreignKeys['assets'],
+				extra_fields: [['folder', 'str']],
+				label: 'auto'
+			})}
+			field="assets"
+			label={m.assets()}
+		/>
+	</div>
+	<TextArea
+		{form}
+		field="observation"
+		label={m.observation()}
+		cacheLock={cacheLocks['observation']}
+		bind:cachedValue={formDataCache['observation']}
+	/>
+{:else}
+	<AutocompleteSelect
+		multiple
+		{form}
+		options={getOptions({ objects: model.foreignKeys['compliance_assessments'] })}
+		field="compliance_assessments"
+		cacheLock={cacheLocks['compliance_assessments']}
+		bind:cachedValue={formDataCache['compliance_assessments']}
+		label={m.complianceAssessment()}
+	/>
+{/if}
diff --git a/frontend/messages/en.json b/frontend/messages/en.json
index bc835ab48..b0434316a 100644
--- a/frontend/messages/en.json
+++ b/frontend/messages/en.json
@@ -1018,5 +1018,7 @@
 	"riskAnalyses": "Risk analyses",
 	"client": "Client",
 	"partner": "Partner",
-	"supplier": "Supplier"
+	"supplier": "Supplier",
+	"referenceEntity": "Reference entity",
+	"referenceEntitySemiColon": "Reference entity:"
 }
diff --git a/frontend/messages/fr.json b/frontend/messages/fr.json
index c45e90e64..fa25b7a05 100644
--- a/frontend/messages/fr.json
+++ b/frontend/messages/fr.json
@@ -1014,5 +1014,7 @@
 	"addStrategicScenario": "Ajouter un scénario stratégique",
 	"client": "Client",
 	"partner": "Partenaire",
-	"supplier": "Fournisseur"
+	"supplier": "Fournisseur",
+	"referenceEntity": "Entité de référence",
+	"referenceEntitySemiColon": "Entité de référence :"
 }
diff --git a/frontend/src/lib/utils/crud.ts b/frontend/src/lib/utils/crud.ts
index fc57fe0a5..24b61e7bb 100644
--- a/frontend/src/lib/utils/crud.ts
+++ b/frontend/src/lib/utils/crud.ts
@@ -614,7 +614,8 @@ export const URL_MODEL_MAP: ModelMap = {
 			{ field: 'authors', urlModel: 'users', urlParams: 'is_third_party=false' },
 			{ field: 'reviewers', urlModel: 'users', urlParams: 'is_third_party=false' },
 			{ field: 'folder', urlModel: 'folders', urlParams: 'content_type=DO' },
-			{ field: 'compliance_assessments', urlModel: 'compliance-assessments' }
+			{ field: 'compliance_assessments', urlModel: 'compliance-assessments' },
+			{ field: 'reference_entity', urlModel: 'entities' }
 		],
 		reverseForeignKeyFields: [{ field: 'ebios_rm_studies', urlModel: 'assets' }]
 	},
diff --git a/frontend/src/lib/utils/schemas.ts b/frontend/src/lib/utils/schemas.ts
index 68d70150a..ded5499a4 100644
--- a/frontend/src/lib/utils/schemas.ts
+++ b/frontend/src/lib/utils/schemas.ts
@@ -402,7 +402,8 @@ export const ebiosRMSchema = z.object({
 	observation: z.string().optional().nullable(),
 	assets: z.string().uuid().optional().array().optional(),
 	folder: z.string(),
-	compliance_assessments: z.string().uuid().optional().array().optional()
+	compliance_assessments: z.string().uuid().optional().array().optional(),
+	reference_entity: z.string().optional()
 });
 
 export const fearedEventsSchema = z.object({
diff --git a/frontend/src/routes/(app)/(internal)/ebios-rm/[id=uuid]/workshop-1/ebios-rm-study/+page.svelte b/frontend/src/routes/(app)/(internal)/ebios-rm/[id=uuid]/workshop-1/ebios-rm-study/+page.svelte
index 7f380d6a1..0186d5079 100644
--- a/frontend/src/routes/(app)/(internal)/ebios-rm/[id=uuid]/workshop-1/ebios-rm-study/+page.svelte
+++ b/frontend/src/routes/(app)/(internal)/ebios-rm/[id=uuid]/workshop-1/ebios-rm-study/+page.svelte
@@ -99,6 +99,16 @@
 				{m.edit()}
 			</a>
 		</div>
+		{#if ebiosRmStudy.reference_entity}
+			<div class="flex justify-center items-center w-full">
+				<span class="text-sm text-gray-500"
+					>{m.referenceEntitySemiColon()}
+					<a class="anchor" href="/entities/{ebiosRmStudy.reference_entity.id}"
+						>{ebiosRmStudy.reference_entity.str}</a
+					>
+				</span>
+			</div>
+		{/if}
 		<div
 			id="activityOne"
 			class="relative p-4 space-y-4 rounded-md w-full flex flex-col items-center