better-grpc 0.1.0 → 0.2.2

package/dist/index.js

@@ -1,5 +1,5 @@
 import { pushable } from 'it-pushable';
-import { ChannelCredentials, createChannel, createClientFactory, createServer } from 'nice-grpc';
+import { ChannelCredentials, createChannel, createClientFactory, createServer, Metadata } from 'nice-grpc';
 import { decode, encode } from '@msgpack/msgpack';
 
 var __create = Object.create;
@@ -348,7 +348,7 @@ var require_metadata = __commonJS({
         }
       }
     }
-    var Metadata = class _Metadata {
+    var Metadata2 = class _Metadata {
       constructor(options = {}) {
         this.internalRepr = /* @__PURE__ */ new Map();
         this.opaqueData = /* @__PURE__ */ new Map();
@@ -540,7 +540,7 @@ var require_metadata = __commonJS({
         return result;
       }
     };
-    exports2.Metadata = Metadata;
+    exports2.Metadata = Metadata2;
     var bufToString = (val) => {
       return Buffer.isBuffer(val) ? val.toString("base64") : val;
     };
@@ -3923,10 +3923,10 @@ var require_client_interceptors = __commonJS({
       const host = options.host;
       const parent = (_b = options.parent) !== null && _b !== void 0 ? _b : null;
       const propagateFlags = options.propagate_flags;
-      const credentials2 = options.credentials;
+      const credentials = options.credentials;
       const call = channel.createCall(path, deadline, host, parent, propagateFlags);
-      if (credentials2) {
-        call.setCredentials(credentials2);
+      if (credentials) {
+        call.setCredentials(credentials);
       }
       return call;
     }
@@ -4085,7 +4085,7 @@ var require_client = __commonJS({
       return ((_a = error.stack) === null || _a === void 0 ? void 0 : _a.split("\n").slice(1).join("\n")) || "no stack trace available";
     }
     var Client = class {
-      constructor(address, credentials2, options = {}) {
+      constructor(address, credentials, options = {}) {
         var _a, _b;
         options = Object.assign({}, options);
         this[INTERCEPTOR_SYMBOL] = (_a = options.interceptors) !== null && _a !== void 0 ? _a : [];
@@ -4102,9 +4102,9 @@ var require_client = __commonJS({
         } else if (options.channelFactoryOverride) {
           const channelFactoryOverride = options.channelFactoryOverride;
           delete options.channelFactoryOverride;
-          this[CHANNEL_SYMBOL] = channelFactoryOverride(address, credentials2, options);
+          this[CHANNEL_SYMBOL] = channelFactoryOverride(address, credentials, options);
         } else {
-          this[CHANNEL_SYMBOL] = new channel_1.ChannelImplementation(address, credentials2, options);
+          this[CHANNEL_SYMBOL] = new channel_1.ChannelImplementation(address, credentials, options);
         }
       }
       close() {
@@ -7715,7 +7715,7 @@ var require_root = __commonJS({
     var util = require_util();
     var Type;
     var parse2;
-    var common2;
+    var common;
     function Root2(options) {
       Namespace.call(this, "", options);
       this.deferred = [];
@@ -7762,7 +7762,7 @@ var require_root = __commonJS({
         var idx = filename2.lastIndexOf("google/protobuf/");
         if (idx > -1) {
           var altname = filename2.substring(idx);
-          if (altname in common2) return altname;
+          if (altname in common) return altname;
         }
         return null;
       }
@@ -7799,14 +7799,14 @@ var require_root = __commonJS({
           return;
         }
         self2.files.push(filename2);
-        if (filename2 in common2) {
+        if (filename2 in common) {
           if (sync) {
-            process2(filename2, common2[filename2]);
+            process2(filename2, common[filename2]);
           } else {
             ++queued;
             setTimeout(function() {
               --queued;
-              process2(filename2, common2[filename2]);
+              process2(filename2, common[filename2]);
             });
           }
           return;
@@ -7946,7 +7946,7 @@ var require_root = __commonJS({
     Root2._configure = function(Type_, parse_, common_) {
       Type = Type_;
       parse2 = parse_;
-      common2 = common_;
+      common = common_;
     };
   }
 });
@@ -9910,16 +9910,16 @@ var require_parse = __commonJS({
 // node_modules/protobufjs/src/common.js
 var require_common = __commonJS({
   "node_modules/protobufjs/src/common.js"(exports2, module2) {
-    module2.exports = common2;
+    module2.exports = common;
     var commonRe = /\/|\./;
-    function common2(name, json) {
+    function common(name, json) {
       if (!commonRe.test(name)) {
         name = "google/protobuf/" + name + ".proto";
         json = { nested: { google: { nested: { protobuf: { nested: json } } } } };
       }
-      common2[name] = json;
+      common[name] = json;
     }
-    common2("any", {
+    common("any", {
       /**
        * Properties of a google.protobuf.Any message.
        * @interface IAny
@@ -9942,7 +9942,7 @@ var require_common = __commonJS({
       }
     });
     var timeType;
-    common2("duration", {
+    common("duration", {
       /**
        * Properties of a google.protobuf.Duration message.
        * @interface IDuration
@@ -9964,7 +9964,7 @@ var require_common = __commonJS({
         }
       }
     });
-    common2("timestamp", {
+    common("timestamp", {
       /**
        * Properties of a google.protobuf.Timestamp message.
        * @interface ITimestamp
@@ -9975,7 +9975,7 @@ var require_common = __commonJS({
        */
       Timestamp: timeType
     });
-    common2("empty", {
+    common("empty", {
       /**
        * Properties of a google.protobuf.Empty message.
        * @interface IEmpty
@@ -9985,7 +9985,7 @@ var require_common = __commonJS({
         fields: {}
       }
     });
-    common2("struct", {
+    common("struct", {
       /**
        * Properties of a google.protobuf.Struct message.
        * @interface IStruct
@@ -10077,7 +10077,7 @@ var require_common = __commonJS({
         }
       }
     });
-    common2("wrappers", {
+    common("wrappers", {
       /**
        * Properties of a google.protobuf.DoubleValue message.
        * @interface IDoubleValue
@@ -10214,7 +10214,7 @@ var require_common = __commonJS({
         }
       }
     });
-    common2("field_mask", {
+    common("field_mask", {
       /**
        * Properties of a google.protobuf.FieldMask message.
        * @interface IDoubleValue
@@ -10232,8 +10232,8 @@ var require_common = __commonJS({
         }
       }
     });
-    common2.get = function get(file) {
-      return common2[file] || null;
+    common.get = function get(file) {
+      return common[file] || null;
     };
   }
 });
@@ -15024,7 +15024,7 @@ var require_single_subchannel_channel = __commonJS({
       getCallNumber() {
         return this.callNumber;
       }
-      setCredentials(credentials2) {
+      setCredentials(credentials) {
         throw new Error("Method not implemented.");
       }
       getAuthContext() {
@@ -15114,7 +15114,7 @@ var require_subchannel = __commonJS({
        * @param credentials The channel credentials used to establish this
        *     connection
        */
-      constructor(channelTarget, subchannelAddress, options, credentials2, connector) {
+      constructor(channelTarget, subchannelAddress, options, credentials, connector) {
         var _a;
         this.channelTarget = channelTarget;
         this.subchannelAddress = subchannelAddress;
@@ -15153,7 +15153,7 @@ var require_subchannel = __commonJS({
         this.channelzRef = (0, channelz_1.registerChannelzSubchannel)(this.subchannelAddressString, () => this.getChannelzInfo(), this.channelzEnabled);
         this.channelzTrace.addTrace("CT_INFO", "Subchannel created");
         this.trace("Subchannel constructed with options " + JSON.stringify(options, void 0, 2));
-        this.secureConnector = credentials2._createSecureConnector(channelTarget, options);
+        this.secureConnector = credentials._createSecureConnector(channelTarget, options);
       }
       getChannelzInfo() {
         return {
@@ -17052,13 +17052,13 @@ var require_load_balancing_call = __commonJS({
     var http2 = __require("http2");
     var TRACER_NAME = "load_balancing_call";
     var LoadBalancingCall = class {
-      constructor(channel, callConfig, methodName, host, credentials2, deadline, callNumber) {
+      constructor(channel, callConfig, methodName, host, credentials, deadline, callNumber) {
         var _a, _b;
         this.channel = channel;
         this.callConfig = callConfig;
         this.methodName = methodName;
         this.host = host;
-        this.credentials = credentials2;
+        this.credentials = credentials;
         this.deadline = deadline;
         this.callNumber = callNumber;
         this.child = null;
@@ -17260,7 +17260,7 @@ var require_load_balancing_call = __commonJS({
           this.pendingHalfClose = true;
         }
       }
-      setCredentials(credentials2) {
+      setCredentials(credentials) {
         throw new Error("Method not implemented.");
       }
       getCallNumber() {
@@ -17541,8 +17541,8 @@ var require_resolving_call = __commonJS({
           this.pendingHalfClose = true;
         }
       }
-      setCredentials(credentials2) {
-        this.credentials = credentials2;
+      setCredentials(credentials) {
+        this.credentials = credentials;
       }
       addStatusWatcher(watcher) {
         this.statusWatchers.push(watcher);
@@ -17636,13 +17636,13 @@ var require_retrying_call = __commonJS({
     var PREVIONS_RPC_ATTEMPTS_METADATA_KEY = "grpc-previous-rpc-attempts";
     var DEFAULT_MAX_ATTEMPTS_LIMIT = 5;
     var RetryingCall = class {
-      constructor(channel, callConfig, methodName, host, credentials2, deadline, callNumber, bufferTracker, retryThrottler) {
+      constructor(channel, callConfig, methodName, host, credentials, deadline, callNumber, bufferTracker, retryThrottler) {
         var _a;
         this.channel = channel;
         this.callConfig = callConfig;
         this.methodName = methodName;
         this.host = host;
-        this.credentials = credentials2;
+        this.credentials = credentials;
         this.deadline = deadline;
         this.callNumber = callNumber;
         this.bufferTracker = bufferTracker;
@@ -18367,9 +18367,9 @@ var require_internal_channel = __commonJS({
       }
     };
     var InternalChannel = class {
-      constructor(target, credentials2, options) {
+      constructor(target, credentials, options) {
         var _a, _b, _c, _d, _e, _f;
-        this.credentials = credentials2;
+        this.credentials = credentials;
         this.options = options;
         this.connectivityState = connectivity_state_1.ConnectivityState.IDLE;
         this.currentPicker = new picker_1.UnavailablePicker();
@@ -18387,7 +18387,7 @@ var require_internal_channel = __commonJS({
         if (typeof target !== "string") {
           throw new TypeError("Channel target must be a string");
         }
-        if (!(credentials2 instanceof channel_credentials_1.ChannelCredentials)) {
+        if (!(credentials instanceof channel_credentials_1.ChannelCredentials)) {
           throw new TypeError("Channel credentials must be a ChannelCredentials object");
         }
         if (options) {
@@ -18669,15 +18669,15 @@ var require_internal_channel = __commonJS({
         this.lastActivityTimestamp = /* @__PURE__ */ new Date();
         this.maybeStartIdleTimer();
       }
-      createLoadBalancingCall(callConfig, method, host, credentials2, deadline) {
+      createLoadBalancingCall(callConfig, method, host, credentials, deadline) {
         const callNumber = (0, call_number_1.getNextCallNumber)();
         this.trace("createLoadBalancingCall [" + callNumber + '] method="' + method + '"');
-        return new load_balancing_call_1.LoadBalancingCall(this, callConfig, method, host, credentials2, deadline, callNumber);
+        return new load_balancing_call_1.LoadBalancingCall(this, callConfig, method, host, credentials, deadline, callNumber);
       }
-      createRetryingCall(callConfig, method, host, credentials2, deadline) {
+      createRetryingCall(callConfig, method, host, credentials, deadline) {
         const callNumber = (0, call_number_1.getNextCallNumber)();
         this.trace("createRetryingCall [" + callNumber + '] method="' + method + '"');
-        return new retrying_call_1.RetryingCall(this, callConfig, method, host, credentials2, deadline, callNumber, this.retryBufferTracker, RETRY_THROTTLER_MAP.get(this.getTarget()));
+        return new retrying_call_1.RetryingCall(this, callConfig, method, host, credentials, deadline, callNumber, this.retryBufferTracker, RETRY_THROTTLER_MAP.get(this.getTarget()));
       }
       createResolvingCall(method, deadline, host, parentCall, propagateFlags) {
         const callNumber = (0, call_number_1.getNextCallNumber)();
@@ -18793,11 +18793,11 @@ var require_channel = __commonJS({
     var channel_credentials_1 = require_channel_credentials();
     var internal_channel_1 = require_internal_channel();
     var ChannelImplementation = class {
-      constructor(target, credentials2, options) {
+      constructor(target, credentials, options) {
         if (typeof target !== "string") {
           throw new TypeError("Channel target must be a string");
         }
-        if (!(credentials2 instanceof channel_credentials_1.ChannelCredentials)) {
+        if (!(credentials instanceof channel_credentials_1.ChannelCredentials)) {
           throw new TypeError("Channel credentials must be a ChannelCredentials object");
         }
         if (options) {
@@ -18805,7 +18805,7 @@ var require_channel = __commonJS({
             throw new TypeError("Channel options must be an object");
           }
         }
-        this.internalChannel = new internal_channel_1.InternalChannel(target, credentials2, options);
+        this.internalChannel = new internal_channel_1.InternalChannel(target, credentials, options);
       }
       close() {
         this.internalChannel.close();
@@ -19332,8 +19332,8 @@ var require_server_credentials = __commonJS({
         return this.childCredentials._getSecureContextOptions();
       }
     };
-    function createServerCredentialsWithInterceptors(credentials2, interceptors) {
-      return new InterceptorServerCredentials(credentials2, interceptors);
+    function createServerCredentialsWithInterceptors(credentials, interceptors) {
+      return new InterceptorServerCredentials(credentials, interceptors);
     }
   }
 });
@@ -20729,11 +20729,11 @@ var require_server = __commonJS({
         experimentalUnregisterListenerFromChannelz(channelzRef) {
           (0, channelz_1.unregisterChannelzRef)(channelzRef);
         }
-        createHttp2Server(credentials2) {
+        createHttp2Server(credentials) {
           let http2Server;
-          if (credentials2._isSecure()) {
-            const constructorOptions = credentials2._getConstructorOptions();
-            const contextOptions = credentials2._getSecureContextOptions();
+          if (credentials._isSecure()) {
+            const constructorOptions = credentials._getConstructorOptions();
+            const contextOptions = credentials._getSecureContextOptions();
             const secureServerOptions = Object.assign(Object.assign(Object.assign(Object.assign({}, this.commonServerOptions), constructorOptions), contextOptions), { enableTrace: this.options["grpc-node.tls_enable_trace"] === 1 });
             let areCredentialsValid = contextOptions !== null;
             this.trace("Initial credentials valid: " + areCredentialsValid);
@@ -20762,15 +20762,15 @@ var require_server = __commonJS({
               areCredentialsValid = options !== null;
               this.trace("Post-update credentials valid: " + areCredentialsValid);
             };
-            credentials2._addWatcher(credsWatcher);
+            credentials._addWatcher(credsWatcher);
             http2Server.on("close", () => {
-              credentials2._removeWatcher(credsWatcher);
+              credentials._removeWatcher(credsWatcher);
             });
           } else {
             http2Server = http2.createServer(this.commonServerOptions);
           }
           http2Server.setTimeout(0, noop);
-          this._setupHandlers(http2Server, credentials2._getInterceptors());
+          this._setupHandlers(http2Server, credentials._getInterceptors());
           return http2Server;
         }
         bindOneAddress(address, boundPortObject) {
@@ -21005,14 +21005,14 @@ var require_server = __commonJS({
          * @param channelzRef
          * @returns
          */
-        experimentalCreateConnectionInjectorWithChannelzRef(credentials2, channelzRef, ownsChannelzRef = false) {
-          if (credentials2 === null || !(credentials2 instanceof server_credentials_1.ServerCredentials)) {
+        experimentalCreateConnectionInjectorWithChannelzRef(credentials, channelzRef, ownsChannelzRef = false) {
+          if (credentials === null || !(credentials instanceof server_credentials_1.ServerCredentials)) {
             throw new TypeError("creds must be a ServerCredentials object");
           }
           if (this.channelzEnabled) {
             this.listenerChildrenTracker.refChild(channelzRef);
           }
-          const server2 = this.createHttp2Server(credentials2);
+          const server2 = this.createHttp2Server(credentials);
           const sessionsSet = /* @__PURE__ */ new Set();
           this.http2Servers.set(server2, {
             channelzRef,
@@ -21042,12 +21042,12 @@ var require_server = __commonJS({
             }
           };
         }
-        createConnectionInjector(credentials2) {
-          if (credentials2 === null || !(credentials2 instanceof server_credentials_1.ServerCredentials)) {
+        createConnectionInjector(credentials) {
+          if (credentials === null || !(credentials instanceof server_credentials_1.ServerCredentials)) {
             throw new TypeError("creds must be a ServerCredentials object");
           }
           const channelzRef = this.registerInjectorToChannelz();
-          return this.experimentalCreateConnectionInjectorWithChannelzRef(credentials2, channelzRef, true);
+          return this.experimentalCreateConnectionInjectorWithChannelzRef(credentials, channelzRef, true);
         }
         closeServer(server2, callback) {
           this.trace("Closing server with address " + JSON.stringify(server2.address()));
@@ -24041,10 +24041,19 @@ var require_src3 = __commonJS({
 
 // src/core/rpc-signatures.ts
 function server() {
-  return {
+  const descriptor = {
     serviceType: "server",
     methodType: "unary"
   };
+  const contextFn = (context) => {
+    descriptor.config = {
+      metadata: context.metadata !== void 0,
+      ack: false
+    };
+    return descriptor;
+  };
+  Object.assign(contextFn, descriptor);
+  return contextFn;
 }
 function client() {
   return {
@@ -24052,6 +24061,21 @@ function client() {
     methodType: "unary"
   };
 }
+function bidi(..._) {
+  const descriptor = {
+    serviceType: "bidi",
+    methodType: "bidi"
+  };
+  const configFn = (config) => {
+    descriptor.config = {
+      metadata: config.metadata !== void 0,
+      ack: config.ack ?? false
+    };
+    return descriptor;
+  };
+  Object.assign(configFn, descriptor);
+  return configFn;
+}
 
 // src/core/service.ts
 var ServiceImpl = class {
@@ -24112,18 +24136,18 @@ function loadProtoFromString(source) {
 function encodeRequestMessage(id, value) {
   return {
     id,
-    value: encode(value)
+    value: value === void 0 ? void 0 : encode(value)
   };
 }
 function decodeRequestMessage(message) {
   const id = message.id;
-  const mapData = message.value === void 0 ? [] : decode(message.value);
+  const mapData = message.value === void 0 ? void 0 : decode(message.value);
   return [id, mapData];
 }
 function encodeResponseMessage(id, value) {
   return {
     id,
-    value: encode(value)
+    value: value === void 0 ? void 0 : encode(value)
   };
 }
 function decodeResponseMessage(message) {
@@ -24131,6 +24155,18 @@ function decodeResponseMessage(message) {
   const mapData = message.value === void 0 ? void 0 : decode(message.value);
   return [id, mapData];
 }
+function encodeMetadata(raw) {
+  const m = new Metadata();
+  m.set("better-grpc", JSON.stringify(raw));
+  return m;
+}
+function decodeMetadata(metadata) {
+  const betterGrpcMetadata = metadata.get("better-grpc");
+  if (!betterGrpcMetadata) {
+    throw new Error("Failed to decode metadata: 'better-grpc' key not found in gRPC metadata");
+  }
+  return JSON.parse(betterGrpcMetadata);
+}
 
 // src/runtime/proto-builder.ts
 function buildProtoString(services) {
@@ -24179,13 +24215,18 @@ var GrpcClient = class {
   clientFactory;
   proto;
   clients = /* @__PURE__ */ new Map();
+  pushableStreams = {};
+  pendingStreams = /* @__PURE__ */ new Map();
+  pendingBidi = /* @__PURE__ */ new Map();
+  // bidi that is waiting for context
+  pendingBidiAck = /* @__PURE__ */ new Map();
   constructor(address, serviceImpls) {
     this.address = address;
     this.serviceImpls = serviceImpls;
     this.proto = loadProtoFromString(buildProtoString(serviceImpls));
     const useSSL = !address.includes("localhost") && !address.includes("127.0.0.1") && !address.includes("0.0.0.0");
-    const credentials2 = useSSL ? ChannelCredentials.createSsl() : grpc.credentials.createInsecure();
-    this.channel = createChannel(address, credentials2, {
+    const credentials = useSSL ? ChannelCredentials.createSsl() : ChannelCredentials.createInsecure();
+    this.channel = createChannel(address, credentials, {
       "grpc.max_receive_message_length": 10 * 1024 * 1024,
       "grpc.max_send_message_length": 10 * 1024 * 1024,
       "grpc.keepalive_time_ms": 3e4,
@@ -24203,27 +24244,90 @@ var GrpcClient = class {
       this.clients.set(serviceImpl.serviceClass.serviceName, client2);
     }
   }
+  setStream(serviceName, methodName, stream) {
+    if (!this.pushableStreams[serviceName]) {
+      this.pushableStreams[serviceName] = {};
+    }
+    this.pushableStreams[serviceName][methodName.toUpperCase()] = stream;
+    if (this.pendingStreams.has(`${serviceName}.${methodName.toUpperCase()}`)) {
+      const resolve = this.pendingStreams.get(`${serviceName}.${methodName.toUpperCase()}`);
+      resolve?.(stream);
+      this.pendingStreams.delete(`${serviceName}.${methodName.toUpperCase()}`);
+    }
+  }
+  async getStream(serviceName, methodName) {
+    const stream = this.pushableStreams[serviceName]?.[methodName.toUpperCase()];
+    if (!stream) {
+      return new Promise((resolve) => {
+        this.pendingStreams.set(`${serviceName}.${methodName.toUpperCase()}`, resolve);
+      });
+    }
+    return stream;
+  }
   watching() {
     for (const serviceImpl of this.serviceImpls) {
       const client2 = this.clients.get(serviceImpl.serviceClass.serviceName);
       for (const [name, descriptor] of Object.entries(serviceImpl.methods())) {
-        switch (descriptor.serviceType) {
-          case "client":
-            switch (descriptor.methodType) {
-              case "unary": {
-                const incomingStream = pushable({ objectMode: true });
-                const incomingMessages = client2[name.toUpperCase()](incomingStream);
-                (async () => {
+        switch (`${descriptor.serviceType}:${descriptor.methodType}`) {
+          case "client:unary": {
+            const incomingStream = pushable({ objectMode: true });
+            const incomingMessages = client2[name.toUpperCase()](incomingStream);
+            (async () => {
+              try {
+                for await (const message of incomingMessages) {
+                  const [id, value] = decodeRequestMessage(message);
+                  const responseValue = await serviceImpl.implementation[name](...value ?? []);
+                  incomingStream.push(encodeResponseMessage(id, responseValue));
+                }
+              } finally {
+                incomingStream.end();
+              }
+            })();
+            break;
+          }
+          case "bidi:bidi": {
+            const setupBidi = (context) => {
+              const outStream = pushable({ objectMode: true });
+              const inStream = pushable({ objectMode: true });
+              this.setStream(
+                `${serviceImpl.serviceClass.serviceName}_OUT`,
+                name.toUpperCase(),
+                outStream
+              );
+              this.setStream(`${serviceImpl.serviceClass.serviceName}_IN`, name.toUpperCase(), inStream);
+              const incomingMessages = context ? client2[name.toUpperCase()](outStream, { metadata: encodeMetadata(context.metadata) }) : client2[name.toUpperCase()](outStream);
+              (async () => {
+                try {
                   for await (const message of incomingMessages) {
                     const [id, value] = decodeRequestMessage(message);
-                    const responseValue = await serviceImpl.implementation[name](...value);
-                    incomingStream.push(encodeResponseMessage(id, responseValue));
+                    if (id && descriptor.config?.ack && value === void 0) {
+                      this.pendingBidiAck.get(id)?.();
+                      this.pendingBidiAck.delete(id);
+                    } else {
+                      inStream.push(value ?? []);
+                      if (id && descriptor.config?.ack) {
+                        outStream.push(encodeRequestMessage(id, void 0));
+                      }
+                    }
                   }
-                })();
-                break;
-              }
+                } finally {
+                  inStream.end();
+                  outStream.end();
+                }
+              })();
+            };
+            if (descriptor.config?.metadata) {
+              (async () => {
+                const context = await new Promise((resolve) => {
+                  this.pendingBidi.set(`${serviceImpl.serviceClass.serviceName}.${name}`, resolve);
+                });
+                setupBidi(context);
+              })();
+            } else {
+              setupBidi(void 0);
             }
             break;
+          }
         }
       }
     }
@@ -24233,13 +24337,72 @@ var GrpcClient = class {
       const serviceCallableInstance = {};
       for (const [name, descriptor] of Object.entries(serviceImpl.methods())) {
         switch (`${descriptor.serviceType}:${descriptor.methodType}`) {
-          case "server:unary":
-            serviceCallableInstance[name] = async (...args) => {
-              const response = await this.clients.get(serviceImpl.serviceClass.serviceName)[name.toUpperCase()](encodeRequestMessage(void 0, args));
-              const [_, value] = decodeResponseMessage(response);
-              return value;
+          case "server:unary": {
+            if (descriptor.config?.metadata) {
+              serviceCallableInstance[name] = (...args) => {
+                return {
+                  withMeta: async (metadata) => {
+                    const response = await this.clients.get(serviceImpl.serviceClass.serviceName)[name.toUpperCase()](encodeRequestMessage(void 0, args), {
+                      metadata: encodeMetadata(metadata)
+                    });
+                    const [_, value] = decodeResponseMessage(response);
+                    return value;
+                  }
+                };
+              };
+            } else {
+              serviceCallableInstance[name] = async (...args) => {
+                const response = await this.clients.get(serviceImpl.serviceClass.serviceName)[name.toUpperCase()](encodeRequestMessage(void 0, args));
+                const [_, value] = decodeResponseMessage(response);
+                return value;
+              };
+            }
+            break;
+          }
+          case "bidi:bidi": {
+            async function* generator(client2) {
+              const inStream = await client2.getStream(
+                `${serviceImpl.serviceClass.serviceName}_IN`,
+                name.toUpperCase()
+              );
+              yield* inStream;
+            }
+            const emitFn = async (...args) => {
+              const outStream = await this.getStream(
+                `${serviceImpl.serviceClass.serviceName}_OUT`,
+                name.toUpperCase()
+              );
+              const ackId = descriptor.config?.ack ? crypto.randomUUID() : void 0;
+              outStream.push(encodeRequestMessage(ackId, args));
+              if (ackId) {
+                return new Promise((resolve) => {
+                  this.pendingBidiAck.set(ackId, resolve);
+                });
+              }
             };
+            const context = {
+              context: async (ctx) => {
+                const resolve = this.pendingBidi.get(`${serviceImpl.serviceClass.serviceName}.${name}`);
+                if (resolve) {
+                  resolve(ctx);
+                  this.pendingBidi.delete(`${serviceImpl.serviceClass.serviceName}.${name}`);
+                }
+              }
+            };
+            const hybrid = Object.assign(emitFn, {
+              [Symbol.asyncIterator]: () => {
+                const iterator = generator(this);
+                return {
+                  next: iterator.next.bind(iterator),
+                  return: iterator.return.bind(iterator),
+                  throw: iterator.throw.bind(iterator)
+                };
+              }
+            });
+            Object.assign(hybrid, context);
+            serviceCallableInstance[name] = hybrid;
             break;
+          }
         }
       }
       Object.defineProperty(this, serviceImpl.serviceClass.serviceName, {
@@ -24291,9 +24454,10 @@ function createServiceImpl(serviceImpl, grpcServer) {
   for (const [name, descriptor] of Object.entries(serviceImpl.methods())) {
     switch (`${descriptor.serviceType}:${descriptor.methodType}`) {
       case "server:unary":
-        grpcImpl[name.toUpperCase()] = async (req) => {
+        grpcImpl[name.toUpperCase()] = async (req, ctx) => {
           const [_, value] = decodeRequestMessage(req);
-          const result = await serviceImpl.implementation[name](...value);
+          const args = descriptor.config?.metadata ? [{ metadata: decodeMetadata(ctx.metadata) }, ...value ?? []] : value ?? [];
+          const result = await serviceImpl.implementation[name](...args);
           return encodeResponseMessage(void 0, result);
         };
         break;
@@ -24302,19 +24466,55 @@ function createServiceImpl(serviceImpl, grpcServer) {
           const stream = pushable({ objectMode: true });
           grpcServer.setStream(serviceImpl.serviceClass.serviceName, name, stream);
           (async () => {
-            for await (const message of incomingStream) {
-              const [id, value] = decodeResponseMessage(message);
-              if (id) {
-                grpcServer.resolveResponse(id, value);
-              } else {
-                throw new Error(`Invalid response message: ${message}`);
+            try {
+              for await (const message of incomingStream) {
+                const [id, value] = decodeResponseMessage(message);
+                if (id) {
+                  grpcServer.resolveResponse(id, value);
+                } else {
+                  throw new Error(`Invalid response message: ${message}`);
+                }
               }
+            } finally {
+              stream.end();
             }
-            stream.end();
           })();
           yield* stream;
         };
         break;
+      case "bidi:bidi":
+        grpcImpl[name.toUpperCase()] = async function* (incomingStream, ctx) {
+          if (descriptor.config?.metadata) {
+            grpcServer.setContext(serviceImpl.serviceClass.serviceName, name, {
+              metadata: decodeMetadata(ctx.metadata)
+            });
+          }
+          const outStream = pushable({ objectMode: true });
+          const inStream = pushable({ objectMode: true });
+          grpcServer.setStream(`${serviceImpl.serviceClass.serviceName}_OUT`, name, outStream);
+          grpcServer.setStream(`${serviceImpl.serviceClass.serviceName}_IN`, name, inStream);
+          (async () => {
+            try {
+              for await (const message of incomingStream) {
+                const [id, value] = decodeResponseMessage(message);
+                if (id && descriptor.config?.ack && value === void 0) {
+                  grpcServer.pendingBidiAck.get(id)?.();
+                  grpcServer.pendingBidiAck.delete(id);
+                } else {
+                  inStream.push(value ?? []);
+                  if (id && descriptor.config?.ack) {
+                    outStream.push(encodeRequestMessage(id, void 0));
+                  }
+                }
+              }
+            } finally {
+              inStream.end();
+              outStream.end();
+            }
+          })();
+          yield* outStream;
+        };
+        break;
       default:
         throw new Error(`Unknown method descriptor: ${descriptor} for ${name}`);
     }
@@ -24331,6 +24531,9 @@ var GrpcServer = class {
   pushableStreams = {};
   pendingRequests = /* @__PURE__ */ new Map();
   pendingStreams = /* @__PURE__ */ new Map();
+  contexts = /* @__PURE__ */ new Map();
+  pendingContext = /* @__PURE__ */ new Map();
+  pendingBidiAck = /* @__PURE__ */ new Map();
   constructor(address, serviceImpls) {
     this.address = address;
     this.serviceImpls = serviceImpls;
@@ -24347,6 +24550,9 @@ var GrpcServer = class {
     await this.grpcServer.listen(this.address);
   }
   setStream(serviceName, methodName, stream) {
+    if (!this.pushableStreams[serviceName]) {
+      this.pushableStreams[serviceName] = {};
+    }
     this.pushableStreams[serviceName][methodName.toUpperCase()] = stream;
     if (this.pendingStreams.has(`${serviceName}.${methodName.toUpperCase()}`)) {
       const resolve = this.pendingStreams.get(`${serviceName}.${methodName.toUpperCase()}`);
@@ -24355,7 +24561,7 @@ var GrpcServer = class {
     }
   }
   async getStream(serviceName, methodName) {
-    const stream = this.pushableStreams[serviceName][methodName.toUpperCase()];
+    const stream = this.pushableStreams[serviceName]?.[methodName.toUpperCase()];
     if (!stream) {
       return new Promise((resolve) => {
         this.pendingStreams.set(`${serviceName}.${methodName.toUpperCase()}`, resolve);
@@ -24363,6 +24569,23 @@ var GrpcServer = class {
     }
     return stream;
   }
+  setContext(serviceName, methodName, context) {
+    this.contexts.set(`${serviceName}.${methodName.toUpperCase()}`, context);
+    if (this.pendingContext.has(`${serviceName}.${methodName.toUpperCase()}`)) {
+      const resolve = this.pendingContext.get(`${serviceName}.${methodName.toUpperCase()}`);
+      resolve?.(context);
+      this.pendingContext.delete(`${serviceName}.${methodName.toUpperCase()}`);
+    }
+  }
+  async getContext(serviceName, methodName) {
+    const context = this.contexts.get(`${serviceName}.${methodName.toUpperCase()}`);
+    if (!context) {
+      return new Promise((resolve) => {
+        this.pendingContext.set(`${serviceName}.${methodName.toUpperCase()}`, resolve);
+      });
+    }
+    return context;
+  }
   resolveResponse(id, value) {
     const resolve = this.pendingRequests.get(id);
     if (!resolve) {
@@ -24386,6 +24609,44 @@ var GrpcServer = class {
               });
             };
             break;
+          case "bidi:bidi": {
+            async function* generator(server2) {
+              const inStream = await server2.getStream(
+                `${serviceImpl.serviceClass.serviceName}_IN`,
+                name.toUpperCase()
+              );
+              yield* inStream;
+            }
+            const emitFn = async (...args) => {
+              const outStream = await this.getStream(
+                `${serviceImpl.serviceClass.serviceName}_OUT`,
+                name.toUpperCase()
+              );
+              const ackId = descriptor.config?.ack ? crypto.randomUUID() : void 0;
+              outStream.push(encodeRequestMessage(ackId, args));
+              if (ackId) {
+                return new Promise((resolve) => {
+                  this.pendingBidiAck.set(ackId, resolve);
+                });
+              }
+            };
+            const context = {
+              context: this.getContext(serviceImpl.serviceClass.serviceName, name)
+            };
+            const hybrid = Object.assign(emitFn, {
+              [Symbol.asyncIterator]: () => {
+                const iterator = generator(this);
+                return {
+                  next: iterator.next.bind(iterator),
+                  return: iterator.return.bind(iterator),
+                  throw: iterator.throw.bind(iterator)
+                };
+              }
+            });
+            Object.assign(hybrid, context);
+            serviceCallableInstance[name] = hybrid;
+            break;
+          }
         }
       }
       Object.defineProperty(this, serviceImpl.serviceClass.serviceName, {
@@ -24449,6 +24710,6 @@ long/umd/index.js:
    *)
 */
 
-export { Service, client, createGrpcClient, createGrpcServer, server };
+export { Service, bidi, client, createGrpcClient, createGrpcServer, server };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map