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

package/dist/index.cjs

@@ -230,17 +230,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];
@@ -286,7 +278,7 @@ var TdInstrumentationAbstract = class {
 
 //#endregion
 //#region package.json
-var version = "0.1.9";
+var version = "0.1.11";
 
 //#endregion
 //#region src/version.ts
@@ -1228,6 +1220,7 @@ async function httpBodyEncoder({ bodyBuffer, contentEncoding }) {
 * @returns The corresponding DecodedType enum value
 */
 function getDecodedType(contentType) {
+	if (!contentType) return;
 	const contentTypeString = Array.isArray(contentType) && contentType.length > 0 ? contentType[0] : contentType;
 	if (!contentTypeString || typeof contentTypeString !== "string") {
 		logger.debug(`Invalid Content-Type header: ${contentType}`);
@@ -1236,22 +1229,7 @@ function getDecodedType(contentType) {
 	const mainType = contentTypeString.toLowerCase().split(";")[0].trim();
 	return CONTENT_TYPE_MAPPING[mainType];
 }
-const STATIC_ASSET_TYPES = new Set([
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.HTML,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.CSS,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.JAVASCRIPT,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.JPEG,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.PNG,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.GIF,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.WEBP,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.SVG,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.PDF,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.AUDIO,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.VIDEO,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.BINARY,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.ZIP,
-	__use_tusk_drift_schemas_core_json_schema.DecodedType.GZIP
-]);
+const ACCEPTABLE_CONTENT_TYPES = new Set([__use_tusk_drift_schemas_core_json_schema.DecodedType.JSON, __use_tusk_drift_schemas_core_json_schema.DecodedType.PLAIN_TEXT]);
 
 //#endregion
 //#region src/instrumentation/libraries/http/mocks/TdHttpMockSocket.ts
@@ -2831,9 +2809,9 @@ var HttpInstrumentation = class extends TdInstrumentationBase {
 					} : { code: __opentelemetry_api.SpanStatusCode.OK };
 					SpanUtils.setStatus(spanInfo.span, status);
 					const decodedType = getDecodedType(outputValue.headers?.["content-type"] || "");
-					if (decodedType && STATIC_ASSET_TYPES.has(decodedType)) {
+					if (decodedType && !ACCEPTABLE_CONTENT_TYPES.has(decodedType)) {
 						TraceBlockingManager.getInstance().blockTrace(spanInfo.traceId);
-						logger.debug(`[HttpInstrumentation] Blocking trace ${spanInfo.traceId} because it is an static asset response. Decoded type: ${decodedType}`);
+						logger.debug(`[HttpInstrumentation] Blocking trace ${spanInfo.traceId} because it is not an acceptable decoded type: ${decodedType}`, { acceptableContentTypes: Array.from(ACCEPTABLE_CONTENT_TYPES) });
 					}
 					SpanUtils.endSpan(spanInfo.span);
 				} catch (error) {
@@ -9821,9 +9799,9 @@ var NextjsInstrumentation = class extends TdInstrumentationBase {
 			} : { code: __opentelemetry_api.SpanStatusCode.OK };
 			SpanUtils.setStatus(spanInfo.span, status);
 			const decodedType = getDecodedType(outputValue.headers?.["content-type"] || "");
-			if (decodedType && STATIC_ASSET_TYPES.has(decodedType)) {
+			if (decodedType && !ACCEPTABLE_CONTENT_TYPES.has(decodedType)) {
 				TraceBlockingManager.getInstance().blockTrace(spanInfo.traceId);
-				logger.debug(`[NextjsInstrumentation] Blocking trace ${spanInfo.traceId} because it is an static asset response. Decoded type: ${decodedType}`);
+				logger.debug(`[NextjsInstrumentation] Blocking trace ${spanInfo.traceId} because it is not an acceptable decoded type: ${decodedType}`, { acceptableContentTypes: Array.from(ACCEPTABLE_CONTENT_TYPES) });
 			}
 			SpanUtils.endSpan(spanInfo.span);
 			if (self.mode === TuskDriftMode.REPLAY) try {
@@ -12992,6 +12970,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({
@@ -13090,7 +13103,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";