da7aa4055e
* Conform more of the code base to strict null checking * More strict fixes * More strict work * Fix missing optional type * Iterate
57 lines
1.8 KiB
TypeScript
57 lines
1.8 KiB
TypeScript
/*
|
|
Copyright 2020 The Matrix.org Foundation C.I.C.
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License");
|
|
you may not use this file except in compliance with the License.
|
|
You may obtain a copy of the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS,
|
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
See the License for the specific language governing permissions and
|
|
limitations under the License.
|
|
*/
|
|
|
|
import { arrayDiff, arrayIntersection } from "./arrays";
|
|
|
|
/**
|
|
* Determines the keys added, changed, and removed between two Maps.
|
|
* For changes, simple triple equal comparisons are done, not in-depth tree checking.
|
|
* @param a The first Map. Must be defined.
|
|
* @param b The second Map. Must be defined.
|
|
* @returns The difference between the keys of each Map.
|
|
*/
|
|
export function mapDiff<K, V>(a: Map<K, V>, b: Map<K, V>): { changed: K[]; added: K[]; removed: K[] } {
|
|
const aKeys = [...a.keys()];
|
|
const bKeys = [...b.keys()];
|
|
const keyDiff = arrayDiff(aKeys, bKeys);
|
|
const possibleChanges = arrayIntersection(aKeys, bKeys);
|
|
const changes = possibleChanges.filter((k) => a.get(k) !== b.get(k));
|
|
|
|
return { changed: changes, added: keyDiff.added, removed: keyDiff.removed };
|
|
}
|
|
|
|
/**
|
|
* A Map<K, V> with added utility.
|
|
*/
|
|
export class EnhancedMap<K, V> extends Map<K, V> {
|
|
public constructor(entries?: Iterable<[K, V]>) {
|
|
super(entries);
|
|
}
|
|
|
|
public getOrCreate(key: K, def: V): V {
|
|
if (this.has(key)) {
|
|
return this.get(key)!;
|
|
}
|
|
this.set(key, def);
|
|
return def;
|
|
}
|
|
|
|
public remove(key: K): V | undefined {
|
|
const v = this.get(key);
|
|
this.delete(key);
|
|
return v;
|
|
}
|
|
}
|