seroval 0.4.0-alpha.8 → 0.4.0

package/src/tree/sync.ts

@@ -1,63 +1,132 @@
 /* eslint-disable @typescript-eslint/no-use-before-define */
 import assert from '../assert';
-import { isIterable, isPrimitive, constructorCheck } from '../checks';
 import { Feature } from '../compat';
 import { createRef, ParserContext } from '../context';
-import { ServerValue } from '../types';
+import quote from '../quote';
+import { BigIntTypedArrayValue, ServerValue, 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,
   SerovalArrayNode,
-  SerovalDictionaryNode,
   SerovalErrorNode,
   SerovalIterableNode,
   SerovalMapNode,
   SerovalNode,
   SerovalNodeType,
+  SerovalNullConstructorNode,
+  SerovalObjectNode,
+  SerovalObjectRecordNode,
   SerovalSetNode,
 } from './types';
 
-function serializePropertiesSync(
+type ObjectLikeNode = SerovalObjectNode | SerovalNullConstructorNode | SerovalIterableNode;
+
+function generateNodeList(ctx: ParserContext, current: ServerValue[]) {
+  const size = current.length;
+  const nodes = new Array<SerovalNode>(size);
+  const deferred = new Array<ServerValue>(size);
+  let item: ServerValue;
+  for (let i = 0; i < size; i++) {
+    if (i in current) {
+      item = current[i];
+      if (isIterable(item)) {
+        deferred[i] = item;
+      } else {
+        nodes[i] = parse(ctx, item);
+      }
+    }
+  }
+  for (let i = 0; i < size; i++) {
+    if (i in deferred) {
+      nodes[i] = parse(ctx, deferred[i]);
+    }
+  }
+  return nodes;
+}
+
+function generateArrayNode(
   ctx: ParserContext,
-  properties: Record<string, unknown>,
-): SerovalDictionaryNode {
-  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<ServerValue>(size);
+  id: number,
+  current: ServerValue[],
+): SerovalArrayNode {
+  return {
+    t: SerovalNodeType.Array,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    d: undefined,
+    a: generateNodeList(ctx, current),
+    n: undefined,
+  };
+}
+
+function generateMapNode(
+  ctx: ParserContext,
+  id: number,
+  current: Map<ServerValue, ServerValue>,
+): 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<ServerValue>(len);
+  const deferredValue = new Array<ServerValue>(len);
   let deferredSize = 0;
-  let nodesSize = 0;
-  for (const key of keys) {
-    if (isIterable(properties[key])) {
-      deferredKeys[deferredSize] = key;
-      deferredValues[deferredSize] = properties[key] as ServerValue;
+  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] = generateTreeSync(ctx, properties[key] as ServerValue);
-      nodesSize++;
+      keyNodes[nodeSize] = parse(ctx, key);
+      valueNodes[nodeSize] = parse(ctx, value);
+      nodeSize++;
     }
   }
   for (let i = 0; i < deferredSize; i++) {
-    keyNodes[nodesSize + i] = deferredKeys[i];
-    valueNodes[nodesSize + i] = generateTreeSync(ctx, deferredValues[i]);
+    keyNodes[nodeSize + i] = parse(ctx, deferredKey[i]);
+    valueNodes[nodeSize + i] = parse(ctx, deferredValue[i]);
   }
-  return [keyNodes, valueNodes, size];
+  return {
+    t: SerovalNodeType.Map,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    d: { k: keyNodes, v: valueNodes, s: len },
+    a: undefined,
+    n: undefined,
+  };
 }
 
 function generateSetNode(
   ctx: ParserContext,
-  current: Set<ServerValue>,
   id: number,
+  current: Set<ServerValue>,
 ): SerovalSetNode {
   assert(ctx.features & Feature.Set, 'Unsupported type "Set"');
   const len = current.size;
@@ -70,173 +139,253 @@ function generateSetNode(
     if (isIterable(item)) {
       deferred[deferredSize++] = item;
     } else {
-      nodes[nodeSize++] = generateTreeSync(ctx, item);
+      nodes[nodeSize++] = parse(ctx, item);
     }
   }
   // Parse deferred items
   for (let i = 0; i < deferredSize; i++) {
-    nodes[nodeSize + i] = generateTreeSync(ctx, deferred[i]);
+    nodes[nodeSize + i] = parse(ctx, deferred[i]);
   }
-  return [SerovalNodeType.Set, nodes, id];
+  return {
+    t: SerovalNodeType.Set,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    d: undefined,
+    a: nodes,
+    n: undefined,
+  };
 }
 
-function generateMapNode(
+function generateProperties(
   ctx: ParserContext,
-  current: Map<ServerValue, ServerValue>,
-  id: number,
-): 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<ServerValue>(len);
-  const deferredValue = new Array<ServerValue>(len);
+  properties: Record<string, ServerValue>,
+): 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<ServerValue>(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: ServerValue;
+  for (const key of keys) {
+    item = properties[key];
+    if (isIterable(item)) {
+      deferredKeys[deferredSize] = key;
+      deferredValues[deferredSize] = item;
       deferredSize++;
     } else {
-      keyNodes[nodeSize] = generateTreeSync(ctx, key);
-      valueNodes[nodeSize] = generateTreeSync(ctx, value);
-      nodeSize++;
+      keyNodes[nodesSize] = key;
+      valueNodes[nodesSize] = parse(ctx, item);
+      nodesSize++;
     }
   }
   for (let i = 0; i < deferredSize; i++) {
-    keyNodes[nodeSize + i] = generateTreeSync(ctx, deferredKey[i]);
-    valueNodes[nodeSize + i] = generateTreeSync(ctx, deferredValue[i]);
+    keyNodes[nodesSize + i] = deferredKeys[i];
+    valueNodes[nodesSize + i] = parse(ctx, deferredValues[i]);
   }
-  return [SerovalNodeType.Map, [keyNodes, valueNodes, len], id];
+  return {
+    k: keyNodes,
+    v: valueNodes,
+    s: size,
+  };
 }
 
-function generateNodeList(
+function generateIterableNode(
   ctx: ParserContext,
-  current: ServerValue[],
-) {
-  const size = current.length;
-  const nodes = new Array<SerovalNode>(size);
-  const deferred = new Array<ServerValue>(size);
-  for (let i = 0; i < size; i++) {
-    if (i in current) {
-      if (isIterable(current[i])) {
-        deferred[i] = current[i];
-      } else {
-        nodes[i] = generateTreeSync(ctx, current[i]);
-      }
-    }
-  }
-  for (let i = 0; i < size; i++) {
-    if (i in deferred) {
-      nodes[i] = generateTreeSync(ctx, deferred[i]);
-    }
-  }
-  return nodes;
+  id: number,
+  current: Iterable<ServerValue>,
+): SerovalIterableNode {
+  assert(ctx.features & Feature.SymbolIterator, 'Unsupported type "Iterable"');
+  const options = getIterableOptions(current);
+  return {
+    t: SerovalNodeType.Iterable,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    // Parse options first before the items
+    d: options
+      ? generateProperties(ctx, options as Record<string, ServerValue>)
+      : undefined,
+    a: generateNodeList(ctx, Array.from(current)),
+    n: undefined,
+  };
 }
 
-function generateArrayNode(
+function generateObjectNode(
   ctx: ParserContext,
-  current: ServerValue[],
   id: number,
-): SerovalArrayNode {
-  return [SerovalNodeType.Array, generateNodeList(ctx, current), id];
+  current: Record<string, ServerValue> | Iterable<ServerValue>,
+  empty: boolean,
+): ObjectLikeNode {
+  if (Symbol.iterator in current) {
+    return generateIterableNode(ctx, id, current);
+  }
+  return {
+    t: empty ? SerovalNodeType.NullConstructor : SerovalNodeType.Object,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: undefined,
+    d: generateProperties(ctx, current),
+    a: undefined,
+    n: undefined,
+  };
 }
 
 function generateAggregateErrorNode(
   ctx: ParserContext,
-  current: AggregateError,
   id: number,
+  current: AggregateError,
 ): SerovalAggregateErrorNode {
-  const options = getErrorOptions(current);
+  const options = getErrorOptions(ctx, current);
   const optionsNode = options
-    ? serializePropertiesSync(ctx, options)
+    ? generateProperties(ctx, options)
     : undefined;
-  const errorsNode = generateTreeSync(ctx, current.errors as ServerValue);
-  return [SerovalNodeType.AggregateError, [current.message, optionsNode, errorsNode], id];
+  return {
+    t: SerovalNodeType.AggregateError,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: undefined,
+    m: current.message,
+    d: optionsNode,
+    a: undefined,
+    n: parse(ctx, current.errors),
+  };
 }
 
 function generateErrorNode(
   ctx: ParserContext,
-  current: Error,
   id: number,
+  current: Error,
 ): SerovalErrorNode {
-  const options = getErrorOptions(current);
+  const options = getErrorOptions(ctx, current);
   const optionsNode = options
-    ? serializePropertiesSync(ctx, options)
+    ? generateProperties(ctx, options)
     : undefined;
-  return [
-    SerovalNodeType.Error,
-    [getErrorConstructor(current), current.message, optionsNode],
-    id,
-  ];
-}
-
-function generateIterableNode(
-  ctx: ParserContext,
-  current: Iterable<ServerValue>,
-  id: number,
-): SerovalIterableNode {
-  assert(ctx.features & Feature.SymbolIterator, 'Unsupported type "Iterable"');
-  const options = getIterableOptions(current);
-  return [SerovalNodeType.Iterable, [
-    // Parse options first before the items
-    options ? serializePropertiesSync(ctx, options) : undefined,
-    generateNodeList(ctx, Array.from(current)),
-  ], id];
+  return {
+    t: SerovalNodeType.Error,
+    i: id,
+    s: undefined,
+    l: undefined,
+    c: getErrorConstructor(current),
+    m: current.message,
+    d: optionsNode,
+    a: undefined,
+    n: undefined,
+  };
 }
 
-function generateTreeSync(
+function parse(
   ctx: ParserContext,
   current: ServerValue,
 ): SerovalNode {
-  if (isPrimitive(current)) {
-    return [SerovalNodeType.Primitive, serializePrimitive(current)];
-  }
-  // Non-primitive values needs a reference ID
-  // mostly because the values themselves are stateful
-  const id = generateRef(ctx, current);
-  if (Array.isArray(id)) {
-    return id;
-  }
-  const semiPrimitive = generateSemiPrimitiveValue(ctx, current, id);
-  if (semiPrimitive) {
-    return semiPrimitive;
-  }
-  if (constructorCheck<Set<ServerValue>>(current, Set)) {
-    return generateSetNode(ctx, current, id);
-  }
-  if (constructorCheck<Map<ServerValue, ServerValue>>(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 [
-      empty ? SerovalNodeType.NullConstructor : SerovalNodeType.Object,
-      serializePropertiesSync(ctx, current as Record<string, unknown>),
-      id,
-    ];
+  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 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<ServerValue, ServerValue>);
+        case Set:
+          return generateSetNode(ctx, id, current as Set<ServerValue>);
+        case Object:
+          return generateObjectNode(ctx, id, current as Record<string, ServerValue>, false);
+        case undefined:
+          return generateObjectNode(ctx, id, current as Record<string, ServerValue>, 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);
+      }
+      // Generator functions don't have a global constructor
+      if (Symbol.iterator in current) {
+        return generateIterableNode(ctx, id, current);
+      }
+      throw new Error('Unsupported type');
+    }
+    default:
+      throw new Error('Unsupported type');
   }
-  throw new Error('Unsupported value');
 }
 
 export default function parseSync(
   ctx: ParserContext,
   current: ServerValue,
 ) {
-  const result = generateTreeSync(ctx, current);
-  return [result, createRef(ctx, current), result[0] === SerovalNodeType.Object] as const;
+  const result = parse(ctx, current);
+  const isObject = result.t === SerovalNodeType.Object
+    || result.t === SerovalNodeType.Iterable;
+  return [result, createRef(ctx, current, false), isObject] as const;
 }