blob: 863de6874fe7dc7e5f6fd45c063f72f34e286346 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
|
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
/*
* A Map that automatically normalizes keys before accessing the values.
*
* The constructor takes two parameters:
* aNormalize: A function which takes a string and returns the "normalized"
* version of it.
* aIterable: A iterable to prefill the map with, keys will be normalized.
*
* Returns a Map object that will automatically run aNormalize on any operations
* involving keys.
*/
export class NormalizedMap extends Map {
constructor(aNormalize, aIterable = []) {
if (typeof aNormalize != "function") {
throw new Error("NormalizedMap must have a normalize function!");
}
// Create the wrapped Map; use the provided iterable after normalizing the
// keys.
let entries = [...aIterable].map(([key, val]) => [aNormalize(key), val]);
super(entries);
// Note: In derived classes, super() must be called before using 'this'.
this._normalize = aNormalize;
}
// Dummy normalize function.
_normalize(aKey) {
return aKey;
}
// Anything that accepts a key as an input needs to be manually overridden.
delete(key) {
return super.delete(this._normalize(key));
}
get(key) {
return super.get(this._normalize(key));
}
has(key) {
return super.has(this._normalize(key));
}
set(key, val) {
super.set(this._normalize(key), val);
return this;
}
}
|
