add SortableSet as a new collection type

timse · timse · commit 0eaa84748dde · 2017-06-18T10:41:58.000+10:00
diff --git a/lib/util/SortableSet.js b/lib/util/SortableSet.js
@@ -0,0 +1,28 @@
+"use strict";
+
+module.exports = class SortableSet extends Set {
+
+	constructor(initialIterable, defaultSort) {
+		super(initialIterable);
+		this._sortFn = defaultSort;
+	}
+
+	/**
+	 * @param {Function} sortFn - function to sort the set
+	 * @returns {void}
+	 */
+	sortWith(sortFn) {
+		const sortedArray = Array.from(this).sort(sortFn);
+		this.clear();
+		for(let i = 0; i < sortedArray.length; i += 1) {
+			this.add(sortedArray[i]);
+		}
+	}
+
+	/**
+	 * @returns {void}
+	 */
+	sort() {
+		this.sortWith(this._sortFn);
+	}
+};
