@use-tusk/drift-node-sdk 0.1.15 → 0.1.17

package/README.md

@@ -31,6 +31,13 @@ For comprehensive guides and API reference, visit our [full documentation](https
 - [Quick Start Guide](docs/quickstart.md) - Record and replay your first trace
 - [Troubleshooting Guide](docs/troubleshooting.md) - Common issues and solutions
 
+<div align="center">
+
+![Tusk Drift Animated Diagram](images/tusk-drift-animated-diagram-light.gif#gh-light-mode-only)
+![Tusk Drift Animated Diagram](images/tusk-drift-animated-diagram-dark.gif#gh-dark-mode-only)
+
+</div>
+
 ## Requirements
 
 Tusk Drift currently supports the following packages and versions:

package/dist/index.cjs

@@ -7864,7 +7864,10 @@ function getDecodedType(contentType) {
 		logger.debug(`Invalid Content-Type header: ${contentType}`);
 		return;
 	}
-	return CONTENT_TYPE_MAPPING[contentTypeString.toLowerCase().split(";")[0].trim()];
+	const mainType = contentTypeString.toLowerCase().split(";")[0].trim();
+	const decodedType = CONTENT_TYPE_MAPPING[mainType];
+	if (decodedType !== void 0) return decodedType;
+	if (mainType.includes("json")) return DecodedType.JSON;
 }
 const ACCEPTABLE_CONTENT_TYPES = new Set([DecodedType.JSON, DecodedType.PLAIN_TEXT]);
 
@@ -8209,10 +8212,17 @@ var JsonSchemaHelper = class JsonSchemaHelper {
 	static decodeDataWithMerges(data, schemaMerges) {
 		if (!schemaMerges) return data;
 		const decodedData = { ...data };
-		for (const [key, schema] of Object.entries(schemaMerges)) if (schema.encoding && schema.decodedType && data[key] !== void 0) try {
+		for (const [key, schema] of Object.entries(schemaMerges)) if (schema.encoding && data[key] !== void 0) try {
 			let decodedValue = data[key];
-			if (schema.encoding === EncodingType.BASE64 && typeof decodedValue === "string") decodedValue = Buffer.from(decodedValue, "base64").toString("utf8");
-			if (schema.decodedType === DecodedType.JSON && typeof decodedValue === "string") decodedValue = JSON.parse(decodedValue);
+			if (typeof decodedValue === "string") {
+				if (schema.encoding === EncodingType.BASE64) decodedValue = Buffer.from(decodedValue, "base64").toString("utf8");
+				if (schema.decodedType === DecodedType.JSON) decodedValue = JSON.parse(decodedValue);
+				else if (!schema.decodedType) try {
+					decodedValue = JSON.parse(decodedValue);
+				} catch {
+					logger.debug(`[JsonSchemaHelper] Failed to parse JSON for key: ${key}, no decodedType specified`);
+				}
+			}
 			decodedData[key] = decodedValue;
 		} catch (error) {
 			logger.debug(`[JsonSchemaHelper] Failed to decode ${key}:`, error);
@@ -11810,6 +11820,7 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 				const inputValue = {
 					text: queryConfig.text,
 					values: queryConfig.values || [],
+					rowMode: queryConfig.rowMode,
 					clientType
 				};
 				if (self.mode === TuskDriftMode.REPLAY) {
@@ -11924,6 +11935,7 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 		if (typeof firstArg === "object" && firstArg.text) return {
 			text: firstArg.text,
 			values: firstArg.values,
+			rowMode: firstArg.rowMode,
 			callback: firstArg.callback || (typeof args[1] === "function" ? args[1] : void 0)
 		};
 		return null;
@@ -12024,7 +12036,7 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 			logger.warn(`[PgInstrumentation] No mock data found for PG query: ${queryText}`);
 			throw new Error(`[PgInstrumentation] No mock data found for PG query: ${queryText}`);
 		}
-		const processedResult = this.convertPostgresTypes(mockData.result);
+		const processedResult = this.convertPostgresTypes(mockData.result, inputValue.rowMode);
 		if (queryConfig.callback) {
 			process.nextTick(() => {
 				queryConfig.callback(null, processedResult);
@@ -12033,35 +12045,56 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 		} else return Promise.resolve(processedResult);
 	}
 	/**
+	* Convert a single value based on its PostgreSQL data type.
+	*
+	* Reference for data type IDs: https://jdbc.postgresql.org/documentation/publicapi/constant-values.html
+	*/
+	convertPostgresValue(value, dataTypeID) {
+		if (value === null || value === void 0) return value;
+		switch (dataTypeID) {
+			case 1184:
+			case 1114:
+			case 1082:
+				if (typeof value === "string") return new Date(value);
+				break;
+			case 1083:
+			case 1266: break;
+			default: break;
+		}
+		return value;
+	}
+	/**
 	* Convert PostgreSQL string values back to appropriate JavaScript types
 	* based on field metadata from the recorded response.
 	*
 	* Reference for data type IDs: https://jdbc.postgresql.org/documentation/publicapi/constant-values.html
 	*/
-	convertPostgresTypes(result) {
+	convertPostgresTypes(result, rowMode) {
 		if (!result || !result.fields || !result.rows) return result;
+		if (rowMode === "array") {
+			const convertedRows$1 = result.rows.map((row) => {
+				if (!Array.isArray(row)) return row;
+				return row.map((value, index) => {
+					const field = result.fields[index];
+					if (!field) return value;
+					return this.convertPostgresValue(value, field.dataTypeID);
+				});
+			});
+			return {
+				...result,
+				rows: convertedRows$1
+			};
+		}
 		const fieldTypeMap = {};
 		result.fields.forEach((field) => {
 			fieldTypeMap[field.name] = field.dataTypeID;
 		});
-		const convertedRows = result.rows.map((row, rowIndex) => {
+		const convertedRows = result.rows.map((row) => {
 			const convertedRow = { ...row };
 			Object.keys(row).forEach((fieldName) => {
 				const dataTypeID = fieldTypeMap[fieldName];
 				const value = row[fieldName];
-				if (value === null || value === void 0) return;
-				switch (dataTypeID) {
-					case 1184:
-					case 1114:
-						if (typeof value === "string") convertedRow[fieldName] = new Date(value);
-						break;
-					case 1082:
-						if (typeof value === "string") convertedRow[fieldName] = new Date(value);
-						break;
-					case 1083:
-					case 1266: break;
-					default: break;
-				}
+				convertedRow[fieldName] = this.convertPostgresValue(value, dataTypeID);
 			});
 			return convertedRow;
 		});
@@ -13289,6 +13322,12 @@ var MysqlInstrumentation = class extends TdInstrumentationBase {
 				logger.debug(`[MysqlInstrumentation] Wrapped Connection.prototype.resume`);
 			}
 		}
+		if (ConnectionClass.prototype && ConnectionClass.prototype.destroy) {
+			if (!isWrapped$1(ConnectionClass.prototype.destroy)) {
+				this._wrap(ConnectionClass.prototype, "destroy", this._getDestroyPatchFn());
+				logger.debug(`[MysqlInstrumentation] Wrapped Connection.prototype.destroy`);
+			}
+		}
 		this.markModuleAsPatched(ConnectionClass);
 		logger.debug(`[MysqlInstrumentation] Connection class patching complete`);
 		return ConnectionClass;
@@ -13703,6 +13742,19 @@ var MysqlInstrumentation = class extends TdInstrumentationBase {
 		};
 	}
 	/**
+	* Get wrapper function for destroy method
+	*/
+	_getDestroyPatchFn() {
+		const self = this;
+		return (originalDestroy) => {
+			return function destroy() {
+				if (self.mode === TuskDriftMode.REPLAY) return;
+				else if (self.mode === TuskDriftMode.RECORD) return originalDestroy.apply(this, arguments);
+				else return originalDestroy.apply(this, arguments);
+			};
+		};
+	}
+	/**
 	* Get wrapper function for Pool.end method
 	*/
 	_getPoolEndPatchFn() {
@@ -32015,7 +32067,7 @@ var require_src = /* @__PURE__ */ __commonJS({ "node_modules/@opentelemetry/sdk-
 //#endregion
 //#region package.json
 var import_src$1 = /* @__PURE__ */ __toESM(require_src(), 1);
-var version = "0.1.15";
+var version = "0.1.17";
 
 //#endregion
 //#region src/version.ts