@nmtjs/gateway 0.15.0-beta.3 → 0.15.0-beta.4

package/package.json

@@ -6,26 +6,27 @@
   },
   "dependencies": {
     "hookable": "6.0.0-rc.1",
-    "@nmtjs/common": "0.15.0-beta.3",
-    "@nmtjs/core": "0.15.0-beta.3",
-    "@nmtjs/type": "0.15.0-beta.3",
-    "@nmtjs/protocol": "0.15.0-beta.3"
+    "@nmtjs/common": "0.15.0-beta.4",
+    "@nmtjs/type": "0.15.0-beta.4",
+    "@nmtjs/core": "0.15.0-beta.4",
+    "@nmtjs/protocol": "0.15.0-beta.4"
   },
   "peerDependencies": {
-    "@nmtjs/common": "0.15.0-beta.3",
-    "@nmtjs/type": "0.15.0-beta.3",
-    "@nmtjs/protocol": "0.15.0-beta.3",
-    "@nmtjs/core": "0.15.0-beta.3"
+    "@nmtjs/common": "0.15.0-beta.4",
+    "@nmtjs/type": "0.15.0-beta.4",
+    "@nmtjs/core": "0.15.0-beta.4",
+    "@nmtjs/protocol": "0.15.0-beta.4"
   },
   "devDependencies": {
-    "@nmtjs/_tests": "0.15.0-beta.3"
+    "@nmtjs/_tests": "0.15.0-beta.4"
   },
   "files": [
     "dist",
+    "src",
     "LICENSE.md",
     "README.md"
   ],
-  "version": "0.15.0-beta.3",
+  "version": "0.15.0-beta.4",
   "scripts": {
     "clean-build": "rm -rf ./dist",
     "build": "tsc --declaration --sourcemap",

package/src/api.ts

@@ -0,0 +1,26 @@
+import type { Container, MetadataStore } from '@nmtjs/core'
+
+import type { GatewayConnection } from './connections.ts'
+
+export type GatewayApiCallOptions = {
+  connection: GatewayConnection
+  procedure: string
+  container: Container
+  payload: any
+  signal: AbortSignal
+  metadata?: (store: MetadataStore) => void
+}
+
+export type GatewayApiCallResult = unknown
+
+export interface GatewayApi {
+  call(options: GatewayApiCallOptions): Promise<GatewayApiCallResult>
+}
+
+export function isAsyncIterable(
+  value: GatewayApiCallResult,
+): value is AsyncIterable<unknown> | Iterable<unknown> {
+  return Boolean(
+    value && typeof value === 'object' && Symbol.asyncIterator in value,
+  )
+}

package/src/connections.ts

@@ -0,0 +1,54 @@
+import type { Container } from '@nmtjs/core'
+import type { ConnectionType } from '@nmtjs/protocol'
+import type {
+  BaseServerDecoder,
+  BaseServerEncoder,
+  ProtocolVersionInterface,
+} from '@nmtjs/protocol/server'
+import { MAX_UINT32, throwError } from '@nmtjs/common'
+
+export interface GatewayConnection {
+  readonly id: string
+  readonly type: ConnectionType
+  readonly transport: string
+  readonly protocol: ProtocolVersionInterface
+  readonly identity: string
+  readonly container: Container
+  readonly encoder: BaseServerEncoder
+  readonly decoder: BaseServerDecoder
+  readonly abortController: AbortController
+}
+
+export class ConnectionManager {
+  readonly connections = new Map<string, GatewayConnection>()
+  readonly streamIds = new Map<string, number>()
+
+  add(connection: GatewayConnection) {
+    this.connections.set(connection.id, connection)
+    this.streamIds.set(connection.id, 0)
+  }
+
+  get(id: string) {
+    return this.connections.get(id) ?? throwError('Connection not found')
+  }
+
+  has(id: string) {
+    return this.connections.has(id)
+  }
+
+  remove(id: string) {
+    this.connections.delete(id)
+    this.streamIds.delete(id)
+  }
+
+  getAll() {
+    return this.connections.values()
+  }
+
+  getStreamId(connectionId: string) {
+    let streamId = this.streamIds.get(connectionId)!
+    if (streamId >= MAX_UINT32) streamId = 0
+    this.streamIds.set(connectionId, streamId + 1)
+    return streamId
+  }
+}

package/src/enums.ts

@@ -0,0 +1,15 @@
+export enum GatewayHook {
+  Connect = 'Connect',
+  Disconnect = 'Disconnect',
+}
+
+export enum ProxyableTransportType {
+  WebSocket = 'WebSocket',
+  HTTP = 'HTTP',
+}
+
+export enum StreamTimeout {
+  Pull = 'Pull',
+  Consume = 'Consume',
+  Finish = 'Finish',
+}