From 4a053dc9356259c8d9190d823ce74085625a49a2 Mon Sep 17 00:00:00 2001
From: Nassim Tabchiche <n.tabchiche2@gmail.com>
Date: Wed, 14 Feb 2024 17:17:56 +0100
Subject: [PATCH] Manage policy CRUD operations

---
 .../src/lib/components/Forms/ModelForm.svelte   |  6 ++++--
 frontend/src/lib/utils/crud.ts                  | 17 +++++++++++++++++
 frontend/src/lib/utils/schemas.ts               | 12 ++++++++++++
 frontend/src/lib/utils/table.ts                 |  4 ++++
 frontend/src/lib/utils/types.ts                 |  1 +
 5 files changed, 38 insertions(+), 2 deletions(-)

diff --git a/frontend/src/lib/components/Forms/ModelForm.svelte b/frontend/src/lib/components/Forms/ModelForm.svelte
index d28a68a18..2ad12a4de 100644
--- a/frontend/src/lib/components/Forms/ModelForm.svelte
+++ b/frontend/src/lib/components/Forms/ModelForm.svelte
@@ -144,8 +144,10 @@
 			field="threats"
 			label="Threats"
 		/>
-	{:else if URLModel === 'security-measures'}
-		<Select {form} options={model.selectOptions['category']} field="category" label="Category" />
+	{:else if URLModel === 'security-measures' || URLModel === 'policies'}
+		{#if schema.shape.category}
+			<Select {form} options={model.selectOptions['category']} field="category" label="Category" />
+		{/if}
 		<Select {form} options={model.selectOptions['status']} field="status" label="Status" />
 		<AutocompleteSelect
 			{form}
diff --git a/frontend/src/lib/utils/crud.ts b/frontend/src/lib/utils/crud.ts
index eef2cf201..d207b9ab4 100644
--- a/frontend/src/lib/utils/crud.ts
+++ b/frontend/src/lib/utils/crud.ts
@@ -170,6 +170,23 @@ export const URL_MODEL_MAP: ModelMap = {
 			{ field: 'folder' }
 		]
 	},
+	policies: {
+		name: 'policy',
+		verboseName: 'Policy',
+		verboseNamePlural: 'Policies',
+		foreignKeyFields: [
+			{ field: 'security_function', urlModel: 'security-functions' },
+			{ field: 'folder', urlModel: 'folders' },
+			{ field: 'evidences', urlModel: 'evidences' }
+		],
+		selectFields: [{ field: 'status' }, { field: 'category' }, { field: 'effort' }],
+		filters: [
+			{ field: 'security_function' },
+			{ field: 'status' },
+			{ field: 'effort' },
+			{ field: 'folder' }
+		]
+	},
 	'risk-acceptances': {
 		name: 'riskacceptance',
 		verboseName: 'Risk acceptance',
diff --git a/frontend/src/lib/utils/schemas.ts b/frontend/src/lib/utils/schemas.ts
index a93055352..500bbadf9 100644
--- a/frontend/src/lib/utils/schemas.ts
+++ b/frontend/src/lib/utils/schemas.ts
@@ -98,6 +98,17 @@ export const SecurityMeasureSchema = baseNamedObject({
 	security_function: z.string().optional()
 });
 
+export const PolicySchema = baseNamedObject({
+	status: z.string().optional(),
+	evidences: z.string().optional().array().optional(),
+	eta: z.string().optional().nullable(),
+	expiry_date: z.string().optional().nullable(),
+	link: z.string().url().optional().nullable(),
+	effort: z.string().optional(),
+	folder: z.string(),
+	security_function: z.string().optional()
+});
+
 export const RiskAcceptanceSchema = baseNamedObject({
 	folder: z.string(),
 	expiry_date: z.string().optional().nullable(),
@@ -181,6 +192,7 @@ const SCHEMA_MAP: Record<string, AnyZodObject> = {
 	threats: ThreatSchema,
 	'risk-scenarios': RiskScenarioSchema,
 	'security-measures': SecurityMeasureSchema,
+	policies: PolicySchema,
 	'risk-acceptances': RiskAcceptanceSchema,
 	'security-functions': SecurityFunctionSchema,
 	assets: AssetSchema,
diff --git a/frontend/src/lib/utils/table.ts b/frontend/src/lib/utils/table.ts
index f1ccf135d..5299b2760 100644
--- a/frontend/src/lib/utils/table.ts
+++ b/frontend/src/lib/utils/table.ts
@@ -49,6 +49,10 @@ export const listViewFields = {
 		head: ['Name', 'Description', 'Category', 'ETA', 'Domain', 'Security function'],
 		body: ['name', 'description', 'category', 'eta', 'folder', 'security_function']
 	},
+	policies: {
+		head: ['Name', 'Description', 'ETA', 'Domain', 'Security function'],
+		body: ['name', 'description', 'eta', 'folder', 'security_function']
+	},
 	'security-functions': {
 		head: ['Ref', 'Name', 'Description', 'Category', 'Provider', 'Domain'],
 		body: ['ref_id', 'name', 'description', 'category', 'provider', 'folder'],
diff --git a/frontend/src/lib/utils/types.ts b/frontend/src/lib/utils/types.ts
index 91598628e..0a7f574cf 100644
--- a/frontend/src/lib/utils/types.ts
+++ b/frontend/src/lib/utils/types.ts
@@ -24,6 +24,7 @@ export const URL_MODEL = [
 	'threats',
 	'risk-scenarios',
 	'security-measures',
+	'policies',
 	'risk-acceptances',
 	'security-functions',
 	'assets',