|
15 | 15 | // specific language governing permissions and limitations |
16 | 16 | // under the License. |
17 | 17 |
|
18 | | -const { PrimitiveType, NonPrimitiveType, RemoteType } = require('./protocolType') |
| 18 | +const { PrimitiveType, NonPrimitiveType, RemoteType, SpecialNumberType } = require('./protocolType') |
19 | 19 |
|
20 | 20 | const TYPE_CONSTANT = 'type' |
21 | 21 | const VALUE_CONSTANT = 'value' |
@@ -189,6 +189,88 @@ class LocalValue { |
189 | 189 | return new ReferenceValue(handle, sharedId) |
190 | 190 | } |
191 | 191 |
|
| 192 | + static getArgument(argument) { |
| 193 | + let localValue = null |
| 194 | + |
| 195 | + if ( |
| 196 | + argument === SpecialNumberType.NAN || |
| 197 | + argument === SpecialNumberType.MINUS_ZERO || |
| 198 | + argument === SpecialNumberType.INFINITY || |
| 199 | + argument === SpecialNumberType.MINUS_INFINITY |
| 200 | + ) { |
| 201 | + localValue = LocalValue.createSpecialNumberValue(argument) |
| 202 | + return localValue |
| 203 | + } |
| 204 | + |
| 205 | + const type = typeof argument |
| 206 | + |
| 207 | + switch (type) { |
| 208 | + case PrimitiveType.STRING: |
| 209 | + localValue = LocalValue.createStringValue(argument) |
| 210 | + break |
| 211 | + case PrimitiveType.NUMBER: |
| 212 | + localValue = LocalValue.createNumberValue(argument) |
| 213 | + break |
| 214 | + case PrimitiveType.BOOLEAN: |
| 215 | + localValue = LocalValue.createBooleanValue(argument) |
| 216 | + break |
| 217 | + case PrimitiveType.BIGINT: |
| 218 | + localValue = LocalValue.createBigIntValue(argument.toString()) |
| 219 | + break |
| 220 | + case PrimitiveType.UNDEFINED: |
| 221 | + localValue = LocalValue.createUndefinedValue() |
| 222 | + break |
| 223 | + case NonPrimitiveType.OBJECT: |
| 224 | + if (argument === null) { |
| 225 | + localValue = LocalValue.createNullValue() |
| 226 | + break |
| 227 | + } |
| 228 | + if (argument instanceof Date) { |
| 229 | + localValue = LocalValue.createDateValue(argument) |
| 230 | + } else if (argument instanceof Map) { |
| 231 | + const map = [] |
| 232 | + |
| 233 | + argument.forEach((value, key) => { |
| 234 | + let objectKey |
| 235 | + if (typeof key === 'string') { |
| 236 | + objectKey = key |
| 237 | + } else { |
| 238 | + objectKey = LocalValue.getArgument(key) |
| 239 | + } |
| 240 | + const objectValue = LocalValue.getArgument(value) |
| 241 | + map.push([objectKey, objectValue]) |
| 242 | + }) |
| 243 | + localValue = new LocalValue(NonPrimitiveType.MAP, map) |
| 244 | + } else if (argument instanceof Set) { |
| 245 | + const set = [] |
| 246 | + argument.forEach((value) => { |
| 247 | + set.push(LocalValue.getArgument(value)) |
| 248 | + }) |
| 249 | + localValue = LocalValue.createSetValue(set) |
| 250 | + } else if (argument instanceof Array) { |
| 251 | + const arr = [] |
| 252 | + argument.forEach((value) => { |
| 253 | + arr.push(LocalValue.getArgument(value)) |
| 254 | + }) |
| 255 | + localValue = LocalValue.createArrayValue(arr) |
| 256 | + } else if (argument instanceof RegExp) { |
| 257 | + localValue = LocalValue.createRegularExpressionValue({ |
| 258 | + pattern: argument.source, |
| 259 | + flags: argument.flags, |
| 260 | + }) |
| 261 | + } else { |
| 262 | + let value = [] |
| 263 | + Object.entries(argument).forEach((entry) => { |
| 264 | + value.push([LocalValue.getArgument(entry[0]), LocalValue.getArgument(entry[1])]) |
| 265 | + }) |
| 266 | + localValue = new LocalValue(NonPrimitiveType.OBJECT, value) |
| 267 | + } |
| 268 | + break |
| 269 | + } |
| 270 | + |
| 271 | + return localValue |
| 272 | + } |
| 273 | + |
192 | 274 | asMap() { |
193 | 275 | let toReturn = {} |
194 | 276 | toReturn[TYPE_CONSTANT] = this.type |
@@ -246,7 +328,7 @@ class RemoteValue { |
246 | 328 | } |
247 | 329 |
|
248 | 330 | deserializeValue(value, type) { |
249 | | - if ([NonPrimitiveType.MAP, NonPrimitiveType.OBJECT].includes(type)) { |
| 331 | + if (type === NonPrimitiveType.OBJECT) { |
250 | 332 | return Object.fromEntries(value) |
251 | 333 | } else if (type === NonPrimitiveType.REGULAR_EXPRESSION) { |
252 | 334 | return new RegExpValue(value.pattern, value.flags) |
|
0 commit comments