ts-proto/build/src/encode.js

[](https://www.npmjs.com/package/ts-proto) [](https://github.com/stephenh/ts-proto/actions)

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.generateEncoder = generateEncoder;
exports.generateDecoder = generateDecoder;
const ts_poet_1 = require("ts-poet");
const types_1 = require("./types");
const options_1 = require("./options");
const utils_1 = require("./utils");
function generateEncoder(ctx, typeName) {
    const name = (0, types_1.wrapperTypeName)(typeName);
    if (!name) {
        return (0, ts_poet_1.code) `${(0, types_1.messageToTypeName)(ctx, typeName, { keepValueType: true })}.encode(value).finish()`;
    }
    if (name == "Timestamp") {
        const TimestampValue = (0, utils_1.impProto)(ctx.options, "google/protobuf/timestamp", (0, utils_1.wrapTypeName)(ctx.options, name));
        let value = (0, ts_poet_1.code) `value`;
        if (ctx.options.useDate === options_1.DateOption.DATE ||
            ctx.options.useDate === options_1.DateOption.STRING ||
            ctx.options.useDate === options_1.DateOption.STRING_NANO ||
            ctx.options.useDate === options_1.DateOption.TEMPORAL) {
            value = (0, ts_poet_1.code) `${ctx.utils.toTimestamp}(${value})`;
        }
        return (0, ts_poet_1.code) `${TimestampValue}.encode(${value}).finish()`;
    }
    if (name == "Struct" || name == "Value") {
        const StructType = (0, utils_1.impProto)(ctx.options, "google/protobuf/struct", (0, utils_1.wrapTypeName)(ctx.options, name));
        return (0, ts_poet_1.code) `${StructType}.encode(${StructType}.wrap(value)).finish()`;
    }
    if (name == "ListValue") {
        const ListValueType = (0, utils_1.impProto)(ctx.options, "google/protobuf/struct", (0, utils_1.wrapTypeName)(ctx.options, name));
        return (0, ts_poet_1.code) `${ListValueType}.encode({values: value ?? []}).finish()`;
    }
    const TypeValue = (0, utils_1.impProto)(ctx.options, "google/protobuf/wrappers", (0, utils_1.wrapTypeName)(ctx.options, name));
    switch (name) {
        case "StringValue":
            return (0, ts_poet_1.code) `${TypeValue}.encode({value: value ?? ""}).finish()`;
        case "Int32Value":
        case "UInt32Value":
        case "DoubleValue":
        case "FloatValue":
            return (0, ts_poet_1.code) `${TypeValue}.encode({value: value ?? 0}).finish()`;
        case "Int64Value":
        case "UInt64Value":
            if (ctx.options.forceLong === options_1.LongOption.LONG) {
                return (0, ts_poet_1.code) `${TypeValue}.encode({value: value ? value.toNumber(): 0}).finish()`;
            }
            return (0, ts_poet_1.code) `${TypeValue}.encode({value: value ?? 0 }).finish()`;
        case "BoolValue":
            return (0, ts_poet_1.code) `${TypeValue}.encode({value: value ?? false}).finish()`;
        case "BytesValue":
            return (0, ts_poet_1.code) `${TypeValue}.encode({value: value ?? new Uint8Array(0)}).finish()`;
    }
    throw new Error(`unknown wrapper type: ${name}`);
}
function generateDecoder(ctx, typeName) {
    const { options } = ctx;
    let name = (0, types_1.wrapperTypeName)(typeName);
    if (!name) {
        return (0, ts_poet_1.code) `${(0, types_1.messageToTypeName)(ctx, typeName, { keepValueType: true })}.decode(value)`;
    }
    let TypeValue;
    if (name == "Timestamp") {
        TypeValue = (0, utils_1.impProto)(ctx.options, "google/protobuf/timestamp", (0, utils_1.wrapTypeName)(ctx.options, name));
        const decoder = (0, ts_poet_1.code) `${TypeValue}.decode(value)`;
        if (ctx.options.useDate === options_1.DateOption.DATE ||
            ctx.options.useDate === options_1.DateOption.STRING ||
            ctx.options.useDate === options_1.DateOption.STRING_NANO ||
            ctx.options.useDate === options_1.DateOption.TEMPORAL) {
            return (0, ts_poet_1.code) `${ctx.utils.fromTimestamp}(${decoder})`;
        }
        return decoder;
    }
    if (name == "Struct" || name == "ListValue" || name == "Value") {
        TypeValue = (0, utils_1.impProto)(ctx.options, "google/protobuf/struct", (0, utils_1.wrapTypeName)(ctx.options, name));
        return (0, ts_poet_1.code) `${TypeValue}.unwrap(${TypeValue}.decode(value))`;
    }
    TypeValue = (0, utils_1.impProto)(ctx.options, "google/protobuf/wrappers", (0, utils_1.wrapTypeName)(ctx.options, name));
    return (0, ts_poet_1.code) `${TypeValue}.decode(value).value`;
}