From 55d501ac6866901ab54bd04949e2a15cdffa50e2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Niels=20Lyngs=C3=B8?= <nsl@umbraco.dk>
Date: Wed, 29 Mar 2023 15:48:57 +0200
Subject: [PATCH] arrayState updateOne method + test

---
 .../libs/observable-api/array-state.test.ts   | 13 +++++++
 .../libs/observable-api/array-state.ts        | 34 ++++++++++++++++++-
 2 files changed, 46 insertions(+), 1 deletion(-)

diff --git a/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.test.ts b/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.test.ts
index 6964402153..854ec3ee92 100644
--- a/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.test.ts
+++ b/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.test.ts
@@ -69,6 +69,19 @@ describe('ArrayState', () => {
 		});
 	});
 
+	it('partially update an existing item via updateOne method.', (done) => {
+		const newItem = { another: 'myValue2.2' };
+		subject.updateOne('2', newItem);
+
+		const observer = subject.asObservable();
+		observer.subscribe((value) => {
+			expect(value.length).to.be.equal(initialData.length);
+			expect(value[0].another).to.be.equal('myValue1');
+			expect(value[1].another).to.be.equal('myValue2.2');
+			done();
+		});
+	});
+
 	it('getObservablePart for a specific entry of array', (done) => {
 		const subObserver = subject.getObservablePart((data) => data.find((x) => x.key === '2'));
 		subObserver.subscribe((entry) => {
diff --git a/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.ts b/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.ts
index d421e4d6b5..5c06539f36 100644
--- a/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.ts
+++ b/src/Umbraco.Web.UI.Client/libs/observable-api/array-state.ts
@@ -1,4 +1,5 @@
 import { DeepState } from './deep-state';
+import { partialUpdateFrozenArray } from './partial-update-frozen-array.function';
 import { pushToUniqueArray } from './push-to-unique-array.function';
 
 /**
@@ -11,8 +12,11 @@ import { pushToUniqueArray } from './push-to-unique-array.function';
  * The ArrayState provides methods to append data when the data is an Object.
  */
 export class ArrayState<T> extends DeepState<T[]> {
-	constructor(initialData: T[], private _getUnique?: (entry: T) => unknown) {
+	private _getUnique?: (entry: T) => unknown;
+
+	constructor(initialData: T[], getUniqueMethod?: (entry: T) => unknown) {
 		super(initialData);
+		this._getUnique = getUniqueMethod;
 	}
 
 	/**
@@ -149,4 +153,32 @@ export class ArrayState<T> extends DeepState<T[]> {
 		}
 		return this;
 	}
+
+	/**
+	 * @method updateOne
+	 * @param {unknown} unique - Unique value to find entry to update.
+	 * @param {Partial<T>} entry - new data to be added in this Subject.
+	 * @return {ArrayState<T>} Reference to it self.
+	 * @description - Update a item with some new data, requires the ArrayState to be constructed with a getUnique method.
+	 * @example <caption>Example append some data.</caption>
+	 * const data = [
+	 * 	{ key: 1, value: 'foo'},
+	 * 	{ key: 2, value: 'bar'}
+	 * ];
+	 * const myState = new ArrayState(data, (x) => x.key);
+	 * myState.updateOne(2, {value: 'updated-bar'});
+	 */
+	updateOne(unique: unknown, entry: Partial<T>) {
+		if (!this._getUnique) {
+			throw new Error("Can't partial update an ArrayState without a getUnique method provided when constructed.");
+		}
+		this.next(
+			partialUpdateFrozenArray(
+				this.getValue(),
+				entry,
+				(x) => unique === (this._getUnique as Exclude<typeof this._getUnique, undefined>)(x)
+			)
+		);
+		return this;
+	}
 }