@nmtjs/protocol 0.6.4 → 0.7.0

package/dist/server/protocol.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../lib/server/protocol.ts"],"sourcesContent":["import { type Callback, createPromise, defer, throwError } from '@nmtjs/common'\nimport { type Container, Hook, type Logger, Scope } from '@nmtjs/core'\nimport { concat, decodeNumber, encodeNumber } from '../common/binary.ts'\nimport type { ProtocolBlob, ProtocolBlobMetadata } from '../common/blob.ts'\nimport { ErrorCode, ServerMessageType } from '../common/enums.ts'\nimport type { ProtocolRPC } from '../common/types.ts'\nimport type { ProtocolApi, ProtocolApiCallResult } from './api.ts'\nimport {\n  Connection,\n  ConnectionContext,\n  type ConnectionOptions,\n} from './connection.ts'\nimport type { Format } from './format.ts'\nimport type { ProtocolRegistry } from './registry.ts'\nimport { ProtocolClientStream, ProtocolServerStream } from './stream.ts'\nimport type { Transport } from './transport.ts'\nimport { getFormat, type ResolveFormatParams } from './utils.ts'\n\nexport class ProtocolError extends Error {\n  code: string\n  data?: any\n\n  constructor(code: string, message?: string, data?: any) {\n    super(message)\n    this.code = code\n    this.data = data\n  }\n\n  get message() {\n    return `${this.code} ${super.message}`\n  }\n\n  toString() {\n    return `${this.code} ${this.message}`\n  }\n\n  toJSON() {\n    return {\n      code: this.code,\n      message: this.message,\n      data: this.data,\n    }\n  }\n}\n\nexport class ProtocolConnections {\n  readonly #collection = new Map<\n    string,\n    {\n      connection: Connection\n      context: ConnectionContext\n      transport: Transport\n    }\n  >()\n\n  constructor(\n    private readonly application: {\n      logger: Logger\n      registry: ProtocolRegistry\n      format: Format\n      container: Container\n    },\n  ) {}\n\n  get(connectionId: string) {\n    const connection = this.#collection.get(connectionId)\n    if (!connection) throwError('Connection not found')\n    return connection\n  }\n\n  async add<T>(\n    transport: Transport<any>,\n    options: ConnectionOptions<T>,\n    params: ResolveFormatParams,\n  ) {\n    const connection = new Connection(options)\n    const format = getFormat(this.application.format, params)\n    const container = this.application.container.fork(Scope.Connection)\n    const context = new ConnectionContext(container, format)\n\n    this.#collection.set(connection.id, { connection, context, transport })\n\n    await this.application.registry.hooks.call(\n      Hook.OnConnect,\n      { concurrent: false },\n      connection,\n    )\n\n    return { connection, context }\n  }\n\n  async remove(connectionId: string) {\n    const { connection, context } = this.get(connectionId)\n\n    this.application.registry.hooks.call(\n      Hook.OnDisconnect,\n      { concurrent: true },\n      connection,\n    )\n\n    this.#collection.delete(connectionId)\n\n    const { calls, serverStreams, clientStreams, rpcStreams, container } =\n      context\n\n    for (const call of calls.values()) {\n      call.reject(new Error('Connection closed'))\n    }\n\n    for (const stream of clientStreams.values()) {\n      stream.destroy(new Error('Connection closed'))\n    }\n\n    for (const stream of serverStreams.values()) {\n      stream.destroy(new Error('Connection closed'))\n    }\n\n    for (const stream of rpcStreams.values()) {\n      stream.abort(new Error('Connection closed'))\n    }\n\n    try {\n      await container.dispose()\n    } catch (error) {\n      this.application.logger.error(\n        { error, connection },\n        'Error during closing connection',\n      )\n    }\n  }\n}\n\nexport class ProtocolClientStreams {\n  constructor(private readonly connections: ProtocolConnections) {}\n\n  get(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { clientStreams } = context\n    const stream = clientStreams.get(streamId) ?? throwError('Stream not found')\n    return stream\n  }\n\n  remove(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { clientStreams } = context\n    clientStreams.get(streamId) || throwError('Stream not found')\n    clientStreams.delete(streamId)\n  }\n\n  add(\n    connectionId: string,\n    streamId: number,\n    metadata: ProtocolBlobMetadata,\n    read: Callback,\n  ) {\n    const { context } = this.connections.get(connectionId)\n    const { clientStreams } = context\n    const stream = new ProtocolClientStream(streamId, metadata, { read })\n    clientStreams.set(streamId, stream)\n    return stream\n  }\n\n  push(connectionId: string, streamId: number, chunk: ArrayBuffer) {\n    const stream = this.get(connectionId, streamId)\n    stream.push(Buffer.from(chunk))\n  }\n\n  end(connectionId: string, streamId: number) {\n    const stream = this.get(connectionId, streamId)\n    stream.push(null)\n    this.remove(connectionId, streamId)\n  }\n\n  abort(connectionId: string, streamId: number, error = new Error('Aborted')) {\n    const stream = this.get(connectionId, streamId)\n    stream.destroy(error)\n    this.remove(connectionId, streamId)\n  }\n}\n\nexport class ProtocolServerStreams {\n  constructor(private readonly connections: ProtocolConnections) {}\n\n  get(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { serverStreams } = context\n    const stream = serverStreams.get(streamId) ?? throwError('Stream not found')\n    return stream\n  }\n\n  add(connectionId: string, streamId: number, blob: ProtocolBlob) {\n    const { context } = this.connections.get(connectionId)\n    const { serverStreams } = context\n    const stream = new ProtocolServerStream(streamId, blob)\n    serverStreams.set(streamId, stream)\n    return stream\n  }\n\n  remove(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { serverStreams } = context\n    serverStreams.has(streamId) || throwError('Stream not found')\n    serverStreams.delete(streamId)\n  }\n\n  pull(connectionId: string, streamId: number) {\n    const stream = this.get(connectionId, streamId)\n    stream.resume()\n  }\n\n  abort(connectionId: string, streamId: number, error = new Error('Aborted')) {\n    const stream = this.get(connectionId, streamId)\n    stream.destroy(error)\n    this.remove(connectionId, streamId)\n  }\n}\n\nexport class Protocol {\n  readonly connections: ProtocolConnections\n  readonly clientStreams: ProtocolClientStreams\n  readonly serverStreams: ProtocolServerStreams\n\n  constructor(\n    protected readonly application: {\n      logger: Logger\n      format: Format\n      container: Container\n      registry: ProtocolRegistry\n      api: ProtocolApi\n    },\n  ) {\n    this.connections = new ProtocolConnections(this.application)\n    this.clientStreams = new ProtocolClientStreams(this.connections)\n    this.serverStreams = new ProtocolServerStreams(this.connections)\n  }\n\n  async rpc(\n    connectionId: string,\n    rpc: ProtocolRPC,\n    params: { signal?: AbortSignal } = {},\n  ) {\n    const { connection, context, transport } =\n      this.connections.get(connectionId)\n    const { calls, format } = context\n    const { callId, namespace, procedure, payload } = rpc\n    const abortController = new AbortController()\n    const signal = params.signal\n      ? AbortSignal.any([params.signal, abortController.signal])\n      : abortController.signal\n\n    const call = Object.assign(createPromise<ProtocolApiCallResult>(), {\n      abort: () => abortController.abort(),\n    })\n\n    calls.set(callId, call)\n    call.promise.finally(() => calls.delete(callId))\n\n    const callIdEncoded = encodeNumber(callId, 'Uint32')\n    const container = context.container.fork(Scope.Call)\n\n    try {\n      const response = await this.application.api.call({\n        connection,\n        container,\n        namespace,\n        payload,\n        procedure,\n        signal,\n      })\n\n      const responseEncoded = format.encoder.encodeRPC(\n        {\n          callId,\n          payload: response.output,\n        },\n        {\n          addStream: (blob) => {\n            const id = context.streamId++\n            const stream = this.serverStreams.add(connectionId, id, blob)\n            stream.on('data', (chunk) => {\n              stream.pause()\n              transport.send(\n                connection,\n                ServerMessageType.ServerStreamPush,\n                concat(\n                  encodeNumber(id, 'Uint32'),\n                  Buffer.from(chunk).buffer as ArrayBuffer,\n                ),\n              )\n            })\n            stream.on('error', (err) => {\n              transport.send(\n                connection,\n                ServerMessageType.ServerStreamAbort,\n                encodeNumber(id, 'Uint32'),\n              )\n            })\n            stream.on('end', () => {\n              transport.send(\n                connection,\n                ServerMessageType.ServerStreamEnd,\n                encodeNumber(id, 'Uint32'),\n              )\n            })\n            return stream\n          },\n          getStream: (id) => {\n            return this.clientStreams.get(connectionId, id)\n          },\n        },\n      )\n\n      if ('subscription' in response) {\n        throwError('Unimplemented')\n      } else if ('iterable' in response) {\n        transport.send(\n          connection,\n          ServerMessageType.RpcStreamResponse,\n          responseEncoded,\n        )\n        try {\n          const ab = new AbortController()\n          context.rpcStreams.set(callId, ab)\n          const iterable =\n            typeof response.iterable === 'function'\n              ? response.iterable()\n              : response.iterable\n          for await (const chunk of iterable) {\n            if (ab.signal.aborted) break\n            const chunkEncoded = format.encoder.encode(chunk)\n            transport.send(\n              connection,\n              ServerMessageType.RpcStreamChunk,\n              concat(callIdEncoded, chunkEncoded),\n            )\n          }\n        } catch (error) {\n          this.application.logger.error(error)\n          transport.send(\n            connection,\n            ServerMessageType.RpcStreamAbort,\n            callIdEncoded,\n          )\n        } finally {\n          context.rpcStreams.delete(callId)\n          response.onFinish && defer(response.onFinish)\n        }\n      } else {\n        transport.send(\n          connection,\n          ServerMessageType.RpcResponse,\n          responseEncoded,\n        )\n      }\n    } catch (error) {\n      if (error instanceof ProtocolError === false) {\n        this.application.logger.error(\n          { error, connection },\n          'Error during RPC call',\n        )\n\n        // biome-ignore lint/suspicious/noCatchAssign:\n        error = new ProtocolError(\n          ErrorCode.InternalServerError,\n          'Internal server error',\n        )\n      }\n\n      const payload = format.encoder.encodeRPC(\n        { callId, error },\n        {\n          addStream(blob) {\n            throwError('Cannot handle stream for error response')\n          },\n          getStream(id) {\n            throwError('Cannot handle stream for error response')\n          },\n        },\n      )\n      transport.send(connection, ServerMessageType.RpcResponse, payload)\n    } finally {\n      container.dispose().catch((error) => {\n        this.application.logger.error(\n          { error, connection },\n          \"Error during disposing connection's container\",\n        )\n      })\n    }\n  }\n\n  async rpcRaw(\n    connectionId: string,\n    buffer: ArrayBuffer,\n    params: { signal?: AbortSignal } = {},\n  ) {\n    const { connection, context, transport } =\n      this.connections.get(connectionId)\n\n    const { format } = context\n\n    const rpc = format.decoder.decodeRPC(buffer, {\n      addStream: (id, metadata) => {\n        return this.clientStreams.add(connectionId, id, metadata, (size) => {\n          transport.send(\n            connection,\n            ServerMessageType.ClientStreamPull,\n            concat(encodeNumber(id, 'Uint32'), encodeNumber(size, 'Uint32')),\n          )\n        })\n      },\n      getStream: (id) => {\n        return this.serverStreams.get(connectionId, id)\n      },\n    })\n\n    return await this.rpc(connectionId, rpc, params)\n  }\n\n  rpcAbort(connectionId: string, callId: number) {\n    const { context } = this.connections.get(connectionId)\n    const call = context.calls.get(callId) ?? throwError('Call not found')\n    call.abort()\n  }\n\n  rpcAbortRaw(connectionId: string, buffer: ArrayBuffer) {\n    const callId = decodeNumber(buffer, 'Uint32')\n    return this.rpcAbort(connectionId, callId)\n  }\n\n  rpcStreamAbort(connectionId: string, callId: number) {\n    const { context } = this.connections.get(connectionId)\n    const ab =\n      context.rpcStreams.get(callId) ?? throwError('Call stream not found')\n    ab.abort()\n  }\n\n  rpcStreamAbortRaw(connectionId: string, buffer: ArrayBuffer) {\n    const callId = decodeNumber(buffer, 'Uint32')\n    return this.rpcStreamAbort(connectionId, callId)\n  }\n\n  notify(connectionId: string, event, payload) {\n    throw Error('Unimplemented')\n  }\n}\n"],"names":["createPromise","defer","throwError","Hook","Scope","concat","decodeNumber","encodeNumber","ErrorCode","ServerMessageType","Connection","ConnectionContext","ProtocolClientStream","ProtocolServerStream","getFormat","ProtocolError","Error","code","data","constructor","message","toString","toJSON","ProtocolConnections","application","Map","get","connectionId","connection","add","transport","options","params","format","container","fork","context","set","id","registry","hooks","call","OnConnect","concurrent","remove","OnDisconnect","delete","calls","serverStreams","clientStreams","rpcStreams","values","reject","stream","destroy","abort","dispose","error","logger","ProtocolClientStreams","connections","streamId","metadata","read","push","chunk","Buffer","from","end","ProtocolServerStreams","blob","has","pull","resume","Protocol","rpc","callId","namespace","procedure","payload","abortController","AbortController","signal","AbortSignal","any","Object","assign","promise","finally","callIdEncoded","Call","response","api","responseEncoded","encoder","encodeRPC","output","addStream","on","pause","send","ServerStreamPush","buffer","err","ServerStreamAbort","ServerStreamEnd","getStream","RpcStreamResponse","ab","iterable","aborted","chunkEncoded","encode","RpcStreamChunk","RpcStreamAbort","onFinish","RpcResponse","InternalServerError","catch","rpcRaw","decoder","decodeRPC","size","ClientStreamPull","rpcAbort","rpcAbortRaw","rpcStreamAbort","rpcStreamAbortRaw","notify","event"],"mappings":"AAAA,SAAwBA,aAAa,EAAEC,KAAK,EAAEC,UAAU,QAAQ,gBAAe;AAC/E,SAAyBC,IAAI,EAAeC,KAAK,QAAQ,cAAa;AACtE,SAASC,MAAM,EAAEC,YAAY,EAAEC,YAAY,QAAQ,sBAAqB;AAExE,SAASC,SAAS,EAAEC,iBAAiB,QAAQ,qBAAoB;AAGjE,SACEC,UAAU,EACVC,iBAAiB,QAEZ,kBAAiB;AAGxB,SAASC,oBAAoB,EAAEC,oBAAoB,QAAQ,cAAa;AAExE,SAASC,SAAS,QAAkC,aAAY;AAEhE,OAAO,MAAMC,sBAAsBC;IACjCC,KAAY;IACZC,KAAU;IAEVC,YAAYF,IAAY,EAAEG,OAAgB,EAAEF,IAAU,CAAE;QACtD,KAAK,CAACE;QACN,IAAI,CAACH,IAAI,GAAGA;QACZ,IAAI,CAACC,IAAI,GAAGA;IACd;IAEA,IAAIE,UAAU;QACZ,OAAO,CAAC,EAAE,IAAI,CAACH,IAAI,CAAC,CAAC,EAAE,KAAK,CAACG,QAAQ,CAAC;IACxC;IAEAC,WAAW;QACT,OAAO,CAAC,EAAE,IAAI,CAACJ,IAAI,CAAC,CAAC,EAAE,IAAI,CAACG,OAAO,CAAC,CAAC;IACvC;IAEAE,SAAS;QACP,OAAO;YACLL,MAAM,IAAI,CAACA,IAAI;YACfG,SAAS,IAAI,CAACA,OAAO;YACrBF,MAAM,IAAI,CAACA,IAAI;QACjB;IACF;AACF;AAEA,OAAO,MAAMK;;IACF,CAAA,UAAW,CAOjB;IAEHJ,YACE,AAAiBK,WAKhB,CACD;aANiBA,cAAAA;aAVV,CAAA,UAAW,GAAG,IAAIC;IAgBxB;IAEHC,IAAIC,YAAoB,EAAE;QACxB,MAAMC,aAAa,IAAI,CAAC,CAAA,UAAW,CAACF,GAAG,CAACC;QACxC,IAAI,CAACC,YAAY1B,WAAW;QAC5B,OAAO0B;IACT;IAEA,MAAMC,IACJC,SAAyB,EACzBC,OAA6B,EAC7BC,MAA2B,EAC3B;QACA,MAAMJ,aAAa,IAAIlB,WAAWqB;QAClC,MAAME,SAASnB,UAAU,IAAI,CAACU,WAAW,CAACS,MAAM,EAAED;QAClD,MAAME,YAAY,IAAI,CAACV,WAAW,CAACU,SAAS,CAACC,IAAI,CAAC/B,MAAMM,UAAU;QAClE,MAAM0B,UAAU,IAAIzB,kBAAkBuB,WAAWD;QAEjD,IAAI,CAAC,CAAA,UAAW,CAACI,GAAG,CAACT,WAAWU,EAAE,EAAE;YAAEV;YAAYQ;YAASN;QAAU;QAErE,MAAM,IAAI,CAACN,WAAW,CAACe,QAAQ,CAACC,KAAK,CAACC,IAAI,CACxCtC,KAAKuC,SAAS,EACd;YAAEC,YAAY;QAAM,GACpBf;QAGF,OAAO;YAAEA;YAAYQ;QAAQ;IAC/B;IAEA,MAAMQ,OAAOjB,YAAoB,EAAE;QACjC,MAAM,EAAEC,UAAU,EAAEQ,OAAO,EAAE,GAAG,IAAI,CAACV,GAAG,CAACC;QAEzC,IAAI,CAACH,WAAW,CAACe,QAAQ,CAACC,KAAK,CAACC,IAAI,CAClCtC,KAAK0C,YAAY,EACjB;YAAEF,YAAY;QAAK,GACnBf;QAGF,IAAI,CAAC,CAAA,UAAW,CAACkB,MAAM,CAACnB;QAExB,MAAM,EAAEoB,KAAK,EAAEC,aAAa,EAAEC,aAAa,EAAEC,UAAU,EAAEhB,SAAS,EAAE,GAClEE;QAEF,KAAK,MAAMK,QAAQM,MAAMI,MAAM,GAAI;YACjCV,KAAKW,MAAM,CAAC,IAAIpC,MAAM;QACxB;QAEA,KAAK,MAAMqC,UAAUJ,cAAcE,MAAM,GAAI;YAC3CE,OAAOC,OAAO,CAAC,IAAItC,MAAM;QAC3B;QAEA,KAAK,MAAMqC,UAAUL,cAAcG,MAAM,GAAI;YAC3CE,OAAOC,OAAO,CAAC,IAAItC,MAAM;QAC3B;QAEA,KAAK,MAAMqC,UAAUH,WAAWC,MAAM,GAAI;YACxCE,OAAOE,KAAK,CAAC,IAAIvC,MAAM;QACzB;QAEA,IAAI;YACF,MAAMkB,UAAUsB,OAAO;QACzB,EAAE,OAAOC,OAAO;YACd,IAAI,CAACjC,WAAW,CAACkC,MAAM,CAACD,KAAK,CAC3B;gBAAEA;gBAAO7B;YAAW,GACpB;QAEJ;IACF;AACF;AAEA,OAAO,MAAM+B;;IACXxC,YAAY,AAAiByC,WAAgC,CAAE;aAAlCA,cAAAA;IAAmC;IAEhElC,IAAIC,YAAoB,EAAEkC,QAAgB,EAAE;QAC1C,MAAM,EAAEzB,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAM,EAAEsB,aAAa,EAAE,GAAGb;QAC1B,MAAMiB,SAASJ,cAAcvB,GAAG,CAACmC,aAAa3D,WAAW;QACzD,OAAOmD;IACT;IAEAT,OAAOjB,YAAoB,EAAEkC,QAAgB,EAAE;QAC7C,MAAM,EAAEzB,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAM,EAAEsB,aAAa,EAAE,GAAGb;QAC1Ba,cAAcvB,GAAG,CAACmC,aAAa3D,WAAW;QAC1C+C,cAAcH,MAAM,CAACe;IACvB;IAEAhC,IACEF,YAAoB,EACpBkC,QAAgB,EAChBC,QAA8B,EAC9BC,IAAc,EACd;QACA,MAAM,EAAE3B,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAM,EAAEsB,aAAa,EAAE,GAAGb;QAC1B,MAAMiB,SAAS,IAAIzC,qBAAqBiD,UAAUC,UAAU;YAAEC;QAAK;QACnEd,cAAcZ,GAAG,CAACwB,UAAUR;QAC5B,OAAOA;IACT;IAEAW,KAAKrC,YAAoB,EAAEkC,QAAgB,EAAEI,KAAkB,EAAE;QAC/D,MAAMZ,SAAS,IAAI,CAAC3B,GAAG,CAACC,cAAckC;QACtCR,OAAOW,IAAI,CAACE,OAAOC,IAAI,CAACF;IAC1B;IAEAG,IAAIzC,YAAoB,EAAEkC,QAAgB,EAAE;QAC1C,MAAMR,SAAS,IAAI,CAAC3B,GAAG,CAACC,cAAckC;QACtCR,OAAOW,IAAI,CAAC;QACZ,IAAI,CAACpB,MAAM,CAACjB,cAAckC;IAC5B;IAEAN,MAAM5B,YAAoB,EAAEkC,QAAgB,EAAEJ,QAAQ,IAAIzC,MAAM,UAAU,EAAE;QAC1E,MAAMqC,SAAS,IAAI,CAAC3B,GAAG,CAACC,cAAckC;QACtCR,OAAOC,OAAO,CAACG;QACf,IAAI,CAACb,MAAM,CAACjB,cAAckC;IAC5B;AACF;AAEA,OAAO,MAAMQ;;IACXlD,YAAY,AAAiByC,WAAgC,CAAE;aAAlCA,cAAAA;IAAmC;IAEhElC,IAAIC,YAAoB,EAAEkC,QAAgB,EAAE;QAC1C,MAAM,EAAEzB,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAM,EAAEqB,aAAa,EAAE,GAAGZ;QAC1B,MAAMiB,SAASL,cAActB,GAAG,CAACmC,aAAa3D,WAAW;QACzD,OAAOmD;IACT;IAEAxB,IAAIF,YAAoB,EAAEkC,QAAgB,EAAES,IAAkB,EAAE;QAC9D,MAAM,EAAElC,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAM,EAAEqB,aAAa,EAAE,GAAGZ;QAC1B,MAAMiB,SAAS,IAAIxC,qBAAqBgD,UAAUS;QAClDtB,cAAcX,GAAG,CAACwB,UAAUR;QAC5B,OAAOA;IACT;IAEAT,OAAOjB,YAAoB,EAAEkC,QAAgB,EAAE;QAC7C,MAAM,EAAEzB,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAM,EAAEqB,aAAa,EAAE,GAAGZ;QAC1BY,cAAcuB,GAAG,CAACV,aAAa3D,WAAW;QAC1C8C,cAAcF,MAAM,CAACe;IACvB;IAEAW,KAAK7C,YAAoB,EAAEkC,QAAgB,EAAE;QAC3C,MAAMR,SAAS,IAAI,CAAC3B,GAAG,CAACC,cAAckC;QACtCR,OAAOoB,MAAM;IACf;IAEAlB,MAAM5B,YAAoB,EAAEkC,QAAgB,EAAEJ,QAAQ,IAAIzC,MAAM,UAAU,EAAE;QAC1E,MAAMqC,SAAS,IAAI,CAAC3B,GAAG,CAACC,cAAckC;QACtCR,OAAOC,OAAO,CAACG;QACf,IAAI,CAACb,MAAM,CAACjB,cAAckC;IAC5B;AACF;AAEA,OAAO,MAAMa;;IACFd,YAAgC;IAChCX,cAAoC;IACpCD,cAAoC;IAE7C7B,YACE,AAAmBK,WAMlB,CACD;aAPmBA,cAAAA;QAQnB,IAAI,CAACoC,WAAW,GAAG,IAAIrC,oBAAoB,IAAI,CAACC,WAAW;QAC3D,IAAI,CAACyB,aAAa,GAAG,IAAIU,sBAAsB,IAAI,CAACC,WAAW;QAC/D,IAAI,CAACZ,aAAa,GAAG,IAAIqB,sBAAsB,IAAI,CAACT,WAAW;IACjE;IAEA,MAAMe,IACJhD,YAAoB,EACpBgD,GAAgB,EAChB3C,SAAmC,CAAC,CAAC,EACrC;QACA,MAAM,EAAEJ,UAAU,EAAEQ,OAAO,EAAEN,SAAS,EAAE,GACtC,IAAI,CAAC8B,WAAW,CAAClC,GAAG,CAACC;QACvB,MAAM,EAAEoB,KAAK,EAAEd,MAAM,EAAE,GAAGG;QAC1B,MAAM,EAAEwC,MAAM,EAAEC,SAAS,EAAEC,SAAS,EAAEC,OAAO,EAAE,GAAGJ;QAClD,MAAMK,kBAAkB,IAAIC;QAC5B,MAAMC,SAASlD,OAAOkD,MAAM,GACxBC,YAAYC,GAAG,CAAC;YAACpD,OAAOkD,MAAM;YAAEF,gBAAgBE,MAAM;SAAC,IACvDF,gBAAgBE,MAAM;QAE1B,MAAMzC,OAAO4C,OAAOC,MAAM,CAACtF,iBAAwC;YACjEuD,OAAO,IAAMyB,gBAAgBzB,KAAK;QACpC;QAEAR,MAAMV,GAAG,CAACuC,QAAQnC;QAClBA,KAAK8C,OAAO,CAACC,OAAO,CAAC,IAAMzC,MAAMD,MAAM,CAAC8B;QAExC,MAAMa,gBAAgBlF,aAAaqE,QAAQ;QAC3C,MAAM1C,YAAYE,QAAQF,SAAS,CAACC,IAAI,CAAC/B,MAAMsF,IAAI;QAEnD,IAAI;YACF,MAAMC,WAAW,MAAM,IAAI,CAACnE,WAAW,CAACoE,GAAG,CAACnD,IAAI,CAAC;gBAC/Cb;gBACAM;gBACA2C;gBACAE;gBACAD;gBACAI;YACF;YAEA,MAAMW,kBAAkB5D,OAAO6D,OAAO,CAACC,SAAS,CAC9C;gBACEnB;gBACAG,SAASY,SAASK,MAAM;YAC1B,GACA;gBACEC,WAAW,CAAC3B;oBACV,MAAMhC,KAAKF,QAAQyB,QAAQ;oBAC3B,MAAMR,SAAS,IAAI,CAACL,aAAa,CAACnB,GAAG,CAACF,cAAcW,IAAIgC;oBACxDjB,OAAO6C,EAAE,CAAC,QAAQ,CAACjC;wBACjBZ,OAAO8C,KAAK;wBACZrE,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkB4F,gBAAgB,EAClChG,OACEE,aAAa+B,IAAI,WACjB4B,OAAOC,IAAI,CAACF,OAAOqC,MAAM;oBAG/B;oBACAjD,OAAO6C,EAAE,CAAC,SAAS,CAACK;wBAClBzE,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkB+F,iBAAiB,EACnCjG,aAAa+B,IAAI;oBAErB;oBACAe,OAAO6C,EAAE,CAAC,OAAO;wBACfpE,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkBgG,eAAe,EACjClG,aAAa+B,IAAI;oBAErB;oBACA,OAAOe;gBACT;gBACAqD,WAAW,CAACpE;oBACV,OAAO,IAAI,CAACW,aAAa,CAACvB,GAAG,CAACC,cAAcW;gBAC9C;YACF;YAGF,IAAI,kBAAkBqD,UAAU;gBAC9BzF,WAAW;YACb,OAAO,IAAI,cAAcyF,UAAU;gBACjC7D,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkBkG,iBAAiB,EACnCd;gBAEF,IAAI;oBACF,MAAMe,KAAK,IAAI3B;oBACf7C,QAAQc,UAAU,CAACb,GAAG,CAACuC,QAAQgC;oBAC/B,MAAMC,WACJ,OAAOlB,SAASkB,QAAQ,KAAK,aACzBlB,SAASkB,QAAQ,KACjBlB,SAASkB,QAAQ;oBACvB,WAAW,MAAM5C,SAAS4C,SAAU;wBAClC,IAAID,GAAG1B,MAAM,CAAC4B,OAAO,EAAE;wBACvB,MAAMC,eAAe9E,OAAO6D,OAAO,CAACkB,MAAM,CAAC/C;wBAC3CnC,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkBwG,cAAc,EAChC5G,OAAOoF,eAAesB;oBAE1B;gBACF,EAAE,OAAOtD,OAAO;oBACd,IAAI,CAACjC,WAAW,CAACkC,MAAM,CAACD,KAAK,CAACA;oBAC9B3B,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkByG,cAAc,EAChCzB;gBAEJ,SAAU;oBACRrD,QAAQc,UAAU,CAACJ,MAAM,CAAC8B;oBAC1Be,SAASwB,QAAQ,IAAIlH,MAAM0F,SAASwB,QAAQ;gBAC9C;YACF,OAAO;gBACLrF,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkB2G,WAAW,EAC7BvB;YAEJ;QACF,EAAE,OAAOpC,OAAO;YACd,IAAIA,iBAAiB1C,kBAAkB,OAAO;gBAC5C,IAAI,CAACS,WAAW,CAACkC,MAAM,CAACD,KAAK,CAC3B;oBAAEA;oBAAO7B;gBAAW,GACpB;gBAIF6B,QAAQ,IAAI1C,cACVP,UAAU6G,mBAAmB,EAC7B;YAEJ;YAEA,MAAMtC,UAAU9C,OAAO6D,OAAO,CAACC,SAAS,CACtC;gBAAEnB;gBAAQnB;YAAM,GAChB;gBACEwC,WAAU3B,IAAI;oBACZpE,WAAW;gBACb;gBACAwG,WAAUpE,EAAE;oBACVpC,WAAW;gBACb;YACF;YAEF4B,UAAUsE,IAAI,CAACxE,YAAYnB,kBAAkB2G,WAAW,EAAErC;QAC5D,SAAU;YACR7C,UAAUsB,OAAO,GAAG8D,KAAK,CAAC,CAAC7D;gBACzB,IAAI,CAACjC,WAAW,CAACkC,MAAM,CAACD,KAAK,CAC3B;oBAAEA;oBAAO7B;gBAAW,GACpB;YAEJ;QACF;IACF;IAEA,MAAM2F,OACJ5F,YAAoB,EACpB2E,MAAmB,EACnBtE,SAAmC,CAAC,CAAC,EACrC;QACA,MAAM,EAAEJ,UAAU,EAAEQ,OAAO,EAAEN,SAAS,EAAE,GACtC,IAAI,CAAC8B,WAAW,CAAClC,GAAG,CAACC;QAEvB,MAAM,EAAEM,MAAM,EAAE,GAAGG;QAEnB,MAAMuC,MAAM1C,OAAOuF,OAAO,CAACC,SAAS,CAACnB,QAAQ;YAC3CL,WAAW,CAAC3D,IAAIwB;gBACd,OAAO,IAAI,CAACb,aAAa,CAACpB,GAAG,CAACF,cAAcW,IAAIwB,UAAU,CAAC4D;oBACzD5F,UAAUsE,IAAI,CACZxE,YACAnB,kBAAkBkH,gBAAgB,EAClCtH,OAAOE,aAAa+B,IAAI,WAAW/B,aAAamH,MAAM;gBAE1D;YACF;YACAhB,WAAW,CAACpE;gBACV,OAAO,IAAI,CAACU,aAAa,CAACtB,GAAG,CAACC,cAAcW;YAC9C;QACF;QAEA,OAAO,MAAM,IAAI,CAACqC,GAAG,CAAChD,cAAcgD,KAAK3C;IAC3C;IAEA4F,SAASjG,YAAoB,EAAEiD,MAAc,EAAE;QAC7C,MAAM,EAAExC,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAMc,OAAOL,QAAQW,KAAK,CAACrB,GAAG,CAACkD,WAAW1E,WAAW;QACrDuC,KAAKc,KAAK;IACZ;IAEAsE,YAAYlG,YAAoB,EAAE2E,MAAmB,EAAE;QACrD,MAAM1B,SAAStE,aAAagG,QAAQ;QACpC,OAAO,IAAI,CAACsB,QAAQ,CAACjG,cAAciD;IACrC;IAEAkD,eAAenG,YAAoB,EAAEiD,MAAc,EAAE;QACnD,MAAM,EAAExC,OAAO,EAAE,GAAG,IAAI,CAACwB,WAAW,CAAClC,GAAG,CAACC;QACzC,MAAMiF,KACJxE,QAAQc,UAAU,CAACxB,GAAG,CAACkD,WAAW1E,WAAW;QAC/C0G,GAAGrD,KAAK;IACV;IAEAwE,kBAAkBpG,YAAoB,EAAE2E,MAAmB,EAAE;QAC3D,MAAM1B,SAAStE,aAAagG,QAAQ;QACpC,OAAO,IAAI,CAACwB,cAAc,CAACnG,cAAciD;IAC3C;IAEAoD,OAAOrG,YAAoB,EAAEsG,KAAK,EAAElD,OAAO,EAAE;QAC3C,MAAM/D,MAAM;IACd;AACF"}
+{"mappings":"AAAA,SAAwB,eAAe,OAAO,kBAAkB,eAAe;AAC/E,SAAyB,MAAmB,aAAa,aAAa;AACtE,SAAS,QAAQ,cAAc,oBAAoB,qBAAqB;AAExE,SAAS,WAAW,yBAAyB,oBAAoB;AAGjE,SACE,YACA,yBAEK,iBAAiB;AAGxB,SAAS,sBAAsB,4BAA4B,aAAa;AAExE,SAAS,iBAA2C,YAAY;AAEhE,OAAO,MAAM,sBAAsB,MAAM;CACvC;CACA;CAEA,YAAYA,MAAcC,SAAkBC,MAAY;AACtD,QAAM,QAAQ;AACd,OAAK,OAAO;AACZ,OAAK,OAAO;CACb;CAED,IAAI,UAAU;AACZ,UAAQ,EAAE,KAAK,KAAK,GAAG,MAAM,QAAQ;CACtC;CAED,WAAW;AACT,UAAQ,EAAE,KAAK,KAAK,GAAG,KAAK,QAAQ;CACrC;CAED,SAAS;AACP,SAAO;GACL,MAAM,KAAK;GACX,SAAS,KAAK;GACd,MAAM,KAAK;EACZ;CACF;AACF;AAED,OAAO,MAAM,oBAAoB;CAC/B,AAASC,cAAc,IAAI;CAS3B,YACmBC,aAMjB;OANiB;CAMf;CAEJ,IAAIC,cAAsB;EACxB,MAAM,aAAa,KAAKF,YAAY,IAAI,aAAa;AACrD,OAAK,WAAY,YAAW,uBAAuB;AACnD,SAAO;CACR;CAED,MAAM,IACJG,WACAC,SACAC,QACA;EACA,MAAM,aAAa,IAAI,WAAW;EAClC,MAAM,SAAS,UAAU,KAAK,YAAY,QAAQ,OAAO;EACzD,MAAM,YAAY,KAAK,YAAY,UAAU,KAAK,MAAM,WAAW;EACnE,MAAM,UAAU,IAAI,kBAAkB,WAAW;AAEjD,OAAKL,YAAY,IAAI,WAAW,IAAI;GAAE;GAAY;GAAS;EAAW,EAAC;AAEvE,QAAM,KAAK,YAAY,SAAS,MAAM,KACpC,KAAK,WACL,EAAE,YAAY,MAAO,GACrB,WACD;AAED,SAAO;GAAE;GAAY;EAAS;CAC/B;CAED,MAAM,OAAOE,cAAsB;EACjC,MAAM,EAAE,YAAY,SAAS,GAAG,KAAK,IAAI,aAAa;AAEtD,OAAK,YAAY,SAAS,MAAM,KAC9B,KAAK,cACL,EAAE,YAAY,KAAM,GACpB,WACD;AAED,OAAKF,YAAY,OAAO,aAAa;EAErC,MAAM,EAAE,OAAO,eAAe,eAAe,YAAY,WAAW,GAClE;AAEF,OAAK,MAAM,QAAQ,MAAM,QAAQ,EAAE;AACjC,QAAK,MAAM,IAAI,MAAM,qBAAqB;EAC3C;AAED,OAAK,MAAM,UAAU,cAAc,QAAQ,EAAE;AAC3C,UAAO,QAAQ,IAAI,MAAM,qBAAqB;EAC/C;AAED,OAAK,MAAM,UAAU,cAAc,QAAQ,EAAE;AAC3C,UAAO,QAAQ,IAAI,MAAM,qBAAqB;EAC/C;AAED,OAAK,MAAM,UAAU,WAAW,QAAQ,EAAE;AACxC,UAAO,MAAM,IAAI,MAAM,qBAAqB;EAC7C;AAED,MAAI;AACF,SAAM,UAAU,SAAS;EAC1B,SAAQ,OAAO;AACd,QAAK,YAAY,OAAO,MACtB;IAAE;IAAO;GAAY,GACrB,kCACD;EACF;CACF;AACF;AAED,OAAO,MAAM,sBAAsB;CACjC,YAA6BM,aAAkC;OAAlC;CAAoC;CAEjE,IAAIJ,cAAsBK,UAAkB;EAC1C,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,EAAE,eAAe,GAAG;EAC1B,MAAM,SAAS,cAAc,IAAI,SAAS,IAAI,WAAW,mBAAmB;AAC5E,SAAO;CACR;CAED,OAAOL,cAAsBK,UAAkB;EAC7C,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,EAAE,eAAe,GAAG;AAC1B,gBAAc,IAAI,SAAS,IAAI,WAAW,mBAAmB;AAC7D,gBAAc,OAAO,SAAS;CAC/B;CAED,IACEL,cACAK,UACAC,UACAC,MACA;EACA,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,EAAE,eAAe,GAAG;EAC1B,MAAM,SAAS,IAAI,qBAAqB,UAAU,UAAU,EAAE,KAAM;AACpE,gBAAc,IAAI,UAAU,OAAO;AACnC,SAAO;CACR;CAED,KAAKP,cAAsBK,UAAkBG,OAAoB;EAC/D,MAAM,SAAS,KAAK,IAAI,cAAc,SAAS;AAC/C,SAAO,KAAK,OAAO,KAAK,MAAM,CAAC;CAChC;CAED,IAAIR,cAAsBK,UAAkB;EAC1C,MAAM,SAAS,KAAK,IAAI,cAAc,SAAS;AAC/C,SAAO,KAAK,KAAK;AACjB,OAAK,OAAO,cAAc,SAAS;CACpC;CAED,MAAML,cAAsBK,UAAkB,QAAQ,IAAI,MAAM,YAAY;EAC1E,MAAM,SAAS,KAAK,IAAI,cAAc,SAAS;AAC/C,SAAO,QAAQ,MAAM;AACrB,OAAK,OAAO,cAAc,SAAS;CACpC;AACF;AAED,OAAO,MAAM,sBAAsB;CACjC,YAA6BD,aAAkC;OAAlC;CAAoC;CAEjE,IAAIJ,cAAsBK,UAAkB;EAC1C,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,EAAE,eAAe,GAAG;EAC1B,MAAM,SAAS,cAAc,IAAI,SAAS,IAAI,WAAW,mBAAmB;AAC5E,SAAO;CACR;CAED,IAAIL,cAAsBK,UAAkBI,MAAoB;EAC9D,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,EAAE,eAAe,GAAG;EAC1B,MAAM,SAAS,IAAI,qBAAqB,UAAU;AAClD,gBAAc,IAAI,UAAU,OAAO;AACnC,SAAO;CACR;CAED,OAAOT,cAAsBK,UAAkB;EAC7C,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,EAAE,eAAe,GAAG;AAC1B,gBAAc,IAAI,SAAS,IAAI,WAAW,mBAAmB;AAC7D,gBAAc,OAAO,SAAS;CAC/B;CAED,KAAKL,cAAsBK,UAAkB;EAC3C,MAAM,SAAS,KAAK,IAAI,cAAc,SAAS;AAC/C,SAAO,QAAQ;CAChB;CAED,MAAML,cAAsBK,UAAkB,QAAQ,IAAI,MAAM,YAAY;EAC1E,MAAM,SAAS,KAAK,IAAI,cAAc,SAAS;AAC/C,SAAO,QAAQ,MAAM;AACrB,OAAK,OAAO,cAAc,SAAS;CACpC;AACF;AAED,OAAO,MAAM,SAAS;CACpB,AAAS;CACT,AAAS;CACT,AAAS;CAET,YACqBK,aAOnB;OAPmB;AAQnB,OAAK,cAAc,IAAI,oBAAoB,KAAK;AAChD,OAAK,gBAAgB,IAAI,sBAAsB,KAAK;AACpD,OAAK,gBAAgB,IAAI,sBAAsB,KAAK;CACrD;CAED,MAAM,IACJV,cACAW,KACAC,SAAmC,CAAE,GACrC;EACA,MAAM,EAAE,YAAY,SAAS,WAAW,GACtC,KAAK,YAAY,IAAI,aAAa;EACpC,MAAM,EAAE,OAAO,QAAQ,GAAG;EAC1B,MAAM,EAAE,QAAQ,WAAW,WAAW,SAAS,GAAG;EAClD,MAAM,kBAAkB,IAAI;EAC5B,MAAM,SAAS,OAAO,SAClB,YAAY,IAAI,CAAC,OAAO,QAAQ,gBAAgB,MAAO,EAAC,GACxD,gBAAgB;EAEpB,MAAM,OAAO,OAAO,OAAO,eAAsC,EAAE,EACjE,OAAO,MAAM,gBAAgB,OAAO,CACrC,EAAC;AAEF,QAAM,IAAI,QAAQ,KAAK;AACvB,OAAK,QAAQ,QAAQ,MAAM,MAAM,OAAO,OAAO,CAAC;EAEhD,MAAM,gBAAgB,aAAa,QAAQ,SAAS;EACpD,MAAM,YAAY,QAAQ,UAAU,KAAK,MAAM,KAAK;AAEpD,MAAI;GACF,MAAM,WAAW,MAAM,KAAK,YAAY,IAAI,KAAK;IAC/C;IACA;IACA;IACA;IACA;IACA;GACD,EAAC;GAEF,MAAM,kBAAkB,OAAO,QAAQ,UACrC;IACE;IACA,SAAS,SAAS;GACnB,GACD;IACE,WAAW,CAAC,SAAS;KACnB,MAAM,KAAK,QAAQ;KACnB,MAAM,SAAS,KAAK,cAAc,IAAI,cAAc,IAAI,KAAK;AAC7D,YAAO,GAAG,QAAQ,CAAC,UAAU;AAC3B,aAAO,OAAO;MACd,MAAM,MAAM,OAAO,KAAK,MAAM;AAC9B,gBAAU,KACR,YACA,kBAAkB,kBAClB,OACE,aAAa,IAAI,SAAS,EAC1B,AAAC,IAAI,OAAuB,MAC1B,IAAI,YACJ,IAAI,aAAa,IAAI,WACtB,CACF,CACF;KACF,EAAC;AACF,YAAO,GAAG,SAAS,CAAC,QAAQ;AAC1B,gBAAU,KACR,YACA,kBAAkB,mBAClB,aAAa,IAAI,SAAS,CAC3B;KACF,EAAC;AACF,YAAO,GAAG,OAAO,MAAM;AACrB,gBAAU,KACR,YACA,kBAAkB,iBAClB,aAAa,IAAI,SAAS,CAC3B;KACF,EAAC;AACF,YAAO;IACR;IACD,WAAW,CAAC,OAAO;AACjB,YAAO,KAAK,cAAc,IAAI,cAAc,GAAG;IAChD;GACF,EACF;AAED,OAAI,kBAAkB,UAAU;AAC9B,eAAW,gBAAgB;GAC5B,WAAU,cAAc,UAAU;AACjC,cAAU,KACR,YACA,kBAAkB,mBAClB,gBACD;AACD,QAAI;KACF,MAAM,KAAK,IAAI;AACf,aAAQ,WAAW,IAAI,QAAQ,GAAG;KAClC,MAAM,kBACG,SAAS,aAAa,aACzB,SAAS,UAAU,GACnB,SAAS;AACf,gBAAW,MAAM,SAAS,UAAU;AAClC,UAAI,GAAG,OAAO,QAAS;MACvB,MAAM,eAAe,OAAO,QAAQ,OAAO,MAAM;AACjD,gBAAU,KACR,YACA,kBAAkB,gBAClB,OAAO,eAAe,aAAa,CACpC;KACF;IACF,SAAQ,OAAO;AACd,UAAK,YAAY,OAAO,MAAM,MAAM;AACpC,eAAU,KACR,YACA,kBAAkB,gBAClB,cACD;IACF,UAAS;AACR,aAAQ,WAAW,OAAO,OAAO;AACjC,cAAS,YAAY,MAAM,SAAS,SAAS;IAC9C;GACF,OAAM;AACL,cAAU,KACR,YACA,kBAAkB,aAClB,gBACD;GACF;EACF,SAAQ,OAAO;AACd,OAAI,iBAAiB,kBAAkB,OAAO;AAC5C,SAAK,YAAY,OAAO,MACtB;KAAE;KAAO;IAAY,GACrB,wBACD;AAGD,YAAQ,IAAI,cACV,UAAU,qBACV;GAEH;GAED,MAAM,UAAU,OAAO,QAAQ,UAC7B;IAAE;IAAQ;GAAO,GACjB;IACE,UAAU,MAAM;AACd,gBAAW,0CAA0C;IACtD;IACD,UAAU,IAAI;AACZ,gBAAW,0CAA0C;IACtD;GACF,EACF;AACD,aAAU,KAAK,YAAY,kBAAkB,aAAa,QAAQ;EACnE,UAAS;AACR,aAAU,SAAS,CAAC,MAAM,CAAC,UAAU;AACnC,SAAK,YAAY,OAAO,MACtB;KAAE;KAAO;IAAY,GACrB,gDACD;GACF,EAAC;EACH;CACF;CAED,MAAM,OACJZ,cACAa,QACAD,SAAmC,CAAE,GACrC;EACA,MAAM,EAAE,YAAY,SAAS,WAAW,GACtC,KAAK,YAAY,IAAI,aAAa;EAEpC,MAAM,EAAE,QAAQ,GAAG;EAEnB,MAAM,MAAM,OAAO,QAAQ,UAAU,QAAQ;GAC3C,WAAW,CAAC,IAAI,aAAa;IAC3B,MAAM,SAAS,KAAK,cAAc,IAChC,cACA,IACA,UACA,CAAC,SAAS;AACR,eAAU,KACR,YACA,kBAAkB,kBAClB,OAAO,aAAa,IAAI,SAAS,EAAE,aAAa,MAAM,SAAS,CAAC,CACjE;IACF,EACF;AACD,WAAO;GACR;GACD,WAAW,CAAC,OAAO;AACjB,WAAO,KAAK,cAAc,IAAI,cAAc,GAAG;GAChD;EACF,EAAC;AAEF,SAAO,MAAM,KAAK,IAAI,cAAc,KAAK,OAAO;CACjD;CAED,SAASZ,cAAsBc,QAAgB;EAC7C,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,OAAO,QAAQ,MAAM,IAAI,OAAO,IAAI,WAAW,iBAAiB;AACtE,OAAK,OAAO;CACb;CAED,YAAYd,cAAsBa,QAAqB;EACrD,MAAM,SAAS,aAAa,QAAQ,SAAS;AAC7C,SAAO,KAAK,SAAS,cAAc,OAAO;CAC3C;CAED,eAAeb,cAAsBc,QAAgB;EACnD,MAAM,EAAE,SAAS,GAAG,KAAK,YAAY,IAAI,aAAa;EACtD,MAAM,KACJ,QAAQ,WAAW,IAAI,OAAO,IAAI,WAAW,wBAAwB;AACvE,KAAG,OAAO;CACX;CAED,kBAAkBd,cAAsBa,QAAqB;EAC3D,MAAM,SAAS,aAAa,QAAQ,SAAS;AAC7C,SAAO,KAAK,eAAe,cAAc,OAAO;CACjD;CAED,OAAOb,cAAsB,OAAO,SAAS;AAC3C,QAAM,MAAM,gBAAgB;CAC7B;AACF","names":["code: string","message?: string","data?: any","#collection","application: {\n      logger: Logger\n      registry: ProtocolRegistry\n      format: Format\n      container: Container\n    }","connectionId: string","transport: Transport<any>","options: ConnectionOptions<T>","params: ResolveFormatParams","connections: ProtocolConnections","streamId: number","metadata: ProtocolBlobMetadata","read: Callback","chunk: ArrayBuffer","blob: ProtocolBlob","application: {\n      logger: Logger\n      format: Format\n      container: Container\n      registry: ProtocolRegistry\n      api: ProtocolApi\n    }","rpc: ProtocolRPC","params: { signal?: AbortSignal }","buffer: ArrayBuffer","callId: number"],"sources":["src/server/protocol.ts"],"sourcesContent":["import { type Callback, createPromise, defer, throwError } from '@nmtjs/common'\nimport { type Container, Hook, type Logger, Scope } from '@nmtjs/core'\nimport { concat, decodeNumber, encodeNumber } from '../common/binary.ts'\nimport type { ProtocolBlob, ProtocolBlobMetadata } from '../common/blob.ts'\nimport { ErrorCode, ServerMessageType } from '../common/enums.ts'\nimport type { ProtocolRPC } from '../common/types.ts'\nimport type { ProtocolApi, ProtocolApiCallResult } from './api.ts'\nimport {\n  Connection,\n  ConnectionContext,\n  type ConnectionOptions,\n} from './connection.ts'\nimport type { Format } from './format.ts'\nimport type { ProtocolRegistry } from './registry.ts'\nimport { ProtocolClientStream, ProtocolServerStream } from './stream.ts'\nimport type { Transport } from './transport.ts'\nimport { getFormat, type ResolveFormatParams } from './utils.ts'\n\nexport class ProtocolError extends Error {\n  code: string\n  data?: any\n\n  constructor(code: string, message?: string, data?: any) {\n    super(message)\n    this.code = code\n    this.data = data\n  }\n\n  get message() {\n    return `${this.code} ${super.message}`\n  }\n\n  toString() {\n    return `${this.code} ${this.message}`\n  }\n\n  toJSON() {\n    return {\n      code: this.code,\n      message: this.message,\n      data: this.data,\n    }\n  }\n}\n\nexport class ProtocolConnections {\n  readonly #collection = new Map<\n    string,\n    {\n      connection: Connection\n      context: ConnectionContext\n      transport: Transport\n    }\n  >()\n\n  constructor(\n    private readonly application: {\n      logger: Logger\n      registry: ProtocolRegistry\n      format: Format\n      container: Container\n    },\n  ) {}\n\n  get(connectionId: string) {\n    const connection = this.#collection.get(connectionId)\n    if (!connection) throwError('Connection not found')\n    return connection\n  }\n\n  async add<T>(\n    transport: Transport<any>,\n    options: ConnectionOptions<T>,\n    params: ResolveFormatParams,\n  ) {\n    const connection = new Connection(options)\n    const format = getFormat(this.application.format, params)\n    const container = this.application.container.fork(Scope.Connection)\n    const context = new ConnectionContext(container, format)\n\n    this.#collection.set(connection.id, { connection, context, transport })\n\n    await this.application.registry.hooks.call(\n      Hook.OnConnect,\n      { concurrent: false },\n      connection,\n    )\n\n    return { connection, context }\n  }\n\n  async remove(connectionId: string) {\n    const { connection, context } = this.get(connectionId)\n\n    this.application.registry.hooks.call(\n      Hook.OnDisconnect,\n      { concurrent: true },\n      connection,\n    )\n\n    this.#collection.delete(connectionId)\n\n    const { calls, serverStreams, clientStreams, rpcStreams, container } =\n      context\n\n    for (const call of calls.values()) {\n      call.abort(new Error('Connection closed'))\n    }\n\n    for (const stream of clientStreams.values()) {\n      stream.destroy(new Error('Connection closed'))\n    }\n\n    for (const stream of serverStreams.values()) {\n      stream.destroy(new Error('Connection closed'))\n    }\n\n    for (const stream of rpcStreams.values()) {\n      stream.abort(new Error('Connection closed'))\n    }\n\n    try {\n      await container.dispose()\n    } catch (error) {\n      this.application.logger.error(\n        { error, connection },\n        'Error during closing connection',\n      )\n    }\n  }\n}\n\nexport class ProtocolClientStreams {\n  constructor(private readonly connections: ProtocolConnections) {}\n\n  get(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { clientStreams } = context\n    const stream = clientStreams.get(streamId) ?? throwError('Stream not found')\n    return stream\n  }\n\n  remove(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { clientStreams } = context\n    clientStreams.get(streamId) || throwError('Stream not found')\n    clientStreams.delete(streamId)\n  }\n\n  add(\n    connectionId: string,\n    streamId: number,\n    metadata: ProtocolBlobMetadata,\n    read: Callback,\n  ) {\n    const { context } = this.connections.get(connectionId)\n    const { clientStreams } = context\n    const stream = new ProtocolClientStream(streamId, metadata, { read })\n    clientStreams.set(streamId, stream)\n    return stream\n  }\n\n  push(connectionId: string, streamId: number, chunk: ArrayBuffer) {\n    const stream = this.get(connectionId, streamId)\n    stream.push(Buffer.from(chunk))\n  }\n\n  end(connectionId: string, streamId: number) {\n    const stream = this.get(connectionId, streamId)\n    stream.push(null)\n    this.remove(connectionId, streamId)\n  }\n\n  abort(connectionId: string, streamId: number, error = new Error('Aborted')) {\n    const stream = this.get(connectionId, streamId)\n    stream.destroy(error)\n    this.remove(connectionId, streamId)\n  }\n}\n\nexport class ProtocolServerStreams {\n  constructor(private readonly connections: ProtocolConnections) {}\n\n  get(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { serverStreams } = context\n    const stream = serverStreams.get(streamId) ?? throwError('Stream not found')\n    return stream\n  }\n\n  add(connectionId: string, streamId: number, blob: ProtocolBlob) {\n    const { context } = this.connections.get(connectionId)\n    const { serverStreams } = context\n    const stream = new ProtocolServerStream(streamId, blob)\n    serverStreams.set(streamId, stream)\n    return stream\n  }\n\n  remove(connectionId: string, streamId: number) {\n    const { context } = this.connections.get(connectionId)\n    const { serverStreams } = context\n    serverStreams.has(streamId) || throwError('Stream not found')\n    serverStreams.delete(streamId)\n  }\n\n  pull(connectionId: string, streamId: number) {\n    const stream = this.get(connectionId, streamId)\n    stream.resume()\n  }\n\n  abort(connectionId: string, streamId: number, error = new Error('Aborted')) {\n    const stream = this.get(connectionId, streamId)\n    stream.destroy(error)\n    this.remove(connectionId, streamId)\n  }\n}\n\nexport class Protocol {\n  readonly connections: ProtocolConnections\n  readonly clientStreams: ProtocolClientStreams\n  readonly serverStreams: ProtocolServerStreams\n\n  constructor(\n    protected readonly application: {\n      logger: Logger\n      format: Format\n      container: Container\n      registry: ProtocolRegistry\n      api: ProtocolApi\n    },\n  ) {\n    this.connections = new ProtocolConnections(this.application)\n    this.clientStreams = new ProtocolClientStreams(this.connections)\n    this.serverStreams = new ProtocolServerStreams(this.connections)\n  }\n\n  async rpc(\n    connectionId: string,\n    rpc: ProtocolRPC,\n    params: { signal?: AbortSignal } = {},\n  ) {\n    const { connection, context, transport } =\n      this.connections.get(connectionId)\n    const { calls, format } = context\n    const { callId, namespace, procedure, payload } = rpc\n    const abortController = new AbortController()\n    const signal = params.signal\n      ? AbortSignal.any([params.signal, abortController.signal])\n      : abortController.signal\n\n    const call = Object.assign(createPromise<ProtocolApiCallResult>(), {\n      abort: () => abortController.abort(),\n    })\n\n    calls.set(callId, call)\n    call.promise.finally(() => calls.delete(callId))\n\n    const callIdEncoded = encodeNumber(callId, 'Uint32')\n    const container = context.container.fork(Scope.Call)\n\n    try {\n      const response = await this.application.api.call({\n        connection,\n        container,\n        namespace,\n        payload,\n        procedure,\n        signal,\n      })\n\n      const responseEncoded = format.encoder.encodeRPC(\n        {\n          callId,\n          payload: response.output,\n        },\n        {\n          addStream: (blob) => {\n            const id = context.streamId++\n            const stream = this.serverStreams.add(connectionId, id, blob)\n            stream.on('data', (chunk) => {\n              stream.pause()\n              const buf = Buffer.from(chunk)\n              transport.send(\n                connection,\n                ServerMessageType.ServerStreamPush,\n                concat(\n                  encodeNumber(id, 'Uint32'),\n                  (buf.buffer as ArrayBuffer).slice(\n                    buf.byteOffset,\n                    buf.byteOffset + buf.byteLength,\n                  ),\n                ),\n              )\n            })\n            stream.on('error', (err) => {\n              transport.send(\n                connection,\n                ServerMessageType.ServerStreamAbort,\n                encodeNumber(id, 'Uint32'),\n              )\n            })\n            stream.on('end', () => {\n              transport.send(\n                connection,\n                ServerMessageType.ServerStreamEnd,\n                encodeNumber(id, 'Uint32'),\n              )\n            })\n            return stream\n          },\n          getStream: (id) => {\n            return this.clientStreams.get(connectionId, id)\n          },\n        },\n      )\n\n      if ('subscription' in response) {\n        throwError('Unimplemented')\n      } else if ('iterable' in response) {\n        transport.send(\n          connection,\n          ServerMessageType.RpcStreamResponse,\n          responseEncoded,\n        )\n        try {\n          const ab = new AbortController()\n          context.rpcStreams.set(callId, ab)\n          const iterable =\n            typeof response.iterable === 'function'\n              ? response.iterable()\n              : response.iterable\n          for await (const chunk of iterable) {\n            if (ab.signal.aborted) break\n            const chunkEncoded = format.encoder.encode(chunk)\n            transport.send(\n              connection,\n              ServerMessageType.RpcStreamChunk,\n              concat(callIdEncoded, chunkEncoded),\n            )\n          }\n        } catch (error) {\n          this.application.logger.error(error)\n          transport.send(\n            connection,\n            ServerMessageType.RpcStreamAbort,\n            callIdEncoded,\n          )\n        } finally {\n          context.rpcStreams.delete(callId)\n          response.onFinish && defer(response.onFinish)\n        }\n      } else {\n        transport.send(\n          connection,\n          ServerMessageType.RpcResponse,\n          responseEncoded,\n        )\n      }\n    } catch (error) {\n      if (error instanceof ProtocolError === false) {\n        this.application.logger.error(\n          { error, connection },\n          'Error during RPC call',\n        )\n\n        // biome-ignore lint/suspicious/noCatchAssign:\n        error = new ProtocolError(\n          ErrorCode.InternalServerError,\n          'Internal server error',\n        )\n      }\n\n      const payload = format.encoder.encodeRPC(\n        { callId, error },\n        {\n          addStream(blob) {\n            throwError('Cannot handle stream for error response')\n          },\n          getStream(id) {\n            throwError('Cannot handle stream for error response')\n          },\n        },\n      )\n      transport.send(connection, ServerMessageType.RpcResponse, payload)\n    } finally {\n      container.dispose().catch((error) => {\n        this.application.logger.error(\n          { error, connection },\n          \"Error during disposing connection's container\",\n        )\n      })\n    }\n  }\n\n  async rpcRaw(\n    connectionId: string,\n    buffer: ArrayBuffer,\n    params: { signal?: AbortSignal } = {},\n  ) {\n    const { connection, context, transport } =\n      this.connections.get(connectionId)\n\n    const { format } = context\n\n    const rpc = format.decoder.decodeRPC(buffer, {\n      addStream: (id, metadata) => {\n        const stream = this.clientStreams.add(\n          connectionId,\n          id,\n          metadata,\n          (size) => {\n            transport.send(\n              connection,\n              ServerMessageType.ClientStreamPull,\n              concat(encodeNumber(id, 'Uint32'), encodeNumber(size, 'Uint32')),\n            )\n          },\n        )\n        return stream\n      },\n      getStream: (id) => {\n        return this.serverStreams.get(connectionId, id)\n      },\n    })\n\n    return await this.rpc(connectionId, rpc, params)\n  }\n\n  rpcAbort(connectionId: string, callId: number) {\n    const { context } = this.connections.get(connectionId)\n    const call = context.calls.get(callId) ?? throwError('Call not found')\n    call.abort()\n  }\n\n  rpcAbortRaw(connectionId: string, buffer: ArrayBuffer) {\n    const callId = decodeNumber(buffer, 'Uint32')\n    return this.rpcAbort(connectionId, callId)\n  }\n\n  rpcStreamAbort(connectionId: string, callId: number) {\n    const { context } = this.connections.get(connectionId)\n    const ab =\n      context.rpcStreams.get(callId) ?? throwError('Call stream not found')\n    ab.abort()\n  }\n\n  rpcStreamAbortRaw(connectionId: string, buffer: ArrayBuffer) {\n    const callId = decodeNumber(buffer, 'Uint32')\n    return this.rpcStreamAbort(connectionId, callId)\n  }\n\n  notify(connectionId: string, event, payload) {\n    throw Error('Unimplemented')\n  }\n}\n"],"version":3}

package/dist/server/registry.js

@@ -1,19 +1,2 @@
-import { Registry } from '@nmtjs/core';
-import { compile } from '@nmtjs/type/compiler';
-export class ProtocolRegistry extends Registry {
-    types = new Set();
-    compiled = new Map();
-    registerType(type) {
-        this.types.add(type);
-    }
-    compile() {
-        for (const type of this.types){
-            this.compiled.set(type, compile(type));
-        }
-    }
-    clear() {
-        super.clear();
-        this.types.clear();
-        this.compiled.clear();
-    }
-}
+import { Registry } from "@nmtjs/core";
+export class ProtocolRegistry extends Registry {}

package/dist/server/registry.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../lib/server/registry.ts"],"sourcesContent":["import { Registry } from '@nmtjs/core'\nimport type { BaseType, BaseTypeAny } from '@nmtjs/type'\nimport { type Compiled, compile } from '@nmtjs/type/compiler'\n\nexport class ProtocolRegistry extends Registry {\n  readonly types = new Set<BaseType>()\n  readonly compiled = new Map<any, Compiled>()\n\n  registerType(type: BaseTypeAny) {\n    this.types.add(type)\n  }\n\n  compile() {\n    for (const type of this.types) {\n      this.compiled.set(type, compile(type))\n    }\n  }\n\n  clear() {\n    super.clear()\n    this.types.clear()\n    this.compiled.clear()\n  }\n}\n"],"names":["Registry","compile","ProtocolRegistry","types","Set","compiled","Map","registerType","type","add","set","clear"],"mappings":"AAAA,SAASA,QAAQ,QAAQ,cAAa;AAEtC,SAAwBC,OAAO,QAAQ,uBAAsB;AAE7D,OAAO,MAAMC,yBAAyBF;IAC3BG,QAAQ,IAAIC,MAAe;IAC3BC,WAAW,IAAIC,MAAoB;IAE5CC,aAAaC,IAAiB,EAAE;QAC9B,IAAI,CAACL,KAAK,CAACM,GAAG,CAACD;IACjB;IAEAP,UAAU;QACR,KAAK,MAAMO,QAAQ,IAAI,CAACL,KAAK,CAAE;YAC7B,IAAI,CAACE,QAAQ,CAACK,GAAG,CAACF,MAAMP,QAAQO;QAClC;IACF;IAEAG,QAAQ;QACN,KAAK,CAACA;QACN,IAAI,CAACR,KAAK,CAACQ,KAAK;QAChB,IAAI,CAACN,QAAQ,CAACM,KAAK;IACrB;AACF"}
+{"mappings":"AAAA,SAAS,gBAAgB,aAAa;AAEtC,OAAO,MAAM,yBAAyB,SAAS,CAAE","names":[],"sources":["src/server/registry.ts"],"sourcesContent":["import { Registry } from '@nmtjs/core'\n\nexport class ProtocolRegistry extends Registry {}\n"],"version":3}

package/dist/server/stream.js

@@ -1,30 +1,28 @@
-import { PassThrough, Readable } from 'node:stream';
-import { ReadableStream } from 'node:stream/web';
+import { PassThrough, Readable } from "node:stream";
+import { ReadableStream } from "node:stream/web";
 export class ProtocolClientStream extends Readable {
-    id;
-    metadata;
-    constructor(id, metadata, options){
-        super(options);
-        this.id = id;
-        this.metadata = metadata;
-    }
+	constructor(id, metadata, options) {
+		super(options);
+		this.id = id;
+		this.metadata = metadata;
+	}
 }
 export class ProtocolServerStream extends PassThrough {
-    id;
-    metadata;
-    constructor(id, blob){
-        let readable;
-        if (blob.source instanceof Readable) {
-            readable = blob.source;
-        } else if (blob.source instanceof ReadableStream) {
-            readable = Readable.fromWeb(blob.source);
-        } else {
-            throw new Error('Invalid source type');
-        }
-        super();
-        this.pause();
-        readable.pipe(this);
-        this.id = id;
-        this.metadata = blob.metadata;
-    }
+	id;
+	metadata;
+	constructor(id, blob) {
+		let readable;
+		if (blob.source instanceof Readable) {
+			readable = blob.source;
+		} else if (blob.source instanceof ReadableStream) {
+			readable = Readable.fromWeb(blob.source);
+		} else {
+			throw new Error("Invalid source type");
+		}
+		super();
+		this.pause();
+		readable.pipe(this);
+		this.id = id;
+		this.metadata = blob.metadata;
+	}
 }

package/dist/server/stream.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../lib/server/stream.ts"],"sourcesContent":["import {\n  PassThrough,\n  Readable,\n  type ReadableOptions,\n  type TransformOptions,\n} from 'node:stream'\nimport { ReadableStream } from 'node:stream/web'\nimport type { ProtocolBlob, ProtocolBlobMetadata } from '../common/blob.ts'\n\nexport class ProtocolClientStream extends Readable {\n  constructor(\n    public readonly id: number,\n    public readonly metadata: ProtocolBlobMetadata,\n    options?: ReadableOptions,\n  ) {\n    super(options)\n  }\n}\n\nexport class ProtocolServerStream extends PassThrough {\n  public readonly id: number\n  public readonly metadata: ProtocolBlobMetadata\n\n  constructor(id: number, blob: ProtocolBlob) {\n    let readable: Readable\n\n    if (blob.source instanceof Readable) {\n      readable = blob.source\n    } else if (blob.source instanceof ReadableStream) {\n      readable = Readable.fromWeb(blob.source as ReadableStream)\n    } else {\n      throw new Error('Invalid source type')\n    }\n\n    super()\n\n    this.pause()\n    readable.pipe(this)\n\n    this.id = id\n    this.metadata = blob.metadata\n  }\n}\n"],"names":["PassThrough","Readable","ReadableStream","ProtocolClientStream","constructor","id","metadata","options","ProtocolServerStream","blob","readable","source","fromWeb","Error","pause","pipe"],"mappings":"AAAA,SACEA,WAAW,EACXC,QAAQ,QAGH,cAAa;AACpB,SAASC,cAAc,QAAQ,kBAAiB;AAGhD,OAAO,MAAMC,6BAA6BF;;;IACxCG,YACE,AAAgBC,EAAU,EAC1B,AAAgBC,QAA8B,EAC9CC,OAAyB,CACzB;QACA,KAAK,CAACA;aAJUF,KAAAA;aACAC,WAAAA;IAIlB;AACF;AAEA,OAAO,MAAME,6BAA6BR;IACxBK,GAAU;IACVC,SAA8B;IAE9CF,YAAYC,EAAU,EAAEI,IAAkB,CAAE;QAC1C,IAAIC;QAEJ,IAAID,KAAKE,MAAM,YAAYV,UAAU;YACnCS,WAAWD,KAAKE,MAAM;QACxB,OAAO,IAAIF,KAAKE,MAAM,YAAYT,gBAAgB;YAChDQ,WAAWT,SAASW,OAAO,CAACH,KAAKE,MAAM;QACzC,OAAO;YACL,MAAM,IAAIE,MAAM;QAClB;QAEA,KAAK;QAEL,IAAI,CAACC,KAAK;QACVJ,SAASK,IAAI,CAAC,IAAI;QAElB,IAAI,CAACV,EAAE,GAAGA;QACV,IAAI,CAACC,QAAQ,GAAGG,KAAKH,QAAQ;IAC/B;AACF"}
+{"mappings":"AAAA,SACE,aACA,gBAGK,aAAa;AACpB,SAAS,sBAAsB,iBAAiB;AAGhD,OAAO,MAAM,6BAA6B,SAAS;CACjD,YACkBA,IACAC,UAChBC,SACA;AACA,QAAM,QAAQ;OAJE;OACA;CAIjB;AACF;AAED,OAAO,MAAM,6BAA6B,YAAY;CACpD,AAAgB;CAChB,AAAgB;CAEhB,YAAYF,IAAYG,MAAoB;EAC1C,IAAIC;AAEJ,MAAI,KAAK,kBAAkB,UAAU;AACnC,cAAW,KAAK;EACjB,WAAU,KAAK,kBAAkB,gBAAgB;AAChD,cAAW,SAAS,QAAQ,KAAK,OAAyB;EAC3D,OAAM;AACL,SAAM,IAAI,MAAM;EACjB;AAED,SAAO;AAEP,OAAK,OAAO;AACZ,WAAS,KAAK,KAAK;AAEnB,OAAK,KAAK;AACV,OAAK,WAAW,KAAK;CACtB;AACF","names":["id: number","metadata: ProtocolBlobMetadata","options?: ReadableOptions","blob: ProtocolBlob","readable: Readable"],"sources":["src/server/stream.ts"],"sourcesContent":["import {\n  PassThrough,\n  Readable,\n  type ReadableOptions,\n  type TransformOptions,\n} from 'node:stream'\nimport { ReadableStream } from 'node:stream/web'\nimport type { ProtocolBlob, ProtocolBlobMetadata } from '../common/blob.ts'\n\nexport class ProtocolClientStream extends Readable {\n  constructor(\n    public readonly id: number,\n    public readonly metadata: ProtocolBlobMetadata,\n    options?: ReadableOptions,\n  ) {\n    super(options)\n  }\n}\n\nexport class ProtocolServerStream extends PassThrough {\n  public readonly id: number\n  public readonly metadata: ProtocolBlobMetadata\n\n  constructor(id: number, blob: ProtocolBlob) {\n    let readable: Readable\n\n    if (blob.source instanceof Readable) {\n      readable = blob.source\n    } else if (blob.source instanceof ReadableStream) {\n      readable = Readable.fromWeb(blob.source as ReadableStream)\n    } else {\n      throw new Error('Invalid source type')\n    }\n\n    super()\n\n    this.pause()\n    readable.pipe(this)\n\n    this.id = id\n    this.metadata = blob.metadata\n  }\n}\n"],"version":3}

package/dist/server/transport.js

@@ -1,7 +1,7 @@
 import { kTransportPlugin } from "./constants.js";
-export const createTransport = (name, init)=>({
-        name,
-        init,
-        [kTransportPlugin]: true
-    });
-export const isTransportPlugin = (plugin)=>kTransportPlugin in plugin;
+export const createTransport = (name, init) => ({
+	name,
+	init,
+	[kTransportPlugin]: true
+});
+export const isTransportPlugin = (plugin) => kTransportPlugin in plugin;

package/dist/server/transport.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../lib/server/transport.ts"],"sourcesContent":["import type { Async } from '@nmtjs/common'\nimport type { BasePlugin, PluginContext } from '@nmtjs/core'\nimport type { ServerMessageType } from '../common/enums.ts'\nimport type { Connection } from './connection.ts'\nimport { kTransportPlugin } from './constants.ts'\nimport type { Protocol } from './protocol.ts'\nimport type { ProtocolRegistry } from './registry.ts'\n\nexport interface Transport<T = unknown> {\n  start: () => Promise<void>\n  stop: () => Promise<void>\n  send: (\n    connection: Connection<T>,\n    messageType: ServerMessageType,\n    buffer: ArrayBuffer,\n  ) => Async<void>\n}\n\nexport interface TransportPluginContext extends PluginContext {\n  protocol: Protocol\n  registry: ProtocolRegistry\n}\n\nexport interface TransportPlugin<Type = unknown, Options = unknown>\n  extends BasePlugin<Transport<Type>, Options, TransportPluginContext> {\n  [kTransportPlugin]: any\n}\n\nexport const createTransport = <Type = unknown, Options = unknown>(\n  name: string,\n  init: TransportPlugin<Type, Options>['init'],\n): TransportPlugin<Type, Options> => ({ name, init, [kTransportPlugin]: true })\n\nexport const isTransportPlugin = (\n  plugin: BasePlugin<any, any, any>,\n): plugin is TransportPlugin => kTransportPlugin in plugin\n"],"names":["kTransportPlugin","createTransport","name","init","isTransportPlugin","plugin"],"mappings":"AAIA,SAASA,gBAAgB,QAAQ,iBAAgB;AAwBjD,OAAO,MAAMC,kBAAkB,CAC7BC,MACAC,OACoC,CAAA;QAAED;QAAMC;QAAM,CAACH,iBAAiB,EAAE;IAAK,CAAA,EAAE;AAE/E,OAAO,MAAMI,oBAAoB,CAC/BC,SAC8BL,oBAAoBK,OAAM"}
+{"mappings":"AAIA,SAAS,wBAAwB,gBAAgB;AAwBjD,OAAO,MAAM,kBAAkB,CAC7BA,MACAC,UACoC;CAAE;CAAM;EAAO,mBAAmB;AAAM;AAE9E,OAAO,MAAM,oBAAoB,CAC/BC,WAC8B,oBAAoB","names":["name: string","init: TransportPlugin<Type, Options>['init']","plugin: BasePlugin<any, any, any>"],"sources":["src/server/transport.ts"],"sourcesContent":["import type { Async } from '@nmtjs/common'\nimport type { BasePlugin, PluginContext } from '@nmtjs/core'\nimport type { ServerMessageType } from '../common/enums.ts'\nimport type { Connection } from './connection.ts'\nimport { kTransportPlugin } from './constants.ts'\nimport type { Protocol } from './protocol.ts'\nimport type { ProtocolRegistry } from './registry.ts'\n\nexport interface Transport<T = unknown> {\n  start: () => Promise<void>\n  stop: () => Promise<void>\n  send: (\n    connection: Connection<T>,\n    messageType: ServerMessageType,\n    buffer: ArrayBuffer,\n  ) => Async<void>\n}\n\nexport interface TransportPluginContext extends PluginContext {\n  protocol: Protocol\n  registry: ProtocolRegistry\n}\n\nexport interface TransportPlugin<Type = unknown, Options = unknown>\n  extends BasePlugin<Transport<Type>, Options, TransportPluginContext> {\n  [kTransportPlugin]: any\n}\n\nexport const createTransport = <Type = unknown, Options = unknown>(\n  name: string,\n  init: TransportPlugin<Type, Options>['init'],\n): TransportPlugin<Type, Options> => ({ name, init, [kTransportPlugin]: true })\n\nexport const isTransportPlugin = (\n  plugin: BasePlugin<any, any, any>,\n): plugin is TransportPlugin => kTransportPlugin in plugin\n"],"version":3}

package/dist/server/utils.js

@@ -1,10 +1,10 @@
-export const getFormat = (format, { acceptType, contentType })=>{
-    const encoder = contentType ? format.supportsEncoder(contentType) : undefined;
-    if (!encoder) throw new Error('Unsupported content-type');
-    const decoder = acceptType ? format.supportsDecoder(acceptType) : undefined;
-    if (!decoder) throw new Error('Unsupported accept');
-    return {
-        encoder,
-        decoder
-    };
+export const getFormat = (format, { acceptType, contentType }) => {
+	const encoder = contentType ? format.supportsEncoder(contentType) : undefined;
+	if (!encoder) throw new Error("Unsupported content-type");
+	const decoder = acceptType ? format.supportsDecoder(acceptType) : undefined;
+	if (!decoder) throw new Error("Unsupported accept");
+	return {
+		encoder,
+		decoder
+	};
 };

package/dist/server/utils.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../../../lib/server/utils.ts"],"sourcesContent":["import type { Format } from './format.ts'\n\nexport type ResolveFormatParams = {\n  contentType?: string | null\n  acceptType?: string | null\n}\n\nexport const getFormat = (\n  format: Format,\n  { acceptType, contentType }: ResolveFormatParams,\n) => {\n  const encoder = contentType ? format.supportsEncoder(contentType) : undefined\n  if (!encoder) throw new Error('Unsupported content-type')\n\n  const decoder = acceptType ? format.supportsDecoder(acceptType) : undefined\n  if (!decoder) throw new Error('Unsupported accept')\n\n  return {\n    encoder,\n    decoder,\n  }\n}\n"],"names":["getFormat","format","acceptType","contentType","encoder","supportsEncoder","undefined","Error","decoder","supportsDecoder"],"mappings":"AAOA,OAAO,MAAMA,YAAY,CACvBC,QACA,EAAEC,UAAU,EAAEC,WAAW,EAAuB;IAEhD,MAAMC,UAAUD,cAAcF,OAAOI,eAAe,CAACF,eAAeG;IACpE,IAAI,CAACF,SAAS,MAAM,IAAIG,MAAM;IAE9B,MAAMC,UAAUN,aAAaD,OAAOQ,eAAe,CAACP,cAAcI;IAClE,IAAI,CAACE,SAAS,MAAM,IAAID,MAAM;IAE9B,OAAO;QACLH;QACAI;IACF;AACF,EAAC"}
+{"mappings":"AAOA,OAAO,MAAM,YAAY,CACvBA,QACA,EAAE,YAAY,aAAkC,KAC7C;CACH,MAAM,UAAU,cAAc,OAAO,gBAAgB,YAAY,GAAG;AACpE,MAAK,QAAS,OAAM,IAAI,MAAM;CAE9B,MAAM,UAAU,aAAa,OAAO,gBAAgB,WAAW,GAAG;AAClE,MAAK,QAAS,OAAM,IAAI,MAAM;AAE9B,QAAO;EACL;EACA;CACD;AACF","names":["format: Format"],"sources":["src/server/utils.ts"],"sourcesContent":["import type { Format } from './format.ts'\n\nexport type ResolveFormatParams = {\n  contentType?: string | null\n  acceptType?: string | null\n}\n\nexport const getFormat = (\n  format: Format,\n  { acceptType, contentType }: ResolveFormatParams,\n) => {\n  const encoder = contentType ? format.supportsEncoder(contentType) : undefined\n  if (!encoder) throw new Error('Unsupported content-type')\n\n  const decoder = acceptType ? format.supportsDecoder(acceptType) : undefined\n  if (!decoder) throw new Error('Unsupported accept')\n\n  return {\n    encoder,\n    decoder,\n  }\n}\n"],"version":3}

package/package.json

@@ -3,36 +3,32 @@
   "type": "module",
   "exports": {
     "./common": {
-      "bun": "./lib/common/index.ts",
-      "default": "./dist/common/index.js",
-      "types": "./lib/common/index.ts"
+      "types": "./src/common/index.ts",
+      "import": "./dist/common/index.js"
     },
     "./server": {
-      "bun": "./lib/server/index.ts",
-      "default": "./dist/server/index.js",
-      "types": "./lib/server/index.ts"
+      "types": "./src/server/index.ts",
+      "import": "./dist/server/index.js"
     },
     "./client": {
-      "bun": "./lib/client/index.ts",
-      "default": "./dist/client/index.js",
-      "types": "./lib/client/index.ts"
+      "types": "./src/client/index.ts",
+      "import": "./dist/client/index.js"
     }
   },
   "dependencies": {
-    "@nmtjs/common": "0.6.4",
-    "@nmtjs/core": "0.6.4",
-    "@nmtjs/type": "0.6.4"
+    "@nmtjs/common": "0.7.0",
+    "@nmtjs/type": "0.7.0",
+    "@nmtjs/core": "0.7.0"
   },
   "files": [
-    "index.ts",
-    "lib",
+    "src",
     "dist",
     "LICENSE.md",
     "README.md"
   ],
-  "version": "0.6.4",
+  "version": "0.7.0",
   "scripts": {
-    "build": "neemata-build --root=./lib './**/*.ts'",
+    "build": "neemata-build --root=./src './**/*.ts'",
     "type-check": "tsc --noEmit"
   }
 }

package/{lib → src}/client/protocol.ts

@@ -81,10 +81,18 @@ export class ProtocolClientStreams {
   }
 
   end(streamId: number) {
-    const stream = this.get(streamId)
-    stream.end()
+    this.get(streamId).end()
     this.remove(streamId)
   }
+
+  clear(error?: Error) {
+    if (error) {
+      for (const stream of this.#collection.values()) {
+        stream.abort(error)
+      }
+    }
+    this.#collection.clear()
+  }
 }
 
 export class ProtocolServerStreams {
@@ -127,12 +135,26 @@ export class ProtocolServerStreams {
     stream.end()
     this.remove(streamId)
   }
+
+  clear(error?: Error) {
+    if (error) {
+      for (const stream of this.#collection.values()) {
+        stream.abort(error)
+      }
+    }
+    this.#collection.clear()
+  }
 }
 
 export interface ProtocolTransport
-  extends EventEmitter<{
-    [K in `${ServerMessageType}`]: [ArrayBuffer]
-  }> {
+  extends EventEmitter<
+    {
+      [K in `${ServerMessageType}`]: [ArrayBuffer]
+    } & {
+      connected: []
+      disconnected: []
+    }
+  > {
   connect(
     auth: any,
     contentType: BaseClientFormat['contentType'],
@@ -354,6 +376,10 @@ export abstract class ProtocolBaseClient<
         }
       },
     )
+
+    this.#transport.on('disconnected', () => {
+      this.#clear()
+    })
   }
 
   async connect(auth: any) {
@@ -361,6 +387,7 @@ export abstract class ProtocolBaseClient<
   }
 
   async disconnect() {
+    this.#clear()
     return await this.#transport.disconnect()
   }
 
@@ -373,6 +400,20 @@ export abstract class ProtocolBaseClient<
     return await this.#transport.send(messageType, buffer)
   }
 
+  async #clear() {
+    const error = new ProtocolError(
+      ErrorCode.ConnectionError,
+      'Connection closed',
+    )
+    for (const call of this.#calls.values()) call.reject(error)
+    this.#calls.clear()
+    this.#serverStreams.clear(error)
+    this.#clientStreams.clear(error)
+    this.#rpcStreams.clear(error)
+    this.#callId = 0
+    this.#streamId = 0
+  }
+
   protected async _call(
     namespace: string,
     procedure: string,

package/{lib → src}/client/stream.ts

@@ -1,5 +1,5 @@
 import { type Callback, defer } from '@nmtjs/common'
-import { encodeText } from '../common/binary.ts'
+import { concat, encodeText } from '../common/binary.ts'
 import type { ProtocolBlobMetadata } from '../common/blob.ts'
 
 export class ProtocolClientBlobStream extends TransformStream<
@@ -38,31 +38,31 @@ export class ProtocolClientBlobStream extends TransformStream<
   }
 
   async read(size: number) {
-    if (this.#queue.byteLength >= size) {
-      const chunk = this.#queue.slice(0, size)
-      const remaining = this.#queue.slice(size)
-      this.#queue = remaining
-      return chunk
-    } else {
+    while (this.#queue.byteLength < size) {
       const { done, value } = await this.#reader.read()
-      if (!done) {
-        const buffer = value as ArrayBuffer
-        const chunk = buffer.slice(0, size)
-        const remaining = buffer.slice(size)
-        this.#queue = remaining
-        return chunk
+      if (done) {
+        if (this.#queue.byteLength > 0) {
+          const chunk = this.#queue
+          this.#queue = new ArrayBuffer(0)
+          return chunk
+        }
+        return null
       }
-      return null
+      const buffer = value as ArrayBuffer
+      this.#queue = concat(this.#queue, buffer)
     }
+    const chunk = this.#queue.slice(0, size)
+    this.#queue = this.#queue.slice(size)
+    return chunk
   }
 
   abort(error = new Error('Stream aborted')) {
     this.#reader.cancel(error)
+    this.source.cancel(error)
   }
 
-  async end() {
-    if (!this.writable.locked) await this.writable.close()
-    else await this.writable.abort(new Error('Stream closed'))
+  end() {
+    return this.source.cancel('Stream ended')
   }
 }
 
@@ -91,7 +91,7 @@ export class ProtocolServerStream<T = any>
     }
   }
 
-  async push(chunk: any) {
+  async push(chunk: T) {
     await this.#writer.write(chunk)
   }
 
@@ -112,12 +112,4 @@ export class ProtocolServerBlobStream extends ProtocolServerStream<ArrayBuffer>
   ) {
     super(start)
   }
-
-  push(chunk: ArrayBuffer) {
-    return super.push(chunk)
-  }
-
-  end() {
-    return super.end()
-  }
 }

package/{lib → src}/server/connection.ts

@@ -1,6 +1,6 @@
 import { randomUUID } from 'node:crypto'
-import type { InteractivePromise } from '@nmtjs/common'
 import type { Container } from '@nmtjs/core'
+import type { InteractivePromise } from '../../../common/src/index.ts'
 import type { ProtocolApiCallResult } from './api.ts'
 import type { BaseServerDecoder, BaseServerEncoder } from './format.ts'
 import type { ProtocolClientStream, ProtocolServerStream } from './stream.ts'

package/{lib → src}/server/protocol.ts

@@ -104,7 +104,7 @@ export class ProtocolConnections {
       context
 
     for (const call of calls.values()) {
-      call.reject(new Error('Connection closed'))
+      call.abort(new Error('Connection closed'))
     }
 
     for (const stream of clientStreams.values()) {
@@ -279,12 +279,16 @@ export class Protocol {
             const stream = this.serverStreams.add(connectionId, id, blob)
             stream.on('data', (chunk) => {
               stream.pause()
+              const buf = Buffer.from(chunk)
               transport.send(
                 connection,
                 ServerMessageType.ServerStreamPush,
                 concat(
                   encodeNumber(id, 'Uint32'),
-                  Buffer.from(chunk).buffer as ArrayBuffer,
+                  (buf.buffer as ArrayBuffer).slice(
+                    buf.byteOffset,
+                    buf.byteOffset + buf.byteLength,
+                  ),
                 ),
               )
             })
@@ -400,13 +404,19 @@ export class Protocol {
 
     const rpc = format.decoder.decodeRPC(buffer, {
       addStream: (id, metadata) => {
-        return this.clientStreams.add(connectionId, id, metadata, (size) => {
-          transport.send(
-            connection,
-            ServerMessageType.ClientStreamPull,
-            concat(encodeNumber(id, 'Uint32'), encodeNumber(size, 'Uint32')),
-          )
-        })
+        const stream = this.clientStreams.add(
+          connectionId,
+          id,
+          metadata,
+          (size) => {
+            transport.send(
+              connection,
+              ServerMessageType.ClientStreamPull,
+              concat(encodeNumber(id, 'Uint32'), encodeNumber(size, 'Uint32')),
+            )
+          },
+        )
+        return stream
       },
       getStream: (id) => {
         return this.serverStreams.get(connectionId, id)

package/src/server/registry.ts

@@ -0,0 +1,3 @@
+import { Registry } from '@nmtjs/core'
+
+export class ProtocolRegistry extends Registry {}

package/lib/server/registry.ts

@@ -1,24 +0,0 @@
-import { Registry } from '@nmtjs/core'
-import type { BaseType, BaseTypeAny } from '@nmtjs/type'
-import { type Compiled, compile } from '@nmtjs/type/compiler'
-
-export class ProtocolRegistry extends Registry {
-  readonly types = new Set<BaseType>()
-  readonly compiled = new Map<any, Compiled>()
-
-  registerType(type: BaseTypeAny) {
-    this.types.add(type)
-  }
-
-  compile() {
-    for (const type of this.types) {
-      this.compiled.set(type, compile(type))
-    }
-  }
-
-  clear() {
-    super.clear()
-    this.types.clear()
-    this.compiled.clear()
-  }
-}