ts-proto 1.167.2 → 1.167.3

package/build/generate-services.js

@@ -98,11 +98,11 @@ function generateRegularRpcMethod(ctx, methodDesc) {
     }
     let encode = (0, ts_poet_1.code) `${rawInputType}.encode(request).finish()`;
     let beforeRequest;
-    if (options.rpcBeforeRequest) {
-        beforeRequest = (0, ts_poet_1.code) `
-    if (this.rpc.beforeRequest) {
-      this.rpc.beforeRequest(this.service, "${methodDesc.name}", request);
-    }`;
+    if (options.rpcBeforeRequest && !methodDesc.clientStreaming) {
+        beforeRequest = generateBeforeRequest(methodDesc.name);
+    }
+    else if (methodDesc.clientStreaming && options.rpcBeforeRequest) {
+        encode = (0, ts_poet_1.code) `{const encodedRequest = ${encode}; ${generateBeforeRequest(methodDesc.name, "encodedRequest")}; return encodedRequest}`;
     }
     let decode = (0, ts_poet_1.code) `${rawOutputType}.decode(${Reader}.create(data))`;
     if (options.rpcAfterResponse) {
@@ -162,28 +162,35 @@ function generateRegularRpcMethod(ctx, methodDesc) {
     }
   `;
 }
+function generateBeforeRequest(methodName, requestVariableName = "request") {
+    return (0, ts_poet_1.code) `
+    if (this.rpc.beforeRequest) {
+      this.rpc.beforeRequest(this.service, "${methodName}", ${requestVariableName});
+    }`;
+}
 function createDefaultServiceReturn(ctx, methodDesc, decode, errorHandler) {
     const { options } = ctx;
     const rawOutputType = (0, types_1.responseType)(ctx, methodDesc, { keepValueType: true });
+    const returnStatement = (0, utils_1.arrowFunction)("data", decode, !options.rpcAfterResponse);
     if (options.returnObservable || methodDesc.serverStreaming) {
         if (options.useAsyncIterable) {
             return (0, ts_poet_1.code) `${rawOutputType}.decodeTransform(result)`;
         }
         else {
-            return (0, ts_poet_1.code) `result.pipe(${(0, ts_poet_1.imp)("map@rxjs/operators")}(data => ${decode}))`;
+            if (errorHandler) {
+                const tc = (0, utils_1.arrowFunction)("data", (0, utils_1.tryCatchBlock)(decode, (0, ts_poet_1.code) `throw error`), !options.rpcAfterResponse);
+                return (0, ts_poet_1.code) `result.pipe(${(0, ts_poet_1.imp)("map@rxjs/operators")}(${tc}))`;
+            }
+            return (0, ts_poet_1.code) `result.pipe(${(0, ts_poet_1.imp)("map@rxjs/operators")}(${returnStatement}))`;
         }
     }
     if (errorHandler) {
-        let tryBlock = decode;
         if (!options.rpcAfterResponse) {
-            tryBlock = (0, ts_poet_1.code) `return ${decode}`;
+            decode = (0, ts_poet_1.code) `return ${decode}`;
         }
-        return (0, ts_poet_1.code) `promise.then(data => { ${(0, utils_1.tryCatchBlock)(tryBlock, (0, ts_poet_1.code) `return Promise.reject(error);`)}}).catch((error) => { ${errorHandler} })`;
-    }
-    else if (options.rpcAfterResponse) {
-        return (0, ts_poet_1.code) `promise.then(data => { ${decode} } )`;
+        return (0, ts_poet_1.code) `promise.then(${(0, utils_1.arrowFunction)("data", (0, utils_1.tryCatchBlock)(decode, (0, ts_poet_1.code) `return Promise.reject(error);`), false)}).catch(${(0, utils_1.arrowFunction)("error", errorHandler, false)})`;
     }
-    return (0, ts_poet_1.code) `promise.then(data => ${decode})`;
+    return (0, ts_poet_1.code) `promise.then(${returnStatement})`;
 }
 function generateServiceClientImpl(ctx, fileDesc, serviceDesc) {
     const { options } = ctx;

package/build/utils.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.tryCatchBlock = exports.impProto = exports.impFile = exports.getPropertyAccessor = exports.safeAccessor = exports.getFieldName = exports.getFieldJsonName = exports.FormattedMethodDescriptor = exports.assertInstanceOf = exports.maybePrefixPackage = exports.prefixDisableLinter = exports.maybeAddComment = exports.upperFirst = exports.lowerFirst = exports.singular = exports.fail = exports.readToBuffer = exports.generateIndexFiles = exports.protoFilesToGenerate = void 0;
+exports.arrowFunction = exports.tryCatchBlock = exports.impProto = exports.impFile = exports.getPropertyAccessor = exports.safeAccessor = exports.getFieldName = exports.getFieldJsonName = exports.FormattedMethodDescriptor = exports.assertInstanceOf = exports.maybePrefixPackage = exports.prefixDisableLinter = exports.maybeAddComment = exports.upperFirst = exports.lowerFirst = exports.singular = exports.fail = exports.readToBuffer = exports.generateIndexFiles = exports.protoFilesToGenerate = void 0;
 const path = require("path");
 const ts_poet_1 = require("ts-poet");
 const options_1 = require("./options");
@@ -254,3 +254,10 @@ function tryCatchBlock(tryBlock, handleErrorBlock) {
   }`;
 }
 exports.tryCatchBlock = tryCatchBlock;
+function arrowFunction(params, body, isOneLine = true) {
+    if (isOneLine) {
+        return (0, ts_poet_1.code) `(${params}) => ${body}`;
+    }
+    return (0, ts_poet_1.code) `(${params}) => { ${body} }`;
+}
+exports.arrowFunction = arrowFunction;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ts-proto",
-  "version": "1.167.2",
+  "version": "1.167.3",
   "description": "",
   "main": "build/plugin.js",
   "repository": "github:stephenh/ts-proto",