Function writeMessage

compile.js:25–61

(ctx, options)

}

function writeMessage(ctx, options) {
    const fields = ctx._proto.fields;

    let code = '\n';

    if (!options.noRead) {
        const readName = `read${ctx._name}`;

        code += `${writeFunctionExport(options, readName)}function ${readName}(pbf, end) {\n`;
        if (fields.length === 0) {
            code += '    pbf.pos = end;\n    return {};\n';
        } else {
            code += `    const obj = ${compileDest(ctx)};\n`;
            code += '    let field;\n';
            code += '    while ((field = pbf.nextField(end))) {\n';

            for (let i = 0; i < fields.length; i++) {
                code += `        ${i ? 'else ' : ''}if (field === ${fields[i].tag}) ${compileFieldRead(ctx, fields[i])}\n`;
            }
            code += '    }\n    return obj;\n';
        }
        code += '}\n';
    }

    if (!options.noWrite) {
        const writeName = `write${ctx._name}`;
        code += `${writeFunctionExport(options, writeName)}function ${writeName}(obj, pbf) {`;
        if (fields.length === 0) {
            code += '}\n';
        } else {
            code += '\n';
            for (const field of fields) code += compileFieldWriteLine(ctx, field);
            code += '}\n';
        }
    }
    return code;
}

function compileFieldWriteLine(ctx, field) {
    const v = `obj.${field.name}`;