protoc-gen-grpc/src/lib/descriptor/partial/Extensions.ts

Protocol Buffers Compiler (protoc) plugin for generating grpc interfaces in TypeScript..

import {FieldDescriptorProto} from 'google-protobuf/google/protobuf/descriptor_pb';
import {Printer} from '../../Printer';
import {EntryMap} from '../../EntryMap';
import {Utility} from '../../Utility';
import {FieldTypes} from './FieldTypes';

export namespace Extension {

    export function print(fileName: string, entryMap: EntryMap, fieldsDescriptor: FieldDescriptorProto, indentLevel: number): string {

        const extensionName = Utility.snakeToCamel(fieldsDescriptor.getName());
        const fieldType = FieldTypes.getFieldType(fieldsDescriptor.getType(), fieldsDescriptor.getTypeName().slice(1), fileName, entryMap);

        const printer = new Printer(indentLevel + 1);
        printer.printEmptyLn();
        printer.printLn(`export const ${extensionName}: jspb.ExtensionFieldInfo<${fieldType}>;`);
        return printer.output;
    }

}