From 840ea29c39e43bed5d5fa0d7e1dcb582b4c709a1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jesper=20M=C3=B8ller=20Jensen?=
 <26099018+JesmoDev@users.noreply.github.com>
Date: Wed, 8 Nov 2023 16:46:56 +1300
Subject: [PATCH] update create methods

---
 .../repository/media-type.repository.ts       | 25 +++++++++++++------
 .../workspace/media-type-workspace.context.ts |  4 ++-
 2 files changed, 20 insertions(+), 9 deletions(-)

diff --git a/src/Umbraco.Web.UI.Client/src/packages/media/media-types/repository/media-type.repository.ts b/src/Umbraco.Web.UI.Client/src/packages/media/media-types/repository/media-type.repository.ts
index f0d411308e..109785e1f6 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/media/media-types/repository/media-type.repository.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/media/media-types/repository/media-type.repository.ts
@@ -9,6 +9,7 @@ import { UmbTreeRepository, UmbTreeDataSource } from '@umbraco-cms/backoffice/re
 import {
 	CreateMediaTypeRequestModel,
 	EntityTreeItemResponseModel,
+	MediaTypeResponseModel,
 	UpdateMediaTypeRequestModel,
 } from '@umbraco-cms/backoffice/backend-api';
 
@@ -160,20 +161,28 @@ export class UmbMediaTypeRepository implements UmbTreeRepository<EntityTreeItemR
 	}
 
 	async create(mediaType: CreateMediaTypeRequestModel) {
+		if (!mediaType || !mediaType.id) throw new Error('Document Type is missing');
 		await this.#init;
 
-		if (!mediaType.name) {
-			throw new Error('Name is missing');
-		}
-
-		const { data, error } = await this.#detailSource.insert(mediaType);
+		const { error } = await this.#detailSource.insert(mediaType);
 
 		if (!error) {
-			const notification = { data: { message: `Media type '${mediaType.name}' created` } };
-			this.#notificationContext?.peek('positive', notification);
+			//TODO: Model mismatch. FIX
+			this.#detailStore?.append(mediaType as unknown as MediaTypeResponseModel);
+
+			const treeItem = {
+				type: 'media-type',
+				parentId: null,
+				name: mediaType.name,
+				id: mediaType.id,
+				isFolder: false,
+				isContainer: false,
+				hasChildren: false,
+			};
+			this.#treeStore?.appendItems([treeItem]);
 		}
 
-		return { data, error };
+		return { error };
 	}
 
 	async move() {
diff --git a/src/Umbraco.Web.UI.Client/src/packages/media/media-types/workspace/media-type-workspace.context.ts b/src/Umbraco.Web.UI.Client/src/packages/media/media-types/workspace/media-type-workspace.context.ts
index 0e3256e1a6..aacb0dedbb 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/media/media-types/workspace/media-type-workspace.context.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/media/media-types/workspace/media-type-workspace.context.ts
@@ -54,7 +54,9 @@ export class UmbMediaTypeWorkspaceContext
 		if (!data) return;
 
 		this.setIsNew(true);
-		this.#data.next(data);
+
+		//TODO: Model mismatch. FIX
+		this.#data.next(data as unknown as MediaTypeResponseModel);
 	}
 
 	async save() {