From 31fd7c96dcc68777374296e7e4a9f31453b8840e Mon Sep 17 00:00:00 2001
From: Mads Rasmussen <madsr@hey.com>
Date: Thu, 7 Dec 2023 10:59:29 +0100
Subject: [PATCH] add util to create media tree item

---
 .../src/mocks/data/utils.ts                    | 18 ++++++++++++------
 1 file changed, 12 insertions(+), 6 deletions(-)

diff --git a/src/Umbraco.Web.UI.Client/src/mocks/data/utils.ts b/src/Umbraco.Web.UI.Client/src/mocks/data/utils.ts
index 0e3d447613..79a7876e20 100644
--- a/src/Umbraco.Web.UI.Client/src/mocks/data/utils.ts
+++ b/src/Umbraco.Web.UI.Client/src/mocks/data/utils.ts
@@ -11,7 +11,9 @@ import type {
 	FileItemResponseModelBaseModel,
 	MediaTypeResponseModel,
 	MediaTypeTreeItemResponseModel,
+	MediaTreeItemResponseModel,
 } from '@umbraco-cms/backoffice/backend-api';
+import { UmbMediaDetailModel } from '@umbraco-cms/backoffice/media';
 
 export const createEntityTreeItem = (item: any): EntityTreeItemResponseModel => {
 	return {
@@ -31,8 +33,7 @@ export const createFolderTreeItem = (item: any): FolderTreeItemResponseModel =>
 	};
 };
 
-// TODO: remove isTrashed type extension when we have found a solution to trashed items
-export const createContentTreeItem = (item: any): ContentTreeItemResponseModel & { isTrashed: boolean } => {
+export const createContentTreeItem = (item: any): ContentTreeItemResponseModel => {
 	// TODO: There we have to adapt to variants as part of the tree model:
 	return {
 		...createEntityTreeItem(item),
@@ -41,10 +42,7 @@ export const createContentTreeItem = (item: any): ContentTreeItemResponseModel &
 	};
 };
 
-// TODO: remove isTrashed type extension when we have found a solution to trashed items
-export const createDocumentTreeItem = (
-	item: DocumentResponseModel,
-): DocumentTreeItemResponseModel & { isTrashed: boolean } => {
+export const createDocumentTreeItem = (item: DocumentResponseModel): DocumentTreeItemResponseModel => {
 	return {
 		...createContentTreeItem(item),
 		type: 'document',
@@ -60,6 +58,14 @@ export const createDocumentTreeItem = (
 	};
 };
 
+export const createMediaTreeItem = (item: UmbMediaDetailModel): MediaTreeItemResponseModel => {
+	return {
+		...createContentTreeItem(item),
+		type: 'media',
+		icon: 'media', // TODO: Should get this from media type...
+	};
+};
+
 export const createDocumentTypeTreeItem = (item: DocumentTypeResponseModel): DocumentTypeTreeItemResponseModel => {
 	return {
 		...createEntityTreeItem(item),