@drarzter/kafka-client 0.5.6 → 0.6.3

package/dist/index.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/nest/kafka.module.ts","../src/nest/kafka.constants.ts","../src/nest/kafka.explorer.ts","../src/nest/kafka.decorator.ts","../src/nest/kafka.health.ts"],"sourcesContent":["import { Module, DynamicModule, Provider, Logger } from \"@nestjs/common\";\nimport { DiscoveryModule } from \"@nestjs/core\";\nimport {\n  KafkaClient,\n  ClientId,\n  GroupId,\n  TopicMapConstraint,\n  KafkaInstrumentation,\n  KafkaClientOptions,\n} from \"../client/kafka.client\";\nimport { getKafkaClientToken } from \"./kafka.constants\";\nimport { KafkaExplorer } from \"./kafka.explorer\";\n\n/** Shared configuration fields for both `register()` and `registerAsync()`. */\ninterface KafkaModuleBaseOptions {\n  /** Optional name for multi-client setups. Must match `@InjectKafkaClient(name)`. */\n  name?: string;\n  /** If true, makes KAFKA_CLIENT available globally without importing KafkaModule in every feature module. */\n  isGlobal?: boolean;\n}\n\n/** Synchronous configuration for `KafkaModule.register()`. */\nexport interface KafkaModuleOptions extends KafkaModuleBaseOptions {\n  /** Unique Kafka client identifier. */\n  clientId: ClientId;\n  /** Consumer group identifier. */\n  groupId: GroupId;\n  /** List of Kafka broker addresses. */\n  brokers: string[];\n  /** Auto-create topics via admin on first use (send/consume). Useful for development. */\n  autoCreateTopics?: boolean;\n  /** When `true`, string topic keys are validated against any schema previously registered via a TopicDescriptor. Default: `true`. */\n  strictSchemas?: boolean;\n  /** Number of partitions for auto-created topics. Default: `1`. */\n  numPartitions?: number;\n  /** Client-wide instrumentation hooks (e.g. OTel). Applied to both send and consume paths. */\n  instrumentation?: KafkaInstrumentation[];\n  /** Called when a message is dropped without being sent to a DLQ. @see `KafkaClientOptions.onMessageLost` */\n  onMessageLost?: KafkaClientOptions[\"onMessageLost\"];\n  /** Called whenever a consumer group rebalance occurs. @see `KafkaClientOptions.onRebalance` */\n  onRebalance?: KafkaClientOptions[\"onRebalance\"];\n}\n\n/** Async configuration for `KafkaModule.registerAsync()` with dependency injection. */\nexport interface KafkaModuleAsyncOptions extends KafkaModuleBaseOptions {\n  imports?: any[];\n  useFactory: (\n    ...args: any[]\n  ) => KafkaModuleOptions | Promise<KafkaModuleOptions>;\n  inject?: any[];\n}\n\n/**\n * NestJS dynamic module for registering type-safe Kafka clients.\n * Use `register()` for static config or `registerAsync()` for DI-based config.\n */\n@Module({})\nexport class KafkaModule {\n  /** Register a Kafka client with static options. */\n  static register<T extends TopicMapConstraint<T>>(\n    options: KafkaModuleOptions,\n  ): DynamicModule {\n    const token = getKafkaClientToken(options.name);\n\n    const kafkaClientProvider: Provider = {\n      provide: token,\n      useFactory: () => KafkaModule.buildClient<T>(options),\n    };\n\n    return {\n      global: options.isGlobal ?? false,\n      module: KafkaModule,\n      imports: [DiscoveryModule],\n      providers: [\n        kafkaClientProvider,\n        KafkaModule.buildDestroyProvider(token),\n        KafkaExplorer,\n      ],\n      exports: [kafkaClientProvider],\n    };\n  }\n\n  /** Register a Kafka client with async/factory-based options. */\n  static registerAsync<T extends TopicMapConstraint<T>>(\n    asyncOptions: KafkaModuleAsyncOptions,\n  ): DynamicModule {\n    const token = getKafkaClientToken(asyncOptions.name);\n\n    const kafkaClientProvider: Provider = {\n      provide: token,\n      useFactory: async (...args: any[]): Promise<KafkaClient<T>> =>\n        KafkaModule.buildClient<T>(await asyncOptions.useFactory(...args)),\n      inject: asyncOptions.inject || [],\n    };\n\n    return {\n      global: asyncOptions.isGlobal ?? false,\n      module: KafkaModule,\n      imports: [...(asyncOptions.imports || []), DiscoveryModule],\n      providers: [\n        kafkaClientProvider,\n        KafkaModule.buildDestroyProvider(token),\n        KafkaExplorer,\n      ],\n      exports: [kafkaClientProvider],\n    };\n  }\n\n  private static async buildClient<T extends TopicMapConstraint<T>>(\n    options: KafkaModuleOptions,\n  ): Promise<KafkaClient<T>> {\n    const client = new KafkaClient<T>(\n      options.clientId,\n      options.groupId,\n      options.brokers,\n      {\n        autoCreateTopics: options.autoCreateTopics,\n        strictSchemas: options.strictSchemas,\n        numPartitions: options.numPartitions,\n        instrumentation: options.instrumentation,\n        onMessageLost: options.onMessageLost,\n        onRebalance: options.onRebalance,\n        logger: new Logger(`KafkaClient:${options.clientId}`),\n      },\n    );\n    await client.connectProducer();\n    return client;\n  }\n\n  private static buildDestroyProvider<T extends TopicMapConstraint<T>>(\n    token: string,\n  ): Provider {\n    return {\n      provide: `${token}_DESTROY`,\n      useFactory: (client: KafkaClient<T>) => ({\n        onModuleDestroy: () => client.disconnect(),\n      }),\n      inject: [token],\n    };\n  }\n}\n","/** Default DI token for the Kafka client. */\nexport const KAFKA_CLIENT = \"KAFKA_CLIENT\";\n\n/** Returns the DI token for a named (or default) Kafka client instance. */\nexport const getKafkaClientToken = (name?: string): string =>\n  name ? `KAFKA_CLIENT_${name}` : KAFKA_CLIENT;\n","import { Inject, Injectable, OnModuleInit, Logger } from \"@nestjs/common\";\nimport { DiscoveryService, ModuleRef } from \"@nestjs/core\";\nimport { KafkaClient } from \"../client/kafka.client\";\nimport {\n  KAFKA_SUBSCRIBER_METADATA,\n  KafkaSubscriberMetadata,\n} from \"./kafka.decorator\";\nimport { getKafkaClientToken } from \"./kafka.constants\";\n\ninterface SubscriberEntry extends KafkaSubscriberMetadata {\n  methodName: string | symbol;\n}\n\n/** Discovers `@SubscribeTo()` decorators and wires them to their Kafka clients on startup. */\n@Injectable()\nexport class KafkaExplorer implements OnModuleInit {\n  private readonly logger = new Logger(KafkaExplorer.name);\n\n  constructor(\n    @Inject(DiscoveryService)\n    private readonly discoveryService: DiscoveryService,\n    @Inject(ModuleRef)\n    private readonly moduleRef: ModuleRef,\n  ) {}\n\n  async onModuleInit() {\n    const providers = this.discoveryService.getProviders();\n\n    for (const wrapper of providers) {\n      const { instance } = wrapper;\n      if (!instance || typeof instance !== \"object\") continue;\n\n      const metadata: SubscriberEntry[] | undefined = Reflect.getMetadata(\n        KAFKA_SUBSCRIBER_METADATA,\n        instance.constructor,\n      );\n\n      if (!metadata || metadata.length === 0) continue;\n\n      for (const entry of metadata) {\n        const token = getKafkaClientToken(entry.clientName);\n        let client: KafkaClient<any>;\n\n        try {\n          client = this.moduleRef.get(token, { strict: false });\n        } catch {\n          this.logger.error(\n            `KafkaClient \"${entry.clientName || \"default\"}\" not found for @SubscribeTo on ${instance.constructor.name}.${String(entry.methodName)}`,\n          );\n          continue;\n        }\n\n        const handler = (instance as any)[entry.methodName].bind(instance);\n\n        const consumerOptions = { ...entry.options };\n        if (entry.schemas) {\n          consumerOptions.schemas = entry.schemas;\n        }\n\n        if (entry.batch) {\n          await client.startBatchConsumer(\n            entry.topics as any,\n            async (envelopes: any[], meta: any) => {\n              await handler(envelopes, meta);\n            },\n            consumerOptions,\n          );\n        } else {\n          await client.startConsumer(\n            entry.topics as any,\n            async (envelope: any) => {\n              await handler(envelope);\n            },\n            consumerOptions,\n          );\n        }\n\n        this.logger.log(\n          `Registered @SubscribeTo(${entry.topics.join(\", \")})${entry.batch ? \" [batch]\" : \"\"} on ${instance.constructor.name}.${String(entry.methodName)}`,\n        );\n      }\n    }\n  }\n}\n","import { Inject } from \"@nestjs/common\";\nimport { getKafkaClientToken } from \"./kafka.constants\";\nimport { ConsumerOptions } from \"../client/kafka.client\";\nimport { TopicDescriptor, SchemaLike } from \"../client/message/topic\";\n\nexport const KAFKA_SUBSCRIBER_METADATA = \"KAFKA_SUBSCRIBER_METADATA\";\n\nexport interface KafkaSubscriberMetadata {\n  topics: string[];\n  schemas?: Map<string, SchemaLike>;\n  options?: ConsumerOptions;\n  clientName?: string;\n  batch?: boolean;\n  methodName?: string | symbol;\n}\n\n/** Inject a `KafkaClient` instance. Pass a name to target a specific named client. */\nexport const InjectKafkaClient = (name?: string): ParameterDecorator =>\n  Inject(getKafkaClientToken(name));\n\n/**\n * Decorator that auto-subscribes a method to Kafka topics on module init.\n * The decorated method receives `(message, topic)` for each consumed message.\n */\nexport const SubscribeTo = (\n  topics:\n    | string\n    | string[]\n    | TopicDescriptor\n    | TopicDescriptor[]\n    | (string | TopicDescriptor)[],\n  options?: ConsumerOptions & { clientName?: string; batch?: boolean },\n): MethodDecorator => {\n  const arr = Array.isArray(topics) ? topics : [topics];\n  const topicsArray = arr.map((t) => (typeof t === \"string\" ? t : t.__topic));\n\n  // Extract schemas from descriptors that have them\n  const schemas = new Map<string, SchemaLike>();\n  for (const t of arr) {\n    if (typeof t !== \"string\" && t.__schema) {\n      schemas.set(t.__topic, t.__schema);\n    }\n  }\n\n  const { clientName, batch, ...consumerOptions } = options || {};\n\n  return (target, propertyKey, _descriptor) => {\n    const existing: KafkaSubscriberMetadata[] =\n      Reflect.getMetadata(KAFKA_SUBSCRIBER_METADATA, target.constructor) || [];\n\n    Reflect.defineMetadata(\n      KAFKA_SUBSCRIBER_METADATA,\n      [\n        ...existing,\n        {\n          topics: topicsArray,\n          schemas: schemas.size > 0 ? schemas : undefined,\n          options: Object.keys(consumerOptions).length\n            ? consumerOptions\n            : undefined,\n          clientName,\n          batch,\n          methodName: propertyKey,\n        },\n      ],\n      target.constructor,\n    );\n  };\n};\n","import { Injectable } from \"@nestjs/common\";\nimport { KafkaClient, TopicMapConstraint } from \"../client/kafka.client\";\n\n/** Result returned by `KafkaHealthIndicator.check()`. */\nexport type KafkaHealthResult =\n  | { status: \"up\"; clientId: string; topics: string[] }\n  | { status: \"down\"; clientId: string; error: string };\n\n/** Health check service. Call `check(client)` to verify broker connectivity. */\n@Injectable()\nexport class KafkaHealthIndicator {\n  async check<T extends TopicMapConstraint<T>>(\n    client: KafkaClient<T>,\n  ): Promise<KafkaHealthResult> {\n    try {\n      return await client.checkStatus();\n    } catch (error) {\n      return {\n        status: \"down\",\n        clientId: client.clientId,\n        error: error instanceof Error ? error.message : String(error),\n      };\n    }\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AAAA,SAAS,QAAiC,UAAAA,eAAc;AACxD,SAAS,uBAAuB;;;ACAzB,IAAM,eAAe;AAGrB,IAAM,sBAAsB,CAAC,SAClC,OAAO,gBAAgB,IAAI,KAAK;;;ACLlC,SAAS,UAAAC,SAAQ,YAA0B,cAAc;AACzD,SAAS,kBAAkB,iBAAiB;;;ACD5C,SAAS,cAAc;AAKhB,IAAM,4BAA4B;AAYlC,IAAM,oBAAoB,CAAC,SAChC,OAAO,oBAAoB,IAAI,CAAC;AAM3B,IAAM,cAAc,CACzB,QAMA,YACoB;AACpB,QAAM,MAAM,MAAM,QAAQ,MAAM,IAAI,SAAS,CAAC,MAAM;AACpD,QAAM,cAAc,IAAI,IAAI,CAAC,MAAO,OAAO,MAAM,WAAW,IAAI,EAAE,OAAQ;AAG1E,QAAM,UAAU,oBAAI,IAAwB;AAC5C,aAAW,KAAK,KAAK;AACnB,QAAI,OAAO,MAAM,YAAY,EAAE,UAAU;AACvC,cAAQ,IAAI,EAAE,SAAS,EAAE,QAAQ;AAAA,IACnC;AAAA,EACF;AAEA,QAAM,EAAE,YAAY,OAAO,GAAG,gBAAgB,IAAI,WAAW,CAAC;AAE9D,SAAO,CAAC,QAAQ,aAAa,gBAAgB;AAC3C,UAAM,WACJ,QAAQ,YAAY,2BAA2B,OAAO,WAAW,KAAK,CAAC;AAEzE,YAAQ;AAAA,MACN;AAAA,MACA;AAAA,QACE,GAAG;AAAA,QACH;AAAA,UACE,QAAQ;AAAA,UACR,SAAS,QAAQ,OAAO,IAAI,UAAU;AAAA,UACtC,SAAS,OAAO,KAAK,eAAe,EAAE,SAClC,kBACA;AAAA,UACJ;AAAA,UACA;AAAA,UACA,YAAY;AAAA,QACd;AAAA,MACF;AAAA,MACA,OAAO;AAAA,IACT;AAAA,EACF;AACF;;;ADrDO,IAAM,gBAAN,MAA4C;AAAA,EAGjD,YAEmB,kBAEA,WACjB;AAHiB;AAEA;AAAA,EAChB;AAAA,EAPc,SAAS,IAAI,OAAO,cAAc,IAAI;AAAA,EASvD,MAAM,eAAe;AACnB,UAAM,YAAY,KAAK,iBAAiB,aAAa;AAErD,eAAW,WAAW,WAAW;AAC/B,YAAM,EAAE,SAAS,IAAI;AACrB,UAAI,CAAC,YAAY,OAAO,aAAa,SAAU;AAE/C,YAAM,WAA0C,QAAQ;AAAA,QACtD;AAAA,QACA,SAAS;AAAA,MACX;AAEA,UAAI,CAAC,YAAY,SAAS,WAAW,EAAG;AAExC,iBAAW,SAAS,UAAU;AAC5B,cAAM,QAAQ,oBAAoB,MAAM,UAAU;AAClD,YAAI;AAEJ,YAAI;AACF,mBAAS,KAAK,UAAU,IAAI,OAAO,EAAE,QAAQ,MAAM,CAAC;AAAA,QACtD,QAAQ;AACN,eAAK,OAAO;AAAA,YACV,gBAAgB,MAAM,cAAc,SAAS,mCAAmC,SAAS,YAAY,IAAI,IAAI,OAAO,MAAM,UAAU,CAAC;AAAA,UACvI;AACA;AAAA,QACF;AAEA,cAAM,UAAW,SAAiB,MAAM,UAAU,EAAE,KAAK,QAAQ;AAEjE,cAAM,kBAAkB,EAAE,GAAG,MAAM,QAAQ;AAC3C,YAAI,MAAM,SAAS;AACjB,0BAAgB,UAAU,MAAM;AAAA,QAClC;AAEA,YAAI,MAAM,OAAO;AACf,gBAAM,OAAO;AAAA,YACX,MAAM;AAAA,YACN,OAAO,WAAkB,SAAc;AACrC,oBAAM,QAAQ,WAAW,IAAI;AAAA,YAC/B;AAAA,YACA;AAAA,UACF;AAAA,QACF,OAAO;AACL,gBAAM,OAAO;AAAA,YACX,MAAM;AAAA,YACN,OAAO,aAAkB;AACvB,oBAAM,QAAQ,QAAQ;AAAA,YACxB;AAAA,YACA;AAAA,UACF;AAAA,QACF;AAEA,aAAK,OAAO;AAAA,UACV,2BAA2B,MAAM,OAAO,KAAK,IAAI,CAAC,IAAI,MAAM,QAAQ,aAAa,EAAE,OAAO,SAAS,YAAY,IAAI,IAAI,OAAO,MAAM,UAAU,CAAC;AAAA,QACjJ;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AApEa,gBAAN;AAAA,EADN,WAAW;AAAA,EAKP,mBAAAC,QAAO,gBAAgB;AAAA,EAEvB,mBAAAA,QAAO,SAAS;AAAA,GANR;;;AF0CN,IAAM,cAAN,MAAkB;AAAA;AAAA,EAEvB,OAAO,SACL,SACe;AACf,UAAM,QAAQ,oBAAoB,QAAQ,IAAI;AAE9C,UAAM,sBAAgC;AAAA,MACpC,SAAS;AAAA,MACT,YAAY,MAAM,YAAY,YAAe,OAAO;AAAA,IACtD;AAEA,WAAO;AAAA,MACL,QAAQ,QAAQ,YAAY;AAAA,MAC5B,QAAQ;AAAA,MACR,SAAS,CAAC,eAAe;AAAA,MACzB,WAAW;AAAA,QACT;AAAA,QACA,YAAY,qBAAqB,KAAK;AAAA,QACtC;AAAA,MACF;AAAA,MACA,SAAS,CAAC,mBAAmB;AAAA,IAC/B;AAAA,EACF;AAAA;AAAA,EAGA,OAAO,cACL,cACe;AACf,UAAM,QAAQ,oBAAoB,aAAa,IAAI;AAEnD,UAAM,sBAAgC;AAAA,MACpC,SAAS;AAAA,MACT,YAAY,UAAU,SACpB,YAAY,YAAe,MAAM,aAAa,WAAW,GAAG,IAAI,CAAC;AAAA,MACnE,QAAQ,aAAa,UAAU,CAAC;AAAA,IAClC;AAEA,WAAO;AAAA,MACL,QAAQ,aAAa,YAAY;AAAA,MACjC,QAAQ;AAAA,MACR,SAAS,CAAC,GAAI,aAAa,WAAW,CAAC,GAAI,eAAe;AAAA,MAC1D,WAAW;AAAA,QACT;AAAA,QACA,YAAY,qBAAqB,KAAK;AAAA,QACtC;AAAA,MACF;AAAA,MACA,SAAS,CAAC,mBAAmB;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,aAAqB,YACnB,SACyB;AACzB,UAAM,SAAS,IAAI;AAAA,MACjB,QAAQ;AAAA,MACR,QAAQ;AAAA,MACR,QAAQ;AAAA,MACR;AAAA,QACE,kBAAkB,QAAQ;AAAA,QAC1B,eAAe,QAAQ;AAAA,QACvB,eAAe,QAAQ;AAAA,QACvB,iBAAiB,QAAQ;AAAA,QACzB,eAAe,QAAQ;AAAA,QACvB,aAAa,QAAQ;AAAA,QACrB,QAAQ,IAAIC,QAAO,eAAe,QAAQ,QAAQ,EAAE;AAAA,MACtD;AAAA,IACF;AACA,UAAM,OAAO,gBAAgB;AAC7B,WAAO;AAAA,EACT;AAAA,EAEA,OAAe,qBACb,OACU;AACV,WAAO;AAAA,MACL,SAAS,GAAG,KAAK;AAAA,MACjB,YAAY,CAAC,YAA4B;AAAA,QACvC,iBAAiB,MAAM,OAAO,WAAW;AAAA,MAC3C;AAAA,MACA,QAAQ,CAAC,KAAK;AAAA,IAChB;AAAA,EACF;AACF;AAnFa,cAAN;AAAA,EADN,OAAO,CAAC,CAAC;AAAA,GACG;;;AIzDb,SAAS,cAAAC,mBAAkB;AAUpB,IAAM,uBAAN,MAA2B;AAAA,EAChC,MAAM,MACJ,QAC4B;AAC5B,QAAI;AACF,aAAO,MAAM,OAAO,YAAY;AAAA,IAClC,SAAS,OAAO;AACd,aAAO;AAAA,QACL,QAAQ;AAAA,QACR,UAAU,OAAO;AAAA,QACjB,OAAO,iBAAiB,QAAQ,MAAM,UAAU,OAAO,KAAK;AAAA,MAC9D;AAAA,IACF;AAAA,EACF;AACF;AAda,uBAAN;AAAA,EADNC,YAAW;AAAA,GACC;","names":["Logger","Inject","Inject","Logger","Injectable","Injectable"]}
+{"version":3,"sources":["../src/nest/kafka.module.ts","../src/nest/kafka.constants.ts","../src/nest/kafka.explorer.ts","../src/nest/kafka.decorator.ts","../src/nest/kafka.health.ts"],"sourcesContent":["import { Module, DynamicModule, Provider, Logger } from \"@nestjs/common\";\nimport { DiscoveryModule } from \"@nestjs/core\";\nimport {\n  KafkaClient,\n  ClientId,\n  GroupId,\n  TopicMapConstraint,\n  KafkaInstrumentation,\n  KafkaClientOptions,\n} from \"../client/kafka.client\";\nimport { getKafkaClientToken } from \"./kafka.constants\";\nimport { KafkaExplorer } from \"./kafka.explorer\";\n\n/** Shared configuration fields for both `register()` and `registerAsync()`. */\ninterface KafkaModuleBaseOptions {\n  /** Optional name for multi-client setups. Must match `@InjectKafkaClient(name)`. */\n  name?: string;\n  /** If true, makes KAFKA_CLIENT available globally without importing KafkaModule in every feature module. */\n  isGlobal?: boolean;\n}\n\n/** Synchronous configuration for `KafkaModule.register()`. */\nexport interface KafkaModuleOptions extends KafkaModuleBaseOptions {\n  /** Unique Kafka client identifier. */\n  clientId: ClientId;\n  /** Consumer group identifier. */\n  groupId: GroupId;\n  /** List of Kafka broker addresses. */\n  brokers: string[];\n  /** Auto-create topics via admin on first use (send/consume). Useful for development. */\n  autoCreateTopics?: boolean;\n  /** When `true`, string topic keys are validated against any schema previously registered via a TopicDescriptor. Default: `true`. */\n  strictSchemas?: boolean;\n  /** Number of partitions for auto-created topics. Default: `1`. */\n  numPartitions?: number;\n  /** Client-wide instrumentation hooks (e.g. OTel). Applied to both send and consume paths. */\n  instrumentation?: KafkaInstrumentation[];\n  /** Called when a message is dropped without being sent to a DLQ. @see `KafkaClientOptions.onMessageLost` */\n  onMessageLost?: KafkaClientOptions[\"onMessageLost\"];\n  /** Called whenever a consumer group rebalance occurs. @see `KafkaClientOptions.onRebalance` */\n  onRebalance?: KafkaClientOptions[\"onRebalance\"];\n}\n\n/** Async configuration for `KafkaModule.registerAsync()` with dependency injection. */\nexport interface KafkaModuleAsyncOptions extends KafkaModuleBaseOptions {\n  imports?: any[];\n  useFactory: (\n    ...args: any[]\n  ) => KafkaModuleOptions | Promise<KafkaModuleOptions>;\n  inject?: any[];\n}\n\n/**\n * NestJS dynamic module for registering type-safe Kafka clients.\n * Use `register()` for static config or `registerAsync()` for DI-based config.\n */\n@Module({})\nexport class KafkaModule {\n  /** Register a Kafka client with static options. */\n  static register<T extends TopicMapConstraint<T>>(\n    options: KafkaModuleOptions,\n  ): DynamicModule {\n    const token = getKafkaClientToken(options.name);\n\n    const kafkaClientProvider: Provider = {\n      provide: token,\n      useFactory: () => KafkaModule.buildClient<T>(options),\n    };\n\n    return {\n      global: options.isGlobal ?? false,\n      module: KafkaModule,\n      imports: [DiscoveryModule],\n      providers: [\n        kafkaClientProvider,\n        KafkaModule.buildDestroyProvider(token),\n        KafkaExplorer,\n      ],\n      exports: [kafkaClientProvider],\n    };\n  }\n\n  /** Register a Kafka client with async/factory-based options. */\n  static registerAsync<T extends TopicMapConstraint<T>>(\n    asyncOptions: KafkaModuleAsyncOptions,\n  ): DynamicModule {\n    const token = getKafkaClientToken(asyncOptions.name);\n\n    const kafkaClientProvider: Provider = {\n      provide: token,\n      useFactory: async (...args: any[]): Promise<KafkaClient<T>> =>\n        KafkaModule.buildClient<T>(await asyncOptions.useFactory(...args)),\n      inject: asyncOptions.inject || [],\n    };\n\n    return {\n      global: asyncOptions.isGlobal ?? false,\n      module: KafkaModule,\n      imports: [...(asyncOptions.imports || []), DiscoveryModule],\n      providers: [\n        kafkaClientProvider,\n        KafkaModule.buildDestroyProvider(token),\n        KafkaExplorer,\n      ],\n      exports: [kafkaClientProvider],\n    };\n  }\n\n  private static async buildClient<T extends TopicMapConstraint<T>>(\n    options: KafkaModuleOptions,\n  ): Promise<KafkaClient<T>> {\n    const client = new KafkaClient<T>(\n      options.clientId,\n      options.groupId,\n      options.brokers,\n      {\n        autoCreateTopics: options.autoCreateTopics,\n        strictSchemas: options.strictSchemas,\n        numPartitions: options.numPartitions,\n        instrumentation: options.instrumentation,\n        onMessageLost: options.onMessageLost,\n        onRebalance: options.onRebalance,\n        logger: new Logger(`KafkaClient:${options.clientId}`),\n      },\n    );\n    await client.connectProducer();\n    return client;\n  }\n\n  private static buildDestroyProvider<T extends TopicMapConstraint<T>>(\n    token: string,\n  ): Provider {\n    return {\n      provide: `${token}_DESTROY`,\n      useFactory: (client: KafkaClient<T>) => ({\n        onModuleDestroy: () => client.disconnect(),\n      }),\n      inject: [token],\n    };\n  }\n}\n","/** Default DI token for the Kafka client. */\nexport const KAFKA_CLIENT = \"KAFKA_CLIENT\";\n\n/** Returns the DI token for a named (or default) Kafka client instance. */\nexport const getKafkaClientToken = (name?: string): string =>\n  name ? `KAFKA_CLIENT_${name}` : KAFKA_CLIENT;\n","import { Inject, Injectable, OnModuleInit, Logger } from \"@nestjs/common\";\nimport { DiscoveryService, ModuleRef } from \"@nestjs/core\";\nimport { KafkaClient } from \"../client/kafka.client\";\nimport {\n  KAFKA_SUBSCRIBER_METADATA,\n  KafkaSubscriberMetadata,\n} from \"./kafka.decorator\";\nimport { getKafkaClientToken } from \"./kafka.constants\";\n\ninterface SubscriberEntry extends KafkaSubscriberMetadata {\n  methodName: string | symbol;\n}\n\n/** Discovers `@SubscribeTo()` decorators and wires them to their Kafka clients on startup. */\n@Injectable()\nexport class KafkaExplorer implements OnModuleInit {\n  private readonly logger = new Logger(KafkaExplorer.name);\n\n  constructor(\n    @Inject(DiscoveryService)\n    private readonly discoveryService: DiscoveryService,\n    @Inject(ModuleRef)\n    private readonly moduleRef: ModuleRef,\n  ) {}\n\n  async onModuleInit() {\n    const providers = this.discoveryService.getProviders();\n\n    for (const wrapper of providers) {\n      const { instance } = wrapper;\n      if (!instance || typeof instance !== \"object\") continue;\n\n      const metadata: SubscriberEntry[] | undefined = Reflect.getMetadata(\n        KAFKA_SUBSCRIBER_METADATA,\n        instance.constructor,\n      );\n\n      if (!metadata || metadata.length === 0) continue;\n\n      for (const entry of metadata) {\n        const token = getKafkaClientToken(entry.clientName);\n        let client: KafkaClient<any>;\n\n        try {\n          client = this.moduleRef.get(token, { strict: false });\n        } catch {\n          this.logger.error(\n            `KafkaClient \"${entry.clientName || \"default\"}\" not found for @SubscribeTo on ${instance.constructor.name}.${String(entry.methodName)}`,\n          );\n          continue;\n        }\n\n        const handler = (instance as any)[entry.methodName].bind(instance);\n\n        const consumerOptions = { ...entry.options };\n        if (entry.schemas) {\n          consumerOptions.schemas = entry.schemas;\n        }\n\n        if (entry.batch) {\n          await client.startBatchConsumer(\n            entry.topics as any,\n            async (envelopes: any[], meta: any) => {\n              await handler(envelopes, meta);\n            },\n            consumerOptions,\n          );\n        } else {\n          await client.startConsumer(\n            entry.topics as any,\n            async (envelope: any) => {\n              await handler(envelope);\n            },\n            consumerOptions,\n          );\n        }\n\n        this.logger.log(\n          `Registered @SubscribeTo(${entry.topics.join(\", \")})${entry.batch ? \" [batch]\" : \"\"} on ${instance.constructor.name}.${String(entry.methodName)}`,\n        );\n      }\n    }\n  }\n}\n","import { Inject } from \"@nestjs/common\";\nimport { getKafkaClientToken } from \"./kafka.constants\";\nimport { ConsumerOptions } from \"../client/kafka.client\";\nimport { TopicDescriptor, SchemaLike } from \"../client/message/topic\";\n\nexport const KAFKA_SUBSCRIBER_METADATA = \"KAFKA_SUBSCRIBER_METADATA\";\n\nexport interface KafkaSubscriberMetadata {\n  topics: string[];\n  schemas?: Map<string, SchemaLike>;\n  options?: ConsumerOptions;\n  clientName?: string;\n  batch?: boolean;\n  methodName?: string | symbol;\n}\n\n/** Inject a `KafkaClient` instance. Pass a name to target a specific named client. */\nexport const InjectKafkaClient = (name?: string): ParameterDecorator =>\n  Inject(getKafkaClientToken(name));\n\n/**\n * Decorator that auto-subscribes a method to Kafka topics on module init.\n * The decorated method receives `(message, topic)` for each consumed message.\n */\nexport const SubscribeTo = (\n  topics:\n    | string\n    | string[]\n    | TopicDescriptor\n    | TopicDescriptor[]\n    | (string | TopicDescriptor)[],\n  options?: ConsumerOptions & { clientName?: string; batch?: boolean },\n): MethodDecorator => {\n  const arr = Array.isArray(topics) ? topics : [topics];\n  const topicsArray = arr.map((t) => (typeof t === \"string\" ? t : t.__topic));\n\n  // Extract schemas from descriptors that have them\n  const schemas = new Map<string, SchemaLike>();\n  for (const t of arr) {\n    if (typeof t !== \"string\" && t.__schema) {\n      schemas.set(t.__topic, t.__schema);\n    }\n  }\n\n  const { clientName, batch, ...consumerOptions } = options || {};\n\n  return (target, propertyKey, _descriptor) => {\n    const existing: KafkaSubscriberMetadata[] =\n      Reflect.getMetadata(KAFKA_SUBSCRIBER_METADATA, target.constructor) || [];\n\n    Reflect.defineMetadata(\n      KAFKA_SUBSCRIBER_METADATA,\n      [\n        ...existing,\n        {\n          topics: topicsArray,\n          schemas: schemas.size > 0 ? schemas : undefined,\n          options: Object.keys(consumerOptions).length\n            ? consumerOptions\n            : undefined,\n          clientName,\n          batch,\n          methodName: propertyKey,\n        },\n      ],\n      target.constructor,\n    );\n  };\n};\n","import { Injectable } from \"@nestjs/common\";\nimport type { IKafkaClient, KafkaHealthResult, TopicMapConstraint } from \"../client/types\";\nexport type { KafkaHealthResult } from \"../client/types\";\n\n/** Health check service. Call `check(client)` to verify broker connectivity. */\n@Injectable()\nexport class KafkaHealthIndicator {\n  async check<T extends TopicMapConstraint<T>>(\n    client: IKafkaClient<T>,\n  ): Promise<KafkaHealthResult> {\n    return client.checkStatus();\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;AAAA,SAAS,QAAiC,UAAAA,eAAc;AACxD,SAAS,uBAAuB;;;ACAzB,IAAM,eAAe;AAGrB,IAAM,sBAAsB,CAAC,SAClC,OAAO,gBAAgB,IAAI,KAAK;;;ACLlC,SAAS,UAAAC,SAAQ,YAA0B,cAAc;AACzD,SAAS,kBAAkB,iBAAiB;;;ACD5C,SAAS,cAAc;AAKhB,IAAM,4BAA4B;AAYlC,IAAM,oBAAoB,CAAC,SAChC,OAAO,oBAAoB,IAAI,CAAC;AAM3B,IAAM,cAAc,CACzB,QAMA,YACoB;AACpB,QAAM,MAAM,MAAM,QAAQ,MAAM,IAAI,SAAS,CAAC,MAAM;AACpD,QAAM,cAAc,IAAI,IAAI,CAAC,MAAO,OAAO,MAAM,WAAW,IAAI,EAAE,OAAQ;AAG1E,QAAM,UAAU,oBAAI,IAAwB;AAC5C,aAAW,KAAK,KAAK;AACnB,QAAI,OAAO,MAAM,YAAY,EAAE,UAAU;AACvC,cAAQ,IAAI,EAAE,SAAS,EAAE,QAAQ;AAAA,IACnC;AAAA,EACF;AAEA,QAAM,EAAE,YAAY,OAAO,GAAG,gBAAgB,IAAI,WAAW,CAAC;AAE9D,SAAO,CAAC,QAAQ,aAAa,gBAAgB;AAC3C,UAAM,WACJ,QAAQ,YAAY,2BAA2B,OAAO,WAAW,KAAK,CAAC;AAEzE,YAAQ;AAAA,MACN;AAAA,MACA;AAAA,QACE,GAAG;AAAA,QACH;AAAA,UACE,QAAQ;AAAA,UACR,SAAS,QAAQ,OAAO,IAAI,UAAU;AAAA,UACtC,SAAS,OAAO,KAAK,eAAe,EAAE,SAClC,kBACA;AAAA,UACJ;AAAA,UACA;AAAA,UACA,YAAY;AAAA,QACd;AAAA,MACF;AAAA,MACA,OAAO;AAAA,IACT;AAAA,EACF;AACF;;;ADrDO,IAAM,gBAAN,MAA4C;AAAA,EAGjD,YAEmB,kBAEA,WACjB;AAHiB;AAEA;AAAA,EAChB;AAAA,EAPc,SAAS,IAAI,OAAO,cAAc,IAAI;AAAA,EASvD,MAAM,eAAe;AACnB,UAAM,YAAY,KAAK,iBAAiB,aAAa;AAErD,eAAW,WAAW,WAAW;AAC/B,YAAM,EAAE,SAAS,IAAI;AACrB,UAAI,CAAC,YAAY,OAAO,aAAa,SAAU;AAE/C,YAAM,WAA0C,QAAQ;AAAA,QACtD;AAAA,QACA,SAAS;AAAA,MACX;AAEA,UAAI,CAAC,YAAY,SAAS,WAAW,EAAG;AAExC,iBAAW,SAAS,UAAU;AAC5B,cAAM,QAAQ,oBAAoB,MAAM,UAAU;AAClD,YAAI;AAEJ,YAAI;AACF,mBAAS,KAAK,UAAU,IAAI,OAAO,EAAE,QAAQ,MAAM,CAAC;AAAA,QACtD,QAAQ;AACN,eAAK,OAAO;AAAA,YACV,gBAAgB,MAAM,cAAc,SAAS,mCAAmC,SAAS,YAAY,IAAI,IAAI,OAAO,MAAM,UAAU,CAAC;AAAA,UACvI;AACA;AAAA,QACF;AAEA,cAAM,UAAW,SAAiB,MAAM,UAAU,EAAE,KAAK,QAAQ;AAEjE,cAAM,kBAAkB,EAAE,GAAG,MAAM,QAAQ;AAC3C,YAAI,MAAM,SAAS;AACjB,0BAAgB,UAAU,MAAM;AAAA,QAClC;AAEA,YAAI,MAAM,OAAO;AACf,gBAAM,OAAO;AAAA,YACX,MAAM;AAAA,YACN,OAAO,WAAkB,SAAc;AACrC,oBAAM,QAAQ,WAAW,IAAI;AAAA,YAC/B;AAAA,YACA;AAAA,UACF;AAAA,QACF,OAAO;AACL,gBAAM,OAAO;AAAA,YACX,MAAM;AAAA,YACN,OAAO,aAAkB;AACvB,oBAAM,QAAQ,QAAQ;AAAA,YACxB;AAAA,YACA;AAAA,UACF;AAAA,QACF;AAEA,aAAK,OAAO;AAAA,UACV,2BAA2B,MAAM,OAAO,KAAK,IAAI,CAAC,IAAI,MAAM,QAAQ,aAAa,EAAE,OAAO,SAAS,YAAY,IAAI,IAAI,OAAO,MAAM,UAAU,CAAC;AAAA,QACjJ;AAAA,MACF;AAAA,IACF;AAAA,EACF;AACF;AApEa,gBAAN;AAAA,EADN,WAAW;AAAA,EAKP,mBAAAC,QAAO,gBAAgB;AAAA,EAEvB,mBAAAA,QAAO,SAAS;AAAA,GANR;;;AF0CN,IAAM,cAAN,MAAkB;AAAA;AAAA,EAEvB,OAAO,SACL,SACe;AACf,UAAM,QAAQ,oBAAoB,QAAQ,IAAI;AAE9C,UAAM,sBAAgC;AAAA,MACpC,SAAS;AAAA,MACT,YAAY,MAAM,YAAY,YAAe,OAAO;AAAA,IACtD;AAEA,WAAO;AAAA,MACL,QAAQ,QAAQ,YAAY;AAAA,MAC5B,QAAQ;AAAA,MACR,SAAS,CAAC,eAAe;AAAA,MACzB,WAAW;AAAA,QACT;AAAA,QACA,YAAY,qBAAqB,KAAK;AAAA,QACtC;AAAA,MACF;AAAA,MACA,SAAS,CAAC,mBAAmB;AAAA,IAC/B;AAAA,EACF;AAAA;AAAA,EAGA,OAAO,cACL,cACe;AACf,UAAM,QAAQ,oBAAoB,aAAa,IAAI;AAEnD,UAAM,sBAAgC;AAAA,MACpC,SAAS;AAAA,MACT,YAAY,UAAU,SACpB,YAAY,YAAe,MAAM,aAAa,WAAW,GAAG,IAAI,CAAC;AAAA,MACnE,QAAQ,aAAa,UAAU,CAAC;AAAA,IAClC;AAEA,WAAO;AAAA,MACL,QAAQ,aAAa,YAAY;AAAA,MACjC,QAAQ;AAAA,MACR,SAAS,CAAC,GAAI,aAAa,WAAW,CAAC,GAAI,eAAe;AAAA,MAC1D,WAAW;AAAA,QACT;AAAA,QACA,YAAY,qBAAqB,KAAK;AAAA,QACtC;AAAA,MACF;AAAA,MACA,SAAS,CAAC,mBAAmB;AAAA,IAC/B;AAAA,EACF;AAAA,EAEA,aAAqB,YACnB,SACyB;AACzB,UAAM,SAAS,IAAI;AAAA,MACjB,QAAQ;AAAA,MACR,QAAQ;AAAA,MACR,QAAQ;AAAA,MACR;AAAA,QACE,kBAAkB,QAAQ;AAAA,QAC1B,eAAe,QAAQ;AAAA,QACvB,eAAe,QAAQ;AAAA,QACvB,iBAAiB,QAAQ;AAAA,QACzB,eAAe,QAAQ;AAAA,QACvB,aAAa,QAAQ;AAAA,QACrB,QAAQ,IAAIC,QAAO,eAAe,QAAQ,QAAQ,EAAE;AAAA,MACtD;AAAA,IACF;AACA,UAAM,OAAO,gBAAgB;AAC7B,WAAO;AAAA,EACT;AAAA,EAEA,OAAe,qBACb,OACU;AACV,WAAO;AAAA,MACL,SAAS,GAAG,KAAK;AAAA,MACjB,YAAY,CAAC,YAA4B;AAAA,QACvC,iBAAiB,MAAM,OAAO,WAAW;AAAA,MAC3C;AAAA,MACA,QAAQ,CAAC,KAAK;AAAA,IAChB;AAAA,EACF;AACF;AAnFa,cAAN;AAAA,EADN,OAAO,CAAC,CAAC;AAAA,GACG;;;AIzDb,SAAS,cAAAC,mBAAkB;AAMpB,IAAM,uBAAN,MAA2B;AAAA,EAChC,MAAM,MACJ,QAC4B;AAC5B,WAAO,OAAO,YAAY;AAAA,EAC5B;AACF;AANa,uBAAN;AAAA,EADNC,YAAW;AAAA,GACC;","names":["Logger","Inject","Inject","Logger","Injectable","Injectable"]}

package/dist/otel.d.mts

@@ -1,4 +1,4 @@
-import { K as KafkaInstrumentation } from './envelope-LeO5e3ob.mjs';
+import { K as KafkaInstrumentation } from './types-zFbQH_Cy.mjs';
 
 /**
  * Create a `KafkaInstrumentation` that automatically propagates

package/dist/otel.d.ts

@@ -1,4 +1,4 @@
-import { K as KafkaInstrumentation } from './envelope-LeO5e3ob.js';
+import { K as KafkaInstrumentation } from './types-zFbQH_Cy.js';
 
 /**
  * Create a `KafkaInstrumentation` that automatically propagates

package/dist/otel.js

@@ -26,8 +26,9 @@ module.exports = __toCommonJS(otel_exports);
 var import_api = require("@opentelemetry/api");
 function otelInstrumentation() {
   const tracer = import_api.trace.getTracer("@drarzter/kafka-client");
+  const activeSpans = /* @__PURE__ */ new Map();
   return {
-    beforeSend(topic, headers) {
+    beforeSend(_topic, headers) {
       import_api.propagation.inject(import_api.context.active(), headers);
     },
     afterSend(_topic) {
@@ -48,10 +49,20 @@ function otelInstrumentation() {
         },
         parentCtx
       );
-      return () => span.end();
+      const spanCtx = import_api.trace.setSpan(parentCtx, span);
+      activeSpans.set(envelope.eventId, span);
+      return {
+        cleanup() {
+          span.end();
+          activeSpans.delete(envelope.eventId);
+        },
+        wrap(fn) {
+          return import_api.context.with(spanCtx, fn);
+        }
+      };
     },
-    onConsumeError(_envelope, error) {
-      const span = import_api.trace.getActiveSpan();
+    onConsumeError(envelope, error) {
+      const span = activeSpans.get(envelope.eventId);
       if (span) {
         span.setStatus({ code: import_api.SpanStatusCode.ERROR, message: error.message });
         span.recordException(error);

package/dist/otel.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/otel.ts"],"sourcesContent":["import {\n  trace,\n  context,\n  propagation,\n  SpanKind,\n  SpanStatusCode,\n} from \"@opentelemetry/api\";\nimport type { KafkaInstrumentation } from \"./client/types\";\nimport type { EventEnvelope } from \"./client/message/envelope\";\n\n/**\n * Create a `KafkaInstrumentation` that automatically propagates\n * W3C Trace Context via Kafka headers.\n *\n * Requires `@opentelemetry/api` as a peer dependency.\n *\n * **Send path:** injects `traceparent` into message headers from the\n * active OpenTelemetry context.\n *\n * **Consume path:** extracts `traceparent` from message headers,\n * starts a `CONSUMER` span as a child of the extracted context,\n * and ends it when the handler completes.\n *\n * @example\n * ```ts\n * import { otelInstrumentation } from '@drarzter/kafka-client/otel';\n *\n * const kafka = new KafkaClient('my-app', 'my-group', brokers, {\n *   instrumentation: [otelInstrumentation()],\n * });\n * ```\n */\nexport function otelInstrumentation(): KafkaInstrumentation {\n  const tracer = trace.getTracer(\"@drarzter/kafka-client\");\n\n  return {\n    beforeSend(topic: string, headers: Record<string, string>) {\n      propagation.inject(context.active(), headers);\n    },\n\n    afterSend(_topic: string) {\n      // Span management for producers is left to the caller's OTel setup.\n      // We only inject context — creating producer spans here would be\n      // inaccurate since buildSendPayload runs synchronously per-message.\n    },\n\n    beforeConsume(envelope: EventEnvelope<any>) {\n      const parentCtx = propagation.extract(context.active(), envelope.headers);\n      const span = tracer.startSpan(\n        `kafka.consume ${envelope.topic}`,\n        {\n          kind: SpanKind.CONSUMER,\n          attributes: {\n            \"messaging.system\": \"kafka\",\n            \"messaging.destination.name\": envelope.topic,\n            \"messaging.message.id\": envelope.eventId,\n            \"messaging.kafka.partition\": envelope.partition,\n            \"messaging.kafka.offset\": envelope.offset,\n          },\n        },\n        parentCtx,\n      );\n      return () => span.end();\n    },\n\n    onConsumeError(_envelope: EventEnvelope<any>, error: Error) {\n      const span = trace.getActiveSpan();\n      if (span) {\n        span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });\n        span.recordException(error);\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,iBAMO;AA0BA,SAAS,sBAA4C;AAC1D,QAAM,SAAS,iBAAM,UAAU,wBAAwB;AAEvD,SAAO;AAAA,IACL,WAAW,OAAe,SAAiC;AACzD,6BAAY,OAAO,mBAAQ,OAAO,GAAG,OAAO;AAAA,IAC9C;AAAA,IAEA,UAAU,QAAgB;AAAA,IAI1B;AAAA,IAEA,cAAc,UAA8B;AAC1C,YAAM,YAAY,uBAAY,QAAQ,mBAAQ,OAAO,GAAG,SAAS,OAAO;AACxE,YAAM,OAAO,OAAO;AAAA,QAClB,iBAAiB,SAAS,KAAK;AAAA,QAC/B;AAAA,UACE,MAAM,oBAAS;AAAA,UACf,YAAY;AAAA,YACV,oBAAoB;AAAA,YACpB,8BAA8B,SAAS;AAAA,YACvC,wBAAwB,SAAS;AAAA,YACjC,6BAA6B,SAAS;AAAA,YACtC,0BAA0B,SAAS;AAAA,UACrC;AAAA,QACF;AAAA,QACA;AAAA,MACF;AACA,aAAO,MAAM,KAAK,IAAI;AAAA,IACxB;AAAA,IAEA,eAAe,WAA+B,OAAc;AAC1D,YAAM,OAAO,iBAAM,cAAc;AACjC,UAAI,MAAM;AACR,aAAK,UAAU,EAAE,MAAM,0BAAe,OAAO,SAAS,MAAM,QAAQ,CAAC;AACrE,aAAK,gBAAgB,KAAK;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AACF;","names":[]}
+{"version":3,"sources":["../src/otel.ts"],"sourcesContent":["import {\n  trace,\n  context,\n  propagation,\n  SpanKind,\n  SpanStatusCode,\n} from \"@opentelemetry/api\";\nimport type { BeforeConsumeResult, KafkaInstrumentation } from \"./client/types\";\nimport type { EventEnvelope } from \"./client/message/envelope\";\n\n/**\n * Create a `KafkaInstrumentation` that automatically propagates\n * W3C Trace Context via Kafka headers.\n *\n * Requires `@opentelemetry/api` as a peer dependency.\n *\n * **Send path:** injects `traceparent` into message headers from the\n * active OpenTelemetry context.\n *\n * **Consume path:** extracts `traceparent` from message headers,\n * starts a `CONSUMER` span as a child of the extracted context,\n * and ends it when the handler completes.\n *\n * @example\n * ```ts\n * import { otelInstrumentation } from '@drarzter/kafka-client/otel';\n *\n * const kafka = new KafkaClient('my-app', 'my-group', brokers, {\n *   instrumentation: [otelInstrumentation()],\n * });\n * ```\n */\nexport function otelInstrumentation(): KafkaInstrumentation {\n  const tracer = trace.getTracer(\"@drarzter/kafka-client\");\n  const activeSpans = new Map<string, ReturnType<typeof tracer.startSpan>>();\n\n  return {\n    beforeSend(_topic: string, headers: Record<string, string>) {\n      propagation.inject(context.active(), headers);\n    },\n\n    afterSend(_topic: string) {\n      // Span management for producers is left to the caller's OTel setup.\n      // We only inject context — creating producer spans here would be\n      // inaccurate since buildSendPayload runs synchronously per-message.\n    },\n\n    beforeConsume(envelope: EventEnvelope<any>): BeforeConsumeResult {\n      const parentCtx = propagation.extract(context.active(), envelope.headers);\n      const span = tracer.startSpan(\n        `kafka.consume ${envelope.topic}`,\n        {\n          kind: SpanKind.CONSUMER,\n          attributes: {\n            \"messaging.system\": \"kafka\",\n            \"messaging.destination.name\": envelope.topic,\n            \"messaging.message.id\": envelope.eventId,\n            \"messaging.kafka.partition\": envelope.partition,\n            \"messaging.kafka.offset\": envelope.offset,\n          },\n        },\n        parentCtx,\n      );\n      const spanCtx = trace.setSpan(parentCtx, span);\n      activeSpans.set(envelope.eventId, span);\n      return {\n        cleanup() {\n          span.end();\n          activeSpans.delete(envelope.eventId);\n        },\n        wrap(fn) {\n          return context.with(spanCtx, fn);\n        },\n      };\n    },\n\n    onConsumeError(envelope: EventEnvelope<any>, error: Error) {\n      const span = activeSpans.get(envelope.eventId);\n      if (span) {\n        span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });\n        span.recordException(error);\n      }\n    },\n  };\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,iBAMO;AA0BA,SAAS,sBAA4C;AAC1D,QAAM,SAAS,iBAAM,UAAU,wBAAwB;AACvD,QAAM,cAAc,oBAAI,IAAiD;AAEzE,SAAO;AAAA,IACL,WAAW,QAAgB,SAAiC;AAC1D,6BAAY,OAAO,mBAAQ,OAAO,GAAG,OAAO;AAAA,IAC9C;AAAA,IAEA,UAAU,QAAgB;AAAA,IAI1B;AAAA,IAEA,cAAc,UAAmD;AAC/D,YAAM,YAAY,uBAAY,QAAQ,mBAAQ,OAAO,GAAG,SAAS,OAAO;AACxE,YAAM,OAAO,OAAO;AAAA,QAClB,iBAAiB,SAAS,KAAK;AAAA,QAC/B;AAAA,UACE,MAAM,oBAAS;AAAA,UACf,YAAY;AAAA,YACV,oBAAoB;AAAA,YACpB,8BAA8B,SAAS;AAAA,YACvC,wBAAwB,SAAS;AAAA,YACjC,6BAA6B,SAAS;AAAA,YACtC,0BAA0B,SAAS;AAAA,UACrC;AAAA,QACF;AAAA,QACA;AAAA,MACF;AACA,YAAM,UAAU,iBAAM,QAAQ,WAAW,IAAI;AAC7C,kBAAY,IAAI,SAAS,SAAS,IAAI;AACtC,aAAO;AAAA,QACL,UAAU;AACR,eAAK,IAAI;AACT,sBAAY,OAAO,SAAS,OAAO;AAAA,QACrC;AAAA,QACA,KAAK,IAAI;AACP,iBAAO,mBAAQ,KAAK,SAAS,EAAE;AAAA,QACjC;AAAA,MACF;AAAA,IACF;AAAA,IAEA,eAAe,UAA8B,OAAc;AACzD,YAAM,OAAO,YAAY,IAAI,SAAS,OAAO;AAC7C,UAAI,MAAM;AACR,aAAK,UAAU,EAAE,MAAM,0BAAe,OAAO,SAAS,MAAM,QAAQ,CAAC;AACrE,aAAK,gBAAgB,KAAK;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/dist/otel.mjs

@@ -10,8 +10,9 @@ import {
 } from "@opentelemetry/api";
 function otelInstrumentation() {
   const tracer = trace.getTracer("@drarzter/kafka-client");
+  const activeSpans = /* @__PURE__ */ new Map();
   return {
-    beforeSend(topic, headers) {
+    beforeSend(_topic, headers) {
       propagation.inject(context.active(), headers);
     },
     afterSend(_topic) {
@@ -32,10 +33,20 @@ function otelInstrumentation() {
         },
         parentCtx
       );
-      return () => span.end();
+      const spanCtx = trace.setSpan(parentCtx, span);
+      activeSpans.set(envelope.eventId, span);
+      return {
+        cleanup() {
+          span.end();
+          activeSpans.delete(envelope.eventId);
+        },
+        wrap(fn) {
+          return context.with(spanCtx, fn);
+        }
+      };
     },
-    onConsumeError(_envelope, error) {
-      const span = trace.getActiveSpan();
+    onConsumeError(envelope, error) {
+      const span = activeSpans.get(envelope.eventId);
       if (span) {
         span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });
         span.recordException(error);

package/dist/otel.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/otel.ts"],"sourcesContent":["import {\n  trace,\n  context,\n  propagation,\n  SpanKind,\n  SpanStatusCode,\n} from \"@opentelemetry/api\";\nimport type { KafkaInstrumentation } from \"./client/types\";\nimport type { EventEnvelope } from \"./client/message/envelope\";\n\n/**\n * Create a `KafkaInstrumentation` that automatically propagates\n * W3C Trace Context via Kafka headers.\n *\n * Requires `@opentelemetry/api` as a peer dependency.\n *\n * **Send path:** injects `traceparent` into message headers from the\n * active OpenTelemetry context.\n *\n * **Consume path:** extracts `traceparent` from message headers,\n * starts a `CONSUMER` span as a child of the extracted context,\n * and ends it when the handler completes.\n *\n * @example\n * ```ts\n * import { otelInstrumentation } from '@drarzter/kafka-client/otel';\n *\n * const kafka = new KafkaClient('my-app', 'my-group', brokers, {\n *   instrumentation: [otelInstrumentation()],\n * });\n * ```\n */\nexport function otelInstrumentation(): KafkaInstrumentation {\n  const tracer = trace.getTracer(\"@drarzter/kafka-client\");\n\n  return {\n    beforeSend(topic: string, headers: Record<string, string>) {\n      propagation.inject(context.active(), headers);\n    },\n\n    afterSend(_topic: string) {\n      // Span management for producers is left to the caller's OTel setup.\n      // We only inject context — creating producer spans here would be\n      // inaccurate since buildSendPayload runs synchronously per-message.\n    },\n\n    beforeConsume(envelope: EventEnvelope<any>) {\n      const parentCtx = propagation.extract(context.active(), envelope.headers);\n      const span = tracer.startSpan(\n        `kafka.consume ${envelope.topic}`,\n        {\n          kind: SpanKind.CONSUMER,\n          attributes: {\n            \"messaging.system\": \"kafka\",\n            \"messaging.destination.name\": envelope.topic,\n            \"messaging.message.id\": envelope.eventId,\n            \"messaging.kafka.partition\": envelope.partition,\n            \"messaging.kafka.offset\": envelope.offset,\n          },\n        },\n        parentCtx,\n      );\n      return () => span.end();\n    },\n\n    onConsumeError(_envelope: EventEnvelope<any>, error: Error) {\n      const span = trace.getActiveSpan();\n      if (span) {\n        span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });\n        span.recordException(error);\n      }\n    },\n  };\n}\n"],"mappings":";;;AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AA0BA,SAAS,sBAA4C;AAC1D,QAAM,SAAS,MAAM,UAAU,wBAAwB;AAEvD,SAAO;AAAA,IACL,WAAW,OAAe,SAAiC;AACzD,kBAAY,OAAO,QAAQ,OAAO,GAAG,OAAO;AAAA,IAC9C;AAAA,IAEA,UAAU,QAAgB;AAAA,IAI1B;AAAA,IAEA,cAAc,UAA8B;AAC1C,YAAM,YAAY,YAAY,QAAQ,QAAQ,OAAO,GAAG,SAAS,OAAO;AACxE,YAAM,OAAO,OAAO;AAAA,QAClB,iBAAiB,SAAS,KAAK;AAAA,QAC/B;AAAA,UACE,MAAM,SAAS;AAAA,UACf,YAAY;AAAA,YACV,oBAAoB;AAAA,YACpB,8BAA8B,SAAS;AAAA,YACvC,wBAAwB,SAAS;AAAA,YACjC,6BAA6B,SAAS;AAAA,YACtC,0BAA0B,SAAS;AAAA,UACrC;AAAA,QACF;AAAA,QACA;AAAA,MACF;AACA,aAAO,MAAM,KAAK,IAAI;AAAA,IACxB;AAAA,IAEA,eAAe,WAA+B,OAAc;AAC1D,YAAM,OAAO,MAAM,cAAc;AACjC,UAAI,MAAM;AACR,aAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,MAAM,QAAQ,CAAC;AACrE,aAAK,gBAAgB,KAAK;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AACF;","names":[]}
+{"version":3,"sources":["../src/otel.ts"],"sourcesContent":["import {\n  trace,\n  context,\n  propagation,\n  SpanKind,\n  SpanStatusCode,\n} from \"@opentelemetry/api\";\nimport type { BeforeConsumeResult, KafkaInstrumentation } from \"./client/types\";\nimport type { EventEnvelope } from \"./client/message/envelope\";\n\n/**\n * Create a `KafkaInstrumentation` that automatically propagates\n * W3C Trace Context via Kafka headers.\n *\n * Requires `@opentelemetry/api` as a peer dependency.\n *\n * **Send path:** injects `traceparent` into message headers from the\n * active OpenTelemetry context.\n *\n * **Consume path:** extracts `traceparent` from message headers,\n * starts a `CONSUMER` span as a child of the extracted context,\n * and ends it when the handler completes.\n *\n * @example\n * ```ts\n * import { otelInstrumentation } from '@drarzter/kafka-client/otel';\n *\n * const kafka = new KafkaClient('my-app', 'my-group', brokers, {\n *   instrumentation: [otelInstrumentation()],\n * });\n * ```\n */\nexport function otelInstrumentation(): KafkaInstrumentation {\n  const tracer = trace.getTracer(\"@drarzter/kafka-client\");\n  const activeSpans = new Map<string, ReturnType<typeof tracer.startSpan>>();\n\n  return {\n    beforeSend(_topic: string, headers: Record<string, string>) {\n      propagation.inject(context.active(), headers);\n    },\n\n    afterSend(_topic: string) {\n      // Span management for producers is left to the caller's OTel setup.\n      // We only inject context — creating producer spans here would be\n      // inaccurate since buildSendPayload runs synchronously per-message.\n    },\n\n    beforeConsume(envelope: EventEnvelope<any>): BeforeConsumeResult {\n      const parentCtx = propagation.extract(context.active(), envelope.headers);\n      const span = tracer.startSpan(\n        `kafka.consume ${envelope.topic}`,\n        {\n          kind: SpanKind.CONSUMER,\n          attributes: {\n            \"messaging.system\": \"kafka\",\n            \"messaging.destination.name\": envelope.topic,\n            \"messaging.message.id\": envelope.eventId,\n            \"messaging.kafka.partition\": envelope.partition,\n            \"messaging.kafka.offset\": envelope.offset,\n          },\n        },\n        parentCtx,\n      );\n      const spanCtx = trace.setSpan(parentCtx, span);\n      activeSpans.set(envelope.eventId, span);\n      return {\n        cleanup() {\n          span.end();\n          activeSpans.delete(envelope.eventId);\n        },\n        wrap(fn) {\n          return context.with(spanCtx, fn);\n        },\n      };\n    },\n\n    onConsumeError(envelope: EventEnvelope<any>, error: Error) {\n      const span = activeSpans.get(envelope.eventId);\n      if (span) {\n        span.setStatus({ code: SpanStatusCode.ERROR, message: error.message });\n        span.recordException(error);\n      }\n    },\n  };\n}\n"],"mappings":";;;AAAA;AAAA,EACE;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,EACA;AAAA,OACK;AA0BA,SAAS,sBAA4C;AAC1D,QAAM,SAAS,MAAM,UAAU,wBAAwB;AACvD,QAAM,cAAc,oBAAI,IAAiD;AAEzE,SAAO;AAAA,IACL,WAAW,QAAgB,SAAiC;AAC1D,kBAAY,OAAO,QAAQ,OAAO,GAAG,OAAO;AAAA,IAC9C;AAAA,IAEA,UAAU,QAAgB;AAAA,IAI1B;AAAA,IAEA,cAAc,UAAmD;AAC/D,YAAM,YAAY,YAAY,QAAQ,QAAQ,OAAO,GAAG,SAAS,OAAO;AACxE,YAAM,OAAO,OAAO;AAAA,QAClB,iBAAiB,SAAS,KAAK;AAAA,QAC/B;AAAA,UACE,MAAM,SAAS;AAAA,UACf,YAAY;AAAA,YACV,oBAAoB;AAAA,YACpB,8BAA8B,SAAS;AAAA,YACvC,wBAAwB,SAAS;AAAA,YACjC,6BAA6B,SAAS;AAAA,YACtC,0BAA0B,SAAS;AAAA,UACrC;AAAA,QACF;AAAA,QACA;AAAA,MACF;AACA,YAAM,UAAU,MAAM,QAAQ,WAAW,IAAI;AAC7C,kBAAY,IAAI,SAAS,SAAS,IAAI;AACtC,aAAO;AAAA,QACL,UAAU;AACR,eAAK,IAAI;AACT,sBAAY,OAAO,SAAS,OAAO;AAAA,QACrC;AAAA,QACA,KAAK,IAAI;AACP,iBAAO,QAAQ,KAAK,SAAS,EAAE;AAAA,QACjC;AAAA,MACF;AAAA,IACF;AAAA,IAEA,eAAe,UAA8B,OAAc;AACzD,YAAM,OAAO,YAAY,IAAI,SAAS,OAAO;AAC7C,UAAI,MAAM;AACR,aAAK,UAAU,EAAE,MAAM,eAAe,OAAO,SAAS,MAAM,QAAQ,CAAC;AACrE,aAAK,gBAAgB,KAAK;AAAA,MAC5B;AAAA,IACF;AAAA,EACF;AACF;","names":[]}

package/dist/testing.d.mts

@@ -1,4 +1,4 @@
-import { T as TopicMapConstraint, I as IKafkaClient } from './envelope-LeO5e3ob.mjs';
+import { T as TopicMapConstraint, I as IKafkaClient } from './types-zFbQH_Cy.mjs';
 
 /**
  * Fully typed mock of `IKafkaClient<T>` where every method is a mock function.

package/dist/testing.d.ts

@@ -1,4 +1,4 @@
-import { T as TopicMapConstraint, I as IKafkaClient } from './envelope-LeO5e3ob.js';
+import { T as TopicMapConstraint, I as IKafkaClient } from './types-zFbQH_Cy.js';
 
 /**
  * Fully typed mock of `IKafkaClient<T>` where every method is a mock function.

package/dist/testing.js

@@ -76,7 +76,8 @@ function createMockKafkaClient(mockFactory) {
       stop: mock().mockResolvedValue(void 0)
     }),
     stopConsumer: resolved(void 0),
-    disconnect: resolved(void 0)
+    disconnect: resolved(void 0),
+    enableGracefulShutdown: mock()
   };
 }
 

package/dist/testing.js.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/testing.ts","../src/testing/mock-client.ts","../src/testing/test-container.ts"],"sourcesContent":["export * from \"./testing/index\";\n","import type { IKafkaClient, TopicMapConstraint } from \"../client/types\";\n\n/**\n * Fully typed mock of `IKafkaClient<T>` where every method is a mock function.\n * Compatible with Jest, Vitest, or any framework whose `fn()` returns\n * an object with `.mock`, `.mockResolvedValue`, etc.\n */\nexport type MockKafkaClient<T extends TopicMapConstraint<T>> = {\n  [K in keyof IKafkaClient<T>]: IKafkaClient<T>[K] & Record<string, any>;\n};\n\n/** Factory that creates a no-op mock function (e.g. `() => jest.fn()`). */\nexport type MockFactory = () => (...args: any[]) => any;\n\nfunction detectMockFactory(): MockFactory {\n  // Jest and Vitest inject their globals (`jest` / `vi`) as module-scope\n  // bindings, not as properties of `globalThis`. The only reliable way to\n  // detect them without a hard import is via `eval`, which evaluates in the\n  // current module scope where those bindings are available.\n  try {\n    if (eval(\"typeof jest === 'object' && typeof jest.fn === 'function'\")) {\n      return () => eval(\"jest.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  try {\n    if (eval(\"typeof vi === 'object' && typeof vi.fn === 'function'\")) {\n      return () => eval(\"vi.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  throw new Error(\n    \"createMockKafkaClient: no mock framework detected (jest/vitest). \" +\n      \"Pass a custom mockFactory.\",\n  );\n}\n\n/**\n * Create a fully typed mock implementing every `IKafkaClient<T>` method.\n * Useful for unit-testing services that depend on `KafkaClient` without\n * touching a real broker.\n *\n * Auto-detects Jest (`jest.fn()`) or Vitest (`vi.fn()`). Pass a custom\n * `mockFactory` for other frameworks.\n *\n * All methods resolve to sensible defaults:\n * - `checkStatus()` → `{ status: 'up', clientId: 'mock-client', topics: [] }`\n * - `getClientId()` → `\"mock-client\"`\n * - void methods → `undefined`\n *\n * @example\n * ```ts\n * const kafka = createMockKafkaClient<MyTopics>();\n *\n * const service = new OrdersService(kafka);\n * await service.createOrder();\n *\n * expect(kafka.sendMessage).toHaveBeenCalledWith(\n *   'order.created',\n *   expect.objectContaining({ orderId: '123' }),\n * );\n * ```\n */\nexport function createMockKafkaClient<T extends TopicMapConstraint<T>>(\n  mockFactory?: MockFactory,\n): MockKafkaClient<T> {\n  const fn = mockFactory ?? detectMockFactory();\n\n  const mock = () => fn() as any;\n  const resolved = (value: unknown) => mock().mockResolvedValue(value);\n  const returning = (value: unknown) => mock().mockReturnValue(value);\n\n  return {\n    checkStatus: resolved({\n      status: \"up\",\n      clientId: \"mock-client\",\n      topics: [],\n    }),\n    getConsumerLag: resolved([]),\n    getClientId: returning(\"mock-client\"),\n    sendMessage: resolved(undefined),\n    sendBatch: resolved(undefined),\n    transaction: mock().mockImplementation(\n      async (cb: (ctx: Record<string, unknown>) => Promise<void>) => {\n        const ctx = {\n          send: resolved(undefined),\n          sendBatch: resolved(undefined),\n        };\n        await cb(ctx);\n      },\n    ),\n    startConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    startBatchConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    stopConsumer: resolved(undefined),\n    disconnect: resolved(undefined),\n  } as unknown as MockKafkaClient<T>;\n}\n","import {\n  KafkaContainer,\n  type StartedKafkaContainer,\n} from \"@testcontainers/kafka\";\nimport { KafkaJS } from \"@confluentinc/kafka-javascript\";\nconst { Kafka, logLevel: KafkaLogLevel } = KafkaJS;\n\n/** Options for `KafkaTestContainer`. */\nexport interface KafkaTestContainerOptions {\n  /** Docker image. Default: `\"confluentinc/cp-kafka:7.7.0\"`. */\n  image?: string;\n  /** Warm up the transactional coordinator on start. Default: `true`. */\n  transactionWarmup?: boolean;\n  /** Topics to pre-create. Each entry can be a string (1 partition) or `{ topic, numPartitions }`. */\n  topics?: Array<string | { topic: string; numPartitions?: number }>;\n}\n\n/**\n * Thin wrapper around `@testcontainers/kafka` that starts a single-node\n * KRaft Kafka container and exposes `brokers` for use with `KafkaClient`.\n *\n * Handles common setup pain points:\n * - Transaction coordinator warmup (avoids transactional producer hangs)\n * - Topic pre-creation (avoids race conditions)\n *\n * @example\n * ```ts\n * const container = new KafkaTestContainer({ topics: ['orders', 'payments'] });\n * const brokers = await container.start();\n *\n * const kafka = new KafkaClient('test', 'test-group', brokers);\n * // ... run tests ...\n *\n * await container.stop();\n * ```\n *\n * @example Jest lifecycle\n * ```ts\n * let container: KafkaTestContainer;\n * let brokers: string[];\n *\n * beforeAll(async () => {\n *   container = new KafkaTestContainer({ topics: ['orders'] });\n *   brokers = await container.start();\n * }, 120_000);\n *\n * afterAll(() => container.stop());\n * ```\n */\nexport class KafkaTestContainer {\n  private container: StartedKafkaContainer | undefined;\n  private readonly image: string;\n  private readonly transactionWarmup: boolean;\n  private readonly topics: Array<\n    string | { topic: string; numPartitions?: number }\n  >;\n\n  constructor(options?: KafkaTestContainerOptions) {\n    this.image = options?.image ?? \"confluentinc/cp-kafka:7.7.0\";\n    this.transactionWarmup = options?.transactionWarmup ?? true;\n    this.topics = options?.topics ?? [];\n  }\n\n  /**\n   * Start the Kafka container, pre-create topics, and optionally warm up\n   * the transaction coordinator.\n   *\n   * @returns Broker connection strings, e.g. `[\"localhost:55123\"]`.\n   */\n  async start(): Promise<string[]> {\n    this.container = await new KafkaContainer(this.image)\n      .withKraft()\n      .withExposedPorts(9093)\n      .withEnvironment({\n        KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: \"1\",\n        KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: \"1\",\n      })\n      .start();\n\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    const brokers = [`${host}:${port}`];\n\n    const kafka = new Kafka({\n      kafkaJS: {\n        clientId: \"test-container-setup\",\n        brokers,\n        logLevel: KafkaLogLevel.NOTHING,\n      },\n    });\n\n    if (this.topics.length > 0) {\n      const admin = kafka.admin();\n      await admin.connect();\n      await admin.createTopics({\n        topics: this.topics.map((t) =>\n          typeof t === \"string\"\n            ? { topic: t, numPartitions: 1 }\n            : { topic: t.topic, numPartitions: t.numPartitions ?? 1 },\n        ),\n      });\n      await admin.disconnect();\n    }\n\n    if (this.transactionWarmup) {\n      const warmupKafka = new Kafka({\n        kafkaJS: {\n          clientId: \"test-container-warmup\",\n          brokers,\n          logLevel: KafkaLogLevel.NOTHING,\n        },\n      });\n      const txProducer = warmupKafka.producer({\n        kafkaJS: {\n          transactionalId: \"test-container-warmup-tx\",\n          idempotent: true,\n          maxInFlightRequests: 1,\n        },\n      });\n      await txProducer.connect();\n      const tx = await txProducer.transaction();\n      await tx.abort();\n      await txProducer.disconnect();\n    }\n\n    return brokers;\n  }\n\n  /** Stop and remove the container. */\n  async stop(): Promise<void> {\n    await this.container?.stop();\n    this.container = undefined;\n  }\n\n  /** Broker connection strings. Throws if container is not started. */\n  get brokers(): string[] {\n    if (!this.container) {\n      throw new Error(\"KafkaTestContainer is not started. Call start() first.\");\n    }\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    return [`${host}:${port}`];\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACcA,SAAS,oBAAiC;AAKxC,MAAI;AACF,QAAI,KAAK,2DAA2D,GAAG;AACrE,aAAO,MAAM,KAAK,WAAW;AAAA,IAC/B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,MAAI;AACF,QAAI,KAAK,uDAAuD,GAAG;AACjE,aAAO,MAAM,KAAK,SAAS;AAAA,IAC7B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,QAAM,IAAI;AAAA,IACR;AAAA,EAEF;AACF;AA4BO,SAAS,sBACd,aACoB;AACpB,QAAM,KAAK,eAAe,kBAAkB;AAE5C,QAAM,OAAO,MAAM,GAAG;AACtB,QAAM,WAAW,CAAC,UAAmB,KAAK,EAAE,kBAAkB,KAAK;AACnE,QAAM,YAAY,CAAC,UAAmB,KAAK,EAAE,gBAAgB,KAAK;AAElE,SAAO;AAAA,IACL,aAAa,SAAS;AAAA,MACpB,QAAQ;AAAA,MACR,UAAU;AAAA,MACV,QAAQ,CAAC;AAAA,IACX,CAAC;AAAA,IACD,gBAAgB,SAAS,CAAC,CAAC;AAAA,IAC3B,aAAa,UAAU,aAAa;AAAA,IACpC,aAAa,SAAS,MAAS;AAAA,IAC/B,WAAW,SAAS,MAAS;AAAA,IAC7B,aAAa,KAAK,EAAE;AAAA,MAClB,OAAO,OAAwD;AAC7D,cAAM,MAAM;AAAA,UACV,MAAM,SAAS,MAAS;AAAA,UACxB,WAAW,SAAS,MAAS;AAAA,QAC/B;AACA,cAAM,GAAG,GAAG;AAAA,MACd;AAAA,IACF;AAAA,IACA,eAAe,SAAS;AAAA,MACtB,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,oBAAoB,SAAS;AAAA,MAC3B,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,cAAc,SAAS,MAAS;AAAA,IAChC,YAAY,SAAS,MAAS;AAAA,EAChC;AACF;;;ACxGA,mBAGO;AACP,8BAAwB;AACxB,IAAM,EAAE,OAAO,UAAU,cAAc,IAAI;AA4CpC,IAAM,qBAAN,MAAyB;AAAA,EACtB;AAAA,EACS;AAAA,EACA;AAAA,EACA;AAAA,EAIjB,YAAY,SAAqC;AAC/C,SAAK,QAAQ,SAAS,SAAS;AAC/B,SAAK,oBAAoB,SAAS,qBAAqB;AACvD,SAAK,SAAS,SAAS,UAAU,CAAC;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,QAA2B;AAC/B,SAAK,YAAY,MAAM,IAAI,4BAAe,KAAK,KAAK,EACjD,UAAU,EACV,iBAAiB,IAAI,EACrB,gBAAgB;AAAA,MACf,gDAAgD;AAAA,MAChD,qCAAqC;AAAA,IACvC,CAAC,EACA,MAAM;AAET,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,UAAM,UAAU,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAElC,UAAM,QAAQ,IAAI,MAAM;AAAA,MACtB,SAAS;AAAA,QACP,UAAU;AAAA,QACV;AAAA,QACA,UAAU,cAAc;AAAA,MAC1B;AAAA,IACF,CAAC;AAED,QAAI,KAAK,OAAO,SAAS,GAAG;AAC1B,YAAM,QAAQ,MAAM,MAAM;AAC1B,YAAM,MAAM,QAAQ;AACpB,YAAM,MAAM,aAAa;AAAA,QACvB,QAAQ,KAAK,OAAO;AAAA,UAAI,CAAC,MACvB,OAAO,MAAM,WACT,EAAE,OAAO,GAAG,eAAe,EAAE,IAC7B,EAAE,OAAO,EAAE,OAAO,eAAe,EAAE,iBAAiB,EAAE;AAAA,QAC5D;AAAA,MACF,CAAC;AACD,YAAM,MAAM,WAAW;AAAA,IACzB;AAEA,QAAI,KAAK,mBAAmB;AAC1B,YAAM,cAAc,IAAI,MAAM;AAAA,QAC5B,SAAS;AAAA,UACP,UAAU;AAAA,UACV;AAAA,UACA,UAAU,cAAc;AAAA,QAC1B;AAAA,MACF,CAAC;AACD,YAAM,aAAa,YAAY,SAAS;AAAA,QACtC,SAAS;AAAA,UACP,iBAAiB;AAAA,UACjB,YAAY;AAAA,UACZ,qBAAqB;AAAA,QACvB;AAAA,MACF,CAAC;AACD,YAAM,WAAW,QAAQ;AACzB,YAAM,KAAK,MAAM,WAAW,YAAY;AACxC,YAAM,GAAG,MAAM;AACf,YAAM,WAAW,WAAW;AAAA,IAC9B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,MAAM,OAAsB;AAC1B,UAAM,KAAK,WAAW,KAAK;AAC3B,SAAK,YAAY;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,UAAoB;AACtB,QAAI,CAAC,KAAK,WAAW;AACnB,YAAM,IAAI,MAAM,wDAAwD;AAAA,IAC1E;AACA,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,WAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAAA,EAC3B;AACF;","names":[]}
+{"version":3,"sources":["../src/testing.ts","../src/testing/mock-client.ts","../src/testing/test-container.ts"],"sourcesContent":["export * from \"./testing/index\";\n","import type { IKafkaClient, TopicMapConstraint } from \"../client/types\";\n\n/**\n * Fully typed mock of `IKafkaClient<T>` where every method is a mock function.\n * Compatible with Jest, Vitest, or any framework whose `fn()` returns\n * an object with `.mock`, `.mockResolvedValue`, etc.\n */\nexport type MockKafkaClient<T extends TopicMapConstraint<T>> = {\n  [K in keyof IKafkaClient<T>]: IKafkaClient<T>[K] & Record<string, any>;\n};\n\n/** Factory that creates a no-op mock function (e.g. `() => jest.fn()`). */\nexport type MockFactory = () => (...args: any[]) => any;\n\nfunction detectMockFactory(): MockFactory {\n  // Jest and Vitest inject their globals (`jest` / `vi`) as module-scope\n  // bindings, not as properties of `globalThis`. The only reliable way to\n  // detect them without a hard import is via `eval`, which evaluates in the\n  // current module scope where those bindings are available.\n  try {\n    if (eval(\"typeof jest === 'object' && typeof jest.fn === 'function'\")) {\n      return () => eval(\"jest.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  try {\n    if (eval(\"typeof vi === 'object' && typeof vi.fn === 'function'\")) {\n      return () => eval(\"vi.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  throw new Error(\n    \"createMockKafkaClient: no mock framework detected (jest/vitest). \" +\n      \"Pass a custom mockFactory.\",\n  );\n}\n\n/**\n * Create a fully typed mock implementing every `IKafkaClient<T>` method.\n * Useful for unit-testing services that depend on `KafkaClient` without\n * touching a real broker.\n *\n * Auto-detects Jest (`jest.fn()`) or Vitest (`vi.fn()`). Pass a custom\n * `mockFactory` for other frameworks.\n *\n * All methods resolve to sensible defaults:\n * - `checkStatus()` → `{ status: 'up', clientId: 'mock-client', topics: [] }`\n * - `getClientId()` → `\"mock-client\"`\n * - void methods → `undefined`\n *\n * @example\n * ```ts\n * const kafka = createMockKafkaClient<MyTopics>();\n *\n * const service = new OrdersService(kafka);\n * await service.createOrder();\n *\n * expect(kafka.sendMessage).toHaveBeenCalledWith(\n *   'order.created',\n *   expect.objectContaining({ orderId: '123' }),\n * );\n * ```\n */\nexport function createMockKafkaClient<T extends TopicMapConstraint<T>>(\n  mockFactory?: MockFactory,\n): MockKafkaClient<T> {\n  const fn = mockFactory ?? detectMockFactory();\n\n  const mock = () => fn() as any;\n  const resolved = (value: unknown) => mock().mockResolvedValue(value);\n  const returning = (value: unknown) => mock().mockReturnValue(value);\n\n  return {\n    checkStatus: resolved({\n      status: \"up\",\n      clientId: \"mock-client\",\n      topics: [],\n    }),\n    getConsumerLag: resolved([]),\n    getClientId: returning(\"mock-client\"),\n    sendMessage: resolved(undefined),\n    sendBatch: resolved(undefined),\n    transaction: mock().mockImplementation(\n      async (cb: (ctx: Record<string, unknown>) => Promise<void>) => {\n        const ctx = {\n          send: resolved(undefined),\n          sendBatch: resolved(undefined),\n        };\n        await cb(ctx);\n      },\n    ),\n    startConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    startBatchConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    stopConsumer: resolved(undefined),\n    disconnect: resolved(undefined),\n    enableGracefulShutdown: mock(),\n  } as unknown as MockKafkaClient<T>;\n}\n","import {\n  KafkaContainer,\n  type StartedKafkaContainer,\n} from \"@testcontainers/kafka\";\nimport { KafkaJS } from \"@confluentinc/kafka-javascript\";\nconst { Kafka, logLevel: KafkaLogLevel } = KafkaJS;\n\n/** Options for `KafkaTestContainer`. */\nexport interface KafkaTestContainerOptions {\n  /** Docker image. Default: `\"confluentinc/cp-kafka:7.7.0\"`. */\n  image?: string;\n  /** Warm up the transactional coordinator on start. Default: `true`. */\n  transactionWarmup?: boolean;\n  /** Topics to pre-create. Each entry can be a string (1 partition) or `{ topic, numPartitions }`. */\n  topics?: Array<string | { topic: string; numPartitions?: number }>;\n}\n\n/**\n * Thin wrapper around `@testcontainers/kafka` that starts a single-node\n * KRaft Kafka container and exposes `brokers` for use with `KafkaClient`.\n *\n * Handles common setup pain points:\n * - Transaction coordinator warmup (avoids transactional producer hangs)\n * - Topic pre-creation (avoids race conditions)\n *\n * @example\n * ```ts\n * const container = new KafkaTestContainer({ topics: ['orders', 'payments'] });\n * const brokers = await container.start();\n *\n * const kafka = new KafkaClient('test', 'test-group', brokers);\n * // ... run tests ...\n *\n * await container.stop();\n * ```\n *\n * @example Jest lifecycle\n * ```ts\n * let container: KafkaTestContainer;\n * let brokers: string[];\n *\n * beforeAll(async () => {\n *   container = new KafkaTestContainer({ topics: ['orders'] });\n *   brokers = await container.start();\n * }, 120_000);\n *\n * afterAll(() => container.stop());\n * ```\n */\nexport class KafkaTestContainer {\n  private container: StartedKafkaContainer | undefined;\n  private readonly image: string;\n  private readonly transactionWarmup: boolean;\n  private readonly topics: Array<\n    string | { topic: string; numPartitions?: number }\n  >;\n\n  constructor(options?: KafkaTestContainerOptions) {\n    this.image = options?.image ?? \"confluentinc/cp-kafka:7.7.0\";\n    this.transactionWarmup = options?.transactionWarmup ?? true;\n    this.topics = options?.topics ?? [];\n  }\n\n  /**\n   * Start the Kafka container, pre-create topics, and optionally warm up\n   * the transaction coordinator.\n   *\n   * @returns Broker connection strings, e.g. `[\"localhost:55123\"]`.\n   */\n  async start(): Promise<string[]> {\n    this.container = await new KafkaContainer(this.image)\n      .withKraft()\n      .withExposedPorts(9093)\n      .withEnvironment({\n        KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: \"1\",\n        KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: \"1\",\n      })\n      .start();\n\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    const brokers = [`${host}:${port}`];\n\n    const kafka = new Kafka({\n      kafkaJS: {\n        clientId: \"test-container-setup\",\n        brokers,\n        logLevel: KafkaLogLevel.NOTHING,\n      },\n    });\n\n    if (this.topics.length > 0) {\n      const admin = kafka.admin();\n      await admin.connect();\n      await admin.createTopics({\n        topics: this.topics.map((t) =>\n          typeof t === \"string\"\n            ? { topic: t, numPartitions: 1 }\n            : { topic: t.topic, numPartitions: t.numPartitions ?? 1 },\n        ),\n      });\n      await admin.disconnect();\n    }\n\n    if (this.transactionWarmup) {\n      const warmupKafka = new Kafka({\n        kafkaJS: {\n          clientId: \"test-container-warmup\",\n          brokers,\n          logLevel: KafkaLogLevel.NOTHING,\n        },\n      });\n      const txProducer = warmupKafka.producer({\n        kafkaJS: {\n          transactionalId: \"test-container-warmup-tx\",\n          idempotent: true,\n          maxInFlightRequests: 1,\n        },\n      });\n      await txProducer.connect();\n      const tx = await txProducer.transaction();\n      await tx.abort();\n      await txProducer.disconnect();\n    }\n\n    return brokers;\n  }\n\n  /** Stop and remove the container. */\n  async stop(): Promise<void> {\n    await this.container?.stop();\n    this.container = undefined;\n  }\n\n  /** Broker connection strings. Throws if container is not started. */\n  get brokers(): string[] {\n    if (!this.container) {\n      throw new Error(\"KafkaTestContainer is not started. Call start() first.\");\n    }\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    return [`${host}:${port}`];\n  }\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;;ACcA,SAAS,oBAAiC;AAKxC,MAAI;AACF,QAAI,KAAK,2DAA2D,GAAG;AACrE,aAAO,MAAM,KAAK,WAAW;AAAA,IAC/B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,MAAI;AACF,QAAI,KAAK,uDAAuD,GAAG;AACjE,aAAO,MAAM,KAAK,SAAS;AAAA,IAC7B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,QAAM,IAAI;AAAA,IACR;AAAA,EAEF;AACF;AA4BO,SAAS,sBACd,aACoB;AACpB,QAAM,KAAK,eAAe,kBAAkB;AAE5C,QAAM,OAAO,MAAM,GAAG;AACtB,QAAM,WAAW,CAAC,UAAmB,KAAK,EAAE,kBAAkB,KAAK;AACnE,QAAM,YAAY,CAAC,UAAmB,KAAK,EAAE,gBAAgB,KAAK;AAElE,SAAO;AAAA,IACL,aAAa,SAAS;AAAA,MACpB,QAAQ;AAAA,MACR,UAAU;AAAA,MACV,QAAQ,CAAC;AAAA,IACX,CAAC;AAAA,IACD,gBAAgB,SAAS,CAAC,CAAC;AAAA,IAC3B,aAAa,UAAU,aAAa;AAAA,IACpC,aAAa,SAAS,MAAS;AAAA,IAC/B,WAAW,SAAS,MAAS;AAAA,IAC7B,aAAa,KAAK,EAAE;AAAA,MAClB,OAAO,OAAwD;AAC7D,cAAM,MAAM;AAAA,UACV,MAAM,SAAS,MAAS;AAAA,UACxB,WAAW,SAAS,MAAS;AAAA,QAC/B;AACA,cAAM,GAAG,GAAG;AAAA,MACd;AAAA,IACF;AAAA,IACA,eAAe,SAAS;AAAA,MACtB,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,oBAAoB,SAAS;AAAA,MAC3B,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,cAAc,SAAS,MAAS;AAAA,IAChC,YAAY,SAAS,MAAS;AAAA,IAC9B,wBAAwB,KAAK;AAAA,EAC/B;AACF;;;ACzGA,mBAGO;AACP,8BAAwB;AACxB,IAAM,EAAE,OAAO,UAAU,cAAc,IAAI;AA4CpC,IAAM,qBAAN,MAAyB;AAAA,EACtB;AAAA,EACS;AAAA,EACA;AAAA,EACA;AAAA,EAIjB,YAAY,SAAqC;AAC/C,SAAK,QAAQ,SAAS,SAAS;AAC/B,SAAK,oBAAoB,SAAS,qBAAqB;AACvD,SAAK,SAAS,SAAS,UAAU,CAAC;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,QAA2B;AAC/B,SAAK,YAAY,MAAM,IAAI,4BAAe,KAAK,KAAK,EACjD,UAAU,EACV,iBAAiB,IAAI,EACrB,gBAAgB;AAAA,MACf,gDAAgD;AAAA,MAChD,qCAAqC;AAAA,IACvC,CAAC,EACA,MAAM;AAET,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,UAAM,UAAU,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAElC,UAAM,QAAQ,IAAI,MAAM;AAAA,MACtB,SAAS;AAAA,QACP,UAAU;AAAA,QACV;AAAA,QACA,UAAU,cAAc;AAAA,MAC1B;AAAA,IACF,CAAC;AAED,QAAI,KAAK,OAAO,SAAS,GAAG;AAC1B,YAAM,QAAQ,MAAM,MAAM;AAC1B,YAAM,MAAM,QAAQ;AACpB,YAAM,MAAM,aAAa;AAAA,QACvB,QAAQ,KAAK,OAAO;AAAA,UAAI,CAAC,MACvB,OAAO,MAAM,WACT,EAAE,OAAO,GAAG,eAAe,EAAE,IAC7B,EAAE,OAAO,EAAE,OAAO,eAAe,EAAE,iBAAiB,EAAE;AAAA,QAC5D;AAAA,MACF,CAAC;AACD,YAAM,MAAM,WAAW;AAAA,IACzB;AAEA,QAAI,KAAK,mBAAmB;AAC1B,YAAM,cAAc,IAAI,MAAM;AAAA,QAC5B,SAAS;AAAA,UACP,UAAU;AAAA,UACV;AAAA,UACA,UAAU,cAAc;AAAA,QAC1B;AAAA,MACF,CAAC;AACD,YAAM,aAAa,YAAY,SAAS;AAAA,QACtC,SAAS;AAAA,UACP,iBAAiB;AAAA,UACjB,YAAY;AAAA,UACZ,qBAAqB;AAAA,QACvB;AAAA,MACF,CAAC;AACD,YAAM,WAAW,QAAQ;AACzB,YAAM,KAAK,MAAM,WAAW,YAAY;AACxC,YAAM,GAAG,MAAM;AACf,YAAM,WAAW,WAAW;AAAA,IAC9B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,MAAM,OAAsB;AAC1B,UAAM,KAAK,WAAW,KAAK;AAC3B,SAAK,YAAY;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,UAAoB;AACtB,QAAI,CAAC,KAAK,WAAW;AACnB,YAAM,IAAI,MAAM,wDAAwD;AAAA,IAC1E;AACA,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,WAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAAA,EAC3B;AACF;","names":[]}

package/dist/testing.mjs

@@ -51,7 +51,8 @@ function createMockKafkaClient(mockFactory) {
       stop: mock().mockResolvedValue(void 0)
     }),
     stopConsumer: resolved(void 0),
-    disconnect: resolved(void 0)
+    disconnect: resolved(void 0),
+    enableGracefulShutdown: mock()
   };
 }
 

package/dist/testing.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/testing/mock-client.ts","../src/testing/test-container.ts"],"sourcesContent":["import type { IKafkaClient, TopicMapConstraint } from \"../client/types\";\n\n/**\n * Fully typed mock of `IKafkaClient<T>` where every method is a mock function.\n * Compatible with Jest, Vitest, or any framework whose `fn()` returns\n * an object with `.mock`, `.mockResolvedValue`, etc.\n */\nexport type MockKafkaClient<T extends TopicMapConstraint<T>> = {\n  [K in keyof IKafkaClient<T>]: IKafkaClient<T>[K] & Record<string, any>;\n};\n\n/** Factory that creates a no-op mock function (e.g. `() => jest.fn()`). */\nexport type MockFactory = () => (...args: any[]) => any;\n\nfunction detectMockFactory(): MockFactory {\n  // Jest and Vitest inject their globals (`jest` / `vi`) as module-scope\n  // bindings, not as properties of `globalThis`. The only reliable way to\n  // detect them without a hard import is via `eval`, which evaluates in the\n  // current module scope where those bindings are available.\n  try {\n    if (eval(\"typeof jest === 'object' && typeof jest.fn === 'function'\")) {\n      return () => eval(\"jest.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  try {\n    if (eval(\"typeof vi === 'object' && typeof vi.fn === 'function'\")) {\n      return () => eval(\"vi.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  throw new Error(\n    \"createMockKafkaClient: no mock framework detected (jest/vitest). \" +\n      \"Pass a custom mockFactory.\",\n  );\n}\n\n/**\n * Create a fully typed mock implementing every `IKafkaClient<T>` method.\n * Useful for unit-testing services that depend on `KafkaClient` without\n * touching a real broker.\n *\n * Auto-detects Jest (`jest.fn()`) or Vitest (`vi.fn()`). Pass a custom\n * `mockFactory` for other frameworks.\n *\n * All methods resolve to sensible defaults:\n * - `checkStatus()` → `{ status: 'up', clientId: 'mock-client', topics: [] }`\n * - `getClientId()` → `\"mock-client\"`\n * - void methods → `undefined`\n *\n * @example\n * ```ts\n * const kafka = createMockKafkaClient<MyTopics>();\n *\n * const service = new OrdersService(kafka);\n * await service.createOrder();\n *\n * expect(kafka.sendMessage).toHaveBeenCalledWith(\n *   'order.created',\n *   expect.objectContaining({ orderId: '123' }),\n * );\n * ```\n */\nexport function createMockKafkaClient<T extends TopicMapConstraint<T>>(\n  mockFactory?: MockFactory,\n): MockKafkaClient<T> {\n  const fn = mockFactory ?? detectMockFactory();\n\n  const mock = () => fn() as any;\n  const resolved = (value: unknown) => mock().mockResolvedValue(value);\n  const returning = (value: unknown) => mock().mockReturnValue(value);\n\n  return {\n    checkStatus: resolved({\n      status: \"up\",\n      clientId: \"mock-client\",\n      topics: [],\n    }),\n    getConsumerLag: resolved([]),\n    getClientId: returning(\"mock-client\"),\n    sendMessage: resolved(undefined),\n    sendBatch: resolved(undefined),\n    transaction: mock().mockImplementation(\n      async (cb: (ctx: Record<string, unknown>) => Promise<void>) => {\n        const ctx = {\n          send: resolved(undefined),\n          sendBatch: resolved(undefined),\n        };\n        await cb(ctx);\n      },\n    ),\n    startConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    startBatchConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    stopConsumer: resolved(undefined),\n    disconnect: resolved(undefined),\n  } as unknown as MockKafkaClient<T>;\n}\n","import {\n  KafkaContainer,\n  type StartedKafkaContainer,\n} from \"@testcontainers/kafka\";\nimport { KafkaJS } from \"@confluentinc/kafka-javascript\";\nconst { Kafka, logLevel: KafkaLogLevel } = KafkaJS;\n\n/** Options for `KafkaTestContainer`. */\nexport interface KafkaTestContainerOptions {\n  /** Docker image. Default: `\"confluentinc/cp-kafka:7.7.0\"`. */\n  image?: string;\n  /** Warm up the transactional coordinator on start. Default: `true`. */\n  transactionWarmup?: boolean;\n  /** Topics to pre-create. Each entry can be a string (1 partition) or `{ topic, numPartitions }`. */\n  topics?: Array<string | { topic: string; numPartitions?: number }>;\n}\n\n/**\n * Thin wrapper around `@testcontainers/kafka` that starts a single-node\n * KRaft Kafka container and exposes `brokers` for use with `KafkaClient`.\n *\n * Handles common setup pain points:\n * - Transaction coordinator warmup (avoids transactional producer hangs)\n * - Topic pre-creation (avoids race conditions)\n *\n * @example\n * ```ts\n * const container = new KafkaTestContainer({ topics: ['orders', 'payments'] });\n * const brokers = await container.start();\n *\n * const kafka = new KafkaClient('test', 'test-group', brokers);\n * // ... run tests ...\n *\n * await container.stop();\n * ```\n *\n * @example Jest lifecycle\n * ```ts\n * let container: KafkaTestContainer;\n * let brokers: string[];\n *\n * beforeAll(async () => {\n *   container = new KafkaTestContainer({ topics: ['orders'] });\n *   brokers = await container.start();\n * }, 120_000);\n *\n * afterAll(() => container.stop());\n * ```\n */\nexport class KafkaTestContainer {\n  private container: StartedKafkaContainer | undefined;\n  private readonly image: string;\n  private readonly transactionWarmup: boolean;\n  private readonly topics: Array<\n    string | { topic: string; numPartitions?: number }\n  >;\n\n  constructor(options?: KafkaTestContainerOptions) {\n    this.image = options?.image ?? \"confluentinc/cp-kafka:7.7.0\";\n    this.transactionWarmup = options?.transactionWarmup ?? true;\n    this.topics = options?.topics ?? [];\n  }\n\n  /**\n   * Start the Kafka container, pre-create topics, and optionally warm up\n   * the transaction coordinator.\n   *\n   * @returns Broker connection strings, e.g. `[\"localhost:55123\"]`.\n   */\n  async start(): Promise<string[]> {\n    this.container = await new KafkaContainer(this.image)\n      .withKraft()\n      .withExposedPorts(9093)\n      .withEnvironment({\n        KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: \"1\",\n        KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: \"1\",\n      })\n      .start();\n\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    const brokers = [`${host}:${port}`];\n\n    const kafka = new Kafka({\n      kafkaJS: {\n        clientId: \"test-container-setup\",\n        brokers,\n        logLevel: KafkaLogLevel.NOTHING,\n      },\n    });\n\n    if (this.topics.length > 0) {\n      const admin = kafka.admin();\n      await admin.connect();\n      await admin.createTopics({\n        topics: this.topics.map((t) =>\n          typeof t === \"string\"\n            ? { topic: t, numPartitions: 1 }\n            : { topic: t.topic, numPartitions: t.numPartitions ?? 1 },\n        ),\n      });\n      await admin.disconnect();\n    }\n\n    if (this.transactionWarmup) {\n      const warmupKafka = new Kafka({\n        kafkaJS: {\n          clientId: \"test-container-warmup\",\n          brokers,\n          logLevel: KafkaLogLevel.NOTHING,\n        },\n      });\n      const txProducer = warmupKafka.producer({\n        kafkaJS: {\n          transactionalId: \"test-container-warmup-tx\",\n          idempotent: true,\n          maxInFlightRequests: 1,\n        },\n      });\n      await txProducer.connect();\n      const tx = await txProducer.transaction();\n      await tx.abort();\n      await txProducer.disconnect();\n    }\n\n    return brokers;\n  }\n\n  /** Stop and remove the container. */\n  async stop(): Promise<void> {\n    await this.container?.stop();\n    this.container = undefined;\n  }\n\n  /** Broker connection strings. Throws if container is not started. */\n  get brokers(): string[] {\n    if (!this.container) {\n      throw new Error(\"KafkaTestContainer is not started. Call start() first.\");\n    }\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    return [`${host}:${port}`];\n  }\n}\n"],"mappings":";;;AAcA,SAAS,oBAAiC;AAKxC,MAAI;AACF,QAAI,KAAK,2DAA2D,GAAG;AACrE,aAAO,MAAM,KAAK,WAAW;AAAA,IAC/B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,MAAI;AACF,QAAI,KAAK,uDAAuD,GAAG;AACjE,aAAO,MAAM,KAAK,SAAS;AAAA,IAC7B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,QAAM,IAAI;AAAA,IACR;AAAA,EAEF;AACF;AA4BO,SAAS,sBACd,aACoB;AACpB,QAAM,KAAK,eAAe,kBAAkB;AAE5C,QAAM,OAAO,MAAM,GAAG;AACtB,QAAM,WAAW,CAAC,UAAmB,KAAK,EAAE,kBAAkB,KAAK;AACnE,QAAM,YAAY,CAAC,UAAmB,KAAK,EAAE,gBAAgB,KAAK;AAElE,SAAO;AAAA,IACL,aAAa,SAAS;AAAA,MACpB,QAAQ;AAAA,MACR,UAAU;AAAA,MACV,QAAQ,CAAC;AAAA,IACX,CAAC;AAAA,IACD,gBAAgB,SAAS,CAAC,CAAC;AAAA,IAC3B,aAAa,UAAU,aAAa;AAAA,IACpC,aAAa,SAAS,MAAS;AAAA,IAC/B,WAAW,SAAS,MAAS;AAAA,IAC7B,aAAa,KAAK,EAAE;AAAA,MAClB,OAAO,OAAwD;AAC7D,cAAM,MAAM;AAAA,UACV,MAAM,SAAS,MAAS;AAAA,UACxB,WAAW,SAAS,MAAS;AAAA,QAC/B;AACA,cAAM,GAAG,GAAG;AAAA,MACd;AAAA,IACF;AAAA,IACA,eAAe,SAAS;AAAA,MACtB,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,oBAAoB,SAAS;AAAA,MAC3B,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,cAAc,SAAS,MAAS;AAAA,IAChC,YAAY,SAAS,MAAS;AAAA,EAChC;AACF;;;ACxGA;AAAA,EACE;AAAA,OAEK;AACP,SAAS,eAAe;AACxB,IAAM,EAAE,OAAO,UAAU,cAAc,IAAI;AA4CpC,IAAM,qBAAN,MAAyB;AAAA,EACtB;AAAA,EACS;AAAA,EACA;AAAA,EACA;AAAA,EAIjB,YAAY,SAAqC;AAC/C,SAAK,QAAQ,SAAS,SAAS;AAC/B,SAAK,oBAAoB,SAAS,qBAAqB;AACvD,SAAK,SAAS,SAAS,UAAU,CAAC;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,QAA2B;AAC/B,SAAK,YAAY,MAAM,IAAI,eAAe,KAAK,KAAK,EACjD,UAAU,EACV,iBAAiB,IAAI,EACrB,gBAAgB;AAAA,MACf,gDAAgD;AAAA,MAChD,qCAAqC;AAAA,IACvC,CAAC,EACA,MAAM;AAET,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,UAAM,UAAU,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAElC,UAAM,QAAQ,IAAI,MAAM;AAAA,MACtB,SAAS;AAAA,QACP,UAAU;AAAA,QACV;AAAA,QACA,UAAU,cAAc;AAAA,MAC1B;AAAA,IACF,CAAC;AAED,QAAI,KAAK,OAAO,SAAS,GAAG;AAC1B,YAAM,QAAQ,MAAM,MAAM;AAC1B,YAAM,MAAM,QAAQ;AACpB,YAAM,MAAM,aAAa;AAAA,QACvB,QAAQ,KAAK,OAAO;AAAA,UAAI,CAAC,MACvB,OAAO,MAAM,WACT,EAAE,OAAO,GAAG,eAAe,EAAE,IAC7B,EAAE,OAAO,EAAE,OAAO,eAAe,EAAE,iBAAiB,EAAE;AAAA,QAC5D;AAAA,MACF,CAAC;AACD,YAAM,MAAM,WAAW;AAAA,IACzB;AAEA,QAAI,KAAK,mBAAmB;AAC1B,YAAM,cAAc,IAAI,MAAM;AAAA,QAC5B,SAAS;AAAA,UACP,UAAU;AAAA,UACV;AAAA,UACA,UAAU,cAAc;AAAA,QAC1B;AAAA,MACF,CAAC;AACD,YAAM,aAAa,YAAY,SAAS;AAAA,QACtC,SAAS;AAAA,UACP,iBAAiB;AAAA,UACjB,YAAY;AAAA,UACZ,qBAAqB;AAAA,QACvB;AAAA,MACF,CAAC;AACD,YAAM,WAAW,QAAQ;AACzB,YAAM,KAAK,MAAM,WAAW,YAAY;AACxC,YAAM,GAAG,MAAM;AACf,YAAM,WAAW,WAAW;AAAA,IAC9B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,MAAM,OAAsB;AAC1B,UAAM,KAAK,WAAW,KAAK;AAC3B,SAAK,YAAY;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,UAAoB;AACtB,QAAI,CAAC,KAAK,WAAW;AACnB,YAAM,IAAI,MAAM,wDAAwD;AAAA,IAC1E;AACA,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,WAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAAA,EAC3B;AACF;","names":[]}
+{"version":3,"sources":["../src/testing/mock-client.ts","../src/testing/test-container.ts"],"sourcesContent":["import type { IKafkaClient, TopicMapConstraint } from \"../client/types\";\n\n/**\n * Fully typed mock of `IKafkaClient<T>` where every method is a mock function.\n * Compatible with Jest, Vitest, or any framework whose `fn()` returns\n * an object with `.mock`, `.mockResolvedValue`, etc.\n */\nexport type MockKafkaClient<T extends TopicMapConstraint<T>> = {\n  [K in keyof IKafkaClient<T>]: IKafkaClient<T>[K] & Record<string, any>;\n};\n\n/** Factory that creates a no-op mock function (e.g. `() => jest.fn()`). */\nexport type MockFactory = () => (...args: any[]) => any;\n\nfunction detectMockFactory(): MockFactory {\n  // Jest and Vitest inject their globals (`jest` / `vi`) as module-scope\n  // bindings, not as properties of `globalThis`. The only reliable way to\n  // detect them without a hard import is via `eval`, which evaluates in the\n  // current module scope where those bindings are available.\n  try {\n    if (eval(\"typeof jest === 'object' && typeof jest.fn === 'function'\")) {\n      return () => eval(\"jest.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  try {\n    if (eval(\"typeof vi === 'object' && typeof vi.fn === 'function'\")) {\n      return () => eval(\"vi.fn()\");\n    }\n  } catch {\n    /* not available */\n  }\n  throw new Error(\n    \"createMockKafkaClient: no mock framework detected (jest/vitest). \" +\n      \"Pass a custom mockFactory.\",\n  );\n}\n\n/**\n * Create a fully typed mock implementing every `IKafkaClient<T>` method.\n * Useful for unit-testing services that depend on `KafkaClient` without\n * touching a real broker.\n *\n * Auto-detects Jest (`jest.fn()`) or Vitest (`vi.fn()`). Pass a custom\n * `mockFactory` for other frameworks.\n *\n * All methods resolve to sensible defaults:\n * - `checkStatus()` → `{ status: 'up', clientId: 'mock-client', topics: [] }`\n * - `getClientId()` → `\"mock-client\"`\n * - void methods → `undefined`\n *\n * @example\n * ```ts\n * const kafka = createMockKafkaClient<MyTopics>();\n *\n * const service = new OrdersService(kafka);\n * await service.createOrder();\n *\n * expect(kafka.sendMessage).toHaveBeenCalledWith(\n *   'order.created',\n *   expect.objectContaining({ orderId: '123' }),\n * );\n * ```\n */\nexport function createMockKafkaClient<T extends TopicMapConstraint<T>>(\n  mockFactory?: MockFactory,\n): MockKafkaClient<T> {\n  const fn = mockFactory ?? detectMockFactory();\n\n  const mock = () => fn() as any;\n  const resolved = (value: unknown) => mock().mockResolvedValue(value);\n  const returning = (value: unknown) => mock().mockReturnValue(value);\n\n  return {\n    checkStatus: resolved({\n      status: \"up\",\n      clientId: \"mock-client\",\n      topics: [],\n    }),\n    getConsumerLag: resolved([]),\n    getClientId: returning(\"mock-client\"),\n    sendMessage: resolved(undefined),\n    sendBatch: resolved(undefined),\n    transaction: mock().mockImplementation(\n      async (cb: (ctx: Record<string, unknown>) => Promise<void>) => {\n        const ctx = {\n          send: resolved(undefined),\n          sendBatch: resolved(undefined),\n        };\n        await cb(ctx);\n      },\n    ),\n    startConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    startBatchConsumer: resolved({\n      groupId: \"mock-group\",\n      stop: mock().mockResolvedValue(undefined),\n    }),\n    stopConsumer: resolved(undefined),\n    disconnect: resolved(undefined),\n    enableGracefulShutdown: mock(),\n  } as unknown as MockKafkaClient<T>;\n}\n","import {\n  KafkaContainer,\n  type StartedKafkaContainer,\n} from \"@testcontainers/kafka\";\nimport { KafkaJS } from \"@confluentinc/kafka-javascript\";\nconst { Kafka, logLevel: KafkaLogLevel } = KafkaJS;\n\n/** Options for `KafkaTestContainer`. */\nexport interface KafkaTestContainerOptions {\n  /** Docker image. Default: `\"confluentinc/cp-kafka:7.7.0\"`. */\n  image?: string;\n  /** Warm up the transactional coordinator on start. Default: `true`. */\n  transactionWarmup?: boolean;\n  /** Topics to pre-create. Each entry can be a string (1 partition) or `{ topic, numPartitions }`. */\n  topics?: Array<string | { topic: string; numPartitions?: number }>;\n}\n\n/**\n * Thin wrapper around `@testcontainers/kafka` that starts a single-node\n * KRaft Kafka container and exposes `brokers` for use with `KafkaClient`.\n *\n * Handles common setup pain points:\n * - Transaction coordinator warmup (avoids transactional producer hangs)\n * - Topic pre-creation (avoids race conditions)\n *\n * @example\n * ```ts\n * const container = new KafkaTestContainer({ topics: ['orders', 'payments'] });\n * const brokers = await container.start();\n *\n * const kafka = new KafkaClient('test', 'test-group', brokers);\n * // ... run tests ...\n *\n * await container.stop();\n * ```\n *\n * @example Jest lifecycle\n * ```ts\n * let container: KafkaTestContainer;\n * let brokers: string[];\n *\n * beforeAll(async () => {\n *   container = new KafkaTestContainer({ topics: ['orders'] });\n *   brokers = await container.start();\n * }, 120_000);\n *\n * afterAll(() => container.stop());\n * ```\n */\nexport class KafkaTestContainer {\n  private container: StartedKafkaContainer | undefined;\n  private readonly image: string;\n  private readonly transactionWarmup: boolean;\n  private readonly topics: Array<\n    string | { topic: string; numPartitions?: number }\n  >;\n\n  constructor(options?: KafkaTestContainerOptions) {\n    this.image = options?.image ?? \"confluentinc/cp-kafka:7.7.0\";\n    this.transactionWarmup = options?.transactionWarmup ?? true;\n    this.topics = options?.topics ?? [];\n  }\n\n  /**\n   * Start the Kafka container, pre-create topics, and optionally warm up\n   * the transaction coordinator.\n   *\n   * @returns Broker connection strings, e.g. `[\"localhost:55123\"]`.\n   */\n  async start(): Promise<string[]> {\n    this.container = await new KafkaContainer(this.image)\n      .withKraft()\n      .withExposedPorts(9093)\n      .withEnvironment({\n        KAFKA_TRANSACTION_STATE_LOG_REPLICATION_FACTOR: \"1\",\n        KAFKA_TRANSACTION_STATE_LOG_MIN_ISR: \"1\",\n      })\n      .start();\n\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    const brokers = [`${host}:${port}`];\n\n    const kafka = new Kafka({\n      kafkaJS: {\n        clientId: \"test-container-setup\",\n        brokers,\n        logLevel: KafkaLogLevel.NOTHING,\n      },\n    });\n\n    if (this.topics.length > 0) {\n      const admin = kafka.admin();\n      await admin.connect();\n      await admin.createTopics({\n        topics: this.topics.map((t) =>\n          typeof t === \"string\"\n            ? { topic: t, numPartitions: 1 }\n            : { topic: t.topic, numPartitions: t.numPartitions ?? 1 },\n        ),\n      });\n      await admin.disconnect();\n    }\n\n    if (this.transactionWarmup) {\n      const warmupKafka = new Kafka({\n        kafkaJS: {\n          clientId: \"test-container-warmup\",\n          brokers,\n          logLevel: KafkaLogLevel.NOTHING,\n        },\n      });\n      const txProducer = warmupKafka.producer({\n        kafkaJS: {\n          transactionalId: \"test-container-warmup-tx\",\n          idempotent: true,\n          maxInFlightRequests: 1,\n        },\n      });\n      await txProducer.connect();\n      const tx = await txProducer.transaction();\n      await tx.abort();\n      await txProducer.disconnect();\n    }\n\n    return brokers;\n  }\n\n  /** Stop and remove the container. */\n  async stop(): Promise<void> {\n    await this.container?.stop();\n    this.container = undefined;\n  }\n\n  /** Broker connection strings. Throws if container is not started. */\n  get brokers(): string[] {\n    if (!this.container) {\n      throw new Error(\"KafkaTestContainer is not started. Call start() first.\");\n    }\n    const host = this.container.getHost();\n    const port = this.container.getMappedPort(9093);\n    return [`${host}:${port}`];\n  }\n}\n"],"mappings":";;;AAcA,SAAS,oBAAiC;AAKxC,MAAI;AACF,QAAI,KAAK,2DAA2D,GAAG;AACrE,aAAO,MAAM,KAAK,WAAW;AAAA,IAC/B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,MAAI;AACF,QAAI,KAAK,uDAAuD,GAAG;AACjE,aAAO,MAAM,KAAK,SAAS;AAAA,IAC7B;AAAA,EACF,QAAQ;AAAA,EAER;AACA,QAAM,IAAI;AAAA,IACR;AAAA,EAEF;AACF;AA4BO,SAAS,sBACd,aACoB;AACpB,QAAM,KAAK,eAAe,kBAAkB;AAE5C,QAAM,OAAO,MAAM,GAAG;AACtB,QAAM,WAAW,CAAC,UAAmB,KAAK,EAAE,kBAAkB,KAAK;AACnE,QAAM,YAAY,CAAC,UAAmB,KAAK,EAAE,gBAAgB,KAAK;AAElE,SAAO;AAAA,IACL,aAAa,SAAS;AAAA,MACpB,QAAQ;AAAA,MACR,UAAU;AAAA,MACV,QAAQ,CAAC;AAAA,IACX,CAAC;AAAA,IACD,gBAAgB,SAAS,CAAC,CAAC;AAAA,IAC3B,aAAa,UAAU,aAAa;AAAA,IACpC,aAAa,SAAS,MAAS;AAAA,IAC/B,WAAW,SAAS,MAAS;AAAA,IAC7B,aAAa,KAAK,EAAE;AAAA,MAClB,OAAO,OAAwD;AAC7D,cAAM,MAAM;AAAA,UACV,MAAM,SAAS,MAAS;AAAA,UACxB,WAAW,SAAS,MAAS;AAAA,QAC/B;AACA,cAAM,GAAG,GAAG;AAAA,MACd;AAAA,IACF;AAAA,IACA,eAAe,SAAS;AAAA,MACtB,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,oBAAoB,SAAS;AAAA,MAC3B,SAAS;AAAA,MACT,MAAM,KAAK,EAAE,kBAAkB,MAAS;AAAA,IAC1C,CAAC;AAAA,IACD,cAAc,SAAS,MAAS;AAAA,IAChC,YAAY,SAAS,MAAS;AAAA,IAC9B,wBAAwB,KAAK;AAAA,EAC/B;AACF;;;ACzGA;AAAA,EACE;AAAA,OAEK;AACP,SAAS,eAAe;AACxB,IAAM,EAAE,OAAO,UAAU,cAAc,IAAI;AA4CpC,IAAM,qBAAN,MAAyB;AAAA,EACtB;AAAA,EACS;AAAA,EACA;AAAA,EACA;AAAA,EAIjB,YAAY,SAAqC;AAC/C,SAAK,QAAQ,SAAS,SAAS;AAC/B,SAAK,oBAAoB,SAAS,qBAAqB;AACvD,SAAK,SAAS,SAAS,UAAU,CAAC;AAAA,EACpC;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,QAA2B;AAC/B,SAAK,YAAY,MAAM,IAAI,eAAe,KAAK,KAAK,EACjD,UAAU,EACV,iBAAiB,IAAI,EACrB,gBAAgB;AAAA,MACf,gDAAgD;AAAA,MAChD,qCAAqC;AAAA,IACvC,CAAC,EACA,MAAM;AAET,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,UAAM,UAAU,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAElC,UAAM,QAAQ,IAAI,MAAM;AAAA,MACtB,SAAS;AAAA,QACP,UAAU;AAAA,QACV;AAAA,QACA,UAAU,cAAc;AAAA,MAC1B;AAAA,IACF,CAAC;AAED,QAAI,KAAK,OAAO,SAAS,GAAG;AAC1B,YAAM,QAAQ,MAAM,MAAM;AAC1B,YAAM,MAAM,QAAQ;AACpB,YAAM,MAAM,aAAa;AAAA,QACvB,QAAQ,KAAK,OAAO;AAAA,UAAI,CAAC,MACvB,OAAO,MAAM,WACT,EAAE,OAAO,GAAG,eAAe,EAAE,IAC7B,EAAE,OAAO,EAAE,OAAO,eAAe,EAAE,iBAAiB,EAAE;AAAA,QAC5D;AAAA,MACF,CAAC;AACD,YAAM,MAAM,WAAW;AAAA,IACzB;AAEA,QAAI,KAAK,mBAAmB;AAC1B,YAAM,cAAc,IAAI,MAAM;AAAA,QAC5B,SAAS;AAAA,UACP,UAAU;AAAA,UACV;AAAA,UACA,UAAU,cAAc;AAAA,QAC1B;AAAA,MACF,CAAC;AACD,YAAM,aAAa,YAAY,SAAS;AAAA,QACtC,SAAS;AAAA,UACP,iBAAiB;AAAA,UACjB,YAAY;AAAA,UACZ,qBAAqB;AAAA,QACvB;AAAA,MACF,CAAC;AACD,YAAM,WAAW,QAAQ;AACzB,YAAM,KAAK,MAAM,WAAW,YAAY;AACxC,YAAM,GAAG,MAAM;AACf,YAAM,WAAW,WAAW;AAAA,IAC9B;AAEA,WAAO;AAAA,EACT;AAAA;AAAA,EAGA,MAAM,OAAsB;AAC1B,UAAM,KAAK,WAAW,KAAK;AAC3B,SAAK,YAAY;AAAA,EACnB;AAAA;AAAA,EAGA,IAAI,UAAoB;AACtB,QAAI,CAAC,KAAK,WAAW;AACnB,YAAM,IAAI,MAAM,wDAAwD;AAAA,IAC1E;AACA,UAAM,OAAO,KAAK,UAAU,QAAQ;AACpC,UAAM,OAAO,KAAK,UAAU,cAAc,IAAI;AAC9C,WAAO,CAAC,GAAG,IAAI,IAAI,IAAI,EAAE;AAAA,EAC3B;AACF;","names":[]}