ts-proto 1.144.1 → 1.145.0

package/build/main.js

@@ -957,10 +957,10 @@ function generateEncode(ctx, fullName, messageDesc) {
               }
             `
                     : writeSnippet(`{ ${maybeTypeField} key: key as any, value }`);
-                const optionalAlternative = isOptional ? " || {}" : "";
+                const optionalAlternative = isOptional ? (ctx.options.useMapType ? " || new Map()" : " || {}") : "";
                 if (ctx.options.useMapType) {
                     chunks.push((0, ts_poet_1.code) `
-            message.${fieldName}${optionalAlternative}.forEach((value, key) => {
+            (message.${fieldName}${optionalAlternative}).forEach((value, key) => {
               ${entryWriteSnippet}
             });
           `);
@@ -1212,6 +1212,9 @@ function generateFromJson(ctx, fullName, fullTypeName, messageDesc) {
                         else if ((0, types_1.isLong)(valueField) && options.forceLong === options_1.LongOption.LONG) {
                             return (0, ts_poet_1.code) `Long.fromValue(${from} as Long | string)`;
                         }
+                        else if ((0, types_1.isLong)(valueField) && options.forceLong === options_1.LongOption.BIGINT) {
+                            return (0, ts_poet_1.code) `BigInt(${from} as string | number | bigint | boolean)`;
+                        }
                         else if ((0, types_1.isEnum)(valueField)) {
                             const fromJson = (0, types_1.getEnumMethod)(ctx, valueField.typeName, "FromJSON");
                             return (0, ts_poet_1.code) `${fromJson}(${from})`;
@@ -1250,6 +1253,7 @@ function generateFromJson(ctx, fullName, fullTypeName, messageDesc) {
                 throw new Error(`Unhandled field ${field}`);
             }
         };
+        const noDefaultValue = !options.initializeFieldsAsUndefined && (0, types_1.isOptionalProperty)(field, messageDesc.options, options);
         // and then use the snippet to handle repeated fields if necessary
         if ((_a = canonicalFromJson[fullTypeName]) === null || _a === void 0 ? void 0 : _a[fieldName]) {
             chunks.push((0, ts_poet_1.code) `${fieldName}: ${canonicalFromJson[fullTypeName][fieldName]("object")},`);
@@ -1257,29 +1261,32 @@ function generateFromJson(ctx, fullName, fullTypeName, messageDesc) {
         else if ((0, types_1.isRepeated)(field)) {
             if ((0, types_1.isMapType)(ctx, messageDesc, field)) {
                 const fieldType = (0, types_1.toTypeName)(ctx, messageDesc, field);
-                const i = maybeCastToNumber(ctx, messageDesc, field, "key");
+                const i = convertFromObjectKey(ctx, messageDesc, field, "key");
                 if (ctx.options.useMapType) {
+                    const fallback = noDefaultValue ? "undefined" : "new Map()";
                     chunks.push((0, ts_poet_1.code) `
             ${fieldName}: ${ctx.utils.isObject}(${jsonProperty})
               ? Object.entries(${jsonProperty}).reduce<${fieldType}>((acc, [key, value]) => {
                   acc.set(${i}, ${readSnippet("value")});
                   return acc;
                 }, new Map())
-              : new Map(),
+              : ${fallback},
           `);
                 }
                 else {
+                    const fallback = noDefaultValue ? "undefined" : "{}";
                     chunks.push((0, ts_poet_1.code) `
             ${fieldName}: ${ctx.utils.isObject}(${jsonProperty})
               ? Object.entries(${jsonProperty}).reduce<${fieldType}>((acc, [key, value]) => {
                   acc[${i}] = ${readSnippet("value")};
                   return acc;
                 }, {})
-              : {},
+              : ${fallback},
           `);
                 }
             }
             else {
+                const fallback = noDefaultValue ? "undefined" : "[]";
                 const readValueSnippet = readSnippet("e");
                 if (readValueSnippet.toString() === (0, ts_poet_1.code) `e`.toString()) {
                     chunks.push((0, ts_poet_1.code) `${fieldName}: Array.isArray(${jsonPropertyOptional}) ? [...${jsonProperty}] : [],`);
@@ -1287,7 +1294,7 @@ function generateFromJson(ctx, fullName, fullTypeName, messageDesc) {
                 else {
                     // Explicit `any` type required to make TS with noImplicitAny happy. `object` is also `any` here.
                     chunks.push((0, ts_poet_1.code) `
-            ${fieldName}: Array.isArray(${jsonPropertyOptional}) ? ${jsonProperty}.map((e: any) => ${readValueSnippet}): [],
+            ${fieldName}: Array.isArray(${jsonPropertyOptional}) ? ${jsonProperty}.map((e: any) => ${readValueSnippet}): ${fallback},
           `);
                 }
             }
@@ -1326,7 +1333,7 @@ function generateFromJson(ctx, fullName, fullTypeName, messageDesc) {
       `);
         }
         else {
-            const fallback = (0, types_1.isWithinOneOf)(field) ? "undefined" : (0, types_1.defaultValue)(ctx, field);
+            const fallback = (0, types_1.isWithinOneOf)(field) || noDefaultValue ? "undefined" : (0, types_1.defaultValue)(ctx, field);
             chunks.push((0, ts_poet_1.code) `
         ${fieldName}: ${ctx.utils.isSet}(${jsonProperty})
           ? ${readSnippet(`${jsonProperty}`)}
@@ -1463,12 +1470,13 @@ function generateToJson(ctx, fullName, fullProtobufTypeName, messageDesc) {
         };
         if ((0, types_1.isMapType)(ctx, messageDesc, field)) {
             // Maps might need their values transformed, i.e. bytes --> base64
+            const i = convertToObjectKey(ctx, messageDesc, field, "k");
             if (ctx.options.useMapType) {
                 chunks.push((0, ts_poet_1.code) `
           ${jsonProperty} = {};
           if (message.${fieldName}) {
             message.${fieldName}.forEach((v, k) => {
-              ${jsonProperty}[k] = ${readSnippet("v")};
+              ${jsonProperty}[${i}] = ${readSnippet("v")};
             });
           }
         `);
@@ -1478,7 +1486,7 @@ function generateToJson(ctx, fullName, fullProtobufTypeName, messageDesc) {
           ${jsonProperty} = {};
           if (message.${fieldName}) {
             Object.entries(message.${fieldName}).forEach(([k, v]) => {
-              ${jsonProperty}[k] = ${readSnippet("v")};
+              ${jsonProperty}[${i}] = ${readSnippet("v")};
             });
           }
         `);
@@ -1572,6 +1580,9 @@ function generateFromPartial(ctx, fullName, messageDesc) {
                         else if ((0, types_1.isLong)(valueField) && options.forceLong === options_1.LongOption.LONG) {
                             return (0, ts_poet_1.code) `Long.fromValue(${from})`;
                         }
+                        else if ((0, types_1.isLong)(valueField) && options.forceLong === options_1.LongOption.BIGINT) {
+                            return (0, ts_poet_1.code) `BigInt(${from} as string | number | bigint | boolean)`;
+                        }
                         else {
                             const cstr = (0, case_1.capitalize)(valueType.toCodeString([]));
                             return (0, ts_poet_1.code) `${cstr}(${from})`;
@@ -1607,18 +1618,22 @@ function generateFromPartial(ctx, fullName, messageDesc) {
                 throw new Error(`Unhandled field ${field}`);
             }
         };
+        const noDefaultValue = !options.initializeFieldsAsUndefined && (0, types_1.isOptionalProperty)(field, messageDesc.options, options);
         // and then use the snippet to handle repeated fields if necessary
         if ((0, types_1.isRepeated)(field)) {
             if ((0, types_1.isMapType)(ctx, messageDesc, field)) {
                 const fieldType = (0, types_1.toTypeName)(ctx, messageDesc, field);
-                const i = maybeCastToNumber(ctx, messageDesc, field, "key");
+                const i = convertFromObjectKey(ctx, messageDesc, field, "key");
+                const noValueSnippet = noDefaultValue
+                    ? `(object.${fieldName} === undefined || object.${fieldName} === null) ? undefined : `
+                    : "";
                 if (ctx.options.useMapType) {
                     chunks.push((0, ts_poet_1.code) `
-            message.${fieldName} = (() => {
+            message.${fieldName} = ${noValueSnippet} (() => {
               const m = new Map();
               (object.${fieldName} as ${fieldType} ?? new Map()).forEach((value, key) => {
                 if (value !== undefined) {
-                  m.set(${i}, ${readSnippet("value")});
+                  m.set(key, ${readSnippet("value")});
                 }
               });
               return m;
@@ -1627,7 +1642,7 @@ function generateFromPartial(ctx, fullName, messageDesc) {
                 }
                 else {
                     chunks.push((0, ts_poet_1.code) `
-            message.${fieldName} = Object.entries(object.${fieldName} ?? {}).reduce<${fieldType}>((acc, [key, value]) => {
+            message.${fieldName} = ${noValueSnippet} Object.entries(object.${fieldName} ?? {}).reduce<${fieldType}>((acc, [key, value]) => {
               if (value !== undefined) {
                 acc[${i}] = ${readSnippet("value")};
               }
@@ -1637,8 +1652,9 @@ function generateFromPartial(ctx, fullName, messageDesc) {
                 }
             }
             else {
+                const fallback = noDefaultValue ? "undefined" : "[]";
                 chunks.push((0, ts_poet_1.code) `
-          message.${fieldName} = object.${fieldName}?.map((e) => ${readSnippet("e")}) || [];
+          message.${fieldName} = object.${fieldName}?.map((e) => ${readSnippet("e")}) || ${fallback};
         `);
             }
         }
@@ -1657,11 +1673,11 @@ function generateFromPartial(ctx, fullName, messageDesc) {
         }
         else if (readSnippet(`x`).toCodeString([]) == "x") {
             // An optimized case of the else below that works when `readSnippet` returns the plain input
-            const fallback = (0, types_1.isWithinOneOf)(field) ? "undefined" : (0, types_1.defaultValue)(ctx, field);
+            const fallback = (0, types_1.isWithinOneOf)(field) || noDefaultValue ? "undefined" : (0, types_1.defaultValue)(ctx, field);
             chunks.push((0, ts_poet_1.code) `message.${fieldName} = object.${fieldName} ?? ${fallback};`);
         }
         else {
-            const fallback = (0, types_1.isWithinOneOf)(field) ? "undefined" : (0, types_1.defaultValue)(ctx, field);
+            const fallback = (0, types_1.isWithinOneOf)(field) || noDefaultValue ? "undefined" : (0, types_1.defaultValue)(ctx, field);
             chunks.push((0, ts_poet_1.code) `
         message.${fieldName} = (object.${fieldName} !== undefined && object.${fieldName} !== null)
           ? ${readSnippet(`object.${fieldName}`)}
@@ -1675,13 +1691,47 @@ function generateFromPartial(ctx, fullName, messageDesc) {
     return (0, ts_poet_1.joinCode)(chunks, { on: "\n" });
 }
 exports.contextTypeVar = "Context extends DataLoaders";
-function maybeCastToNumber(ctx, messageDesc, field, variableName) {
-    const { keyType } = (0, types_1.detectMapType)(ctx, messageDesc, field);
+function convertFromObjectKey(ctx, messageDesc, field, variableName) {
+    const { keyType, keyField } = (0, types_1.detectMapType)(ctx, messageDesc, field);
     if (keyType.toCodeString([]) === "string") {
-        return variableName;
+        return (0, ts_poet_1.code) `${variableName}`;
+    }
+    else if ((0, types_1.isLong)(keyField) && ctx.options.useMapType) {
+        if (ctx.options.forceLong === options_1.LongOption.LONG) {
+            return (0, ts_poet_1.code) `${(0, case_1.capitalize)(keyType.toCodeString([]))}.fromValue(${variableName})`;
+        }
+        else if (ctx.options.forceLong === options_1.LongOption.BIGINT) {
+            return (0, ts_poet_1.code) `BigInt(${variableName})`;
+        }
+        else if (ctx.options.forceLong === options_1.LongOption.STRING) {
+            return (0, ts_poet_1.code) `String(${variableName})`;
+        }
+        else {
+            return (0, ts_poet_1.code) `Number(${variableName})`;
+        }
+    }
+    else {
+        return (0, ts_poet_1.code) `Number(${variableName})`;
+    }
+}
+function convertToObjectKey(ctx, messageDesc, field, variableName) {
+    const { keyType, keyField } = (0, types_1.detectMapType)(ctx, messageDesc, field);
+    if (keyType.toCodeString([]) === "string") {
+        return (0, ts_poet_1.code) `${variableName}`;
+    }
+    else if ((0, types_1.isLong)(keyField) && ctx.options.useMapType) {
+        if (ctx.options.forceLong === options_1.LongOption.LONG) {
+            return (0, ts_poet_1.code) `${ctx.utils.longToNumber}(${variableName})`;
+        }
+        else if (ctx.options.forceLong === options_1.LongOption.BIGINT) {
+            return (0, ts_poet_1.code) `${variableName}.toString()`;
+        }
+        else {
+            return (0, ts_poet_1.code) `${variableName}`;
+        }
     }
     else {
-        return `Number(${variableName})`;
+        return (0, ts_poet_1.code) `${variableName}`;
     }
 }
 function maybeReadonly(options) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ts-proto",
-  "version": "1.144.1",
+  "version": "1.145.0",
   "description": "",
   "main": "build/plugin.js",
   "repository": "github:stephenh/ts-proto",