feat: export generic type of wrapped service client

yandex-cloud · Jan 12, 2022 · bec2aca · bec2aca
1 parent 2f424d6
commit bec2aca
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -3,3 +3,4 @@ export * as cloudApi from './generated/yandex/cloud';
 export * from './session';
 export * from './utils/operation';
 export * from './utils/decode-message';
+export { WrappedServiceClientType } from './types';
diff --git a/src/session.ts b/src/session.ts
@@ -7,7 +7,7 @@ import {
     GeneratedServiceClientCtor,
     IamTokenCredentialsConfig,
     OAuthCredentialsConfig,
-    ServiceAccountCredentialsConfig,
+    ServiceAccountCredentialsConfig, WrappedServiceClientType,
     SessionConfig,
 } from './types';
 import { IamTokenService } from './token-service/iam-token-service';
@@ -94,7 +94,7 @@ export class Session {
         return this.config.pollInterval;
     }
 
-    client<S extends ServiceDefinition>(clientClass: GeneratedServiceClientCtor<S>) {
+    client<S extends ServiceDefinition>(clientClass: GeneratedServiceClientCtor<S>): WrappedServiceClientType<S> {
         const endpoint = getServiceClientEndpoint(clientClass);
         const channel = createChannel(endpoint, this.channelCredentials);
 

diff --git a/src/types.ts b/src/types.ts
@@ -1,6 +1,9 @@
 import {
     ChannelCredentials, ChannelOptions, Client, ServiceDefinition,
 } from '@grpc/grpc-js';
+import { RawClient } from 'nice-grpc';
+import { NormalizedServiceDefinition } from 'nice-grpc/lib/service-definitions';
+import { DeadlineOptions } from 'nice-grpc-client-middleware-deadline';
 
 export interface TokenService {
     getToken: () => Promise<string>;
@@ -48,3 +51,5 @@ export type SessionConfig =
     | IamTokenCredentialsConfig
     | ServiceAccountCredentialsConfig
     | GenericCredentialsConfig;
+
+export type WrappedServiceClientType<S extends ServiceDefinition> = RawClient<NormalizedServiceDefinition<S>, DeadlineOptions>;