@use-tusk/drift-node-sdk 0.1.10 → 0.1.11

package/dist/index.d.cts

@@ -206,6 +206,7 @@ interface InitParams {
   env?: string;
   logLevel?: LogLevel;
   transforms?: TransformConfigs;
+  samplingRate?: number;
 }
 interface TuskDriftPublicAPI {
   /**
@@ -217,6 +218,7 @@ interface TuskDriftPublicAPI {
    *   - apiKey: string - Your TuskDrift API key (required)
    *   - env: string - The environment name (e.g., 'development', 'staging', 'production') (required)
    *   - logLevel?: LogLevel - Optional logging level ('silent' | 'error' | 'warn' | 'info' | 'debug'), defaults to 'info'
+   *   - samplingRate?: number - Optional sampling rate (0.0-1.0) for recording requests. Overrides TUSK_SAMPLING_RATE env var and config.yaml. Defaults to 1.0
    *
    * @returns void - Initializes the SDK
    *
@@ -227,7 +229,8 @@ interface TuskDriftPublicAPI {
    * TuskDrift.initialize({
    *   apiKey: 'your-api-key',
    *   env: 'production',
-   *   logLevel: 'debug'
+   *   logLevel: 'debug',
+   *   samplingRate: 1  // Record 100% of requests
    * });
    *
    * ```

package/dist/index.d.ts

@@ -208,6 +208,7 @@ interface InitParams {
   env?: string;
   logLevel?: LogLevel;
   transforms?: TransformConfigs;
+  samplingRate?: number;
 }
 interface TuskDriftPublicAPI {
   /**
@@ -219,6 +220,7 @@ interface TuskDriftPublicAPI {
    *   - apiKey: string - Your TuskDrift API key (required)
    *   - env: string - The environment name (e.g., 'development', 'staging', 'production') (required)
    *   - logLevel?: LogLevel - Optional logging level ('silent' | 'error' | 'warn' | 'info' | 'debug'), defaults to 'info'
+   *   - samplingRate?: number - Optional sampling rate (0.0-1.0) for recording requests. Overrides TUSK_SAMPLING_RATE env var and config.yaml. Defaults to 1.0
    *
    * @returns void - Initializes the SDK
    *
@@ -229,7 +231,8 @@ interface TuskDriftPublicAPI {
    * TuskDrift.initialize({
    *   apiKey: 'your-api-key',
    *   env: 'production',
-   *   logLevel: 'debug'
+   *   logLevel: 'debug',
+   *   samplingRate: 1  // Record 100% of requests
    * });
    *
    * ```

package/dist/index.js

@@ -208,17 +208,9 @@ function withTuskDrift(nextConfig = {}, options = {}) {
 				const originalExternals = webpackConfig.externals;
 				const coreExternals = ["require-in-the-middle", "jsonpath"];
 				const externalsMapping = {};
-				try {
-					const sdkPath = __require.resolve("@use-tusk/drift-node-sdk");
-					const sdkNodeModules = __require("path").resolve(sdkPath, "../..", "node_modules");
-					for (const pkg of coreExternals) {
-						const pkgPath = __require("path").join(sdkNodeModules, pkg);
-						externalsMapping[pkg] = `commonjs ${pkgPath}`;
-						debugLog(debug, `Mapped external ${pkg} -> ${pkgPath}`);
-					}
-				} catch (e) {
-					warn(suppressAllWarnings || false, `Could not resolve SDK path, falling back to regular externals: ${e instanceof Error ? e.message : String(e)}`);
-					for (const pkg of coreExternals) externalsMapping[pkg] = `commonjs ${pkg}`;
+				for (const pkg of coreExternals) {
+					externalsMapping[pkg] = `commonjs ${pkg}`;
+					debugLog(debug, `Mapped external ${pkg} -> commonjs ${pkg}`);
 				}
 				if (!originalExternals) {
 					webpackConfig.externals = [externalsMapping];
@@ -264,7 +256,7 @@ var TdInstrumentationAbstract = class {
 
 //#endregion
 //#region package.json
-var version = "0.1.10";
+var version = "0.1.11";
 
 //#endregion
 //#region src/version.ts
@@ -12956,6 +12948,41 @@ var TuskDriftCore = class TuskDriftCore {
 			default: return TuskDriftMode.DISABLED;
 		}
 	}
+	validateSamplingRate(value, source) {
+		if (typeof value !== "number" || isNaN(value)) {
+			logger.warn(`Invalid sampling rate from ${source}: not a number. Ignoring.`);
+			return false;
+		}
+		if (value < 0 || value > 1) {
+			logger.warn(`Invalid sampling rate from ${source}: ${value}. Must be between 0.0 and 1.0. Ignoring.`);
+			return false;
+		}
+		return true;
+	}
+	determineSamplingRate(initParams) {
+		if (initParams.samplingRate !== void 0) {
+			if (this.validateSamplingRate(initParams.samplingRate, "init params")) {
+				logger.debug(`Using sampling rate from init params: ${initParams.samplingRate}`);
+				return initParams.samplingRate;
+			}
+		}
+		const envSamplingRate = OriginalGlobalUtils.getOriginalProcessEnvVar("TUSK_SAMPLING_RATE");
+		if (envSamplingRate !== void 0) {
+			const parsed = parseFloat(envSamplingRate);
+			if (this.validateSamplingRate(parsed, "TUSK_SAMPLING_RATE env var")) {
+				logger.debug(`Using sampling rate from TUSK_SAMPLING_RATE env var: ${parsed}`);
+				return parsed;
+			}
+		}
+		if (this.config.recording?.sampling_rate !== void 0) {
+			if (this.validateSamplingRate(this.config.recording.sampling_rate, "config.yaml")) {
+				logger.debug(`Using sampling rate from config.yaml: ${this.config.recording.sampling_rate}`);
+				return this.config.recording.sampling_rate;
+			}
+		}
+		logger.debug("Using default sampling rate: 1.0");
+		return 1;
+	}
 	registerDefaultInstrumentations() {
 		const transforms = this.config.transforms ?? this.initParams.transforms;
 		new HttpInstrumentation({
@@ -13054,7 +13081,7 @@ var TuskDriftCore = class TuskDriftCore {
 			logLevel: initParams.logLevel || "info",
 			prefix: "TuskDrift"
 		});
-		this.samplingRate = this.config.recording?.sampling_rate ?? 1;
+		this.samplingRate = this.determineSamplingRate(initParams);
 		this.initParams = initParams;
 		if (!this.initParams.env) {
 			const nodeEnv = OriginalGlobalUtils.getOriginalProcessEnvVar("NODE_ENV") || "development";