grpc-tools-ts/src/lib/format/partial/EnumFormatter.ts

Generate d.ts definitions for generated js files from grpc_tools_node_protoc

import {EnumDescriptorProto} from "google-protobuf/google/protobuf/descriptor_pb";
import {TplEngine} from "../../TplEngine";
import {Utility} from "../../Utility";

export namespace EnumFormatter {

    export function format(enumDescriptor: EnumDescriptorProto, indentLevel: number): string {
        let enumName = enumDescriptor.getName();
        let values: { [key: string]: number } = {};
        enumDescriptor.getValueList().forEach(value => {
            values[value.getName()] = value.getNumber();
        });

        return TplEngine.render('partial/enum', {
            indent: Utility.generateIndent(indentLevel),
            enumName: enumName,
            values: values,
        });
    }

}