359 lines
18 KiB
JavaScript
359 lines
18 KiB
JavaScript
|
"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
|