fix: Fix usageV2 imports, allow sdk source

Author: arcoraven

.changeset/small-moons-behave.md

@@ -0,0 +1,5 @@
+---
+"@thirdweb-dev/service-utils": patch
+---
+
+[service-utils] fix: Add missing CompressionCodecs import

packages/service-utils/package.json

@@ -52,7 +52,6 @@
   },
   "devDependencies": {
     "@cloudflare/workers-types": "4.20250129.0",
-    "@types/lz4js": "^0.2.1",
     "@types/node": "22.13.0",
     "typescript": "5.7.3",
     "vitest": "3.0.4"

packages/service-utils/src/core/usageV2.ts

@@ -1,4 +1,6 @@
-import type { ServiceName } from "../node/index.js";
+import type { ServiceName } from "./services.js";
+
+export type UsageV2Source = ServiceName | "sdk";
 
 export interface UsageV2Event {
   /**
@@ -53,6 +55,6 @@ export interface UsageV2Event {
   [key: string]: boolean | number | string | Date | null | undefined;
 }
 
-export function getTopicName(productName: ServiceName) {
-  return `usage_v2.raw_${productName}`;
+export function getTopicName(source: UsageV2Source) {
+  return `usage_v2.raw_${source}`;
 }

packages/service-utils/src/node/usageV2.ts

@@ -1,15 +1,21 @@
 import { randomUUID } from "node:crypto";
 import { checkServerIdentity } from "node:tls";
 import {
-  CompressionCodecs,
   CompressionTypes,
   Kafka,
   type Producer,
   type ProducerConfig,
 } from "kafkajs";
 import { compress, decompress } from "lz4js";
-import type { ServiceName } from "../core/services.js";
-import { type UsageV2Event, getTopicName } from "../core/usageV2.js";
+import {
+  type UsageV2Event,
+  type UsageV2Source,
+  getTopicName,
+} from "../core/usageV2.js";
+
+// CompressionCodecs is not exported properly in kafkajs. Source: https://github.com/tulios/kafkajs/issues/1391
+import KafkaJS from "kafkajs";
+const { CompressionCodecs } = KafkaJS;
 
 /**
  * Creates a UsageV2Producer which opens a persistent TCP connection.
@@ -40,9 +46,9 @@ export class UsageV2Producer {
      */
     environment: "development" | "production";
     /**
-     * The product "source" where usage is coming from.
+     * The product where usage is coming from.
      */
-    productName: ServiceName;
+    source: UsageV2Source;
     /**
      * Whether to compress the events.
      */
@@ -54,7 +60,7 @@ export class UsageV2Producer {
     const {
       producerName,
       environment,
-      productName,
+      source,
       shouldCompress = true,
       username,
       password,
@@ -78,7 +84,7 @@ export class UsageV2Producer {
       },
     });
 
-    this.topic = getTopicName(productName);
+    this.topic = getTopicName(source);
     this.compression = shouldCompress
       ? CompressionTypes.LZ4
       : CompressionTypes.None;