1 | require('@lwc/synthetic-shadow/dist/synthetic-shadow.js');
|
2 |
|
3 |
|
4 | global.wireAdaptersRegistryHack = new Map();
|
5 | let originalRegisterDecorators;
|
6 |
|
7 | function isValidWireAdapter(adapter) {
|
8 | let isValid = false;
|
9 | if (typeof adapter === 'function') {
|
10 |
|
11 | try {
|
12 | const adapterInstance = new adapter(() => {});
|
13 | isValid =
|
14 | typeof adapterInstance.connect === 'function' &&
|
15 | typeof adapterInstance.update === 'function' &&
|
16 | typeof adapterInstance.disconnect === 'function';
|
17 | } catch (e) {
|
18 | isValid = false;
|
19 | }
|
20 | }
|
21 |
|
22 | return isValid;
|
23 | }
|
24 |
|
25 |
|
26 |
|
27 |
|
28 |
|
29 |
|
30 |
|
31 |
|
32 |
|
33 |
|
34 |
|
35 |
|
36 | function createWireAdapterMockClass(originalAdapter) {
|
37 | const noopAdapter = {
|
38 | connect() {},
|
39 | update() {},
|
40 | disconnect() {},
|
41 | };
|
42 | let baseAdapter;
|
43 | let baseAdapterFn = () => {};
|
44 | const spies = [];
|
45 |
|
46 | if (Object.prototype.hasOwnProperty.call(originalAdapter, 'adapter')) {
|
47 |
|
48 |
|
49 | baseAdapter = originalAdapter.adapter;
|
50 | } else if (isValidWireAdapter(originalAdapter)) {
|
51 |
|
52 | baseAdapter = originalAdapter;
|
53 | }
|
54 |
|
55 | if (typeof originalAdapter === 'function') {
|
56 |
|
57 | baseAdapterFn = originalAdapter;
|
58 | }
|
59 |
|
60 |
|
61 | const newAdapterMock = function (...args) {
|
62 | return baseAdapterFn.call(this, ...args);
|
63 | };
|
64 |
|
65 | newAdapterMock.adapter = class WireAdapterMock {
|
66 | constructor(dataCallback) {
|
67 |
|
68 | this._originalAdapter =
|
69 | spies.length === 0 && baseAdapter ? new baseAdapter(dataCallback) : noopAdapter;
|
70 | this._dataCallback = dataCallback;
|
71 |
|
72 | spies.forEach((spy) => spy.createInstance(this));
|
73 | }
|
74 | connect() {
|
75 | spies.forEach((spy) => spy.connect(this));
|
76 | this._originalAdapter.connect();
|
77 | }
|
78 | update(config) {
|
79 | spies.forEach((spy) => spy.update(this, config));
|
80 | this._originalAdapter.update(config);
|
81 | }
|
82 | disconnect() {
|
83 | spies.forEach((spy) => spy.disconnect(this));
|
84 | this._originalAdapter.disconnect();
|
85 | }
|
86 | emit(value) {
|
87 | this._dataCallback(value);
|
88 | }
|
89 | static spyAdapter(spy) {
|
90 |
|
91 |
|
92 |
|
93 | spies.push(spy);
|
94 | }
|
95 | };
|
96 |
|
97 | return newAdapterMock;
|
98 | }
|
99 |
|
100 | function overriddenRegisterDecorators(Ctor, decoratorsMeta) {
|
101 | const wire = decoratorsMeta.wire || {};
|
102 |
|
103 | Object.keys(wire).forEach((adapterName) => {
|
104 | const adapter = wire[adapterName].adapter;
|
105 | let wireAdapterMock = global.wireAdaptersRegistryHack.get(adapter);
|
106 |
|
107 | if (!wireAdapterMock) {
|
108 |
|
109 |
|
110 |
|
111 |
|
112 |
|
113 | if (!Object.isExtensible(adapter)) {
|
114 |
|
115 |
|
116 | throw new TypeError('Invalid adapterId, it must be extensible.');
|
117 | }
|
118 |
|
119 |
|
120 | wireAdapterMock = createWireAdapterMockClass(adapter);
|
121 |
|
122 | global.wireAdaptersRegistryHack.set(adapter, wireAdapterMock);
|
123 | }
|
124 |
|
125 |
|
126 | wire[adapterName].adapter = wireAdapterMock;
|
127 | });
|
128 |
|
129 | return originalRegisterDecorators(Ctor, decoratorsMeta);
|
130 | }
|
131 |
|
132 | function installRegisterDecoratorsTrap(lwc) {
|
133 | const originalDescriptor = Object.getOwnPropertyDescriptor(lwc, 'registerDecorators');
|
134 |
|
135 | if (originalDescriptor.value === overriddenRegisterDecorators) {
|
136 | return;
|
137 | }
|
138 |
|
139 | originalRegisterDecorators = originalDescriptor.value;
|
140 |
|
141 | const newDescriptor = {
|
142 | ...originalDescriptor,
|
143 | value: overriddenRegisterDecorators,
|
144 | };
|
145 |
|
146 | Object.defineProperty(lwc, 'registerDecorators', newDescriptor);
|
147 | }
|
148 |
|
149 | const lwc = require('@lwc/engine');
|
150 |
|
151 | installRegisterDecoratorsTrap(lwc);
|