@apibara/protocol 2.1.0-beta.37 → 2.1.0-beta.39

package/dist/index.cjs

@@ -2,15 +2,17 @@
 
 const config = require('./shared/protocol.53f81a1e.cjs');
 const codec = require('./codec.cjs');
+const assert = require('node:assert');
+const consola = require('consola');
 const niceGrpc = require('nice-grpc');
 require('protobufjs/minimal.js');
-const assert = require('node:assert');
 require('viem');
 require('long');
 
 function _interopDefaultCompat (e) { return e && typeof e === 'object' && 'default' in e ? e.default : e; }
 
 const assert__default = /*#__PURE__*/_interopDefaultCompat(assert);
+const consola__default = /*#__PURE__*/_interopDefaultCompat(consola);
 
 const index = {
   __proto__: null,
@@ -52,6 +54,27 @@ function createClient(config$1, streamUrl, options = {}) {
   );
   return new GrpcClient(config$1, client);
 }
+function createAuthenticatedClient(config, streamUrl, options) {
+  const dnaToken = process.env.DNA_TOKEN;
+  if (!dnaToken) {
+    consola__default.warn(
+      "DNA_TOKEN environment variable is not set. Trying to connect without authentication."
+    );
+  }
+  return createClient(config, streamUrl, {
+    ...options,
+    defaultCallOptions: {
+      ...options?.defaultCallOptions ?? {},
+      "*": {
+        metadata: niceGrpc.Metadata({
+          Authorization: `Bearer ${dnaToken}`
+        }),
+        // metadata cant be overrided with spread as its a class so we override it fully if user provided it.
+        ...options?.defaultCallOptions?.["*"] ?? {}
+      }
+    }
+  });
+}
 class GrpcClient {
   constructor(config, client) {
     this.config = config;
@@ -206,6 +229,7 @@ exports.StatusRequest = StatusRequest;
 exports.StatusResponse = StatusResponse;
 exports.StreamDataIterable = StreamDataIterable;
 exports.TimeoutError = TimeoutError;
+exports.createAuthenticatedClient = createAuthenticatedClient;
 exports.createClient = createClient;
 exports.proto = index;
 //# sourceMappingURL=index.cjs.map

package/dist/index.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.cjs","sources":["../src/status.ts","../src/client.ts","../src/rate.ts"],"sourcesContent":["import { type CodecType, MessageCodec, OptionalCodec } from \"./codec\";\nimport { Cursor } from \"./common\";\n\n/** The request to the `status` endpoint. */\nexport const StatusRequest = MessageCodec({});\n\nexport type StatusRequest = CodecType<typeof StatusRequest>;\n\n/** The response from the `status` endpoint. */\nexport const StatusResponse = MessageCodec({\n  currentHead: OptionalCodec(Cursor),\n  lastIngested: OptionalCodec(Cursor),\n  finalized: OptionalCodec(Cursor),\n  starting: OptionalCodec(Cursor),\n});\n\nexport type StatusResponse = CodecType<typeof StatusResponse>;\n","import {\n  type ChannelCredentials,\n  type ChannelOptions,\n  type DefaultCallOptions,\n  type NormalizedServiceDefinition,\n  createChannel,\n  createClient as grpcCreateClient,\n} from \"nice-grpc\";\n\nimport * as proto from \"./proto\";\n\nimport assert from \"node:assert\";\nimport type { Codec } from \"./codec\";\nimport type { Cursor } from \"./common\";\nimport type { StreamConfig } from \"./config\";\nimport { StatusRequest, StatusResponse } from \"./status\";\nimport { type StreamDataRequest, StreamDataResponse } from \"./stream\";\n\nexport { ClientError, ServerError, Status, Metadata } from \"nice-grpc\";\n\nconst DEFAULT_TIMEOUT_MS = 45_000;\n\nexport class TimeoutError extends Error {\n  constructor(timeout: number) {\n    super(`No message received in ${timeout}ms`);\n    this.name = \"TimeoutError\";\n  }\n}\n\n/** Client call options. */\nexport interface ClientCallOptions {\n  signal?: AbortSignal;\n}\n\nexport interface StreamDataOptions extends ClientCallOptions {\n  /** Stop at the specified cursor (inclusive). */\n  endingCursor?: Cursor;\n  /** Timeout between messages, in milliseconds. */\n  timeout?: number;\n}\n\n/** DNA client. */\nexport interface Client<TFilter, TBlock> {\n  /** Fetch the DNA stream status. */\n  status(\n    request?: StatusRequest,\n    options?: ClientCallOptions,\n  ): Promise<StatusResponse>;\n\n  /** Start streaming data from the DNA server. */\n  streamData(\n    request: StreamDataRequest<TFilter>,\n    options?: StreamDataOptions,\n  ): AsyncIterable<StreamDataResponse<TBlock>>;\n}\n\nexport type CreateClientOptions = {\n  defaultCallOptions?: DefaultCallOptions<\n    NormalizedServiceDefinition<proto.stream.DnaStreamDefinition>\n  >;\n  credentials?: ChannelCredentials;\n  channelOptions?: ChannelOptions;\n};\n\n/** Create a client connecting to the DNA grpc service. */\nexport function createClient<TFilter, TBlock>(\n  config: StreamConfig<TFilter, TBlock>,\n  streamUrl: string,\n  options: CreateClientOptions = {},\n) {\n  const channel = createChannel(\n    streamUrl,\n    options?.credentials,\n    options?.channelOptions,\n  );\n\n  const client: proto.stream.DnaStreamClient = grpcCreateClient(\n    proto.stream.DnaStreamDefinition,\n    channel,\n    options?.defaultCallOptions,\n  );\n\n  return new GrpcClient(config, client);\n}\n\nexport class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {\n  private encodeRequest;\n\n  constructor(\n    private config: StreamConfig<TFilter, TBlock>,\n    private client: proto.stream.DnaStreamClient,\n  ) {\n    this.encodeRequest = config.Request.encode;\n  }\n\n  async status(request?: StatusRequest, options?: ClientCallOptions) {\n    const response = await this.client.status(\n      StatusRequest.encode(request ?? {}),\n      options,\n    );\n    return StatusResponse.decode(response);\n  }\n\n  streamData(request: StreamDataRequest<TFilter>, options?: StreamDataOptions) {\n    const it = this.client.streamData(this.encodeRequest(request), options);\n    return new StreamDataIterable(it, this.config.Block, options);\n  }\n}\n\nexport class StreamDataIterable<TBlock> {\n  constructor(\n    private it: AsyncIterable<proto.stream.StreamDataResponse>,\n    private schema: Codec<TBlock, Uint8Array>,\n    private options?: StreamDataOptions,\n  ) {}\n\n  [Symbol.asyncIterator](): AsyncIterator<StreamDataResponse<TBlock>> {\n    const inner = this.it[Symbol.asyncIterator]();\n    const schema = StreamDataResponse(this.schema);\n    const decoder = schema.decode;\n    const { endingCursor, timeout = DEFAULT_TIMEOUT_MS } = this.options ?? {};\n    let shouldStop = false;\n\n    let clock: string | number | NodeJS.Timeout | undefined;\n\n    return {\n      async next() {\n        if (shouldStop) {\n          return { done: true, value: undefined };\n        }\n\n        // biome-ignore lint/suspicious/noExplicitAny: any is ok\n        const t: Promise<{ done: boolean; value: any }> = new Promise(\n          (_, reject) => {\n            clock = setTimeout(() => {\n              reject(new TimeoutError(timeout));\n            }, timeout);\n          },\n        );\n\n        try {\n          const { done, value } = await Promise.race([inner.next(), t]);\n\n          clearTimeout(clock);\n\n          if (done || value.message === undefined) {\n            return { done: true, value: undefined };\n          }\n\n          const decodedMessage = decoder(value.message);\n\n          if (endingCursor) {\n            assert(value.message.$case === \"data\");\n            assert(decodedMessage._tag === \"data\");\n\n            const { orderKey, uniqueKey } = endingCursor;\n            const endCursor = decodedMessage.data.endCursor;\n\n            // Check if the orderKey matches\n            if (orderKey === endCursor?.orderKey) {\n              // If a uniqueKey is specified, it must also match\n              if (!uniqueKey || uniqueKey === endCursor.uniqueKey) {\n                shouldStop = true;\n                return { done: false, value: decodedMessage };\n              }\n            }\n          }\n\n          return {\n            done: false,\n            value: decodedMessage,\n          };\n        } finally {\n          clearTimeout(clock);\n        }\n      },\n    };\n  }\n}\n","/** Track data rate using high precision timers. */\nexport class RateGauge {\n  private interval: number;\n  private prev?: bigint;\n  private rateMs?: number;\n  private var: number;\n\n  constructor(intervalSeconds: number) {\n    // Convert seconds to milliseconds.\n    this.interval = intervalSeconds * 1_000;\n    this.var = 0;\n  }\n\n  public record(items: number) {\n    // Compute the exponential moving average of the rate.\n    const prev = this.prev;\n    const now = process.hrtime.bigint();\n    this.prev = now;\n\n    if (!prev) {\n      return;\n    }\n\n    const deltaMs = Number(now - prev) / 1_000_000;\n    // rate in items/ms.\n    const rateMs = items / deltaMs;\n\n    if (this.rateMs === undefined) {\n      this.rateMs = rateMs;\n      this.var = 0;\n      return;\n    }\n\n    const alpha = 1 - Math.exp(-deltaMs / this.interval);\n    this.rateMs = alpha * rateMs + (1 - alpha) * this.rateMs;\n\n    const diff = rateMs - this.rateMs;\n    const incr = alpha * diff;\n    this.var = (1 - alpha) * (this.var + incr * diff);\n  }\n\n  /** Returns the average rate per second. */\n  public average() {\n    if (this.rateMs === undefined) {\n      return undefined;\n    }\n    return this.rateMs * 1_000;\n  }\n\n  /** Returns the variance. */\n  public variance() {\n    return this.var;\n  }\n}\n"],"names":["MessageCodec","OptionalCodec","Cursor","config","createChannel","grpcCreateClient","proto.stream.DnaStreamDefinition","__publicField","StreamDataResponse","assert"],"mappings":";;;;;;;;;;;;;;;;;;;;AAIa,MAAA,aAAA,GAAgBA,kBAAa,CAAA,EAAE,EAAA;AAKrC,MAAM,iBAAiBA,kBAAa,CAAA;AAAA,EACzC,WAAA,EAAaC,oBAAcC,aAAM,CAAA;AAAA,EACjC,YAAA,EAAcD,oBAAcC,aAAM,CAAA;AAAA,EAClC,SAAA,EAAWD,oBAAcC,aAAM,CAAA;AAAA,EAC/B,QAAA,EAAUD,oBAAcC,aAAM,CAAA;AAChC,CAAC;;;;;;;;ACMD,MAAM,kBAAqB,GAAA,IAAA,CAAA;AAEpB,MAAM,qBAAqB,KAAM,CAAA;AAAA,EACtC,YAAY,OAAiB,EAAA;AAC3B,IAAM,KAAA,CAAA,CAAA,uBAAA,EAA0B,OAAO,CAAI,EAAA,CAAA,CAAA,CAAA;AAC3C,IAAA,IAAA,CAAK,IAAO,GAAA,cAAA,CAAA;AAAA,GACd;AACF,CAAA;AAsCO,SAAS,YACd,CAAAC,QAAA,EACA,SACA,EAAA,OAAA,GAA+B,EAC/B,EAAA;AACA,EAAA,MAAM,OAAU,GAAAC,sBAAA;AAAA,IACd,SAAA;AAAA,IACA,OAAS,EAAA,WAAA;AAAA,IACT,OAAS,EAAA,cAAA;AAAA,GACX,CAAA;AAEA,EAAA,MAAM,MAAuC,GAAAC,qBAAA;AAAA,IAC3CC,0BAAa;AAAA,IACb,OAAA;AAAA,IACA,OAAS,EAAA,kBAAA;AAAA,GACX,CAAA;AAEA,EAAO,OAAA,IAAI,UAAW,CAAAH,QAAA,EAAQ,MAAM,CAAA,CAAA;AACtC,CAAA;AAEO,MAAM,UAA+D,CAAA;AAAA,EAG1E,WAAA,CACU,QACA,MACR,EAAA;AAFQ,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AAJV,IAAQI,eAAA,CAAA,IAAA,EAAA,eAAA,CAAA,CAAA;AAMN,IAAK,IAAA,CAAA,aAAA,GAAgB,OAAO,OAAQ,CAAA,MAAA,CAAA;AAAA,GACtC;AAAA,EAEA,MAAM,MAAO,CAAA,OAAA,EAAyB,OAA6B,EAAA;AACjE,IAAM,MAAA,QAAA,GAAW,MAAM,IAAA,CAAK,MAAO,CAAA,MAAA;AAAA,MACjC,aAAc,CAAA,MAAA,CAAO,OAAW,IAAA,EAAE,CAAA;AAAA,MAClC,OAAA;AAAA,KACF,CAAA;AACA,IAAO,OAAA,cAAA,CAAe,OAAO,QAAQ,CAAA,CAAA;AAAA,GACvC;AAAA,EAEA,UAAA,CAAW,SAAqC,OAA6B,EAAA;AAC3E,IAAM,MAAA,EAAA,GAAK,KAAK,MAAO,CAAA,UAAA,CAAW,KAAK,aAAc,CAAA,OAAO,GAAG,OAAO,CAAA,CAAA;AACtE,IAAA,OAAO,IAAI,kBAAmB,CAAA,EAAA,EAAI,IAAK,CAAA,MAAA,CAAO,OAAO,OAAO,CAAA,CAAA;AAAA,GAC9D;AACF,CAAA;AAEO,MAAM,kBAA2B,CAAA;AAAA,EACtC,WAAA,CACU,EACA,EAAA,MAAA,EACA,OACR,EAAA;AAHQ,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA,CAAA;AAAA,GACP;AAAA,EAEH,CAAC,MAAO,CAAA,aAAa,CAA+C,GAAA;AAClE,IAAA,MAAM,KAAQ,GAAA,IAAA,CAAK,EAAG,CAAA,MAAA,CAAO,aAAa,CAAE,EAAA,CAAA;AAC5C,IAAM,MAAA,MAAA,GAASC,yBAAmB,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAC7C,IAAA,MAAM,UAAU,MAAO,CAAA,MAAA,CAAA;AACvB,IAAA,MAAM,EAAE,YAAc,EAAA,OAAA,GAAU,oBAAuB,GAAA,IAAA,CAAK,WAAW,EAAC,CAAA;AACxE,IAAA,IAAI,UAAa,GAAA,KAAA,CAAA;AAEjB,IAAI,IAAA,KAAA,CAAA;AAEJ,IAAO,OAAA;AAAA,MACL,MAAM,IAAO,GAAA;AACX,QAAA,IAAI,UAAY,EAAA;AACd,UAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,SACxC;AAGA,QAAA,MAAM,IAA4C,IAAI,OAAA;AAAA,UACpD,CAAC,GAAG,MAAW,KAAA;AACb,YAAA,KAAA,GAAQ,WAAW,MAAM;AACvB,cAAO,MAAA,CAAA,IAAI,YAAa,CAAA,OAAO,CAAC,CAAA,CAAA;AAAA,eAC/B,OAAO,CAAA,CAAA;AAAA,WACZ;AAAA,SACF,CAAA;AAEA,QAAI,IAAA;AACF,UAAA,MAAM,EAAE,IAAA,EAAM,KAAM,EAAA,GAAI,MAAM,OAAA,CAAQ,IAAK,CAAA,CAAC,KAAM,CAAA,IAAA,EAAQ,EAAA,CAAC,CAAC,CAAA,CAAA;AAE5D,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAElB,UAAI,IAAA,IAAA,IAAQ,KAAM,CAAA,OAAA,KAAY,KAAW,CAAA,EAAA;AACvC,YAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,WACxC;AAEA,UAAM,MAAA,cAAA,GAAiB,OAAQ,CAAA,KAAA,CAAM,OAAO,CAAA,CAAA;AAE5C,UAAA,IAAI,YAAc,EAAA;AAChB,YAAOC,eAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,KAAA,KAAU,MAAM,CAAA,CAAA;AACrC,YAAOA,eAAA,CAAA,cAAA,CAAe,SAAS,MAAM,CAAA,CAAA;AAErC,YAAM,MAAA,EAAE,QAAU,EAAA,SAAA,EAAc,GAAA,YAAA,CAAA;AAChC,YAAM,MAAA,SAAA,GAAY,eAAe,IAAK,CAAA,SAAA,CAAA;AAGtC,YAAI,IAAA,QAAA,KAAa,WAAW,QAAU,EAAA;AAEpC,cAAA,IAAI,CAAC,SAAA,IAAa,SAAc,KAAA,SAAA,CAAU,SAAW,EAAA;AACnD,gBAAa,UAAA,GAAA,IAAA,CAAA;AACb,gBAAA,OAAO,EAAE,IAAA,EAAM,KAAO,EAAA,KAAA,EAAO,cAAe,EAAA,CAAA;AAAA,eAC9C;AAAA,aACF;AAAA,WACF;AAEA,UAAO,OAAA;AAAA,YACL,IAAM,EAAA,KAAA;AAAA,YACN,KAAO,EAAA,cAAA;AAAA,WACT,CAAA;AAAA,SACA,SAAA;AACA,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAAA,SACpB;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AACF;;;;;;;;ACjLO,MAAM,SAAU,CAAA;AAAA,EAMrB,YAAY,eAAyB,EAAA;AALrC,IAAQ,aAAA,CAAA,IAAA,EAAA,UAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,MAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,QAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,KAAA,CAAA,CAAA;AAIN,IAAA,IAAA,CAAK,WAAW,eAAkB,GAAA,GAAA,CAAA;AAClC,IAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AAAA,GACb;AAAA,EAEO,OAAO,KAAe,EAAA;AAE3B,IAAA,MAAM,OAAO,IAAK,CAAA,IAAA,CAAA;AAClB,IAAM,MAAA,GAAA,GAAM,OAAQ,CAAA,MAAA,CAAO,MAAO,EAAA,CAAA;AAClC,IAAA,IAAA,CAAK,IAAO,GAAA,GAAA,CAAA;AAEZ,IAAA,IAAI,CAAC,IAAM,EAAA;AACT,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,OAAU,GAAA,MAAA,CAAO,GAAM,GAAA,IAAI,CAAI,GAAA,GAAA,CAAA;AAErC,IAAA,MAAM,SAAS,KAAQ,GAAA,OAAA,CAAA;AAEvB,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAA,IAAA,CAAK,MAAS,GAAA,MAAA,CAAA;AACd,MAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AACX,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,QAAQ,CAAI,GAAA,IAAA,CAAK,IAAI,CAAC,OAAA,GAAU,KAAK,QAAQ,CAAA,CAAA;AACnD,IAAA,IAAA,CAAK,MAAS,GAAA,KAAA,GAAQ,MAAU,GAAA,CAAA,CAAA,GAAI,SAAS,IAAK,CAAA,MAAA,CAAA;AAElD,IAAM,MAAA,IAAA,GAAO,SAAS,IAAK,CAAA,MAAA,CAAA;AAC3B,IAAA,MAAM,OAAO,KAAQ,GAAA,IAAA,CAAA;AACrB,IAAA,IAAA,CAAK,GAAO,GAAA,CAAA,CAAA,GAAI,KAAU,KAAA,IAAA,CAAK,MAAM,IAAO,GAAA,IAAA,CAAA,CAAA;AAAA,GAC9C;AAAA;AAAA,EAGO,OAAU,GAAA;AACf,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAO,OAAA,KAAA,CAAA,CAAA;AAAA,KACT;AACA,IAAA,OAAO,KAAK,MAAS,GAAA,GAAA,CAAA;AAAA,GACvB;AAAA;AAAA,EAGO,QAAW,GAAA;AAChB,IAAA,OAAO,IAAK,CAAA,GAAA,CAAA;AAAA,GACd;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;"}
+{"version":3,"file":"index.cjs","sources":["../src/status.ts","../src/client.ts","../src/rate.ts"],"sourcesContent":["import { type CodecType, MessageCodec, OptionalCodec } from \"./codec\";\nimport { Cursor } from \"./common\";\n\n/** The request to the `status` endpoint. */\nexport const StatusRequest = MessageCodec({});\n\nexport type StatusRequest = CodecType<typeof StatusRequest>;\n\n/** The response from the `status` endpoint. */\nexport const StatusResponse = MessageCodec({\n  currentHead: OptionalCodec(Cursor),\n  lastIngested: OptionalCodec(Cursor),\n  finalized: OptionalCodec(Cursor),\n  starting: OptionalCodec(Cursor),\n});\n\nexport type StatusResponse = CodecType<typeof StatusResponse>;\n","import assert from \"node:assert\";\n\nimport consola from \"consola\";\nimport {\n  type ChannelCredentials,\n  type ChannelOptions,\n  type DefaultCallOptions,\n  Metadata,\n  type NormalizedServiceDefinition,\n  createChannel,\n  createClient as grpcCreateClient,\n} from \"nice-grpc\";\n\nimport * as proto from \"./proto\";\n\nimport type { Codec } from \"./codec\";\nimport type { Cursor } from \"./common\";\nimport type { StreamConfig } from \"./config\";\nimport { StatusRequest, StatusResponse } from \"./status\";\nimport { type StreamDataRequest, StreamDataResponse } from \"./stream\";\n\nexport { ClientError, ServerError, Status, Metadata } from \"nice-grpc\";\n\nconst DEFAULT_TIMEOUT_MS = 45_000;\n\nexport class TimeoutError extends Error {\n  constructor(timeout: number) {\n    super(`No message received in ${timeout}ms`);\n    this.name = \"TimeoutError\";\n  }\n}\n\n/** Client call options. */\nexport interface ClientCallOptions {\n  signal?: AbortSignal;\n}\n\nexport interface StreamDataOptions extends ClientCallOptions {\n  /** Stop at the specified cursor (inclusive). */\n  endingCursor?: Cursor;\n  /** Timeout between messages, in milliseconds. */\n  timeout?: number;\n}\n\n/** DNA client. */\nexport interface Client<TFilter, TBlock> {\n  /** Fetch the DNA stream status. */\n  status(\n    request?: StatusRequest,\n    options?: ClientCallOptions,\n  ): Promise<StatusResponse>;\n\n  /** Start streaming data from the DNA server. */\n  streamData(\n    request: StreamDataRequest<TFilter>,\n    options?: StreamDataOptions,\n  ): AsyncIterable<StreamDataResponse<TBlock>>;\n}\n\nexport type CreateClientOptions = {\n  defaultCallOptions?: DefaultCallOptions<\n    NormalizedServiceDefinition<proto.stream.DnaStreamDefinition>\n  >;\n  credentials?: ChannelCredentials;\n  channelOptions?: ChannelOptions;\n};\n\n/** Create a client connecting to the DNA grpc service. */\nexport function createClient<TFilter, TBlock>(\n  config: StreamConfig<TFilter, TBlock>,\n  streamUrl: string,\n  options: CreateClientOptions = {},\n) {\n  const channel = createChannel(\n    streamUrl,\n    options?.credentials,\n    options?.channelOptions,\n  );\n\n  const client: proto.stream.DnaStreamClient = grpcCreateClient(\n    proto.stream.DnaStreamDefinition,\n    channel,\n    options?.defaultCallOptions,\n  );\n\n  return new GrpcClient(config, client);\n}\n\nexport function createAuthenticatedClient<TFilter, TBlock>(\n  config: StreamConfig<TFilter, TBlock>,\n  streamUrl: string,\n  options?: CreateClientOptions,\n) {\n  const dnaToken = process.env.DNA_TOKEN;\n  if (!dnaToken) {\n    consola.warn(\n      \"DNA_TOKEN environment variable is not set. Trying to connect without authentication.\",\n    );\n  }\n\n  return createClient(config, streamUrl, {\n    ...options,\n    defaultCallOptions: {\n      ...(options?.defaultCallOptions ?? {}),\n      \"*\": {\n        metadata: Metadata({\n          Authorization: `Bearer ${dnaToken}`,\n        }),\n        // metadata cant be overrided with spread as its a class so we override it fully if user provided it.\n        ...(options?.defaultCallOptions?.[\"*\"] ?? {}),\n      },\n    },\n  });\n}\n\nexport class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {\n  private encodeRequest;\n\n  constructor(\n    private config: StreamConfig<TFilter, TBlock>,\n    private client: proto.stream.DnaStreamClient,\n  ) {\n    this.encodeRequest = config.Request.encode;\n  }\n\n  async status(request?: StatusRequest, options?: ClientCallOptions) {\n    const response = await this.client.status(\n      StatusRequest.encode(request ?? {}),\n      options,\n    );\n    return StatusResponse.decode(response);\n  }\n\n  streamData(request: StreamDataRequest<TFilter>, options?: StreamDataOptions) {\n    const it = this.client.streamData(this.encodeRequest(request), options);\n    return new StreamDataIterable(it, this.config.Block, options);\n  }\n}\n\nexport class StreamDataIterable<TBlock> {\n  constructor(\n    private it: AsyncIterable<proto.stream.StreamDataResponse>,\n    private schema: Codec<TBlock, Uint8Array>,\n    private options?: StreamDataOptions,\n  ) {}\n\n  [Symbol.asyncIterator](): AsyncIterator<StreamDataResponse<TBlock>> {\n    const inner = this.it[Symbol.asyncIterator]();\n    const schema = StreamDataResponse(this.schema);\n    const decoder = schema.decode;\n    const { endingCursor, timeout = DEFAULT_TIMEOUT_MS } = this.options ?? {};\n    let shouldStop = false;\n\n    let clock: string | number | NodeJS.Timeout | undefined;\n\n    return {\n      async next() {\n        if (shouldStop) {\n          return { done: true, value: undefined };\n        }\n\n        // biome-ignore lint/suspicious/noExplicitAny: any is ok\n        const t: Promise<{ done: boolean; value: any }> = new Promise(\n          (_, reject) => {\n            clock = setTimeout(() => {\n              reject(new TimeoutError(timeout));\n            }, timeout);\n          },\n        );\n\n        try {\n          const { done, value } = await Promise.race([inner.next(), t]);\n\n          clearTimeout(clock);\n\n          if (done || value.message === undefined) {\n            return { done: true, value: undefined };\n          }\n\n          const decodedMessage = decoder(value.message);\n\n          if (endingCursor) {\n            assert(value.message.$case === \"data\");\n            assert(decodedMessage._tag === \"data\");\n\n            const { orderKey, uniqueKey } = endingCursor;\n            const endCursor = decodedMessage.data.endCursor;\n\n            // Check if the orderKey matches\n            if (orderKey === endCursor?.orderKey) {\n              // If a uniqueKey is specified, it must also match\n              if (!uniqueKey || uniqueKey === endCursor.uniqueKey) {\n                shouldStop = true;\n                return { done: false, value: decodedMessage };\n              }\n            }\n          }\n\n          return {\n            done: false,\n            value: decodedMessage,\n          };\n        } finally {\n          clearTimeout(clock);\n        }\n      },\n    };\n  }\n}\n","/** Track data rate using high precision timers. */\nexport class RateGauge {\n  private interval: number;\n  private prev?: bigint;\n  private rateMs?: number;\n  private var: number;\n\n  constructor(intervalSeconds: number) {\n    // Convert seconds to milliseconds.\n    this.interval = intervalSeconds * 1_000;\n    this.var = 0;\n  }\n\n  public record(items: number) {\n    // Compute the exponential moving average of the rate.\n    const prev = this.prev;\n    const now = process.hrtime.bigint();\n    this.prev = now;\n\n    if (!prev) {\n      return;\n    }\n\n    const deltaMs = Number(now - prev) / 1_000_000;\n    // rate in items/ms.\n    const rateMs = items / deltaMs;\n\n    if (this.rateMs === undefined) {\n      this.rateMs = rateMs;\n      this.var = 0;\n      return;\n    }\n\n    const alpha = 1 - Math.exp(-deltaMs / this.interval);\n    this.rateMs = alpha * rateMs + (1 - alpha) * this.rateMs;\n\n    const diff = rateMs - this.rateMs;\n    const incr = alpha * diff;\n    this.var = (1 - alpha) * (this.var + incr * diff);\n  }\n\n  /** Returns the average rate per second. */\n  public average() {\n    if (this.rateMs === undefined) {\n      return undefined;\n    }\n    return this.rateMs * 1_000;\n  }\n\n  /** Returns the variance. */\n  public variance() {\n    return this.var;\n  }\n}\n"],"names":["MessageCodec","OptionalCodec","Cursor","config","createChannel","grpcCreateClient","proto.stream.DnaStreamDefinition","consola","Metadata","__publicField","StreamDataResponse","assert"],"mappings":";;;;;;;;;;;;;;;;;;;;;;AAIa,MAAA,aAAA,GAAgBA,kBAAa,CAAA,EAAE,EAAA;AAKrC,MAAM,iBAAiBA,kBAAa,CAAA;AAAA,EACzC,WAAA,EAAaC,oBAAcC,aAAM,CAAA;AAAA,EACjC,YAAA,EAAcD,oBAAcC,aAAM,CAAA;AAAA,EAClC,SAAA,EAAWD,oBAAcC,aAAM,CAAA;AAAA,EAC/B,QAAA,EAAUD,oBAAcC,aAAM,CAAA;AAChC,CAAC;;;;;;;;ACSD,MAAM,kBAAqB,GAAA,IAAA,CAAA;AAEpB,MAAM,qBAAqB,KAAM,CAAA;AAAA,EACtC,YAAY,OAAiB,EAAA;AAC3B,IAAM,KAAA,CAAA,CAAA,uBAAA,EAA0B,OAAO,CAAI,EAAA,CAAA,CAAA,CAAA;AAC3C,IAAA,IAAA,CAAK,IAAO,GAAA,cAAA,CAAA;AAAA,GACd;AACF,CAAA;AAsCO,SAAS,YACd,CAAAC,QAAA,EACA,SACA,EAAA,OAAA,GAA+B,EAC/B,EAAA;AACA,EAAA,MAAM,OAAU,GAAAC,sBAAA;AAAA,IACd,SAAA;AAAA,IACA,OAAS,EAAA,WAAA;AAAA,IACT,OAAS,EAAA,cAAA;AAAA,GACX,CAAA;AAEA,EAAA,MAAM,MAAuC,GAAAC,qBAAA;AAAA,IAC3CC,0BAAa;AAAA,IACb,OAAA;AAAA,IACA,OAAS,EAAA,kBAAA;AAAA,GACX,CAAA;AAEA,EAAO,OAAA,IAAI,UAAW,CAAAH,QAAA,EAAQ,MAAM,CAAA,CAAA;AACtC,CAAA;AAEgB,SAAA,yBAAA,CACd,MACA,EAAA,SAAA,EACA,OACA,EAAA;AACA,EAAM,MAAA,QAAA,GAAW,QAAQ,GAAI,CAAA,SAAA,CAAA;AAC7B,EAAA,IAAI,CAAC,QAAU,EAAA;AACb,IAAQI,gBAAA,CAAA,IAAA;AAAA,MACN,sFAAA;AAAA,KACF,CAAA;AAAA,GACF;AAEA,EAAO,OAAA,YAAA,CAAa,QAAQ,SAAW,EAAA;AAAA,IACrC,GAAG,OAAA;AAAA,IACH,kBAAoB,EAAA;AAAA,MAClB,GAAI,OAAS,EAAA,kBAAA,IAAsB,EAAC;AAAA,MACpC,GAAK,EAAA;AAAA,QACH,UAAUC,iBAAS,CAAA;AAAA,UACjB,aAAA,EAAe,UAAU,QAAQ,CAAA,CAAA;AAAA,SAClC,CAAA;AAAA;AAAA,QAED,GAAI,OAAA,EAAS,kBAAqB,GAAA,GAAG,KAAK,EAAC;AAAA,OAC7C;AAAA,KACF;AAAA,GACD,CAAA,CAAA;AACH,CAAA;AAEO,MAAM,UAA+D,CAAA;AAAA,EAG1E,WAAA,CACU,QACA,MACR,EAAA;AAFQ,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AAJV,IAAQC,eAAA,CAAA,IAAA,EAAA,eAAA,CAAA,CAAA;AAMN,IAAK,IAAA,CAAA,aAAA,GAAgB,OAAO,OAAQ,CAAA,MAAA,CAAA;AAAA,GACtC;AAAA,EAEA,MAAM,MAAO,CAAA,OAAA,EAAyB,OAA6B,EAAA;AACjE,IAAM,MAAA,QAAA,GAAW,MAAM,IAAA,CAAK,MAAO,CAAA,MAAA;AAAA,MACjC,aAAc,CAAA,MAAA,CAAO,OAAW,IAAA,EAAE,CAAA;AAAA,MAClC,OAAA;AAAA,KACF,CAAA;AACA,IAAO,OAAA,cAAA,CAAe,OAAO,QAAQ,CAAA,CAAA;AAAA,GACvC;AAAA,EAEA,UAAA,CAAW,SAAqC,OAA6B,EAAA;AAC3E,IAAM,MAAA,EAAA,GAAK,KAAK,MAAO,CAAA,UAAA,CAAW,KAAK,aAAc,CAAA,OAAO,GAAG,OAAO,CAAA,CAAA;AACtE,IAAA,OAAO,IAAI,kBAAmB,CAAA,EAAA,EAAI,IAAK,CAAA,MAAA,CAAO,OAAO,OAAO,CAAA,CAAA;AAAA,GAC9D;AACF,CAAA;AAEO,MAAM,kBAA2B,CAAA;AAAA,EACtC,WAAA,CACU,EACA,EAAA,MAAA,EACA,OACR,EAAA;AAHQ,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA,CAAA;AAAA,GACP;AAAA,EAEH,CAAC,MAAO,CAAA,aAAa,CAA+C,GAAA;AAClE,IAAA,MAAM,KAAQ,GAAA,IAAA,CAAK,EAAG,CAAA,MAAA,CAAO,aAAa,CAAE,EAAA,CAAA;AAC5C,IAAM,MAAA,MAAA,GAASC,yBAAmB,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAC7C,IAAA,MAAM,UAAU,MAAO,CAAA,MAAA,CAAA;AACvB,IAAA,MAAM,EAAE,YAAc,EAAA,OAAA,GAAU,oBAAuB,GAAA,IAAA,CAAK,WAAW,EAAC,CAAA;AACxE,IAAA,IAAI,UAAa,GAAA,KAAA,CAAA;AAEjB,IAAI,IAAA,KAAA,CAAA;AAEJ,IAAO,OAAA;AAAA,MACL,MAAM,IAAO,GAAA;AACX,QAAA,IAAI,UAAY,EAAA;AACd,UAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,SACxC;AAGA,QAAA,MAAM,IAA4C,IAAI,OAAA;AAAA,UACpD,CAAC,GAAG,MAAW,KAAA;AACb,YAAA,KAAA,GAAQ,WAAW,MAAM;AACvB,cAAO,MAAA,CAAA,IAAI,YAAa,CAAA,OAAO,CAAC,CAAA,CAAA;AAAA,eAC/B,OAAO,CAAA,CAAA;AAAA,WACZ;AAAA,SACF,CAAA;AAEA,QAAI,IAAA;AACF,UAAA,MAAM,EAAE,IAAA,EAAM,KAAM,EAAA,GAAI,MAAM,OAAA,CAAQ,IAAK,CAAA,CAAC,KAAM,CAAA,IAAA,EAAQ,EAAA,CAAC,CAAC,CAAA,CAAA;AAE5D,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAElB,UAAI,IAAA,IAAA,IAAQ,KAAM,CAAA,OAAA,KAAY,KAAW,CAAA,EAAA;AACvC,YAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,WACxC;AAEA,UAAM,MAAA,cAAA,GAAiB,OAAQ,CAAA,KAAA,CAAM,OAAO,CAAA,CAAA;AAE5C,UAAA,IAAI,YAAc,EAAA;AAChB,YAAOC,eAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,KAAA,KAAU,MAAM,CAAA,CAAA;AACrC,YAAOA,eAAA,CAAA,cAAA,CAAe,SAAS,MAAM,CAAA,CAAA;AAErC,YAAM,MAAA,EAAE,QAAU,EAAA,SAAA,EAAc,GAAA,YAAA,CAAA;AAChC,YAAM,MAAA,SAAA,GAAY,eAAe,IAAK,CAAA,SAAA,CAAA;AAGtC,YAAI,IAAA,QAAA,KAAa,WAAW,QAAU,EAAA;AAEpC,cAAA,IAAI,CAAC,SAAA,IAAa,SAAc,KAAA,SAAA,CAAU,SAAW,EAAA;AACnD,gBAAa,UAAA,GAAA,IAAA,CAAA;AACb,gBAAA,OAAO,EAAE,IAAA,EAAM,KAAO,EAAA,KAAA,EAAO,cAAe,EAAA,CAAA;AAAA,eAC9C;AAAA,aACF;AAAA,WACF;AAEA,UAAO,OAAA;AAAA,YACL,IAAM,EAAA,KAAA;AAAA,YACN,KAAO,EAAA,cAAA;AAAA,WACT,CAAA;AAAA,SACA,SAAA;AACA,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAAA,SACpB;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AACF;;;;;;;;AC/MO,MAAM,SAAU,CAAA;AAAA,EAMrB,YAAY,eAAyB,EAAA;AALrC,IAAQ,aAAA,CAAA,IAAA,EAAA,UAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,MAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,QAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,KAAA,CAAA,CAAA;AAIN,IAAA,IAAA,CAAK,WAAW,eAAkB,GAAA,GAAA,CAAA;AAClC,IAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AAAA,GACb;AAAA,EAEO,OAAO,KAAe,EAAA;AAE3B,IAAA,MAAM,OAAO,IAAK,CAAA,IAAA,CAAA;AAClB,IAAM,MAAA,GAAA,GAAM,OAAQ,CAAA,MAAA,CAAO,MAAO,EAAA,CAAA;AAClC,IAAA,IAAA,CAAK,IAAO,GAAA,GAAA,CAAA;AAEZ,IAAA,IAAI,CAAC,IAAM,EAAA;AACT,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,OAAU,GAAA,MAAA,CAAO,GAAM,GAAA,IAAI,CAAI,GAAA,GAAA,CAAA;AAErC,IAAA,MAAM,SAAS,KAAQ,GAAA,OAAA,CAAA;AAEvB,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAA,IAAA,CAAK,MAAS,GAAA,MAAA,CAAA;AACd,MAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AACX,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,QAAQ,CAAI,GAAA,IAAA,CAAK,IAAI,CAAC,OAAA,GAAU,KAAK,QAAQ,CAAA,CAAA;AACnD,IAAA,IAAA,CAAK,MAAS,GAAA,KAAA,GAAQ,MAAU,GAAA,CAAA,CAAA,GAAI,SAAS,IAAK,CAAA,MAAA,CAAA;AAElD,IAAM,MAAA,IAAA,GAAO,SAAS,IAAK,CAAA,MAAA,CAAA;AAC3B,IAAA,MAAM,OAAO,KAAQ,GAAA,IAAA,CAAA;AACrB,IAAA,IAAA,CAAK,GAAO,GAAA,CAAA,CAAA,GAAI,KAAU,KAAA,IAAA,CAAK,MAAM,IAAO,GAAA,IAAA,CAAA,CAAA;AAAA,GAC9C;AAAA;AAAA,EAGO,OAAU,GAAA;AACf,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAO,OAAA,KAAA,CAAA,CAAA;AAAA,KACT;AACA,IAAA,OAAO,KAAK,MAAS,GAAA,GAAA,CAAA;AAAA,GACvB;AAAA;AAAA,EAGO,QAAW,GAAA;AAChB,IAAA,OAAO,IAAK,CAAA,GAAA,CAAA;AAAA,GACd;AACF;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;"}

package/dist/index.d.cts

@@ -1,5 +1,5 @@
-import { s as stream } from './shared/protocol.a5762a90.cjs';
-export { B as Bytes, b as BytesFromUint8Array, w as Client, u as ClientCallOptions, x as CreateClientOptions, C as Cursor, e as CursorFromBytes, c as CursorProto, q as Data, g as DataFinality, h as DataProduction, D as DnaStreamClient, a as DnaStreamDefinition, k as Duration, j as DurationCodec, F as Finalize, G as GrpcClient, H as Heartbeat, I as Invalidate, R as ResponseWithoutData, S as StatusRequest, f as StatusResponse, o as StdErr, m as StdOut, t as StreamConfig, z as StreamDataIterable, v as StreamDataOptions, l as StreamDataRequest, r as StreamDataResponse, p as SystemMessage, T as TimeoutError, y as createClient, d as createCursor, i as isCursor, n as normalizeCursor } from './shared/protocol.a5762a90.cjs';
+import { s as stream } from './shared/protocol.0e734e33.cjs';
+export { B as Bytes, b as BytesFromUint8Array, w as Client, u as ClientCallOptions, x as CreateClientOptions, C as Cursor, e as CursorFromBytes, c as CursorProto, q as Data, g as DataFinality, h as DataProduction, D as DnaStreamClient, a as DnaStreamDefinition, k as Duration, j as DurationCodec, F as Finalize, G as GrpcClient, H as Heartbeat, I as Invalidate, R as ResponseWithoutData, S as StatusRequest, f as StatusResponse, o as StdErr, m as StdOut, t as StreamConfig, A as StreamDataIterable, v as StreamDataOptions, l as StreamDataRequest, r as StreamDataResponse, p as SystemMessage, T as TimeoutError, z as createAuthenticatedClient, y as createClient, d as createCursor, i as isCursor, n as normalizeCursor } from './shared/protocol.0e734e33.cjs';
 import _m0 from 'protobufjs/minimal.js';
 export { ClientError, Metadata, ServerError, Status } from 'nice-grpc';
 import './codec.cjs';

package/dist/index.d.mts

@@ -1,5 +1,5 @@
-import { s as stream } from './shared/protocol.21b07506.mjs';
-export { B as Bytes, b as BytesFromUint8Array, w as Client, u as ClientCallOptions, x as CreateClientOptions, C as Cursor, e as CursorFromBytes, c as CursorProto, q as Data, g as DataFinality, h as DataProduction, D as DnaStreamClient, a as DnaStreamDefinition, k as Duration, j as DurationCodec, F as Finalize, G as GrpcClient, H as Heartbeat, I as Invalidate, R as ResponseWithoutData, S as StatusRequest, f as StatusResponse, o as StdErr, m as StdOut, t as StreamConfig, z as StreamDataIterable, v as StreamDataOptions, l as StreamDataRequest, r as StreamDataResponse, p as SystemMessage, T as TimeoutError, y as createClient, d as createCursor, i as isCursor, n as normalizeCursor } from './shared/protocol.21b07506.mjs';
+import { s as stream } from './shared/protocol.21e66b9e.mjs';
+export { B as Bytes, b as BytesFromUint8Array, w as Client, u as ClientCallOptions, x as CreateClientOptions, C as Cursor, e as CursorFromBytes, c as CursorProto, q as Data, g as DataFinality, h as DataProduction, D as DnaStreamClient, a as DnaStreamDefinition, k as Duration, j as DurationCodec, F as Finalize, G as GrpcClient, H as Heartbeat, I as Invalidate, R as ResponseWithoutData, S as StatusRequest, f as StatusResponse, o as StdErr, m as StdOut, t as StreamConfig, A as StreamDataIterable, v as StreamDataOptions, l as StreamDataRequest, r as StreamDataResponse, p as SystemMessage, T as TimeoutError, z as createAuthenticatedClient, y as createClient, d as createCursor, i as isCursor, n as normalizeCursor } from './shared/protocol.21e66b9e.mjs';
 import _m0 from 'protobufjs/minimal.js';
 export { ClientError, Metadata, ServerError, Status } from 'nice-grpc';
 import './codec.mjs';

package/dist/index.d.ts

@@ -1,5 +1,5 @@
-import { s as stream } from './shared/protocol.526c6532.js';
-export { B as Bytes, b as BytesFromUint8Array, w as Client, u as ClientCallOptions, x as CreateClientOptions, C as Cursor, e as CursorFromBytes, c as CursorProto, q as Data, g as DataFinality, h as DataProduction, D as DnaStreamClient, a as DnaStreamDefinition, k as Duration, j as DurationCodec, F as Finalize, G as GrpcClient, H as Heartbeat, I as Invalidate, R as ResponseWithoutData, S as StatusRequest, f as StatusResponse, o as StdErr, m as StdOut, t as StreamConfig, z as StreamDataIterable, v as StreamDataOptions, l as StreamDataRequest, r as StreamDataResponse, p as SystemMessage, T as TimeoutError, y as createClient, d as createCursor, i as isCursor, n as normalizeCursor } from './shared/protocol.526c6532.js';
+import { s as stream } from './shared/protocol.8fb09325.js';
+export { B as Bytes, b as BytesFromUint8Array, w as Client, u as ClientCallOptions, x as CreateClientOptions, C as Cursor, e as CursorFromBytes, c as CursorProto, q as Data, g as DataFinality, h as DataProduction, D as DnaStreamClient, a as DnaStreamDefinition, k as Duration, j as DurationCodec, F as Finalize, G as GrpcClient, H as Heartbeat, I as Invalidate, R as ResponseWithoutData, S as StatusRequest, f as StatusResponse, o as StdErr, m as StdOut, t as StreamConfig, A as StreamDataIterable, v as StreamDataOptions, l as StreamDataRequest, r as StreamDataResponse, p as SystemMessage, T as TimeoutError, z as createAuthenticatedClient, y as createClient, d as createCursor, i as isCursor, n as normalizeCursor } from './shared/protocol.8fb09325.js';
 import _m0 from 'protobufjs/minimal.js';
 export { ClientError, Metadata, ServerError, Status } from 'nice-grpc';
 import './codec.js';

package/dist/index.mjs

@@ -1,10 +1,11 @@
 import { s as stream, t as testing, C as Cursor, D as DnaStreamDefinition, S as StreamDataResponse } from './shared/protocol.68fdd897.mjs';
 export { B as BytesFromUint8Array, a as CursorFromBytes, k as Data, b as DataFinality, d as DataProduction, e as DurationCodec, F as Finalize, H as Heartbeat, I as Invalidate, R as ResponseWithoutData, h as StdErr, g as StdOut, l as StreamConfig, f as StreamDataRequest, j as SystemMessage, c as createCursor, i as isCursor, n as normalizeCursor } from './shared/protocol.68fdd897.mjs';
 import { MessageCodec, OptionalCodec } from './codec.mjs';
-import { createChannel, createClient as createClient$1 } from 'nice-grpc';
+import assert from 'node:assert';
+import consola from 'consola';
+import { createChannel, createClient as createClient$1, Metadata } from 'nice-grpc';
 export { ClientError, Metadata, ServerError, Status } from 'nice-grpc';
 import 'protobufjs/minimal.js';
-import assert from 'node:assert';
 import 'viem';
 import 'long';
 
@@ -48,6 +49,27 @@ function createClient(config, streamUrl, options = {}) {
   );
   return new GrpcClient(config, client);
 }
+function createAuthenticatedClient(config, streamUrl, options) {
+  const dnaToken = process.env.DNA_TOKEN;
+  if (!dnaToken) {
+    consola.warn(
+      "DNA_TOKEN environment variable is not set. Trying to connect without authentication."
+    );
+  }
+  return createClient(config, streamUrl, {
+    ...options,
+    defaultCallOptions: {
+      ...options?.defaultCallOptions ?? {},
+      "*": {
+        metadata: Metadata({
+          Authorization: `Bearer ${dnaToken}`
+        }),
+        // metadata cant be overrided with spread as its a class so we override it fully if user provided it.
+        ...options?.defaultCallOptions?.["*"] ?? {}
+      }
+    }
+  });
+}
 class GrpcClient {
   constructor(config, client) {
     this.config = config;
@@ -171,5 +193,5 @@ class RateGauge {
   }
 }
 
-export { Cursor, DnaStreamDefinition, GrpcClient, RateGauge, StatusRequest, StatusResponse, StreamDataIterable, StreamDataResponse, TimeoutError, createClient, index as proto };
+export { Cursor, DnaStreamDefinition, GrpcClient, RateGauge, StatusRequest, StatusResponse, StreamDataIterable, StreamDataResponse, TimeoutError, createAuthenticatedClient, createClient, index as proto };
 //# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"file":"index.mjs","sources":["../src/status.ts","../src/client.ts","../src/rate.ts"],"sourcesContent":["import { type CodecType, MessageCodec, OptionalCodec } from \"./codec\";\nimport { Cursor } from \"./common\";\n\n/** The request to the `status` endpoint. */\nexport const StatusRequest = MessageCodec({});\n\nexport type StatusRequest = CodecType<typeof StatusRequest>;\n\n/** The response from the `status` endpoint. */\nexport const StatusResponse = MessageCodec({\n  currentHead: OptionalCodec(Cursor),\n  lastIngested: OptionalCodec(Cursor),\n  finalized: OptionalCodec(Cursor),\n  starting: OptionalCodec(Cursor),\n});\n\nexport type StatusResponse = CodecType<typeof StatusResponse>;\n","import {\n  type ChannelCredentials,\n  type ChannelOptions,\n  type DefaultCallOptions,\n  type NormalizedServiceDefinition,\n  createChannel,\n  createClient as grpcCreateClient,\n} from \"nice-grpc\";\n\nimport * as proto from \"./proto\";\n\nimport assert from \"node:assert\";\nimport type { Codec } from \"./codec\";\nimport type { Cursor } from \"./common\";\nimport type { StreamConfig } from \"./config\";\nimport { StatusRequest, StatusResponse } from \"./status\";\nimport { type StreamDataRequest, StreamDataResponse } from \"./stream\";\n\nexport { ClientError, ServerError, Status, Metadata } from \"nice-grpc\";\n\nconst DEFAULT_TIMEOUT_MS = 45_000;\n\nexport class TimeoutError extends Error {\n  constructor(timeout: number) {\n    super(`No message received in ${timeout}ms`);\n    this.name = \"TimeoutError\";\n  }\n}\n\n/** Client call options. */\nexport interface ClientCallOptions {\n  signal?: AbortSignal;\n}\n\nexport interface StreamDataOptions extends ClientCallOptions {\n  /** Stop at the specified cursor (inclusive). */\n  endingCursor?: Cursor;\n  /** Timeout between messages, in milliseconds. */\n  timeout?: number;\n}\n\n/** DNA client. */\nexport interface Client<TFilter, TBlock> {\n  /** Fetch the DNA stream status. */\n  status(\n    request?: StatusRequest,\n    options?: ClientCallOptions,\n  ): Promise<StatusResponse>;\n\n  /** Start streaming data from the DNA server. */\n  streamData(\n    request: StreamDataRequest<TFilter>,\n    options?: StreamDataOptions,\n  ): AsyncIterable<StreamDataResponse<TBlock>>;\n}\n\nexport type CreateClientOptions = {\n  defaultCallOptions?: DefaultCallOptions<\n    NormalizedServiceDefinition<proto.stream.DnaStreamDefinition>\n  >;\n  credentials?: ChannelCredentials;\n  channelOptions?: ChannelOptions;\n};\n\n/** Create a client connecting to the DNA grpc service. */\nexport function createClient<TFilter, TBlock>(\n  config: StreamConfig<TFilter, TBlock>,\n  streamUrl: string,\n  options: CreateClientOptions = {},\n) {\n  const channel = createChannel(\n    streamUrl,\n    options?.credentials,\n    options?.channelOptions,\n  );\n\n  const client: proto.stream.DnaStreamClient = grpcCreateClient(\n    proto.stream.DnaStreamDefinition,\n    channel,\n    options?.defaultCallOptions,\n  );\n\n  return new GrpcClient(config, client);\n}\n\nexport class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {\n  private encodeRequest;\n\n  constructor(\n    private config: StreamConfig<TFilter, TBlock>,\n    private client: proto.stream.DnaStreamClient,\n  ) {\n    this.encodeRequest = config.Request.encode;\n  }\n\n  async status(request?: StatusRequest, options?: ClientCallOptions) {\n    const response = await this.client.status(\n      StatusRequest.encode(request ?? {}),\n      options,\n    );\n    return StatusResponse.decode(response);\n  }\n\n  streamData(request: StreamDataRequest<TFilter>, options?: StreamDataOptions) {\n    const it = this.client.streamData(this.encodeRequest(request), options);\n    return new StreamDataIterable(it, this.config.Block, options);\n  }\n}\n\nexport class StreamDataIterable<TBlock> {\n  constructor(\n    private it: AsyncIterable<proto.stream.StreamDataResponse>,\n    private schema: Codec<TBlock, Uint8Array>,\n    private options?: StreamDataOptions,\n  ) {}\n\n  [Symbol.asyncIterator](): AsyncIterator<StreamDataResponse<TBlock>> {\n    const inner = this.it[Symbol.asyncIterator]();\n    const schema = StreamDataResponse(this.schema);\n    const decoder = schema.decode;\n    const { endingCursor, timeout = DEFAULT_TIMEOUT_MS } = this.options ?? {};\n    let shouldStop = false;\n\n    let clock: string | number | NodeJS.Timeout | undefined;\n\n    return {\n      async next() {\n        if (shouldStop) {\n          return { done: true, value: undefined };\n        }\n\n        // biome-ignore lint/suspicious/noExplicitAny: any is ok\n        const t: Promise<{ done: boolean; value: any }> = new Promise(\n          (_, reject) => {\n            clock = setTimeout(() => {\n              reject(new TimeoutError(timeout));\n            }, timeout);\n          },\n        );\n\n        try {\n          const { done, value } = await Promise.race([inner.next(), t]);\n\n          clearTimeout(clock);\n\n          if (done || value.message === undefined) {\n            return { done: true, value: undefined };\n          }\n\n          const decodedMessage = decoder(value.message);\n\n          if (endingCursor) {\n            assert(value.message.$case === \"data\");\n            assert(decodedMessage._tag === \"data\");\n\n            const { orderKey, uniqueKey } = endingCursor;\n            const endCursor = decodedMessage.data.endCursor;\n\n            // Check if the orderKey matches\n            if (orderKey === endCursor?.orderKey) {\n              // If a uniqueKey is specified, it must also match\n              if (!uniqueKey || uniqueKey === endCursor.uniqueKey) {\n                shouldStop = true;\n                return { done: false, value: decodedMessage };\n              }\n            }\n          }\n\n          return {\n            done: false,\n            value: decodedMessage,\n          };\n        } finally {\n          clearTimeout(clock);\n        }\n      },\n    };\n  }\n}\n","/** Track data rate using high precision timers. */\nexport class RateGauge {\n  private interval: number;\n  private prev?: bigint;\n  private rateMs?: number;\n  private var: number;\n\n  constructor(intervalSeconds: number) {\n    // Convert seconds to milliseconds.\n    this.interval = intervalSeconds * 1_000;\n    this.var = 0;\n  }\n\n  public record(items: number) {\n    // Compute the exponential moving average of the rate.\n    const prev = this.prev;\n    const now = process.hrtime.bigint();\n    this.prev = now;\n\n    if (!prev) {\n      return;\n    }\n\n    const deltaMs = Number(now - prev) / 1_000_000;\n    // rate in items/ms.\n    const rateMs = items / deltaMs;\n\n    if (this.rateMs === undefined) {\n      this.rateMs = rateMs;\n      this.var = 0;\n      return;\n    }\n\n    const alpha = 1 - Math.exp(-deltaMs / this.interval);\n    this.rateMs = alpha * rateMs + (1 - alpha) * this.rateMs;\n\n    const diff = rateMs - this.rateMs;\n    const incr = alpha * diff;\n    this.var = (1 - alpha) * (this.var + incr * diff);\n  }\n\n  /** Returns the average rate per second. */\n  public average() {\n    if (this.rateMs === undefined) {\n      return undefined;\n    }\n    return this.rateMs * 1_000;\n  }\n\n  /** Returns the variance. */\n  public variance() {\n    return this.var;\n  }\n}\n"],"names":["grpcCreateClient","proto.stream.DnaStreamDefinition","__publicField"],"mappings":";;;;;;;;;;;;;;;;AAIa,MAAA,aAAA,GAAgB,YAAa,CAAA,EAAE,EAAA;AAKrC,MAAM,iBAAiB,YAAa,CAAA;AAAA,EACzC,WAAA,EAAa,cAAc,MAAM,CAAA;AAAA,EACjC,YAAA,EAAc,cAAc,MAAM,CAAA;AAAA,EAClC,SAAA,EAAW,cAAc,MAAM,CAAA;AAAA,EAC/B,QAAA,EAAU,cAAc,MAAM,CAAA;AAChC,CAAC;;;;;;;;ACMD,MAAM,kBAAqB,GAAA,IAAA,CAAA;AAEpB,MAAM,qBAAqB,KAAM,CAAA;AAAA,EACtC,YAAY,OAAiB,EAAA;AAC3B,IAAM,KAAA,CAAA,CAAA,uBAAA,EAA0B,OAAO,CAAI,EAAA,CAAA,CAAA,CAAA;AAC3C,IAAA,IAAA,CAAK,IAAO,GAAA,cAAA,CAAA;AAAA,GACd;AACF,CAAA;AAsCO,SAAS,YACd,CAAA,MAAA,EACA,SACA,EAAA,OAAA,GAA+B,EAC/B,EAAA;AACA,EAAA,MAAM,OAAU,GAAA,aAAA;AAAA,IACd,SAAA;AAAA,IACA,OAAS,EAAA,WAAA;AAAA,IACT,OAAS,EAAA,cAAA;AAAA,GACX,CAAA;AAEA,EAAA,MAAM,MAAuC,GAAAA,cAAA;AAAA,IAC3CC,mBAAa;AAAA,IACb,OAAA;AAAA,IACA,OAAS,EAAA,kBAAA;AAAA,GACX,CAAA;AAEA,EAAO,OAAA,IAAI,UAAW,CAAA,MAAA,EAAQ,MAAM,CAAA,CAAA;AACtC,CAAA;AAEO,MAAM,UAA+D,CAAA;AAAA,EAG1E,WAAA,CACU,QACA,MACR,EAAA;AAFQ,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AAJV,IAAQC,eAAA,CAAA,IAAA,EAAA,eAAA,CAAA,CAAA;AAMN,IAAK,IAAA,CAAA,aAAA,GAAgB,OAAO,OAAQ,CAAA,MAAA,CAAA;AAAA,GACtC;AAAA,EAEA,MAAM,MAAO,CAAA,OAAA,EAAyB,OAA6B,EAAA;AACjE,IAAM,MAAA,QAAA,GAAW,MAAM,IAAA,CAAK,MAAO,CAAA,MAAA;AAAA,MACjC,aAAc,CAAA,MAAA,CAAO,OAAW,IAAA,EAAE,CAAA;AAAA,MAClC,OAAA;AAAA,KACF,CAAA;AACA,IAAO,OAAA,cAAA,CAAe,OAAO,QAAQ,CAAA,CAAA;AAAA,GACvC;AAAA,EAEA,UAAA,CAAW,SAAqC,OAA6B,EAAA;AAC3E,IAAM,MAAA,EAAA,GAAK,KAAK,MAAO,CAAA,UAAA,CAAW,KAAK,aAAc,CAAA,OAAO,GAAG,OAAO,CAAA,CAAA;AACtE,IAAA,OAAO,IAAI,kBAAmB,CAAA,EAAA,EAAI,IAAK,CAAA,MAAA,CAAO,OAAO,OAAO,CAAA,CAAA;AAAA,GAC9D;AACF,CAAA;AAEO,MAAM,kBAA2B,CAAA;AAAA,EACtC,WAAA,CACU,EACA,EAAA,MAAA,EACA,OACR,EAAA;AAHQ,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA,CAAA;AAAA,GACP;AAAA,EAEH,CAAC,MAAO,CAAA,aAAa,CAA+C,GAAA;AAClE,IAAA,MAAM,KAAQ,GAAA,IAAA,CAAK,EAAG,CAAA,MAAA,CAAO,aAAa,CAAE,EAAA,CAAA;AAC5C,IAAM,MAAA,MAAA,GAAS,kBAAmB,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAC7C,IAAA,MAAM,UAAU,MAAO,CAAA,MAAA,CAAA;AACvB,IAAA,MAAM,EAAE,YAAc,EAAA,OAAA,GAAU,oBAAuB,GAAA,IAAA,CAAK,WAAW,EAAC,CAAA;AACxE,IAAA,IAAI,UAAa,GAAA,KAAA,CAAA;AAEjB,IAAI,IAAA,KAAA,CAAA;AAEJ,IAAO,OAAA;AAAA,MACL,MAAM,IAAO,GAAA;AACX,QAAA,IAAI,UAAY,EAAA;AACd,UAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,SACxC;AAGA,QAAA,MAAM,IAA4C,IAAI,OAAA;AAAA,UACpD,CAAC,GAAG,MAAW,KAAA;AACb,YAAA,KAAA,GAAQ,WAAW,MAAM;AACvB,cAAO,MAAA,CAAA,IAAI,YAAa,CAAA,OAAO,CAAC,CAAA,CAAA;AAAA,eAC/B,OAAO,CAAA,CAAA;AAAA,WACZ;AAAA,SACF,CAAA;AAEA,QAAI,IAAA;AACF,UAAA,MAAM,EAAE,IAAA,EAAM,KAAM,EAAA,GAAI,MAAM,OAAA,CAAQ,IAAK,CAAA,CAAC,KAAM,CAAA,IAAA,EAAQ,EAAA,CAAC,CAAC,CAAA,CAAA;AAE5D,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAElB,UAAI,IAAA,IAAA,IAAQ,KAAM,CAAA,OAAA,KAAY,KAAW,CAAA,EAAA;AACvC,YAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,WACxC;AAEA,UAAM,MAAA,cAAA,GAAiB,OAAQ,CAAA,KAAA,CAAM,OAAO,CAAA,CAAA;AAE5C,UAAA,IAAI,YAAc,EAAA;AAChB,YAAO,MAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,KAAA,KAAU,MAAM,CAAA,CAAA;AACrC,YAAO,MAAA,CAAA,cAAA,CAAe,SAAS,MAAM,CAAA,CAAA;AAErC,YAAM,MAAA,EAAE,QAAU,EAAA,SAAA,EAAc,GAAA,YAAA,CAAA;AAChC,YAAM,MAAA,SAAA,GAAY,eAAe,IAAK,CAAA,SAAA,CAAA;AAGtC,YAAI,IAAA,QAAA,KAAa,WAAW,QAAU,EAAA;AAEpC,cAAA,IAAI,CAAC,SAAA,IAAa,SAAc,KAAA,SAAA,CAAU,SAAW,EAAA;AACnD,gBAAa,UAAA,GAAA,IAAA,CAAA;AACb,gBAAA,OAAO,EAAE,IAAA,EAAM,KAAO,EAAA,KAAA,EAAO,cAAe,EAAA,CAAA;AAAA,eAC9C;AAAA,aACF;AAAA,WACF;AAEA,UAAO,OAAA;AAAA,YACL,IAAM,EAAA,KAAA;AAAA,YACN,KAAO,EAAA,cAAA;AAAA,WACT,CAAA;AAAA,SACA,SAAA;AACA,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAAA,SACpB;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AACF;;;;;;;;ACjLO,MAAM,SAAU,CAAA;AAAA,EAMrB,YAAY,eAAyB,EAAA;AALrC,IAAQ,aAAA,CAAA,IAAA,EAAA,UAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,MAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,QAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,KAAA,CAAA,CAAA;AAIN,IAAA,IAAA,CAAK,WAAW,eAAkB,GAAA,GAAA,CAAA;AAClC,IAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AAAA,GACb;AAAA,EAEO,OAAO,KAAe,EAAA;AAE3B,IAAA,MAAM,OAAO,IAAK,CAAA,IAAA,CAAA;AAClB,IAAM,MAAA,GAAA,GAAM,OAAQ,CAAA,MAAA,CAAO,MAAO,EAAA,CAAA;AAClC,IAAA,IAAA,CAAK,IAAO,GAAA,GAAA,CAAA;AAEZ,IAAA,IAAI,CAAC,IAAM,EAAA;AACT,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,OAAU,GAAA,MAAA,CAAO,GAAM,GAAA,IAAI,CAAI,GAAA,GAAA,CAAA;AAErC,IAAA,MAAM,SAAS,KAAQ,GAAA,OAAA,CAAA;AAEvB,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAA,IAAA,CAAK,MAAS,GAAA,MAAA,CAAA;AACd,MAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AACX,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,QAAQ,CAAI,GAAA,IAAA,CAAK,IAAI,CAAC,OAAA,GAAU,KAAK,QAAQ,CAAA,CAAA;AACnD,IAAA,IAAA,CAAK,MAAS,GAAA,KAAA,GAAQ,MAAU,GAAA,CAAA,CAAA,GAAI,SAAS,IAAK,CAAA,MAAA,CAAA;AAElD,IAAM,MAAA,IAAA,GAAO,SAAS,IAAK,CAAA,MAAA,CAAA;AAC3B,IAAA,MAAM,OAAO,KAAQ,GAAA,IAAA,CAAA;AACrB,IAAA,IAAA,CAAK,GAAO,GAAA,CAAA,CAAA,GAAI,KAAU,KAAA,IAAA,CAAK,MAAM,IAAO,GAAA,IAAA,CAAA,CAAA;AAAA,GAC9C;AAAA;AAAA,EAGO,OAAU,GAAA;AACf,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAO,OAAA,KAAA,CAAA,CAAA;AAAA,KACT;AACA,IAAA,OAAO,KAAK,MAAS,GAAA,GAAA,CAAA;AAAA,GACvB;AAAA;AAAA,EAGO,QAAW,GAAA;AAChB,IAAA,OAAO,IAAK,CAAA,GAAA,CAAA;AAAA,GACd;AACF;;;;"}
+{"version":3,"file":"index.mjs","sources":["../src/status.ts","../src/client.ts","../src/rate.ts"],"sourcesContent":["import { type CodecType, MessageCodec, OptionalCodec } from \"./codec\";\nimport { Cursor } from \"./common\";\n\n/** The request to the `status` endpoint. */\nexport const StatusRequest = MessageCodec({});\n\nexport type StatusRequest = CodecType<typeof StatusRequest>;\n\n/** The response from the `status` endpoint. */\nexport const StatusResponse = MessageCodec({\n  currentHead: OptionalCodec(Cursor),\n  lastIngested: OptionalCodec(Cursor),\n  finalized: OptionalCodec(Cursor),\n  starting: OptionalCodec(Cursor),\n});\n\nexport type StatusResponse = CodecType<typeof StatusResponse>;\n","import assert from \"node:assert\";\n\nimport consola from \"consola\";\nimport {\n  type ChannelCredentials,\n  type ChannelOptions,\n  type DefaultCallOptions,\n  Metadata,\n  type NormalizedServiceDefinition,\n  createChannel,\n  createClient as grpcCreateClient,\n} from \"nice-grpc\";\n\nimport * as proto from \"./proto\";\n\nimport type { Codec } from \"./codec\";\nimport type { Cursor } from \"./common\";\nimport type { StreamConfig } from \"./config\";\nimport { StatusRequest, StatusResponse } from \"./status\";\nimport { type StreamDataRequest, StreamDataResponse } from \"./stream\";\n\nexport { ClientError, ServerError, Status, Metadata } from \"nice-grpc\";\n\nconst DEFAULT_TIMEOUT_MS = 45_000;\n\nexport class TimeoutError extends Error {\n  constructor(timeout: number) {\n    super(`No message received in ${timeout}ms`);\n    this.name = \"TimeoutError\";\n  }\n}\n\n/** Client call options. */\nexport interface ClientCallOptions {\n  signal?: AbortSignal;\n}\n\nexport interface StreamDataOptions extends ClientCallOptions {\n  /** Stop at the specified cursor (inclusive). */\n  endingCursor?: Cursor;\n  /** Timeout between messages, in milliseconds. */\n  timeout?: number;\n}\n\n/** DNA client. */\nexport interface Client<TFilter, TBlock> {\n  /** Fetch the DNA stream status. */\n  status(\n    request?: StatusRequest,\n    options?: ClientCallOptions,\n  ): Promise<StatusResponse>;\n\n  /** Start streaming data from the DNA server. */\n  streamData(\n    request: StreamDataRequest<TFilter>,\n    options?: StreamDataOptions,\n  ): AsyncIterable<StreamDataResponse<TBlock>>;\n}\n\nexport type CreateClientOptions = {\n  defaultCallOptions?: DefaultCallOptions<\n    NormalizedServiceDefinition<proto.stream.DnaStreamDefinition>\n  >;\n  credentials?: ChannelCredentials;\n  channelOptions?: ChannelOptions;\n};\n\n/** Create a client connecting to the DNA grpc service. */\nexport function createClient<TFilter, TBlock>(\n  config: StreamConfig<TFilter, TBlock>,\n  streamUrl: string,\n  options: CreateClientOptions = {},\n) {\n  const channel = createChannel(\n    streamUrl,\n    options?.credentials,\n    options?.channelOptions,\n  );\n\n  const client: proto.stream.DnaStreamClient = grpcCreateClient(\n    proto.stream.DnaStreamDefinition,\n    channel,\n    options?.defaultCallOptions,\n  );\n\n  return new GrpcClient(config, client);\n}\n\nexport function createAuthenticatedClient<TFilter, TBlock>(\n  config: StreamConfig<TFilter, TBlock>,\n  streamUrl: string,\n  options?: CreateClientOptions,\n) {\n  const dnaToken = process.env.DNA_TOKEN;\n  if (!dnaToken) {\n    consola.warn(\n      \"DNA_TOKEN environment variable is not set. Trying to connect without authentication.\",\n    );\n  }\n\n  return createClient(config, streamUrl, {\n    ...options,\n    defaultCallOptions: {\n      ...(options?.defaultCallOptions ?? {}),\n      \"*\": {\n        metadata: Metadata({\n          Authorization: `Bearer ${dnaToken}`,\n        }),\n        // metadata cant be overrided with spread as its a class so we override it fully if user provided it.\n        ...(options?.defaultCallOptions?.[\"*\"] ?? {}),\n      },\n    },\n  });\n}\n\nexport class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {\n  private encodeRequest;\n\n  constructor(\n    private config: StreamConfig<TFilter, TBlock>,\n    private client: proto.stream.DnaStreamClient,\n  ) {\n    this.encodeRequest = config.Request.encode;\n  }\n\n  async status(request?: StatusRequest, options?: ClientCallOptions) {\n    const response = await this.client.status(\n      StatusRequest.encode(request ?? {}),\n      options,\n    );\n    return StatusResponse.decode(response);\n  }\n\n  streamData(request: StreamDataRequest<TFilter>, options?: StreamDataOptions) {\n    const it = this.client.streamData(this.encodeRequest(request), options);\n    return new StreamDataIterable(it, this.config.Block, options);\n  }\n}\n\nexport class StreamDataIterable<TBlock> {\n  constructor(\n    private it: AsyncIterable<proto.stream.StreamDataResponse>,\n    private schema: Codec<TBlock, Uint8Array>,\n    private options?: StreamDataOptions,\n  ) {}\n\n  [Symbol.asyncIterator](): AsyncIterator<StreamDataResponse<TBlock>> {\n    const inner = this.it[Symbol.asyncIterator]();\n    const schema = StreamDataResponse(this.schema);\n    const decoder = schema.decode;\n    const { endingCursor, timeout = DEFAULT_TIMEOUT_MS } = this.options ?? {};\n    let shouldStop = false;\n\n    let clock: string | number | NodeJS.Timeout | undefined;\n\n    return {\n      async next() {\n        if (shouldStop) {\n          return { done: true, value: undefined };\n        }\n\n        // biome-ignore lint/suspicious/noExplicitAny: any is ok\n        const t: Promise<{ done: boolean; value: any }> = new Promise(\n          (_, reject) => {\n            clock = setTimeout(() => {\n              reject(new TimeoutError(timeout));\n            }, timeout);\n          },\n        );\n\n        try {\n          const { done, value } = await Promise.race([inner.next(), t]);\n\n          clearTimeout(clock);\n\n          if (done || value.message === undefined) {\n            return { done: true, value: undefined };\n          }\n\n          const decodedMessage = decoder(value.message);\n\n          if (endingCursor) {\n            assert(value.message.$case === \"data\");\n            assert(decodedMessage._tag === \"data\");\n\n            const { orderKey, uniqueKey } = endingCursor;\n            const endCursor = decodedMessage.data.endCursor;\n\n            // Check if the orderKey matches\n            if (orderKey === endCursor?.orderKey) {\n              // If a uniqueKey is specified, it must also match\n              if (!uniqueKey || uniqueKey === endCursor.uniqueKey) {\n                shouldStop = true;\n                return { done: false, value: decodedMessage };\n              }\n            }\n          }\n\n          return {\n            done: false,\n            value: decodedMessage,\n          };\n        } finally {\n          clearTimeout(clock);\n        }\n      },\n    };\n  }\n}\n","/** Track data rate using high precision timers. */\nexport class RateGauge {\n  private interval: number;\n  private prev?: bigint;\n  private rateMs?: number;\n  private var: number;\n\n  constructor(intervalSeconds: number) {\n    // Convert seconds to milliseconds.\n    this.interval = intervalSeconds * 1_000;\n    this.var = 0;\n  }\n\n  public record(items: number) {\n    // Compute the exponential moving average of the rate.\n    const prev = this.prev;\n    const now = process.hrtime.bigint();\n    this.prev = now;\n\n    if (!prev) {\n      return;\n    }\n\n    const deltaMs = Number(now - prev) / 1_000_000;\n    // rate in items/ms.\n    const rateMs = items / deltaMs;\n\n    if (this.rateMs === undefined) {\n      this.rateMs = rateMs;\n      this.var = 0;\n      return;\n    }\n\n    const alpha = 1 - Math.exp(-deltaMs / this.interval);\n    this.rateMs = alpha * rateMs + (1 - alpha) * this.rateMs;\n\n    const diff = rateMs - this.rateMs;\n    const incr = alpha * diff;\n    this.var = (1 - alpha) * (this.var + incr * diff);\n  }\n\n  /** Returns the average rate per second. */\n  public average() {\n    if (this.rateMs === undefined) {\n      return undefined;\n    }\n    return this.rateMs * 1_000;\n  }\n\n  /** Returns the variance. */\n  public variance() {\n    return this.var;\n  }\n}\n"],"names":["grpcCreateClient","proto.stream.DnaStreamDefinition","__publicField"],"mappings":";;;;;;;;;;;;;;;;;AAIa,MAAA,aAAA,GAAgB,YAAa,CAAA,EAAE,EAAA;AAKrC,MAAM,iBAAiB,YAAa,CAAA;AAAA,EACzC,WAAA,EAAa,cAAc,MAAM,CAAA;AAAA,EACjC,YAAA,EAAc,cAAc,MAAM,CAAA;AAAA,EAClC,SAAA,EAAW,cAAc,MAAM,CAAA;AAAA,EAC/B,QAAA,EAAU,cAAc,MAAM,CAAA;AAChC,CAAC;;;;;;;;ACSD,MAAM,kBAAqB,GAAA,IAAA,CAAA;AAEpB,MAAM,qBAAqB,KAAM,CAAA;AAAA,EACtC,YAAY,OAAiB,EAAA;AAC3B,IAAM,KAAA,CAAA,CAAA,uBAAA,EAA0B,OAAO,CAAI,EAAA,CAAA,CAAA,CAAA;AAC3C,IAAA,IAAA,CAAK,IAAO,GAAA,cAAA,CAAA;AAAA,GACd;AACF,CAAA;AAsCO,SAAS,YACd,CAAA,MAAA,EACA,SACA,EAAA,OAAA,GAA+B,EAC/B,EAAA;AACA,EAAA,MAAM,OAAU,GAAA,aAAA;AAAA,IACd,SAAA;AAAA,IACA,OAAS,EAAA,WAAA;AAAA,IACT,OAAS,EAAA,cAAA;AAAA,GACX,CAAA;AAEA,EAAA,MAAM,MAAuC,GAAAA,cAAA;AAAA,IAC3CC,mBAAa;AAAA,IACb,OAAA;AAAA,IACA,OAAS,EAAA,kBAAA;AAAA,GACX,CAAA;AAEA,EAAO,OAAA,IAAI,UAAW,CAAA,MAAA,EAAQ,MAAM,CAAA,CAAA;AACtC,CAAA;AAEgB,SAAA,yBAAA,CACd,MACA,EAAA,SAAA,EACA,OACA,EAAA;AACA,EAAM,MAAA,QAAA,GAAW,QAAQ,GAAI,CAAA,SAAA,CAAA;AAC7B,EAAA,IAAI,CAAC,QAAU,EAAA;AACb,IAAQ,OAAA,CAAA,IAAA;AAAA,MACN,sFAAA;AAAA,KACF,CAAA;AAAA,GACF;AAEA,EAAO,OAAA,YAAA,CAAa,QAAQ,SAAW,EAAA;AAAA,IACrC,GAAG,OAAA;AAAA,IACH,kBAAoB,EAAA;AAAA,MAClB,GAAI,OAAS,EAAA,kBAAA,IAAsB,EAAC;AAAA,MACpC,GAAK,EAAA;AAAA,QACH,UAAU,QAAS,CAAA;AAAA,UACjB,aAAA,EAAe,UAAU,QAAQ,CAAA,CAAA;AAAA,SAClC,CAAA;AAAA;AAAA,QAED,GAAI,OAAA,EAAS,kBAAqB,GAAA,GAAG,KAAK,EAAC;AAAA,OAC7C;AAAA,KACF;AAAA,GACD,CAAA,CAAA;AACH,CAAA;AAEO,MAAM,UAA+D,CAAA;AAAA,EAG1E,WAAA,CACU,QACA,MACR,EAAA;AAFQ,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AAJV,IAAQC,eAAA,CAAA,IAAA,EAAA,eAAA,CAAA,CAAA;AAMN,IAAK,IAAA,CAAA,aAAA,GAAgB,OAAO,OAAQ,CAAA,MAAA,CAAA;AAAA,GACtC;AAAA,EAEA,MAAM,MAAO,CAAA,OAAA,EAAyB,OAA6B,EAAA;AACjE,IAAM,MAAA,QAAA,GAAW,MAAM,IAAA,CAAK,MAAO,CAAA,MAAA;AAAA,MACjC,aAAc,CAAA,MAAA,CAAO,OAAW,IAAA,EAAE,CAAA;AAAA,MAClC,OAAA;AAAA,KACF,CAAA;AACA,IAAO,OAAA,cAAA,CAAe,OAAO,QAAQ,CAAA,CAAA;AAAA,GACvC;AAAA,EAEA,UAAA,CAAW,SAAqC,OAA6B,EAAA;AAC3E,IAAM,MAAA,EAAA,GAAK,KAAK,MAAO,CAAA,UAAA,CAAW,KAAK,aAAc,CAAA,OAAO,GAAG,OAAO,CAAA,CAAA;AACtE,IAAA,OAAO,IAAI,kBAAmB,CAAA,EAAA,EAAI,IAAK,CAAA,MAAA,CAAO,OAAO,OAAO,CAAA,CAAA;AAAA,GAC9D;AACF,CAAA;AAEO,MAAM,kBAA2B,CAAA;AAAA,EACtC,WAAA,CACU,EACA,EAAA,MAAA,EACA,OACR,EAAA;AAHQ,IAAA,IAAA,CAAA,EAAA,GAAA,EAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACA,IAAA,IAAA,CAAA,OAAA,GAAA,OAAA,CAAA;AAAA,GACP;AAAA,EAEH,CAAC,MAAO,CAAA,aAAa,CAA+C,GAAA;AAClE,IAAA,MAAM,KAAQ,GAAA,IAAA,CAAK,EAAG,CAAA,MAAA,CAAO,aAAa,CAAE,EAAA,CAAA;AAC5C,IAAM,MAAA,MAAA,GAAS,kBAAmB,CAAA,IAAA,CAAK,MAAM,CAAA,CAAA;AAC7C,IAAA,MAAM,UAAU,MAAO,CAAA,MAAA,CAAA;AACvB,IAAA,MAAM,EAAE,YAAc,EAAA,OAAA,GAAU,oBAAuB,GAAA,IAAA,CAAK,WAAW,EAAC,CAAA;AACxE,IAAA,IAAI,UAAa,GAAA,KAAA,CAAA;AAEjB,IAAI,IAAA,KAAA,CAAA;AAEJ,IAAO,OAAA;AAAA,MACL,MAAM,IAAO,GAAA;AACX,QAAA,IAAI,UAAY,EAAA;AACd,UAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,SACxC;AAGA,QAAA,MAAM,IAA4C,IAAI,OAAA;AAAA,UACpD,CAAC,GAAG,MAAW,KAAA;AACb,YAAA,KAAA,GAAQ,WAAW,MAAM;AACvB,cAAO,MAAA,CAAA,IAAI,YAAa,CAAA,OAAO,CAAC,CAAA,CAAA;AAAA,eAC/B,OAAO,CAAA,CAAA;AAAA,WACZ;AAAA,SACF,CAAA;AAEA,QAAI,IAAA;AACF,UAAA,MAAM,EAAE,IAAA,EAAM,KAAM,EAAA,GAAI,MAAM,OAAA,CAAQ,IAAK,CAAA,CAAC,KAAM,CAAA,IAAA,EAAQ,EAAA,CAAC,CAAC,CAAA,CAAA;AAE5D,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAElB,UAAI,IAAA,IAAA,IAAQ,KAAM,CAAA,OAAA,KAAY,KAAW,CAAA,EAAA;AACvC,YAAA,OAAO,EAAE,IAAA,EAAM,IAAM,EAAA,KAAA,EAAO,KAAU,CAAA,EAAA,CAAA;AAAA,WACxC;AAEA,UAAM,MAAA,cAAA,GAAiB,OAAQ,CAAA,KAAA,CAAM,OAAO,CAAA,CAAA;AAE5C,UAAA,IAAI,YAAc,EAAA;AAChB,YAAO,MAAA,CAAA,KAAA,CAAM,OAAQ,CAAA,KAAA,KAAU,MAAM,CAAA,CAAA;AACrC,YAAO,MAAA,CAAA,cAAA,CAAe,SAAS,MAAM,CAAA,CAAA;AAErC,YAAM,MAAA,EAAE,QAAU,EAAA,SAAA,EAAc,GAAA,YAAA,CAAA;AAChC,YAAM,MAAA,SAAA,GAAY,eAAe,IAAK,CAAA,SAAA,CAAA;AAGtC,YAAI,IAAA,QAAA,KAAa,WAAW,QAAU,EAAA;AAEpC,cAAA,IAAI,CAAC,SAAA,IAAa,SAAc,KAAA,SAAA,CAAU,SAAW,EAAA;AACnD,gBAAa,UAAA,GAAA,IAAA,CAAA;AACb,gBAAA,OAAO,EAAE,IAAA,EAAM,KAAO,EAAA,KAAA,EAAO,cAAe,EAAA,CAAA;AAAA,eAC9C;AAAA,aACF;AAAA,WACF;AAEA,UAAO,OAAA;AAAA,YACL,IAAM,EAAA,KAAA;AAAA,YACN,KAAO,EAAA,cAAA;AAAA,WACT,CAAA;AAAA,SACA,SAAA;AACA,UAAA,YAAA,CAAa,KAAK,CAAA,CAAA;AAAA,SACpB;AAAA,OACF;AAAA,KACF,CAAA;AAAA,GACF;AACF;;;;;;;;AC/MO,MAAM,SAAU,CAAA;AAAA,EAMrB,YAAY,eAAyB,EAAA;AALrC,IAAQ,aAAA,CAAA,IAAA,EAAA,UAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,MAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,QAAA,CAAA,CAAA;AACR,IAAQ,aAAA,CAAA,IAAA,EAAA,KAAA,CAAA,CAAA;AAIN,IAAA,IAAA,CAAK,WAAW,eAAkB,GAAA,GAAA,CAAA;AAClC,IAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AAAA,GACb;AAAA,EAEO,OAAO,KAAe,EAAA;AAE3B,IAAA,MAAM,OAAO,IAAK,CAAA,IAAA,CAAA;AAClB,IAAM,MAAA,GAAA,GAAM,OAAQ,CAAA,MAAA,CAAO,MAAO,EAAA,CAAA;AAClC,IAAA,IAAA,CAAK,IAAO,GAAA,GAAA,CAAA;AAEZ,IAAA,IAAI,CAAC,IAAM,EAAA;AACT,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,OAAU,GAAA,MAAA,CAAO,GAAM,GAAA,IAAI,CAAI,GAAA,GAAA,CAAA;AAErC,IAAA,MAAM,SAAS,KAAQ,GAAA,OAAA,CAAA;AAEvB,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAA,IAAA,CAAK,MAAS,GAAA,MAAA,CAAA;AACd,MAAA,IAAA,CAAK,GAAM,GAAA,CAAA,CAAA;AACX,MAAA,OAAA;AAAA,KACF;AAEA,IAAA,MAAM,QAAQ,CAAI,GAAA,IAAA,CAAK,IAAI,CAAC,OAAA,GAAU,KAAK,QAAQ,CAAA,CAAA;AACnD,IAAA,IAAA,CAAK,MAAS,GAAA,KAAA,GAAQ,MAAU,GAAA,CAAA,CAAA,GAAI,SAAS,IAAK,CAAA,MAAA,CAAA;AAElD,IAAM,MAAA,IAAA,GAAO,SAAS,IAAK,CAAA,MAAA,CAAA;AAC3B,IAAA,MAAM,OAAO,KAAQ,GAAA,IAAA,CAAA;AACrB,IAAA,IAAA,CAAK,GAAO,GAAA,CAAA,CAAA,GAAI,KAAU,KAAA,IAAA,CAAK,MAAM,IAAO,GAAA,IAAA,CAAA,CAAA;AAAA,GAC9C;AAAA;AAAA,EAGO,OAAU,GAAA;AACf,IAAI,IAAA,IAAA,CAAK,WAAW,KAAW,CAAA,EAAA;AAC7B,MAAO,OAAA,KAAA,CAAA,CAAA;AAAA,KACT;AACA,IAAA,OAAO,KAAK,MAAS,GAAA,GAAA,CAAA;AAAA,GACvB;AAAA;AAAA,EAGO,QAAW,GAAA;AAChB,IAAA,OAAO,IAAK,CAAA,GAAA,CAAA;AAAA,GACd;AACF;;;;"}

package/dist/shared/{protocol.a5762a90.d.cts → protocol.0e734e33.d.cts}

@@ -872,6 +872,7 @@ type CreateClientOptions = {
 };
 /** Create a client connecting to the DNA grpc service. */
 declare function createClient<TFilter, TBlock>(config: StreamConfig<TFilter, TBlock>, streamUrl: string, options?: CreateClientOptions): GrpcClient<TFilter, TBlock>;
+declare function createAuthenticatedClient<TFilter, TBlock>(config: StreamConfig<TFilter, TBlock>, streamUrl: string, options?: CreateClientOptions): GrpcClient<TFilter, TBlock>;
 declare class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {
     private config;
     private client;
@@ -905,4 +906,4 @@ declare class StreamDataIterable<TBlock> {
     [Symbol.asyncIterator](): AsyncIterator<StreamDataResponse<TBlock>>;
 }
 
-export { Cursor$1 as A, type Bytes as B, Cursor as C, type DnaStreamClient as D, DataFinality$1 as E, Finalize as F, GrpcClient as G, Heartbeat as H, Invalidate as I, DataProduction$1 as J, ResponseWithoutData as R, StatusRequest as S, TimeoutError as T, DnaStreamDefinition as a, BytesFromUint8Array as b, type CursorProto as c, createCursor as d, CursorFromBytes as e, StatusResponse as f, DataFinality as g, DataProduction as h, isCursor as i, DurationCodec as j, type Duration as k, StreamDataRequest as l, StdOut as m, normalizeCursor as n, StdErr as o, SystemMessage as p, Data as q, StreamDataResponse as r, stream as s, StreamConfig as t, type ClientCallOptions as u, type StreamDataOptions as v, type Client as w, type CreateClientOptions as x, createClient as y, StreamDataIterable as z };
+export { StreamDataIterable as A, type Bytes as B, Cursor as C, type DnaStreamClient as D, Cursor$1 as E, Finalize as F, GrpcClient as G, Heartbeat as H, Invalidate as I, DataFinality$1 as J, DataProduction$1 as K, ResponseWithoutData as R, StatusRequest as S, TimeoutError as T, DnaStreamDefinition as a, BytesFromUint8Array as b, type CursorProto as c, createCursor as d, CursorFromBytes as e, StatusResponse as f, DataFinality as g, DataProduction as h, isCursor as i, DurationCodec as j, type Duration as k, StreamDataRequest as l, StdOut as m, normalizeCursor as n, StdErr as o, SystemMessage as p, Data as q, StreamDataResponse as r, stream as s, StreamConfig as t, type ClientCallOptions as u, type StreamDataOptions as v, type Client as w, type CreateClientOptions as x, createClient as y, createAuthenticatedClient as z };

package/dist/shared/{protocol.21b07506.d.mts → protocol.21e66b9e.d.mts}

@@ -872,6 +872,7 @@ type CreateClientOptions = {
 };
 /** Create a client connecting to the DNA grpc service. */
 declare function createClient<TFilter, TBlock>(config: StreamConfig<TFilter, TBlock>, streamUrl: string, options?: CreateClientOptions): GrpcClient<TFilter, TBlock>;
+declare function createAuthenticatedClient<TFilter, TBlock>(config: StreamConfig<TFilter, TBlock>, streamUrl: string, options?: CreateClientOptions): GrpcClient<TFilter, TBlock>;
 declare class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {
     private config;
     private client;
@@ -905,4 +906,4 @@ declare class StreamDataIterable<TBlock> {
     [Symbol.asyncIterator](): AsyncIterator<StreamDataResponse<TBlock>>;
 }
 
-export { Cursor$1 as A, type Bytes as B, Cursor as C, type DnaStreamClient as D, DataFinality$1 as E, Finalize as F, GrpcClient as G, Heartbeat as H, Invalidate as I, DataProduction$1 as J, ResponseWithoutData as R, StatusRequest as S, TimeoutError as T, DnaStreamDefinition as a, BytesFromUint8Array as b, type CursorProto as c, createCursor as d, CursorFromBytes as e, StatusResponse as f, DataFinality as g, DataProduction as h, isCursor as i, DurationCodec as j, type Duration as k, StreamDataRequest as l, StdOut as m, normalizeCursor as n, StdErr as o, SystemMessage as p, Data as q, StreamDataResponse as r, stream as s, StreamConfig as t, type ClientCallOptions as u, type StreamDataOptions as v, type Client as w, type CreateClientOptions as x, createClient as y, StreamDataIterable as z };
+export { StreamDataIterable as A, type Bytes as B, Cursor as C, type DnaStreamClient as D, Cursor$1 as E, Finalize as F, GrpcClient as G, Heartbeat as H, Invalidate as I, DataFinality$1 as J, DataProduction$1 as K, ResponseWithoutData as R, StatusRequest as S, TimeoutError as T, DnaStreamDefinition as a, BytesFromUint8Array as b, type CursorProto as c, createCursor as d, CursorFromBytes as e, StatusResponse as f, DataFinality as g, DataProduction as h, isCursor as i, DurationCodec as j, type Duration as k, StreamDataRequest as l, StdOut as m, normalizeCursor as n, StdErr as o, SystemMessage as p, Data as q, StreamDataResponse as r, stream as s, StreamConfig as t, type ClientCallOptions as u, type StreamDataOptions as v, type Client as w, type CreateClientOptions as x, createClient as y, createAuthenticatedClient as z };

package/dist/shared/{protocol.526c6532.d.ts → protocol.8fb09325.d.ts}

@@ -872,6 +872,7 @@ type CreateClientOptions = {
 };
 /** Create a client connecting to the DNA grpc service. */
 declare function createClient<TFilter, TBlock>(config: StreamConfig<TFilter, TBlock>, streamUrl: string, options?: CreateClientOptions): GrpcClient<TFilter, TBlock>;
+declare function createAuthenticatedClient<TFilter, TBlock>(config: StreamConfig<TFilter, TBlock>, streamUrl: string, options?: CreateClientOptions): GrpcClient<TFilter, TBlock>;
 declare class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {
     private config;
     private client;
@@ -905,4 +906,4 @@ declare class StreamDataIterable<TBlock> {
     [Symbol.asyncIterator](): AsyncIterator<StreamDataResponse<TBlock>>;
 }
 
-export { Cursor$1 as A, type Bytes as B, Cursor as C, type DnaStreamClient as D, DataFinality$1 as E, Finalize as F, GrpcClient as G, Heartbeat as H, Invalidate as I, DataProduction$1 as J, ResponseWithoutData as R, StatusRequest as S, TimeoutError as T, DnaStreamDefinition as a, BytesFromUint8Array as b, type CursorProto as c, createCursor as d, CursorFromBytes as e, StatusResponse as f, DataFinality as g, DataProduction as h, isCursor as i, DurationCodec as j, type Duration as k, StreamDataRequest as l, StdOut as m, normalizeCursor as n, StdErr as o, SystemMessage as p, Data as q, StreamDataResponse as r, stream as s, StreamConfig as t, type ClientCallOptions as u, type StreamDataOptions as v, type Client as w, type CreateClientOptions as x, createClient as y, StreamDataIterable as z };
+export { StreamDataIterable as A, type Bytes as B, Cursor as C, type DnaStreamClient as D, Cursor$1 as E, Finalize as F, GrpcClient as G, Heartbeat as H, Invalidate as I, DataFinality$1 as J, DataProduction$1 as K, ResponseWithoutData as R, StatusRequest as S, TimeoutError as T, DnaStreamDefinition as a, BytesFromUint8Array as b, type CursorProto as c, createCursor as d, CursorFromBytes as e, StatusResponse as f, DataFinality as g, DataProduction as h, isCursor as i, DurationCodec as j, type Duration as k, StreamDataRequest as l, StdOut as m, normalizeCursor as n, StdErr as o, SystemMessage as p, Data as q, StreamDataResponse as r, stream as s, StreamConfig as t, type ClientCallOptions as u, type StreamDataOptions as v, type Client as w, type CreateClientOptions as x, createClient as y, createAuthenticatedClient as z };

package/dist/testing/index.d.cts

@@ -1,4 +1,4 @@
-import { w as Client, l as StreamDataRequest, v as StreamDataOptions, r as StreamDataResponse, S as StatusRequest, u as ClientCallOptions, f as StatusResponse, t as StreamConfig, B as Bytes, A as Cursor, E as DataFinality, J as DataProduction } from '../shared/protocol.a5762a90.cjs';
+import { w as Client, l as StreamDataRequest, v as StreamDataOptions, r as StreamDataResponse, S as StatusRequest, u as ClientCallOptions, f as StatusResponse, t as StreamConfig, B as Bytes, E as Cursor, J as DataFinality, K as DataProduction } from '../shared/protocol.0e734e33.cjs';
 import { MessageCodec, Codec, CodecType } from '../codec.cjs';
 import 'nice-grpc-common';
 import 'protobufjs/minimal.js';

package/dist/testing/index.d.mts

@@ -1,4 +1,4 @@
-import { w as Client, l as StreamDataRequest, v as StreamDataOptions, r as StreamDataResponse, S as StatusRequest, u as ClientCallOptions, f as StatusResponse, t as StreamConfig, B as Bytes, A as Cursor, E as DataFinality, J as DataProduction } from '../shared/protocol.21b07506.mjs';
+import { w as Client, l as StreamDataRequest, v as StreamDataOptions, r as StreamDataResponse, S as StatusRequest, u as ClientCallOptions, f as StatusResponse, t as StreamConfig, B as Bytes, E as Cursor, J as DataFinality, K as DataProduction } from '../shared/protocol.21e66b9e.mjs';
 import { MessageCodec, Codec, CodecType } from '../codec.mjs';
 import 'nice-grpc-common';
 import 'protobufjs/minimal.js';

package/dist/testing/index.d.ts

@@ -1,4 +1,4 @@
-import { w as Client, l as StreamDataRequest, v as StreamDataOptions, r as StreamDataResponse, S as StatusRequest, u as ClientCallOptions, f as StatusResponse, t as StreamConfig, B as Bytes, A as Cursor, E as DataFinality, J as DataProduction } from '../shared/protocol.526c6532.js';
+import { w as Client, l as StreamDataRequest, v as StreamDataOptions, r as StreamDataResponse, S as StatusRequest, u as ClientCallOptions, f as StatusResponse, t as StreamConfig, B as Bytes, E as Cursor, J as DataFinality, K as DataProduction } from '../shared/protocol.8fb09325.js';
 import { MessageCodec, Codec, CodecType } from '../codec.js';
 import 'nice-grpc-common';
 import 'protobufjs/minimal.js';

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@apibara/protocol",
-  "version": "2.1.0-beta.37",
+  "version": "2.1.0-beta.39",
   "type": "module",
   "files": [
     "dist",
@@ -47,6 +47,7 @@
     "vitest": "^1.6.0"
   },
   "dependencies": {
+    "consola": "^3.4.2",
     "long": "^5.2.3",
     "nice-grpc": "^2.1.8",
     "nice-grpc-common": "^2.0.2",

package/src/client.ts

@@ -1,7 +1,11 @@
+import assert from "node:assert";
+
+import consola from "consola";
 import {
   type ChannelCredentials,
   type ChannelOptions,
   type DefaultCallOptions,
+  Metadata,
   type NormalizedServiceDefinition,
   createChannel,
   createClient as grpcCreateClient,
@@ -9,7 +13,6 @@ import {
 
 import * as proto from "./proto";
 
-import assert from "node:assert";
 import type { Codec } from "./codec";
 import type { Cursor } from "./common";
 import type { StreamConfig } from "./config";
@@ -83,6 +86,33 @@ export function createClient<TFilter, TBlock>(
   return new GrpcClient(config, client);
 }
 
+export function createAuthenticatedClient<TFilter, TBlock>(
+  config: StreamConfig<TFilter, TBlock>,
+  streamUrl: string,
+  options?: CreateClientOptions,
+) {
+  const dnaToken = process.env.DNA_TOKEN;
+  if (!dnaToken) {
+    consola.warn(
+      "DNA_TOKEN environment variable is not set. Trying to connect without authentication.",
+    );
+  }
+
+  return createClient(config, streamUrl, {
+    ...options,
+    defaultCallOptions: {
+      ...(options?.defaultCallOptions ?? {}),
+      "*": {
+        metadata: Metadata({
+          Authorization: `Bearer ${dnaToken}`,
+        }),
+        // metadata cant be overrided with spread as its a class so we override it fully if user provided it.
+        ...(options?.defaultCallOptions?.["*"] ?? {}),
+      },
+    },
+  });
+}
+
 export class GrpcClient<TFilter, TBlock> implements Client<TFilter, TBlock> {
   private encodeRequest;