From 06744ec39b2b20142ffda8d2e2f375cbd06c31c9 Mon Sep 17 00:00:00 2001
From: Mads Rasmussen <madsr@hey.com>
Date: Wed, 12 Apr 2023 17:05:01 +0200
Subject: [PATCH] add picker context base class

---
 .../libs/picker/index.ts                      |  1 +
 .../libs/picker/picker.context.ts             | 94 +++++++++++++++++++
 2 files changed, 95 insertions(+)
 create mode 100644 src/Umbraco.Web.UI.Client/libs/picker/index.ts
 create mode 100644 src/Umbraco.Web.UI.Client/libs/picker/picker.context.ts

diff --git a/src/Umbraco.Web.UI.Client/libs/picker/index.ts b/src/Umbraco.Web.UI.Client/libs/picker/index.ts
new file mode 100644
index 0000000000..49b83a7cbe
--- /dev/null
+++ b/src/Umbraco.Web.UI.Client/libs/picker/index.ts
@@ -0,0 +1 @@
+export * from './picker.context';
diff --git a/src/Umbraco.Web.UI.Client/libs/picker/picker.context.ts b/src/Umbraco.Web.UI.Client/libs/picker/picker.context.ts
new file mode 100644
index 0000000000..0144293104
--- /dev/null
+++ b/src/Umbraco.Web.UI.Client/libs/picker/picker.context.ts
@@ -0,0 +1,94 @@
+import { UmbTreeRepository } from '../repository';
+import { UmbControllerHostElement } from '@umbraco-cms/backoffice/controller';
+import { ArrayState, UmbObserverController } from '@umbraco-cms/backoffice/observable-api';
+import { createExtensionClass, umbExtensionsRegistry } from '@umbraco-cms/backoffice/extensions-api';
+import {
+	UMB_CONFIRM_MODAL,
+	UMB_MODAL_CONTEXT_TOKEN,
+	UmbModalContext,
+	UmbModalToken,
+} from '@umbraco-cms/backoffice/modal';
+import { UmbContextConsumerController } from '@umbraco-cms/backoffice/context-api';
+
+export class UmbPickerContext<RepositoryType extends UmbTreeRepository> {
+	host: UmbControllerHostElement;
+	modalAlias: UmbModalToken | string;
+	repository?: RepositoryType;
+
+	public modalContext?: UmbModalContext;
+
+	#selection = new ArrayState<string>([]);
+	selection = this.#selection.asObservable();
+
+	#items = new ArrayState<any>([]);
+	items = this.#items.asObservable();
+
+	max = Infinity;
+	min = 0;
+
+	constructor(host: UmbControllerHostElement, repositoryAlias: string, modalAlias: UmbModalToken | string) {
+		this.host = host;
+		this.modalAlias = modalAlias;
+
+		// TODO: unsure a method can't be called before everything is initialized
+		new UmbObserverController(
+			this.host,
+
+			// TODO: this code is reused in multiple places, so it should be extracted to a function
+			umbExtensionsRegistry.getByTypeAndAlias('repository', repositoryAlias),
+			async (repositoryManifest) => {
+				if (!repositoryManifest) return;
+
+				try {
+					const result = await createExtensionClass<RepositoryType>(repositoryManifest, [this.host]);
+					this.repository = result;
+				} catch (error) {
+					throw new Error('Could not create repository with alias: ' + repositoryAlias + '');
+				}
+			}
+		);
+
+		new UmbContextConsumerController(this.host, UMB_MODAL_CONTEXT_TOKEN, (instance) => {
+			this.modalContext = instance;
+		});
+	}
+
+	getSelection() {
+		return this.#selection.value;
+	}
+
+	setSelection(selection: string[]) {
+		this.#selection.next(selection);
+	}
+
+	openPicker() {
+		if (!this.modalContext) throw new Error('Modal context is not initialized');
+
+		const modalHandler = this.modalContext.open(this.modalAlias, {
+			multiple: this.max === 1 ? false : true,
+			selection: [...this.getSelection()],
+		});
+
+		modalHandler?.onSubmit().then(({ selection }: any) => {
+			this.setSelection(selection);
+		});
+	}
+
+	async removeItem(unique: string) {
+		if (!this.repository) throw new Error('Repository is not initialized');
+
+		const { data } = await this.repository.requestTreeItems([unique]);
+		if (!data) throw new Error('Could not find item with unique id: ' + unique);
+
+		const modalHandler = this.modalContext?.open(UMB_CONFIRM_MODAL, {
+			color: 'danger',
+			headline: `Remove ${data[0].name}?`,
+			content: 'Are you sure you want to remove this item',
+			confirmLabel: 'Remove',
+		});
+
+		await modalHandler?.onSubmit();
+		const newSelection = this.getSelection().filter((value) => value !== unique);
+		this.setSelection(newSelection);
+	}
+}