ts-protoc-gen/lib/service/grpcweb.js

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
var Printer_1 = require("../Printer");
var CodePrinter_1 = require("../CodePrinter");
var common_1 = require("./common");
function generateGrpcWebService(filename, descriptor, exportMap) {
    return [
        common_1.createFile(generateTypeScriptDefinition(descriptor, exportMap), filename + "_service.d.ts"),
        common_1.createFile(generateJavaScript(descriptor, exportMap), filename + "_service.js"),
    ];
}
exports.generateGrpcWebService = generateGrpcWebService;
function generateTypeScriptDefinition(fileDescriptor, exportMap) {
    var serviceDescriptor = new common_1.GrpcServiceDescriptor(fileDescriptor, exportMap);
    var printer = new Printer_1.Printer(0);
    printer.printLn("// package: " + serviceDescriptor.packageName);
    printer.printLn("// file: " + serviceDescriptor.filename);
    printer.printEmptyLn();
    if (serviceDescriptor.services.length === 0) {
        return printer.getOutput();
    }
    serviceDescriptor.imports
        .forEach(function (importDescriptor) {
        printer.printLn("import * as " + importDescriptor.namespace + " from \"" + importDescriptor.path + "\";");
    });
    printer.printLn("import {grpc} from \"@improbable-eng/grpc-web\";");
    printer.printEmptyLn();
    serviceDescriptor.services
        .forEach(function (service) {
        service.methods.forEach(function (method) {
            printer.printLn("type " + method.serviceName + method.nameAsPascalCase + " = {");
            printer.printIndentedLn("readonly methodName: string;");
            printer.printIndentedLn("readonly service: typeof " + method.serviceName + ";");
            printer.printIndentedLn("readonly requestStream: " + method.requestStream + ";");
            printer.printIndentedLn("readonly responseStream: " + method.responseStream + ";");
            printer.printIndentedLn("readonly requestType: typeof " + method.requestType + ";");
            printer.printIndentedLn("readonly responseType: typeof " + method.responseType + ";");
            printer.printLn("};");
            printer.printEmptyLn();
        });
        printer.printLn("export class " + service.name + " {");
        printer.printIndentedLn("static readonly serviceName: string;");
        service.methods.forEach(function (method) {
            printer.printIndentedLn("static readonly " + method.nameAsPascalCase + ": " + method.serviceName + method.nameAsPascalCase + ";");
        });
        printer.printLn("}");
        printer.printEmptyLn();
    });
    printer.printLn("export type ServiceError = { message: string, code: number; metadata: grpc.Metadata }");
    printer.printLn("export type Status = { details: string, code: number; metadata: grpc.Metadata }");
    printer.printEmptyLn();
    printer.printLn("interface UnaryResponse {");
    printer.printIndentedLn("cancel(): void;");
    printer.printLn("}");
    printer.printLn("interface ResponseStream<T> {");
    printer.printIndentedLn("cancel(): void;");
    printer.printIndentedLn("on(type: 'data', handler: (message: T) => void): ResponseStream<T>;");
    printer.printIndentedLn("on(type: 'end', handler: (status?: Status) => void): ResponseStream<T>;");
    printer.printIndentedLn("on(type: 'status', handler: (status: Status) => void): ResponseStream<T>;");
    printer.printLn("}");
    printer.printLn("interface RequestStream<T> {");
    printer.printIndentedLn("write(message: T): RequestStream<T>;");
    printer.printIndentedLn("end(): void;");
    printer.printIndentedLn("cancel(): void;");
    printer.printIndentedLn("on(type: 'end', handler: (status?: Status) => void): RequestStream<T>;");
    printer.printIndentedLn("on(type: 'status', handler: (status: Status) => void): RequestStream<T>;");
    printer.printLn("}");
    printer.printLn("interface BidirectionalStream<ReqT, ResT> {");
    printer.printIndentedLn("write(message: ReqT): BidirectionalStream<ReqT, ResT>;");
    printer.printIndentedLn("end(): void;");
    printer.printIndentedLn("cancel(): void;");
    printer.printIndentedLn("on(type: 'data', handler: (message: ResT) => void): BidirectionalStream<ReqT, ResT>;");
    printer.printIndentedLn("on(type: 'end', handler: (status?: Status) => void): BidirectionalStream<ReqT, ResT>;");
    printer.printIndentedLn("on(type: 'status', handler: (status: Status) => void): BidirectionalStream<ReqT, ResT>;");
    printer.printLn("}");
    printer.printEmptyLn();
    serviceDescriptor.services
        .forEach(function (service) {
        printServiceStubTypes(printer, service);
        printer.printEmptyLn();
    });
    return printer.getOutput();
}
function generateJavaScript(fileDescriptor, exportMap) {
    var serviceDescriptor = new common_1.GrpcServiceDescriptor(fileDescriptor, exportMap);
    var printer = new Printer_1.Printer(0);
    printer.printLn("// package: " + serviceDescriptor.packageName);
    printer.printLn("// file: " + serviceDescriptor.filename);
    printer.printEmptyLn();
    if (serviceDescriptor.services.length === 0) {
        return printer.getOutput();
    }
    serviceDescriptor.imports
        .forEach(function (importDescriptor) {
        printer.printLn("var " + importDescriptor.namespace + " = require(\"" + importDescriptor.path + "\");");
    });
    printer.printLn("var grpc = require(\"@improbable-eng/grpc-web\").grpc;");
    printer.printEmptyLn();
    serviceDescriptor.services
        .forEach(function (service) {
        printer.printLn("var " + service.name + " = (function () {");
        printer.printIndentedLn("function " + service.name + "() {}");
        printer.printIndentedLn(service.name + ".serviceName = \"" + service.qualifiedName + "\";");
        printer.printIndentedLn("return " + service.name + ";");
        printer.printLn("}());");
        printer.printEmptyLn();
        service.methods
            .forEach(function (method) {
            printer.printLn(method.serviceName + "." + method.nameAsPascalCase + " = {");
            printer.printIndentedLn("methodName: \"" + method.nameAsPascalCase + "\",");
            printer.printIndentedLn("service: " + method.serviceName + ",");
            printer.printIndentedLn("requestStream: " + method.requestStream + ",");
            printer.printIndentedLn("responseStream: " + method.responseStream + ",");
            printer.printIndentedLn("requestType: " + method.requestType + ",");
            printer.printIndentedLn("responseType: " + method.responseType);
            printer.printLn("};");
            printer.printEmptyLn();
        });
        printer.printLn("exports." + service.name + " = " + service.name + ";");
        printer.printEmptyLn();
        printServiceStub(printer, service);
        printer.printEmptyLn();
    });
    return printer.getOutput();
}
function printServiceStub(methodPrinter, service) {
    var printer = new CodePrinter_1.CodePrinter(0, methodPrinter);
    printer
        .printLn("function " + service.name + "Client(serviceHost, options) {")
        .indent().printLn("this.serviceHost = serviceHost;")
        .printLn("this.options = options || {};")
        .dedent().printLn("}")
        .printEmptyLn();
    service.methods.forEach(function (method) {
        if (method.requestStream && method.responseStream) {
            printBidirectionalStubMethod(printer, method);
        }
        else if (method.requestStream) {
            printClientStreamStubMethod(printer, method);
        }
        else if (method.responseStream) {
            printServerStreamStubMethod(printer, method);
        }
        else {
            printUnaryStubMethod(printer, method);
        }
        printer.printEmptyLn();
    });
    printer.printLn("exports." + service.name + "Client = " + service.name + "Client;");
}
function printUnaryStubMethod(printer, method) {
    printer
        .printLn(method.serviceName + "Client.prototype." + method.nameAsCamelCase + " = function " + method.functionName + "(requestMessage, metadata, callback) {")
        .indent().printLn("if (arguments.length === 2) {")
        .indent().printLn("callback = arguments[1];")
        .dedent().printLn("}")
        .printLn("var client = grpc.unary(" + method.serviceName + "." + method.nameAsPascalCase + ", {")
        .indent().printLn("request: requestMessage,")
        .printLn("host: this.serviceHost,")
        .printLn("metadata: metadata,")
        .printLn("transport: this.options.transport,")
        .printLn("debug: this.options.debug,")
        .printLn("onEnd: function (response) {")
        .indent().printLn("if (callback) {")
        .indent().printLn("if (response.status !== grpc.Code.OK) {")
        .indent().printLn("var err = new Error(response.statusMessage);")
        .printLn("err.code = response.status;")
        .printLn("err.metadata = response.trailers;")
        .printLn("callback(err, null);")
        .dedent().printLn("} else {")
        .indent().printLn("callback(null, response.message);")
        .dedent().printLn("}")
        .dedent().printLn("}")
        .dedent().printLn("}")
        .dedent().printLn("});")
        .printLn("return {")
        .indent().printLn("cancel: function () {")
        .indent().printLn("callback = null;")
        .printLn("client.close();")
        .dedent().printLn("}")
        .dedent().printLn("};")
        .dedent().printLn("};");
}
function printServerStreamStubMethod(printer, method) {
    printer
        .printLn(method.serviceName + "Client.prototype." + method.nameAsCamelCase + " = function " + method.functionName + "(requestMessage, metadata) {")
        .indent().printLn("var listeners = {")
        .indent().printLn("data: [],")
        .printLn("end: [],")
        .printLn("status: []")
        .dedent().printLn("};")
        .printLn("var client = grpc.invoke(" + method.serviceName + "." + method.nameAsPascalCase + ", {")
        .indent().printLn("request: requestMessage,")
        .printLn("host: this.serviceHost,")
        .printLn("metadata: metadata,")
        .printLn("transport: this.options.transport,")
        .printLn("debug: this.options.debug,")
        .printLn("onMessage: function (responseMessage) {")
        .indent().printLn("listeners.data.forEach(function (handler) {")
        .indent().printLn("handler(responseMessage);")
        .dedent().printLn("});")
        .dedent().printLn("},")
        .printLn("onEnd: function (status, statusMessage, trailers) {")
        .indent().printLn("listeners.status.forEach(function (handler) {")
        .indent().printLn("handler({ code: status, details: statusMessage, metadata: trailers });")
        .dedent().printLn("});")
        .printLn("listeners.end.forEach(function (handler) {")
        .indent().printLn("handler({ code: status, details: statusMessage, metadata: trailers });")
        .dedent().printLn("});")
        .printLn("listeners = null;")
        .dedent().printLn("}")
        .dedent().printLn("});")
        .printLn("return {")
        .indent().printLn("on: function (type, handler) {")
        .indent().printLn("listeners[type].push(handler);")
        .printLn("return this;")
        .dedent().printLn("},")
        .printLn("cancel: function () {")
        .indent().printLn("listeners = null;")
        .printLn("client.close();")
        .dedent().printLn("}")
        .dedent().printLn("};")
        .dedent().printLn("};");
}
function printClientStreamStubMethod(printer, method) {
    printer
        .printLn(method.serviceName + "Client.prototype." + method.nameAsCamelCase + " = function " + method.functionName + "(metadata) {")
        .indent().printLn("var listeners = {")
        .indent().printLn("end: [],")
        .printLn("status: []")
        .dedent().printLn("};")
        .printLn("var client = grpc.client(" + method.serviceName + "." + method.nameAsPascalCase + ", {")
        .indent().printLn("host: this.serviceHost,")
        .printLn("metadata: metadata,")
        .printLn("transport: this.options.transport")
        .dedent().printLn("});")
        .printLn("client.onEnd(function (status, statusMessage, trailers) {")
        .indent().printLn("listeners.status.forEach(function (handler) {")
        .indent().printLn("handler({ code: status, details: statusMessage, metadata: trailers });")
        .dedent().printLn("});")
        .printLn("listeners.end.forEach(function (handler) {")
        .indent().printLn("handler({ code: status, details: statusMessage, metadata: trailers });")
        .dedent().printLn("});")
        .printLn("listeners = null;")
        .dedent().printLn("});")
        .printLn("return {")
        .indent().printLn("on: function (type, handler) {")
        .indent().printLn("listeners[type].push(handler);")
        .printLn("return this;")
        .dedent().printLn("},")
        .printLn("write: function (requestMessage) {")
        .indent().printLn("if (!client.started) {")
        .indent().printLn("client.start(metadata);")
        .dedent().printLn("}")
        .printLn("client.send(requestMessage);")
        .printLn("return this;")
        .dedent().printLn("},")
        .printLn("end: function () {")
        .indent().printLn("client.finishSend();")
        .dedent().printLn("},")
        .printLn("cancel: function () {")
        .indent().printLn("listeners = null;")
        .printLn("client.close();")
        .dedent().printLn("}")
        .dedent().printLn("};")
        .dedent().printLn("};");
}
function printBidirectionalStubMethod(printer, method) {
    printer
        .printLn(method.serviceName + "Client.prototype." + method.nameAsCamelCase + " = function " + method.functionName + "(metadata) {")
        .indent().printLn("var listeners = {")
        .indent().printLn("data: [],")
        .printLn("end: [],")
        .printLn("status: []")
        .dedent().printLn("};")
        .printLn("var client = grpc.client(" + method.serviceName + "." + method.nameAsPascalCase + ", {")
        .indent().printLn("host: this.serviceHost,")
        .printLn("metadata: metadata,")
        .printLn("transport: this.options.transport")
        .dedent().printLn("});")
        .printLn("client.onEnd(function (status, statusMessage, trailers) {")
        .indent().printLn("listeners.status.forEach(function (handler) {")
        .indent().printLn("handler({ code: status, details: statusMessage, metadata: trailers });")
        .dedent().printLn("});")
        .printLn("listeners.end.forEach(function (handler) {")
        .indent().printLn("handler({ code: status, details: statusMessage, metadata: trailers });")
        .dedent().printLn("});")
        .printLn("listeners = null;")
        .dedent().printLn("});")
        .printLn("client.onMessage(function (message) {")
        .indent().printLn("listeners.data.forEach(function (handler) {")
        .indent().printLn("handler(message);")
        .dedent().printLn("})")
        .dedent().printLn("});")
        .printLn("client.start(metadata);")
        .printLn("return {")
        .indent().printLn("on: function (type, handler) {")
        .indent().printLn("listeners[type].push(handler);")
        .printLn("return this;")
        .dedent().printLn("},")
        .printLn("write: function (requestMessage) {")
        .indent().printLn("client.send(requestMessage);")
        .printLn("return this;")
        .dedent().printLn("},")
        .printLn("end: function () {")
        .indent().printLn("client.finishSend();")
        .dedent().printLn("},")
        .printLn("cancel: function () {")
        .indent().printLn("listeners = null;")
        .printLn("client.close();")
        .dedent().printLn("}")
        .dedent().printLn("};")
        .dedent().printLn("};");
}
function printServiceStubTypes(methodPrinter, service) {
    var printer = new CodePrinter_1.CodePrinter(0, methodPrinter);
    printer
        .printLn("export class " + service.name + "Client {")
        .indent().printLn("readonly serviceHost: string;")
        .printEmptyLn()
        .printLn("constructor(serviceHost: string, options?: grpc.RpcOptions);");
    service.methods.forEach(function (method) {
        if (method.requestStream && method.responseStream) {
            printBidirectionalStubMethodTypes(printer, method);
        }
        else if (method.requestStream) {
            printClientStreamStubMethodTypes(printer, method);
        }
        else if (method.responseStream) {
            printServerStreamStubMethodTypes(printer, method);
        }
        else {
            printUnaryStubMethodTypes(printer, method);
        }
    });
    printer.dedent().printLn("}");
}
function printUnaryStubMethodTypes(printer, method) {
    printer
        .printLn(method.nameAsCamelCase + "(")
        .indent().printLn("requestMessage: " + method.requestType + ",")
        .printLn("metadata: grpc.Metadata,")
        .printLn("callback: (error: ServiceError|null, responseMessage: " + method.responseType + "|null) => void")
        .dedent().printLn("): UnaryResponse;")
        .printLn(method.nameAsCamelCase + "(")
        .indent().printLn("requestMessage: " + method.requestType + ",")
        .printLn("callback: (error: ServiceError|null, responseMessage: " + method.responseType + "|null) => void")
        .dedent().printLn("): UnaryResponse;");
}
function printServerStreamStubMethodTypes(printer, method) {
    printer.printLn(method.nameAsCamelCase + "(requestMessage: " + method.requestType + ", metadata?: grpc.Metadata): ResponseStream<" + method.responseType + ">;");
}
function printClientStreamStubMethodTypes(printer, method) {
    printer.printLn(method.nameAsCamelCase + "(metadata?: grpc.Metadata): RequestStream<" + method.requestType + ">;");
}
function printBidirectionalStubMethodTypes(printer, method) {
    printer.printLn(method.nameAsCamelCase + "(metadata?: grpc.Metadata): BidirectionalStream<" + method.requestType + ", " + method.responseType + ">;");
}
//# sourceMappingURL=grpcweb.js.map