ts-proto/build/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.generateDecoder = exports.generateEncoder = void 0;
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 = types_1.wrapperTypeName(typeName);
    if (!name) {
        return ts_poet_1.code `${types_1.messageToTypeName(ctx, typeName, { keepValueType: true })}.encode(value).finish()`;
    }
    if (name == 'Timestamp') {
        const TimestampValue = utils_1.impProto(ctx.options, 'google/protobuf/timestamp', name);
        return ts_poet_1.code `${TimestampValue}.encode(${ctx.utils.toTimestamp}(value)).finish()`;
    }
    if (name == 'Struct') {
        const StructType = utils_1.impProto(ctx.options, 'google/protobuf/struct', name);
        return ts_poet_1.code `${StructType}.encode(${StructType}.wrap(value)).finish()`;
    }
    if (name == 'ListValue') {
        const ListValueType = utils_1.impProto(ctx.options, 'google/protobuf/struct', name);
        return ts_poet_1.code `${ListValueType}.encode({values: value ?? []}).finish()`;
    }
    const TypeValue = utils_1.impProto(ctx.options, 'google/protobuf/wrappers', name);
    switch (name) {
        case 'StringValue':
            return ts_poet_1.code `${TypeValue}.encode({value: value ?? ""}).finish()`;
        case 'Int32Value':
        case 'UInt32Value':
        case 'DoubleValue':
        case 'FloatValue':
            return ts_poet_1.code `${TypeValue}.encode({value: value ?? 0}).finish()`;
        case 'Int64Value':
        case 'UInt64Value':
            if (ctx.options.forceLong === options_1.LongOption.LONG) {
                return ts_poet_1.code `${TypeValue}.encode({value: value ? value.toNumber(): 0}).finish()`;
            }
            return ts_poet_1.code `${TypeValue}.encode({value: value ?? 0 }).finish()`;
        case 'BoolValue':
            return ts_poet_1.code `${TypeValue}.encode({value: value ?? false}).finish()`;
        case 'BytesValue':
            return ts_poet_1.code `${TypeValue}.encode({value: value ?? new Uint8Array()}).finish()`;
    }
    throw new Error(`unknown wrapper type: ${name}`);
}
exports.generateEncoder = generateEncoder;
function generateDecoder(ctx, typeName) {
    const { options } = ctx;
    let name = types_1.wrapperTypeName(typeName);
    if (!name) {
        return ts_poet_1.code `${types_1.messageToTypeName(ctx, typeName, { keepValueType: true })}.decode(value)`;
    }
    let TypeValue;
    if (name == 'Timestamp') {
        TypeValue = utils_1.impProto(ctx.options, 'google/protobuf/timestamp', name);
        return ts_poet_1.code `${TypeValue}.decode(value)`;
    }
    if (name == 'Struct' || name == 'ListValue') {
        TypeValue = utils_1.impProto(ctx.options, 'google/protobuf/struct', name);
        return ts_poet_1.code `${TypeValue}.unwrap(${TypeValue}.decode(value))`;
    }
    TypeValue = utils_1.impProto(ctx.options, 'google/protobuf/wrappers', name);
    return ts_poet_1.code `${TypeValue}.decode(value).value`;
}
exports.generateDecoder = generateDecoder;