seroval 0.5.1 → 0.7.0

package/src/tree/serialize.ts

@@ -6,7 +6,7 @@ import {
   getRefParam,
   markRef,
 } from '../context';
-import quote from '../quote';
+import { GLOBAL_KEY } from './reference';
 import { isValidIdentifier } from './shared';
 import { SYMBOL_STRING } from './symbols';
 import {
@@ -22,19 +22,32 @@ import {
   SerovalObjectNode,
   SerovalObjectRecordNode,
   SerovalPromiseNode,
-  SerovalReferenceNode,
+  SerovalIndexedValueNode,
   SerovalSetNode,
   SerovalTypedArrayNode,
+  SerovalArrayBufferNode,
+  SerovalDataViewNode,
+  SerovalBlobNode,
+  SerovalFileNode,
+  SerovalHeadersNode,
+  SerovalRegExpNode,
+  SerovalDateNode,
+  SerovalURLNode,
+  SerovalURLSearchParamsNode,
+  SerovalReferenceNode,
+  SerovalFormDataNode,
 } from './types';
 
 function getAssignmentExpression(assignment: Assignment): string {
   switch (assignment.t) {
     case 'index':
       return assignment.s + '=' + assignment.v;
-    case 'map':
-      return assignment.s + '.set(' + assignment.k + ',' + assignment.v + ')';
     case 'set':
+      return assignment.s + '.set(' + assignment.k + ',' + assignment.v + ')';
+    case 'add':
       return assignment.s + '.add(' + assignment.v + ')';
+    case 'append':
+      return assignment.s + '.append(' + assignment.k + ',' + assignment.v + ')';
     default:
       return '';
   }
@@ -65,11 +78,11 @@ function mergeAssignments(assignments: Assignment[]) {
             current = item;
           }
           break;
-        case 'map':
+        case 'set':
           if (item.s === prev.s) {
             // Maps has chaining methods, merge if source is the same
             current = {
-              t: 'map',
+              t: 'set',
               s: getAssignmentExpression(current),
               k: item.k,
               v: item.v,
@@ -80,11 +93,11 @@ function mergeAssignments(assignments: Assignment[]) {
             current = item;
           }
           break;
-        case 'set':
+        case 'add':
           if (item.s === prev.s) {
             // Sets has chaining methods too
             current = {
-              t: 'set',
+              t: 'add',
               s: getAssignmentExpression(current),
               k: undefined,
               v: item.v,
@@ -95,6 +108,11 @@ function mergeAssignments(assignments: Assignment[]) {
             current = item;
           }
           break;
+        case 'append':
+          // Different assignment, push current
+          newAssignments.push(current);
+          current = item;
+          break;
         default:
           break;
       }
@@ -145,29 +163,41 @@ function createAssignment(
   });
 }
 
-function createSetAdd(
+function createAddAssignment(
   ctx: SerializationContext,
   ref: number,
   value: string,
 ) {
-  markRef(ctx, ref);
   ctx.assignments.push({
-    t: 'set',
+    t: 'add',
     s: getRefParam(ctx, ref),
     k: undefined,
     v: value,
   });
 }
 
-function createMapSet(
+function createSetAssignment(
+  ctx: SerializationContext,
+  ref: number,
+  key: string,
+  value: string,
+) {
+  ctx.assignments.push({
+    t: 'set',
+    s: getRefParam(ctx, ref),
+    k: key,
+    v: value,
+  });
+}
+
+function createAppendAssignment(
   ctx: SerializationContext,
   ref: number,
   key: string,
   value: string,
 ) {
-  markRef(ctx, ref);
   ctx.assignments.push({
-    t: 'map',
+    t: 'append',
     s: getRefParam(ctx, ref),
     k: key,
     v: value,
@@ -180,7 +210,6 @@ function createArrayAssign(
   index: number | string,
   value: string,
 ) {
-  markRef(ctx, ref);
   createAssignment(ctx, getRefParam(ctx, ref) + '[' + index + ']', value);
 }
 
@@ -194,7 +223,7 @@ function createObjectAssign(
   createAssignment(ctx, getRefParam(ctx, ref) + '.' + key, value);
 }
 
-function assignRef(
+function assignIndexedValue(
   ctx: SerializationContext,
   index: number,
   value: string,
@@ -205,16 +234,20 @@ function assignRef(
   return value;
 }
 
-function isReferenceInStack(
+function isIndexedValueInStack(
   ctx: SerializationContext,
   node: SerovalNode,
-): node is SerovalReferenceNode {
-  return node.t === SerovalNodeType.Reference && ctx.stack.includes(node.i);
+): node is SerovalIndexedValueNode {
+  return node.t === SerovalNodeType.IndexedValue && ctx.stack.includes(node.i);
 }
 
+type SerovalNodeListNode =
+  | SerovalArrayNode
+  | SerovalIterableNode;
+
 function serializeNodeList(
   ctx: SerializationContext,
-  node: SerovalArrayNode | SerovalIterableNode | SerovalAggregateErrorNode,
+  node: SerovalNodeListNode,
 ) {
   // This is different than Map and Set
   // because we also need to serialize
@@ -232,7 +265,8 @@ function serializeNodeList(
     // Check if index is a hole
     if (item) {
       // Check if item is a parent
-      if (isReferenceInStack(ctx, item)) {
+      if (isIndexedValueInStack(ctx, item)) {
+        markRef(ctx, node.i);
         createArrayAssign(ctx, node.i, i, getRefParam(ctx, item.i));
         isHoley = true;
       } else {
@@ -253,7 +287,7 @@ function serializeArray(
   ctx.stack.push(node.i);
   const result = serializeNodeList(ctx, node);
   ctx.stack.pop();
-  return assignRef(ctx, node.i, result);
+  return assignIndexedValue(ctx, node.i, result);
 }
 
 function serializeProperties(
@@ -272,23 +306,27 @@ function serializeProperties(
   let isIdentifier: boolean;
   let refParam: string;
   let hasPrev = false;
-  for (let i = 0; i < node.s; i++) {
-    key = node.k[i];
-    val = node.v[i];
+  const keys = node.k;
+  const values = node.v;
+  for (let i = 0, len = node.s; i < len; i++) {
+    key = keys[i];
+    val = values[i];
     check = Number(key);
     // Test if key is a valid number or JS identifier
     // so that we don't have to serialize the key and wrap with brackets
     isIdentifier = check >= 0 || isValidIdentifier(key);
-    if (isReferenceInStack(ctx, val)) {
+    if (isIndexedValueInStack(ctx, val)) {
       refParam = getRefParam(ctx, val.i);
       if (isIdentifier && Number.isNaN(check)) {
+        markRef(ctx, sourceID);
         createObjectAssign(ctx, sourceID, key, refParam);
       } else {
-        createArrayAssign(ctx, sourceID, isIdentifier ? key : ('"' + quote(key) + '"'), refParam);
+        markRef(ctx, sourceID);
+        createArrayAssign(ctx, sourceID, isIdentifier ? key : ('"' + key + '"'), refParam);
       }
     } else {
       result += (hasPrev ? ',' : '')
-        + (isIdentifier ? key : ('"' + quote(key) + '"'))
+        + (isIdentifier ? key : ('"' + key + '"'))
         + ':' + serializeTree(ctx, val);
       hasPrev = true;
     }
@@ -323,12 +361,14 @@ function serializeAssignments(
   let check: number;
   let parentAssignment: Assignment[];
   let isIdentifier: boolean;
-  for (let i = 0; i < node.s; i++) {
+  const keys = node.k;
+  const values = node.v;
+  for (let i = 0, len = node.s; i < len; i++) {
     parentStack = ctx.stack;
     ctx.stack = [];
-    refParam = serializeTree(ctx, node.v[i]);
+    refParam = serializeTree(ctx, values[i]);
     ctx.stack = parentStack;
-    key = node.k[i];
+    key = keys[i];
     check = Number(key);
     parentAssignment = ctx.assignments;
     ctx.assignments = mainAssignments;
@@ -338,7 +378,7 @@ function serializeAssignments(
     if (isIdentifier && Number.isNaN(check)) {
       createObjectAssign(ctx, sourceID, key, refParam);
     } else {
-      createArrayAssign(ctx, sourceID, isIdentifier ? key : ('"' + quote(key) + '"'), refParam);
+      createArrayAssign(ctx, sourceID, isIdentifier ? key : ('"' + key + '"'), refParam);
     }
     ctx.assignments = parentAssignment;
   }
@@ -359,11 +399,11 @@ function serializeDictionary(
       markRef(ctx, i);
       const assignments = serializeAssignments(ctx, i, d);
       if (assignments) {
-        return '(' + assignRef(ctx, i, init) + ',' + assignments + getRefParam(ctx, i) + ')';
+        return '(' + assignIndexedValue(ctx, i, init) + ',' + assignments + getRefParam(ctx, i) + ')';
       }
     }
   }
-  return assignRef(ctx, i, init);
+  return assignIndexedValue(ctx, i, init);
 }
 
 function serializeNullConstructor(
@@ -377,7 +417,7 @@ function serializeObject(
   ctx: SerializationContext,
   node: SerovalObjectNode,
 ) {
-  return assignRef(ctx, node.i, serializeProperties(ctx, node.i, node.d));
+  return assignIndexedValue(ctx, node.i, serializeProperties(ctx, node.i, node.d));
 }
 
 function serializeSet(
@@ -393,8 +433,9 @@ function serializeSet(
     let hasPrev = false;
     for (let i = 0; i < size; i++) {
       item = node.a[i];
-      if (isReferenceInStack(ctx, item)) {
-        createSetAdd(ctx, node.i, getRefParam(ctx, item.i));
+      if (isIndexedValueInStack(ctx, item)) {
+        markRef(ctx, node.i);
+        createAddAssignment(ctx, node.i, getRefParam(ctx, item.i));
       } else {
         // Push directly
         result += (hasPrev ? ',' : '') + serializeTree(ctx, item);
@@ -406,7 +447,7 @@ function serializeSet(
       serialized += '([' + result + '])';
     }
   }
-  return assignRef(ctx, node.i, serialized);
+  return assignIndexedValue(ctx, node.i, serialized);
 }
 
 function serializeMap(
@@ -427,16 +468,17 @@ function serializeMap(
       // Check if key is a parent
       key = node.d.k[i];
       val = node.d.v[i];
-      if (isReferenceInStack(ctx, key)) {
+      if (isIndexedValueInStack(ctx, key)) {
         // Create reference for the map instance
         keyRef = getRefParam(ctx, key.i);
+        markRef(ctx, node.i);
         // Check if value is a parent
-        if (isReferenceInStack(ctx, val)) {
+        if (isIndexedValueInStack(ctx, val)) {
           valueRef = getRefParam(ctx, val.i);
           // Register an assignment since
           // both key and value are a parent of this
           // Map instance
-          createMapSet(ctx, node.i, keyRef, valueRef);
+          createSetAssignment(ctx, node.i, keyRef, valueRef);
         } else {
           // Reset the stack
           // This is required because the serialized
@@ -445,16 +487,17 @@ function serializeMap(
           // assignment
           parent = ctx.stack;
           ctx.stack = [];
-          createMapSet(ctx, node.i, keyRef, serializeTree(ctx, val));
+          createSetAssignment(ctx, node.i, keyRef, serializeTree(ctx, val));
           ctx.stack = parent;
         }
-      } else if (isReferenceInStack(ctx, val)) {
+      } else if (isIndexedValueInStack(ctx, val)) {
         // Create ref for the Map instance
         valueRef = getRefParam(ctx, val.i);
+        markRef(ctx, node.i);
         // Reset stack for the key serialization
         parent = ctx.stack;
         ctx.stack = [];
-        createMapSet(ctx, node.i, serializeTree(ctx, key), valueRef);
+        createSetAssignment(ctx, node.i, serializeTree(ctx, key), valueRef);
         ctx.stack = parent;
       } else {
         result += (hasPrev ? ',[' : '[') + serializeTree(ctx, key) + ',' + serializeTree(ctx, val) + ']';
@@ -469,7 +512,7 @@ function serializeMap(
       serialized += '([' + result + '])';
     }
   }
-  return assignRef(ctx, node.i, serialized);
+  return assignIndexedValue(ctx, node.i, serialized);
 }
 
 function serializeAggregateError(
@@ -478,7 +521,7 @@ function serializeAggregateError(
 ) {
   // Serialize the required arguments
   ctx.stack.push(node.i);
-  const serialized = 'new AggregateError(' + serializeNodeList(ctx, node) + ',"' + quote(node.m) + '")';
+  const serialized = 'new AggregateError([],"' + node.m + '")';
   ctx.stack.pop();
   // `AggregateError` might've been extended
   // either through class or custom properties
@@ -490,7 +533,7 @@ function serializeError(
   ctx: SerializationContext,
   node: SerovalErrorNode,
 ) {
-  const serialized = 'new ' + node.c + '("' + quote(node.m) + '")';
+  const serialized = 'new ' + node.c + '("' + node.m + '")';
   return serializeDictionary(ctx, node.i, node.d, serialized);
 }
 
@@ -500,7 +543,7 @@ function serializePromise(
 ) {
   let serialized: string;
   // Check if resolved value is a parent expression
-  if (isReferenceInStack(ctx, node.f)) {
+  if (isIndexedValueInStack(ctx, node.f)) {
     // A Promise trick, reference the value
     // inside the `then` expression so that
     // the Promise evaluates after the parent
@@ -518,20 +561,30 @@ function serializePromise(
     // just inline the value/reference here
     serialized = 'Promise.resolve(' + result + ')';
   }
-  return assignRef(ctx, node.i, serialized);
+  return assignIndexedValue(ctx, node.i, serialized);
+}
+
+function serializeArrayBuffer(
+  ctx: SerializationContext,
+  node: SerovalArrayBufferNode,
+) {
+  let result = 'new Uint8Array(';
+  if (node.s.length) {
+    result += '[';
+    for (let i = 0, len = node.s.length; i < len; i++) {
+      result += ((i > 0) ? ',' : '') + node.s[i];
+    }
+    result += ']';
+  }
+  return assignIndexedValue(ctx, node.i, result + ').buffer');
 }
 
 function serializeTypedArray(
   ctx: SerializationContext,
   node: SerovalTypedArrayNode | SerovalBigIntTypedArrayNode,
 ) {
-  let result = '';
-  const isBigInt = node.t === SerovalNodeType.BigIntTypedArray;
-  for (let i = 0, len = node.s.length; i < len; i++) {
-    result += (i !== 0 ? ',' : '') + node.s[i] + (isBigInt ? 'n' : '');
-  }
-  const args = '[' + result + ']' + (node.l !== 0 ? (',' + node.l) : '');
-  return assignRef(ctx, node.i, 'new ' + node.c + '(' + args + ')');
+  const args = serializeTree(ctx, node.f) + ',' + node.b + ',' + node.l;
+  return assignIndexedValue(ctx, node.i, 'new ' + node.c + '(' + args + ')');
 }
 
 function serializeIterable(
@@ -558,6 +611,112 @@ function serializeIterable(
   return serializeDictionary(ctx, node.i, node.d, serialized);
 }
 
+function serializeDate(
+  ctx: SerializationContext,
+  node: SerovalDateNode,
+) {
+  return assignIndexedValue(ctx, node.i, 'new Date("' + node.s + '")');
+}
+
+function serializeRegExp(
+  ctx: SerializationContext,
+  node: SerovalRegExpNode,
+) {
+  return assignIndexedValue(ctx, node.i, '/' + node.c + '/' + node.m);
+}
+
+function serializeURL(
+  ctx: SerializationContext,
+  node: SerovalURLNode,
+) {
+  return assignIndexedValue(ctx, node.i, 'new URL("' + node.s + '")');
+}
+
+function serializeURLSearchParams(
+  ctx: SerializationContext,
+  node: SerovalURLSearchParamsNode,
+) {
+  return assignIndexedValue(ctx, node.i, node.s ? 'new URLSearchParams("' + node.s + '")' : 'new URLSearchParams');
+}
+
+function serializeReference(
+  ctx: SerializationContext,
+  node: SerovalReferenceNode,
+) {
+  return assignIndexedValue(ctx, node.i, GLOBAL_KEY + '.get("' + node.s + '")');
+}
+
+function serializeDataView(
+  ctx: SerializationContext,
+  node: SerovalDataViewNode,
+) {
+  const args = serializeTree(ctx, node.f) + ',' + node.b + ',' + node.l;
+  return assignIndexedValue(ctx, node.i, 'new DataView(' + args + ')');
+}
+
+function serializeBlob(
+  ctx: SerializationContext,
+  node: SerovalBlobNode,
+) {
+  const args = '[' + serializeTree(ctx, node.f) + '],{type:"' + node.c + '"}';
+  return assignIndexedValue(ctx, node.i, 'new Blob(' + args + ')');
+}
+
+function serializeFile(
+  ctx: SerializationContext,
+  node: SerovalFileNode,
+) {
+  const options = '{type:"' + node.c + '",lastModified:' + node.b + '}';
+  const args = '[' + serializeTree(ctx, node.f) + '],"' + node.m + '",' + options;
+  return assignIndexedValue(ctx, node.i, 'new File(' + args + ')');
+}
+
+function serializeHeaders(
+  ctx: SerializationContext,
+  node: SerovalHeadersNode,
+) {
+  return assignIndexedValue(ctx, node.i, 'new Headers(' + serializeProperties(ctx, node.i, node.d) + ')');
+}
+
+function serializeFormDataEntries(
+  ctx: SerializationContext,
+  node: SerovalFormDataNode,
+) {
+  ctx.stack.push(node.i);
+  const mainAssignments: Assignment[] = [];
+  let parentStack: number[];
+  let value: string;
+  let key: string;
+  let parentAssignment: Assignment[];
+  for (let i = 0; i < node.d.s; i++) {
+    parentStack = ctx.stack;
+    ctx.stack = [];
+    value = serializeTree(ctx, node.d.v[i]);
+    key = node.d.k[i];
+    ctx.stack = parentStack;
+    parentAssignment = ctx.assignments;
+    ctx.assignments = mainAssignments;
+    createAppendAssignment(ctx, node.i, '"' + key + '"', value);
+    ctx.assignments = parentAssignment;
+  }
+  ctx.stack.pop();
+  return resolveAssignments(mainAssignments);
+}
+
+function serializeFormData(
+  ctx: SerializationContext,
+  node: SerovalFormDataNode,
+) {
+  if (node.d.s) {
+    markRef(ctx, node.i);
+  }
+  const result = assignIndexedValue(ctx, node.i, 'new FormData()');
+  if (node.d.s) {
+    return '(' + result + ',' + serializeFormDataEntries(ctx, node) + getRefParam(ctx, node.i) + ')';
+  }
+  return result;
+}
+
 export default function serializeTree(
   ctx: SerializationContext,
   node: SerovalNode,
@@ -583,7 +742,7 @@ export default function serializeTree(
       return 'NaN';
     case SerovalNodeType.BigInt:
       return node.s + 'n';
-    case SerovalNodeType.Reference:
+    case SerovalNodeType.IndexedValue:
       return getRefParam(ctx, node.i);
     case SerovalNodeType.Array:
       return serializeArray(ctx, node);
@@ -592,16 +751,20 @@ export default function serializeTree(
     case SerovalNodeType.NullConstructor:
       return serializeNullConstructor(ctx, node);
     case SerovalNodeType.Date:
-      return assignRef(ctx, node.i, 'new Date("' + node.s + '")');
+      return serializeDate(ctx, node);
     case SerovalNodeType.RegExp:
-      return assignRef(ctx, node.i, '/' + node.c + '/' + node.m);
+      return serializeRegExp(ctx, node);
     case SerovalNodeType.Set:
       return serializeSet(ctx, node);
     case SerovalNodeType.Map:
       return serializeMap(ctx, node);
+    case SerovalNodeType.ArrayBuffer:
+      return serializeArrayBuffer(ctx, node);
     case SerovalNodeType.BigIntTypedArray:
     case SerovalNodeType.TypedArray:
       return serializeTypedArray(ctx, node);
+    case SerovalNodeType.DataView:
+      return serializeDataView(ctx, node);
     case SerovalNodeType.AggregateError:
       return serializeAggregateError(ctx, node);
     case SerovalNodeType.Error:
@@ -612,6 +775,20 @@ export default function serializeTree(
       return serializePromise(ctx, node);
     case SerovalNodeType.WKSymbol:
       return SYMBOL_STRING[node.s];
+    case SerovalNodeType.URL:
+      return serializeURL(ctx, node);
+    case SerovalNodeType.URLSearchParams:
+      return serializeURLSearchParams(ctx, node);
+    case SerovalNodeType.Reference:
+      return serializeReference(ctx, node);
+    case SerovalNodeType.Blob:
+      return serializeBlob(ctx, node);
+    case SerovalNodeType.File:
+      return serializeFile(ctx, node);
+    case SerovalNodeType.Headers:
+      return serializeHeaders(ctx, node);
+    case SerovalNodeType.FormData:
+      return serializeFormData(ctx, node);
     default:
       throw new Error('Unsupported type');
   }