seroval 0.4.0-alpha.9 → 0.4.1

package/src/tree/async.ts

@@ -1,22 +1,33 @@
 /* eslint-disable no-await-in-loop */
 /* eslint-disable @typescript-eslint/no-use-before-define */
 import assert from '../assert';
-import {
-  isIterable,
-  isPrimitive,
-  constructorCheck,
-  isPromise,
-} from '../checks';
 import { Feature } from '../compat';
 import { createRef, ParserContext } from '../context';
-import { AsyncServerValue } from '../types';
+import quote from '../quote';
+import {
+  AsyncServerValue,
+  BigIntTypedArrayValue,
+  TypedArrayValue,
+} from '../types';
+import {
+  createBigIntNode,
+  createBigIntTypedArrayNode,
+  createDateNode,
+  createPrimitiveNode,
+  createReferenceNode,
+  createRegExpNode,
+  createTypedArrayNode,
+  FALSE_NODE,
+  NEG_ZERO_NODE,
+  NULL_NODE,
+  TRUE_NODE,
+  UNDEFINED_NODE,
+} from './primitives';
 import {
-  generateRef,
-  generateSemiPrimitiveValue,
   getErrorConstructor,
   getErrorOptions,
   getIterableOptions,
-  serializePrimitive,
+  isIterable,
 } from './shared';
 import {
   SerovalAggregateErrorNode,
@@ -26,48 +37,114 @@ import {
   SerovalMapNode,
   SerovalNode,
   SerovalNodeType,
+  SerovalNullConstructorNode,
+  SerovalObjectNode,
   SerovalObjectRecordNode,
+  SerovalPromiseNode,
   SerovalSetNode,
 } from './types';
 
-async function serializePropertiesAsync(
+type ObjectLikeNode =
+  | SerovalObjectNode
+  | SerovalNullConstructorNode
+  | SerovalIterableNode
+  | SerovalPromiseNode;
+
+type ObjectLikeValue =
+  | Record<string, AsyncServerValue>
+  | Iterable<AsyncServerValue>
+  | PromiseLike<AsyncServerValue>;
+
+async function generateNodeList(
   ctx: ParserContext,
-  properties: Record<string, unknown>,
-): Promise<SerovalObjectRecordNode> {
-  const keys = Object.keys(properties);
-  const size = keys.length;
-  const keyNodes = new Array<string>(size);
-  const valueNodes = new Array<SerovalNode>(size);
-  const deferredKeys = new Array<string>(size);
-  const deferredValues = new Array<AsyncServerValue>(size);
+  current: AsyncServerValue[],
+) {
+  const size = current.length;
+  const nodes = new Array<SerovalNode>(size);
+  const deferred = new Array<AsyncServerValue>(size);
+  let item: AsyncServerValue;
+  for (let i = 0; i < size; i++) {
+    if (i in current) {
+      item = current[i];
+      if (isIterable(item)) {
+        deferred[i] = item;
+      } else {
+        nodes[i] = await parse(ctx, item);
+      }
+    }
+  }
+  for (let i = 0; i < size; i++) {
+    if (i in deferred) {
+      nodes[i] = await parse(ctx, deferred[i]);
+    }
+  }
+  return nodes;
+}
+
+async function generateArrayNode(
+  ctx: ParserContext,
+  id: number,
+  current: AsyncServerValue[],
+): Promise<SerovalArrayNode> {
+  return {
+    t: SerovalNodeType.Array,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    d: undefined,
+    a: await generateNodeList(ctx, current),
+    n: undefined,
+  };
+}
+
+async function generateMapNode(
+  ctx: ParserContext,
+  id: number,
+  current: Map<AsyncServerValue, AsyncServerValue>,
+): Promise<SerovalMapNode> {
+  assert(ctx.features & Feature.Map, 'Unsupported type "Map"');
+  const len = current.size;
+  const keyNodes = new Array<SerovalNode>(len);
+  const valueNodes = new Array<SerovalNode>(len);
+  const deferredKey = new Array<AsyncServerValue>(len);
+  const deferredValue = new Array<AsyncServerValue>(len);
   let deferredSize = 0;
-  let nodesSize = 0;
-  for (const key of keys) {
-    if (isIterable(properties[key])) {
-      deferredKeys[deferredSize] = key;
-      deferredValues[deferredSize] = properties[key] as AsyncServerValue;
+  let nodeSize = 0;
+  for (const [key, value] of current.entries()) {
+    // Either key or value might be an iterable
+    if (isIterable(key) || isIterable(value)) {
+      deferredKey[deferredSize] = key;
+      deferredValue[deferredSize] = value;
       deferredSize++;
     } else {
-      keyNodes[nodesSize] = key;
-      valueNodes[nodesSize] = await generateTreeAsync(ctx, properties[key] as AsyncServerValue);
-      nodesSize++;
+      keyNodes[nodeSize] = await parse(ctx, key);
+      valueNodes[nodeSize] = await parse(ctx, value);
+      nodeSize++;
     }
   }
   for (let i = 0; i < deferredSize; i++) {
-    keyNodes[nodesSize + i] = deferredKeys[i];
-    valueNodes[nodesSize + i] = await generateTreeAsync(ctx, deferredValues[i]);
+    keyNodes[nodeSize + i] = await parse(ctx, deferredKey[i]);
+    valueNodes[nodeSize + i] = await parse(ctx, deferredValue[i]);
   }
   return {
-    k: keyNodes,
-    v: valueNodes,
-    s: size,
+    t: SerovalNodeType.Map,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    d: { k: keyNodes, v: valueNodes, s: len },
+    a: undefined,
+    n: undefined,
   };
 }
 
 async function generateSetNode(
   ctx: ParserContext,
-  current: Set<AsyncServerValue>,
   id: number,
+  current: Set<AsyncServerValue>,
 ): Promise<SerovalSetNode> {
   assert(ctx.features & Feature.Set, 'Unsupported type "Set"');
   const len = current.size;
@@ -80,258 +157,303 @@ async function generateSetNode(
     if (isIterable(item)) {
       deferred[deferredSize++] = item;
     } else {
-      nodes[nodeSize++] = await generateTreeAsync(ctx, item);
+      nodes[nodeSize++] = await parse(ctx, item);
     }
   }
   // Parse deferred items
   for (let i = 0; i < deferredSize; i++) {
-    nodes[nodeSize + i] = await generateTreeAsync(ctx, deferred[i]);
+    nodes[nodeSize + i] = await parse(ctx, deferred[i]);
   }
   return {
     t: SerovalNodeType.Set,
     i: id,
-    a: nodes,
     s: undefined,
     l: undefined,
-    m: undefined,
     c: undefined,
+    m: undefined,
     d: undefined,
+    a: nodes,
     n: undefined,
   };
 }
 
-async function generateMapNode(
+async function generateProperties(
   ctx: ParserContext,
-  current: Map<AsyncServerValue, AsyncServerValue>,
-  id: number,
-): Promise<SerovalMapNode> {
-  assert(ctx.features & Feature.Map, 'Unsupported type "Map"');
-  const len = current.size;
-  const keyNodes = new Array<SerovalNode>(len);
-  const valueNodes = new Array<SerovalNode>(len);
-  const deferredKey = new Array<AsyncServerValue>(len);
-  const deferredValue = new Array<AsyncServerValue>(len);
+  properties: Record<string, AsyncServerValue>,
+): Promise<SerovalObjectRecordNode> {
+  const keys = Object.keys(properties);
+  const size = keys.length;
+  const keyNodes = new Array<string>(size);
+  const valueNodes = new Array<SerovalNode>(size);
+  const deferredKeys = new Array<string>(size);
+  const deferredValues = new Array<AsyncServerValue>(size);
   let deferredSize = 0;
-  let nodeSize = 0;
-  for (const [key, value] of current.entries()) {
-    // Either key or value might be an iterable
-    if (isIterable(key) || isIterable(value)) {
-      deferredKey[deferredSize] = key;
-      deferredValue[deferredSize] = value;
+  let nodesSize = 0;
+  let item: AsyncServerValue;
+  for (const key of keys) {
+    item = properties[key];
+    if (isIterable(item)) {
+      deferredKeys[deferredSize] = key;
+      deferredValues[deferredSize] = item;
       deferredSize++;
     } else {
-      keyNodes[nodeSize] = await generateTreeAsync(ctx, key);
-      valueNodes[nodeSize] = await generateTreeAsync(ctx, value);
-      nodeSize++;
+      keyNodes[nodesSize] = key;
+      valueNodes[nodesSize] = await parse(ctx, item);
+      nodesSize++;
     }
   }
   for (let i = 0; i < deferredSize; i++) {
-    keyNodes[nodeSize + i] = await generateTreeAsync(ctx, deferredKey[i]);
-    valueNodes[nodeSize + i] = await generateTreeAsync(ctx, deferredValue[i]);
+    keyNodes[nodesSize + i] = deferredKeys[i];
+    valueNodes[nodesSize + i] = await parse(ctx, deferredValues[i]);
   }
   return {
-    t: SerovalNodeType.Map,
+    k: keyNodes,
+    v: valueNodes,
+    s: size,
+  };
+}
+
+async function generateIterableNode(
+  ctx: ParserContext,
+  id: number,
+  current: Iterable<AsyncServerValue>,
+): Promise<SerovalIterableNode> {
+  assert(ctx.features & Feature.SymbolIterator, 'Unsupported type "Iterable"');
+  const options = getIterableOptions(current);
+  return {
+    t: SerovalNodeType.Iterable,
     i: id,
-    a: undefined,
     s: undefined,
     l: undefined,
-    m: undefined,
     c: undefined,
-    d: { k: keyNodes, v: valueNodes, s: len },
+    m: undefined,
+    // Parse options first before the items
+    d: options
+      ? await generateProperties(ctx, options as Record<string, AsyncServerValue>)
+      : undefined,
+    a: await generateNodeList(ctx, Array.from(current)),
     n: undefined,
   };
 }
 
-async function generateNodeList(
+async function generatePromiseNode(
   ctx: ParserContext,
-  current: AsyncServerValue[],
-) {
-  const size = current.length;
-  const nodes = new Array<SerovalNode>(size);
-  const deferred = new Array<AsyncServerValue>(size);
-  for (let i = 0; i < size; i++) {
-    if (i in current) {
-      if (isIterable(current[i])) {
-        deferred[i] = current[i];
-      } else {
-        nodes[i] = await generateTreeAsync(ctx, current[i]);
-      }
-    }
-  }
-  for (let i = 0; i < size; i++) {
-    if (i in deferred) {
-      nodes[i] = await generateTreeAsync(ctx, deferred[i]);
-    }
-  }
-  return nodes;
+  id: number,
+  current: PromiseLike<AsyncServerValue>,
+): Promise<SerovalPromiseNode> {
+  assert(ctx.features & Feature.Promise, 'Unsupported type "Promise"');
+  return current.then(async (value) => ({
+    t: SerovalNodeType.Promise,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    // Parse options first before the items
+    d: undefined,
+    a: undefined,
+    n: await parse(ctx, value),
+  }));
 }
 
-async function generateArrayNode(
+async function generateObjectNode(
   ctx: ParserContext,
-  current: AsyncServerValue[],
   id: number,
-): Promise<SerovalArrayNode> {
+  current: ObjectLikeValue,
+  empty: boolean,
+): Promise<ObjectLikeNode> {
+  if (Symbol.iterator in current) {
+    return generateIterableNode(ctx, id, current);
+  }
+  if ('then' in current && typeof current.then === 'function') {
+    return generatePromiseNode(ctx, id, current as PromiseLike<AsyncServerValue>);
+  }
   return {
-    t: SerovalNodeType.Array,
+    t: empty ? SerovalNodeType.NullConstructor : SerovalNodeType.Object,
     i: id,
-    a: await generateNodeList(ctx, current),
     s: undefined,
     l: undefined,
-    m: undefined,
     c: undefined,
-    d: undefined,
+    m: undefined,
+    d: await generateProperties(ctx, current as Record<string, AsyncServerValue>),
+    a: undefined,
     n: undefined,
   };
 }
 
 async function generateAggregateErrorNode(
   ctx: ParserContext,
-  current: AggregateError,
   id: number,
+  current: AggregateError,
 ): Promise<SerovalAggregateErrorNode> {
-  const options = getErrorOptions(current);
+  const options = getErrorOptions(ctx, current);
   const optionsNode = options
-    ? await serializePropertiesAsync(ctx, options)
+    ? await generateProperties(ctx, options)
     : undefined;
   return {
     t: SerovalNodeType.AggregateError,
     i: id,
-    a: undefined,
     s: undefined,
     l: undefined,
-    m: current.message,
     c: undefined,
+    m: current.message,
     d: optionsNode,
-    n: await generateTreeAsync(ctx, current.errors as AsyncServerValue),
+    a: undefined,
+    n: await parse(ctx, current.errors),
   };
 }
 
 async function generateErrorNode(
   ctx: ParserContext,
-  current: Error,
   id: number,
+  current: Error,
 ): Promise<SerovalErrorNode> {
-  const options = getErrorOptions(current);
+  const options = getErrorOptions(ctx, current);
   const optionsNode = options
-    ? await serializePropertiesAsync(ctx, options)
+    ? await generateProperties(ctx, options)
     : undefined;
   return {
     t: SerovalNodeType.Error,
     i: id,
-    a: undefined,
     s: undefined,
     l: undefined,
-    m: current.message,
     c: getErrorConstructor(current),
+    m: current.message,
     d: optionsNode,
+    a: undefined,
     n: undefined,
   };
 }
 
-async function generateIterableNode(
-  ctx: ParserContext,
-  current: Iterable<AsyncServerValue>,
-  id: number,
-): Promise<SerovalIterableNode> {
-  assert(ctx.features & Feature.SymbolIterator, 'Unsupported type "Iterable"');
-  const options = getIterableOptions(current);
-  return {
-    t: SerovalNodeType.Iterable,
-    i: id,
-    s: undefined,
-    l: undefined,
-    m: undefined,
-    c: undefined,
-    // Parse options first before the items
-    d: options ? await serializePropertiesAsync(ctx, options) : undefined,
-    a: await generateNodeList(ctx, Array.from(current)),
-    n: undefined,
-  };
-}
-
-async function generateTreeAsync(
+async function parse(
   ctx: ParserContext,
   current: AsyncServerValue,
 ): Promise<SerovalNode> {
-  if (isPrimitive(current)) {
-    return {
-      t: SerovalNodeType.Primitive,
-      i: undefined,
-      s: serializePrimitive(current),
-      l: undefined,
-      m: undefined,
-      c: undefined,
-      // Parse options first before the items
-      d: undefined,
-      a: undefined,
-      n: undefined,
-    };
-  }
-  // Non-primitive values needs a reference ID
-  // mostly because the values themselves are stateful
-  const id = generateRef(ctx, current);
-  if (typeof id !== 'number') {
-    return id;
-  }
-  const semiPrimitive = generateSemiPrimitiveValue(ctx, current, id);
-  if (semiPrimitive) {
-    return semiPrimitive;
-  }
-  if (isPromise(current)) {
-    assert(ctx.features & Feature.Promise, 'Unsupported type "Promise"');
-    return current.then(async (value) => ({
-      t: SerovalNodeType.Promise,
-      i: id,
-      s: undefined,
-      l: undefined,
-      m: undefined,
-      c: undefined,
-      // Parse options first before the items
-      d: undefined,
-      a: undefined,
-      n: await generateTreeAsync(ctx, value),
-    }));
-  }
-  if (constructorCheck<Set<AsyncServerValue>>(current, Set)) {
-    return generateSetNode(ctx, current, id);
-  }
-  if (constructorCheck<Map<AsyncServerValue, AsyncServerValue>>(current, Map)) {
-    return generateMapNode(ctx, current, id);
-  }
-  if (Array.isArray(current)) {
-    return generateArrayNode(ctx, current, id);
-  }
-  if (current instanceof AggregateError && ctx.features & Feature.AggregateError) {
-    return generateAggregateErrorNode(ctx, current, id);
-  }
-  if (current instanceof Error) {
-    return generateErrorNode(ctx, current, id);
-  }
-  if (isIterable(current)) {
-    return generateIterableNode(ctx, current, id);
-  }
-  const empty = current.constructor == null;
-  if (current.constructor === Object || empty) {
-    return {
-      t: empty ? SerovalNodeType.NullConstructor : SerovalNodeType.Object,
-      i: id,
-      s: undefined,
-      l: undefined,
-      m: undefined,
-      c: undefined,
-      // Parse options first before the items
-      d: await serializePropertiesAsync(ctx, current as Record<string, unknown>),
-      a: undefined,
-      n: undefined,
-    };
+  switch (typeof current) {
+    case 'boolean':
+      return current ? TRUE_NODE : FALSE_NODE;
+    case 'undefined':
+      return UNDEFINED_NODE;
+    case 'string':
+      return createPrimitiveNode(quote(current));
+    case 'number':
+      if (Object.is(current, -0)) {
+        return NEG_ZERO_NODE;
+      }
+      if (Object.is(current, Infinity)) {
+        return createPrimitiveNode('1/0');
+      }
+      if (Object.is(current, -Infinity)) {
+        return createPrimitiveNode('-1/0');
+      }
+      return createPrimitiveNode(current);
+    case 'bigint':
+      return createBigIntNode(ctx, current);
+    case 'object': {
+      if (!current) {
+        return NULL_NODE;
+      }
+      // Non-primitive values needs a reference ID
+      // mostly because the values themselves are stateful
+      const id = createRef(ctx, current, true);
+      if (ctx.markedRefs.has(id)) {
+        return createReferenceNode(id);
+      }
+      if (Array.isArray(current)) {
+        return generateArrayNode(ctx, id, current);
+      }
+      switch (current.constructor) {
+        case Date:
+          return createDateNode(id, current as Date);
+        case RegExp:
+          return createRegExpNode(id, current as RegExp);
+        case Promise:
+          return generatePromiseNode(ctx, id, current as Promise<AsyncServerValue>);
+        case Int8Array:
+        case Int16Array:
+        case Int32Array:
+        case Uint8Array:
+        case Uint16Array:
+        case Uint32Array:
+        case Uint8ClampedArray:
+        case Float32Array:
+        case Float64Array:
+          return createTypedArrayNode(ctx, id, current as TypedArrayValue);
+        case BigInt64Array:
+        case BigUint64Array:
+          return createBigIntTypedArrayNode(ctx, id, current as BigIntTypedArrayValue);
+        case Map:
+          return generateMapNode(
+            ctx,
+            id,
+            current as Map<AsyncServerValue, AsyncServerValue>,
+          );
+        case Set:
+          return generateSetNode(
+            ctx,
+            id,
+            current as Set<AsyncServerValue>,
+          );
+        case Object:
+          return generateObjectNode(
+            ctx,
+            id,
+            current as Record<string, AsyncServerValue>,
+            false,
+          );
+        case undefined:
+          return generateObjectNode(
+            ctx,
+            id,
+            current as Record<string, AsyncServerValue>,
+            true,
+          );
+        case AggregateError:
+          if (ctx.features & Feature.AggregateError) {
+            return generateAggregateErrorNode(ctx, id, current as AggregateError);
+          }
+          return generateErrorNode(ctx, id, current as AggregateError);
+        case Error:
+        case EvalError:
+        case RangeError:
+        case ReferenceError:
+        case SyntaxError:
+        case TypeError:
+        case URIError:
+          return generateErrorNode(ctx, id, current as Error);
+        default:
+          break;
+      }
+      if (current instanceof AggregateError && ctx.features & Feature.AggregateError) {
+        return generateAggregateErrorNode(ctx, id, current);
+      }
+      if (current instanceof Error) {
+        return generateErrorNode(ctx, id, current);
+      }
+      if (current instanceof Promise) {
+        return generatePromiseNode(ctx, id, current);
+      }
+      // Generator functions don't have a global constructor
+      if (Symbol.iterator in current) {
+        return generateIterableNode(ctx, id, current);
+      }
+      // For Promise-like objects
+      if ('then' in current && typeof current.then === 'function') {
+        return generatePromiseNode(ctx, id, current as PromiseLike<AsyncServerValue>);
+      }
+      throw new Error('Unsupported type');
+    }
+    default:
+      throw new Error('Unsupported type');
   }
-  throw new Error('Unsupported value');
 }
 
 export default async function parseAsync(
   ctx: ParserContext,
   current: AsyncServerValue,
 ) {
-  const result = await generateTreeAsync(ctx, current);
-  return [result, createRef(ctx, current), result.t === SerovalNodeType.Object] as const;
+  const result = await parse(ctx, current);
+  const isObject = result.t === SerovalNodeType.Object
+    || result.t === SerovalNodeType.Iterable;
+  return [result, createRef(ctx, current, false), isObject] as const;
 }