conditional export (#1003)

stephenh · Feb 16, 2024 · a605a24 · a605a24
1 parent 9fdce05
commit a605a24
Showing 1 changed file with 13 additions and 7 deletions.
diff --git a/src/main.ts b/src/main.ts
@@ -436,7 +436,7 @@ export function makeUtils(options: Options): Utils {
     ...makeTimestampMethods(options, longs, bytes),
     ...longs,
     ...makeComparisonUtils(),
-    ...makeNiceGrpcServerStreamingMethodResult(),
+    ...makeNiceGrpcServerStreamingMethodResult(options),
     ...makeGrpcWebErrorClass(bytes),
     ...makeExtensionClass(options),
     ...makeAssertionUtils(bytes),
@@ -872,14 +872,20 @@ function makeComparisonUtils() {
   return { isObject, isSet };
 }
 
-function makeNiceGrpcServerStreamingMethodResult() {
+function makeNiceGrpcServerStreamingMethodResult(options: Options) {
   const NiceGrpcServerStreamingMethodResult = conditionalOutput(
     "ServerStreamingMethodResult",
-    code`
-      export type ServerStreamingMethodResult<Response> = {
-        [Symbol.asyncIterator](): AsyncIterator<Response, void>;
-      };
-    `,
+    options.outputIndex
+      ? code`
+        type ServerStreamingMethodResult<Response> = {
+          [Symbol.asyncIterator](): AsyncIterator<Response, void>;
+        };
+      `
+      : code`
+        export type ServerStreamingMethodResult<Response> = {
+          [Symbol.asyncIterator](): AsyncIterator<Response, void>;
+        };
+      `,
   );
 
   return { NiceGrpcServerStreamingMethodResult };