|
1 | 1 | import { Element, StringElement } from 'minim';
|
2 |
| -import stampit from 'stampit'; |
3 | 2 | import ShortUniqueId from 'short-unique-id';
|
4 | 3 |
|
5 | 4 | import ElementIdentityError from './errors/ElementIdentityError';
|
6 | 5 | import { isElement, isStringElement } from '../predicates';
|
7 | 6 |
|
8 |
| -export interface IdentityManager<T extends Element = Element> { |
9 |
| - length: number; |
10 |
| - uuid: ShortUniqueId; |
11 |
| - identityMap: WeakMap<T, StringElement>; |
| 7 | +export class IdentityManager { |
| 8 | + protected readonly uuid: ShortUniqueId; |
12 | 9 |
|
13 |
| - identify(this: IdentityManager<T>, element: T): StringElement; |
14 |
| - forget(this: IdentityManager<T>, element: T): boolean; |
15 |
| - generateId(this: IdentityManager<T>): string; |
16 |
| -} |
| 10 | + protected readonly identityMap: WeakMap<Element, StringElement>; |
17 | 11 |
|
18 |
| -// @TODO([email protected]): transforming this stamp to class will break backward compatibility |
19 |
| -export const IdentityManager: stampit.Stamp<IdentityManager> = stampit({ |
20 |
| - props: { |
21 |
| - uuid: null, |
22 |
| - length: null, |
23 |
| - identityMap: null, |
24 |
| - }, |
25 |
| - init(this: IdentityManager, { length = 6 } = {}) { |
26 |
| - this.length = 6; |
| 12 | + constructor({ length = 6 } = {}) { |
27 | 13 | this.uuid = new ShortUniqueId({ length });
|
28 | 14 | this.identityMap = new WeakMap();
|
29 |
| - }, |
30 |
| - methods: { |
31 |
| - identify<T extends Element>(this: IdentityManager, element: T): StringElement { |
32 |
| - if (!isElement(element)) { |
33 |
| - throw new ElementIdentityError( |
34 |
| - 'Cannot not identify the element. `element` is neither structurally compatible nor a subclass of an Element class.', |
35 |
| - { |
36 |
| - value: element, |
37 |
| - }, |
38 |
| - ); |
39 |
| - } |
40 |
| - |
41 |
| - // use already assigned identity |
42 |
| - if ( |
43 |
| - element.meta.hasKey('id') && |
44 |
| - isStringElement(element.meta.get('id')) && |
45 |
| - !element.meta.get('id').equals('') |
46 |
| - ) { |
47 |
| - return element.id; |
48 |
| - } |
49 |
| - |
50 |
| - // assign identity in immutable way |
51 |
| - if (this.identityMap.has(element)) { |
52 |
| - return this.identityMap.get(element)!; |
53 |
| - } |
54 |
| - |
55 |
| - // return element identity |
56 |
| - const id = new StringElement(this.generateId()); |
57 |
| - this.identityMap.set(element, id); |
58 |
| - return id; |
59 |
| - }, |
60 |
| - |
61 |
| - forget<T extends Element>(this: IdentityManager, element: T): boolean { |
62 |
| - if (this.identityMap.has(element)) { |
63 |
| - this.identityMap.delete(element); |
64 |
| - return true; |
65 |
| - } |
66 |
| - return false; |
67 |
| - }, |
68 |
| - |
69 |
| - generateId(this: IdentityManager): string { |
70 |
| - return this.uuid.randomUUID(); |
71 |
| - }, |
72 |
| - }, |
73 |
| -}); |
| 15 | + } |
| 16 | + |
| 17 | + identify<T extends Element>(this: IdentityManager, element: T): StringElement { |
| 18 | + if (!isElement(element)) { |
| 19 | + throw new ElementIdentityError( |
| 20 | + 'Cannot not identify the element. `element` is neither structurally compatible nor a subclass of an Element class.', |
| 21 | + { |
| 22 | + value: element, |
| 23 | + }, |
| 24 | + ); |
| 25 | + } |
| 26 | + |
| 27 | + // use already assigned identity |
| 28 | + if ( |
| 29 | + element.meta.hasKey('id') && |
| 30 | + isStringElement(element.meta.get('id')) && |
| 31 | + !element.meta.get('id').equals('') |
| 32 | + ) { |
| 33 | + return element.id; |
| 34 | + } |
| 35 | + |
| 36 | + // assign identity in immutable way |
| 37 | + if (this.identityMap.has(element)) { |
| 38 | + return this.identityMap.get(element)!; |
| 39 | + } |
| 40 | + |
| 41 | + // return element identity |
| 42 | + const id = new StringElement(this.generateId()); |
| 43 | + this.identityMap.set(element, id); |
| 44 | + return id; |
| 45 | + } |
| 46 | + |
| 47 | + forget<T extends Element>(element: T): boolean { |
| 48 | + if (this.identityMap.has(element)) { |
| 49 | + this.identityMap.delete(element); |
| 50 | + return true; |
| 51 | + } |
| 52 | + return false; |
| 53 | + } |
| 54 | + |
| 55 | + generateId() { |
| 56 | + return this.uuid.randomUUID(); |
| 57 | + } |
| 58 | +} |
74 | 59 |
|
75 |
| -export const defaultIdentityManager = IdentityManager({ length: 6 }); |
| 60 | +export const defaultIdentityManager = new IdentityManager(); |
0 commit comments