ts-protoc-gen/lib/ts/message.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var util_1 = require("../util");
var descriptor_pb_1 = require("google-protobuf/google/protobuf/descriptor_pb");
var FieldTypes_1 = require("./FieldTypes");
var Printer_1 = require("../Printer");
var enum_1 = require("./enum");
var oneof_1 = require("./oneof");
var extensions_1 = require("./extensions");
var JSType = descriptor_pb_1.FieldOptions.JSType;
function hasFieldPresence(field, fileDescriptor) {
    if (field.getLabel() === descriptor_pb_1.FieldDescriptorProto.Label.LABEL_REPEATED) {
        return false;
    }
    if (field.hasOneofIndex()) {
        return true;
    }
    if (field.getType() === FieldTypes_1.MESSAGE_TYPE) {
        return true;
    }
    if (util_1.isProto2(fileDescriptor)) {
        return true;
    }
    if (field.getProto3Optional()) {
        return true;
    }
    return false;
}
function jsGetterName(name) {
    return name === "Extension" || name === "JsPbMessageId" ? name + "$" : name;
}
function printMessage(fileName, exportMap, messageDescriptor, indentLevel, fileDescriptor) {
    var messageName = messageDescriptor.getName();
    var messageOptions = messageDescriptor.getOptions();
    if (messageOptions !== undefined && messageOptions.getMapEntry()) {
        return "";
    }
    var objectTypeName = "AsObject";
    var toObjectType = new Printer_1.Printer(indentLevel + 1);
    toObjectType.printLn("export type " + objectTypeName + " = {");
    var printer = new Printer_1.Printer(indentLevel);
    printer.printEmptyLn();
    printer.printLn("export class " + messageName + " extends jspb.Message {");
    var oneOfGroups = [];
    var syntheticOneOfGroups = [];
    messageDescriptor.getFieldList().forEach(function (field) {
        if (field.hasOneofIndex()) {
            var oneOfIndex = field.getOneofIndex();
            if (oneOfIndex === undefined) {
                util_1.throwError("Missing one_of index");
            }
            else if (field.getProto3Optional()) {
                syntheticOneOfGroups[oneOfIndex] = true;
            }
            else {
                var existing = oneOfGroups[oneOfIndex];
                if (existing === undefined) {
                    existing = [];
                    oneOfGroups[oneOfIndex] = existing;
                }
                existing.push(field);
            }
        }
        var fieldName = field.getName() || util_1.throwError("Missing field name");
        var snakeCaseName = util_1.stripPrefix(fieldName.toLowerCase(), "_");
        var camelCaseName = util_1.snakeToCamel(snakeCaseName);
        var withUppercase = util_1.uppercaseFirst(camelCaseName);
        var type = field.getType() || util_1.throwError("Missing field type");
        var exportType;
        if (type === FieldTypes_1.MESSAGE_TYPE) {
            var fieldTypeName = field.getTypeName() || util_1.throwError("Missing field type name for message field: " + fieldName);
            var fullTypeName = fieldTypeName.slice(1);
            var fieldMessageType = exportMap.getMessage(fullTypeName);
            if (fieldMessageType === undefined) {
                throw new Error("No message export for: " + fullTypeName);
            }
            if (fieldMessageType.messageOptions !== undefined && fieldMessageType.messageOptions.getMapEntry()) {
                var keyTuple = fieldMessageType.mapFieldOptions.key;
                var keyType = keyTuple[0];
                var keyTypeName = FieldTypes_1.getFieldType(keyType, keyTuple[1], fileName, exportMap);
                var valueTuple = fieldMessageType.mapFieldOptions.value;
                var valueType = valueTuple[0];
                var valueTypeName = FieldTypes_1.getFieldType(valueType, valueTuple[1], fileName, exportMap);
                if (valueType === FieldTypes_1.BYTES_TYPE) {
                    valueTypeName = "Uint8Array | string";
                }
                if (valueType === FieldTypes_1.ENUM_TYPE) {
                    valueTypeName = valueTypeName + "[keyof " + valueTypeName + "]";
                }
                printer.printIndentedLn("get" + withUppercase + "Map(): jspb.Map<" + keyTypeName + ", " + valueTypeName + ">;");
                printer.printIndentedLn("clear" + withUppercase + "Map(): void;");
                toObjectType.printIndentedLn(camelCaseName + "Map: Array<[" + keyTypeName + (keyType === FieldTypes_1.MESSAGE_TYPE ? ".AsObject" : "") + ", " + valueTypeName + (valueType === FieldTypes_1.MESSAGE_TYPE ? ".AsObject" : "") + "]>,");
                return;
            }
            var withinNamespace = util_1.withinNamespaceFromExportEntry(fullTypeName, fieldMessageType);
            if (fieldMessageType.fileName === fileName) {
                exportType = withinNamespace;
            }
            else {
                exportType = util_1.filePathToPseudoNamespace(fieldMessageType.fileName) + "." + withinNamespace;
            }
        }
        else if (type === FieldTypes_1.ENUM_TYPE) {
            var fieldTypeName = field.getTypeName() || util_1.throwError("Missing field type name for message field: " + fieldName);
            var fullTypeName = fieldTypeName.slice(1);
            var fieldEnumType = exportMap.getEnum(fullTypeName);
            if (fieldEnumType === undefined) {
                throw new Error("No enum export for: " + fullTypeName);
            }
            var withinNamespace = util_1.withinNamespaceFromExportEntry(fullTypeName, fieldEnumType);
            if (fieldEnumType.fileName === fileName) {
                exportType = withinNamespace;
            }
            else {
                exportType = util_1.filePathToPseudoNamespace(fieldEnumType.fileName) + "." + withinNamespace;
            }
            exportType = exportType + "Map[keyof " + exportType + "Map]";
        }
        else {
            var fieldOptions = field.getOptions();
            if (fieldOptions && fieldOptions.hasJstype()) {
                switch (fieldOptions.getJstype()) {
                    case JSType.JS_NUMBER:
                        exportType = "number";
                        break;
                    case JSType.JS_STRING:
                        exportType = "string";
                        break;
                    default:
                        exportType = FieldTypes_1.getTypeName(type);
                }
            }
            else {
                exportType = FieldTypes_1.getTypeName(type);
            }
        }
        var hasClearMethod = false;
        function printClearIfNotPresent() {
            if (!hasClearMethod) {
                hasClearMethod = true;
                printer.printIndentedLn("clear" + jsGetterName(withUppercase) + (field.getLabel() === descriptor_pb_1.FieldDescriptorProto.Label.LABEL_REPEATED ? "List" : "") + "(): void;");
            }
        }
        if (hasFieldPresence(field, fileDescriptor)) {
            printer.printIndentedLn("has" + jsGetterName(withUppercase) + "(): boolean;");
            printClearIfNotPresent();
        }
        function printRepeatedAddMethod(valueType) {
            var optionalValue = field.getType() === FieldTypes_1.MESSAGE_TYPE;
            printer.printIndentedLn("add" + withUppercase + "(value" + (optionalValue ? "?" : "") + ": " + valueType + ", index?: number): " + valueType + ";");
        }
        if (field.getLabel() === descriptor_pb_1.FieldDescriptorProto.Label.LABEL_REPEATED) {
            printClearIfNotPresent();
            if (type === FieldTypes_1.BYTES_TYPE) {
                toObjectType.printIndentedLn(camelCaseName + "List: Array<Uint8Array | string>,");
                printer.printIndentedLn("get" + withUppercase + "List(): Array<Uint8Array | string>;");
                printer.printIndentedLn("get" + withUppercase + "List_asU8(): Array<Uint8Array>;");
                printer.printIndentedLn("get" + withUppercase + "List_asB64(): Array<string>;");
                printer.printIndentedLn("set" + withUppercase + "List(value: Array<Uint8Array | string>): void;");
                printRepeatedAddMethod("Uint8Array | string");
            }
            else {
                toObjectType.printIndentedLn(camelCaseName + "List: Array<" + exportType + (type === FieldTypes_1.MESSAGE_TYPE ? ".AsObject" : "") + ">,");
                printer.printIndentedLn("get" + withUppercase + "List(): Array<" + exportType + ">;");
                printer.printIndentedLn("set" + withUppercase + "List(value: Array<" + exportType + ">): void;");
                printRepeatedAddMethod(exportType);
            }
        }
        else {
            if (type === FieldTypes_1.BYTES_TYPE) {
                toObjectType.printIndentedLn(camelCaseName + ": Uint8Array | string,");
                printer.printIndentedLn("get" + jsGetterName(withUppercase) + "(): Uint8Array | string;");
                printer.printIndentedLn("get" + withUppercase + "_asU8(): Uint8Array;");
                printer.printIndentedLn("get" + withUppercase + "_asB64(): string;");
                printer.printIndentedLn("set" + jsGetterName(withUppercase) + "(value: Uint8Array | string): void;");
            }
            else {
                var fieldObjectType = exportType;
                var canBeUndefined = false;
                if (type === FieldTypes_1.MESSAGE_TYPE) {
                    fieldObjectType += ".AsObject";
                    if (!util_1.isProto2(fileDescriptor) || (field.getLabel() === descriptor_pb_1.FieldDescriptorProto.Label.LABEL_OPTIONAL)) {
                        canBeUndefined = true;
                    }
                }
                else {
                    if (util_1.isProto2(fileDescriptor)) {
                        canBeUndefined = true;
                    }
                }
                var fieldObjectName = util_1.normaliseFieldObjectName(camelCaseName);
                toObjectType.printIndentedLn("" + fieldObjectName + (canBeUndefined ? "?" : "") + ": " + fieldObjectType + ",");
                printer.printIndentedLn("get" + jsGetterName(withUppercase) + "(): " + exportType + (canBeUndefined ? " | undefined" : "") + ";");
                printer.printIndentedLn("set" + jsGetterName(withUppercase) + "(value" + (type === FieldTypes_1.MESSAGE_TYPE ? "?" : "") + ": " + exportType + "): void;");
            }
        }
        printer.printEmptyLn();
    });
    toObjectType.printLn("}");
    messageDescriptor.getOneofDeclList().forEach(function (oneOfDecl, index) {
        var oneOfDeclName = oneOfDecl.getName() || util_1.throwError("Missing one_of name");
        if (!syntheticOneOfGroups[index]) {
            printer.printIndentedLn("get" + util_1.oneOfName(oneOfDeclName) + "Case(): " + messageName + "." + util_1.oneOfName(oneOfDeclName) + "Case;");
        }
    });
    printer.printIndentedLn("serializeBinary(): Uint8Array;");
    printer.printIndentedLn("toObject(includeInstance?: boolean): " + messageName + "." + objectTypeName + ";");
    printer.printIndentedLn("static toObject(includeInstance: boolean, msg: " + messageName + "): " + messageName + "." + objectTypeName + ";");
    printer.printIndentedLn("static extensions: {[key: number]: jspb.ExtensionFieldInfo<jspb.Message>};");
    printer.printIndentedLn("static extensionsBinary: {[key: number]: jspb.ExtensionFieldBinaryInfo<jspb.Message>};");
    printer.printIndentedLn("static serializeBinaryToWriter(message: " + messageName + ", writer: jspb.BinaryWriter): void;");
    printer.printIndentedLn("static deserializeBinary(bytes: Uint8Array): " + messageName + ";");
    printer.printIndentedLn("static deserializeBinaryFromReader(message: " + messageName + ", reader: jspb.BinaryReader): " + messageName + ";");
    printer.printLn("}");
    printer.printEmptyLn();
    printer.printLn("export namespace " + messageName + " {");
    printer.print(toObjectType.getOutput());
    messageDescriptor.getNestedTypeList().forEach(function (nested) {
        var msgOutput = printMessage(fileName, exportMap, nested, indentLevel + 1, fileDescriptor);
        if (msgOutput !== "") {
            printer.print(msgOutput);
        }
    });
    messageDescriptor.getEnumTypeList().forEach(function (enumType) {
        printer.print("" + enum_1.printEnum(enumType, indentLevel + 1));
    });
    messageDescriptor.getOneofDeclList().forEach(function (oneOfDecl, index) {
        if (!syntheticOneOfGroups[index]) {
            printer.print("" + oneof_1.printOneOfDecl(oneOfDecl, oneOfGroups[index] || [], indentLevel + 1));
        }
    });
    messageDescriptor.getExtensionList().forEach(function (extension) {
        printer.print(extensions_1.printExtension(fileName, exportMap, extension, indentLevel + 1));
    });
    printer.printLn("}");
    return printer.getOutput();
}
exports.printMessage = printMessage;
//# sourceMappingURL=message.js.map