From bea12b7a50b9c7d9f9352d34604a19a48678ceba Mon Sep 17 00:00:00 2001
From: Lee Kelleher <leekelleher@users.noreply.github.com>
Date: Tue, 26 Nov 2024 13:34:40 +0000
Subject: [PATCH] UFM: ContentName, adds support for Media Picker (#17635)

---
 .../content-name/content-name.element.ts      | 64 +++++++++++++------
 1 file changed, 45 insertions(+), 19 deletions(-)

diff --git a/src/Umbraco.Web.UI.Client/src/packages/ufm/components/content-name/content-name.element.ts b/src/Umbraco.Web.UI.Client/src/packages/ufm/components/content-name/content-name.element.ts
index 4335c2bb28..215e53fc72 100644
--- a/src/Umbraco.Web.UI.Client/src/packages/ufm/components/content-name/content-name.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/packages/ufm/components/content-name/content-name.element.ts
@@ -1,9 +1,10 @@
 import { UmbUfmElementBase } from '../ufm-element-base.js';
 import { UMB_UFM_RENDER_CONTEXT } from '../ufm-render/ufm-render.context.js';
 import { customElement, property } from '@umbraco-cms/backoffice/external/lit';
-import { UmbDocumentItemRepository } from '@umbraco-cms/backoffice/document';
-import { UmbMediaItemRepository } from '@umbraco-cms/backoffice/media';
-import { UmbMemberItemRepository } from '@umbraco-cms/backoffice/member';
+import { UmbDocumentItemRepository, UMB_DOCUMENT_ENTITY_TYPE } from '@umbraco-cms/backoffice/document';
+import { UmbId } from '@umbraco-cms/backoffice/id';
+import { UmbMediaItemRepository, UMB_MEDIA_ENTITY_TYPE } from '@umbraco-cms/backoffice/media';
+import { UmbMemberItemRepository, UMB_MEMBER_ENTITY_TYPE } from '@umbraco-cms/backoffice/member';
 
 const elementName = 'ufm-content-name';
 
@@ -25,39 +26,64 @@ export class UmbUfmContentNameElement extends UmbUfmElementBase {
 				async (value) => {
 					const temp =
 						this.alias && typeof value === 'object'
-							? ((value as Record<string, unknown>)[this.alias] as string)
-							: (value as string);
+							? (value as Record<string, unknown>)[this.alias]
+							: (value as unknown);
 
-					const entityType = Array.isArray(temp) && temp.length > 0 ? temp[0].type : null;
-					const uniques = Array.isArray(temp) ? temp.map((x) => x.unique) : temp ? [temp] : [];
+					if (!temp) return;
 
-					if (uniques?.length) {
-						const repository = this.#getRepository(entityType);
-						if (repository) {
-							const { data } = await repository.requestItems(uniques);
-							this.value = data ? data.map((item) => item.name).join(', ') : '';
-							return;
-						}
-					}
+					const entityType = this.#getEntityType(temp);
+					const uniques = this.#getUniques(temp);
 
-					this.value = '';
+					this.value = await this.#getNames(entityType, uniques);
 				},
 				'observeValue',
 			);
 		});
 	}
 
+	#getEntityType(value: unknown) {
+		if (Array.isArray(value) && value.length > 0) {
+			const item = value[0];
+			if (item.type) return item.type;
+			if (item.mediaKey) return UMB_MEDIA_ENTITY_TYPE;
+		}
+
+		return null;
+	}
+
+	#getUniques(value: unknown) {
+		if (Array.isArray(value)) {
+			return value.map((x) => x.unique ?? x.mediaKey ?? x).filter((x) => UmbId.validate(x));
+		}
+
+		return typeof value === 'string' && UmbId.validate(value) ? [value] : [];
+	}
+
+	async #getNames(entityType: string, uniques?: Array<string>) {
+		if (uniques?.length) {
+			const repository = this.#getRepository(entityType);
+			if (repository) {
+				const { data } = await repository.requestItems(uniques);
+				if (Array.isArray(data) && data.length > 0) {
+					return data.map((item) => item.name).join(', ');
+				}
+			}
+		}
+
+		return '';
+	}
+
 	#getRepository(entityType?: string | null) {
 		switch (entityType) {
-			case 'media':
+			case UMB_MEDIA_ENTITY_TYPE:
 				if (!this.#mediaRepository) this.#mediaRepository = new UmbMediaItemRepository(this);
 				return this.#mediaRepository;
 
-			case 'member':
+			case UMB_MEMBER_ENTITY_TYPE:
 				if (!this.#memberRepository) this.#memberRepository = new UmbMemberItemRepository(this);
 				return this.#memberRepository;
 
-			case 'document':
+			case UMB_DOCUMENT_ENTITY_TYPE:
 			default:
 				if (!this.#documentRepository) this.#documentRepository = new UmbDocumentItemRepository(this);
 				return this.#documentRepository;