Implements GetDebugVisualizationDataExtractor

data-extraction/src/js/Extractors/GetDebugVisualizationDataExtractor.ts

@@ -0,0 +1,27 @@
+import { ExtractedData } from "../../DataExtractionResult";
+import { DataExtractor, ExtractionCollector } from "..";
+
+export class GetDebugVisualizationDataExtractor
+	implements DataExtractor<ExtractedData> {
+	readonly id = "get-debug-visualization";
+	getExtractions(
+		data: unknown,
+		extractionCollector: ExtractionCollector<ExtractedData>,
+		evalFn: <TEval>(expression: string) => TEval
+	): void {
+		if (
+			typeof data === "object" &&
+			data &&
+			"getDebugVisualization" in data
+		) {
+			extractionCollector.addExtraction({
+				id: this.id,
+				name: "getDebugVisualization",
+				priority: 500,
+				extractData() {
+					return (data as any).getDebugVisualization();
+				},
+			});
+		}
+	}
+}

data-extraction/src/js/Extractors/index.ts

@@ -1,14 +1,20 @@
 import { getDataExtractorApi } from "../DataExtractorApi";
 import { TypeScriptAstDataExtractor } from "./TypeScriptDataExtractors";
 import { AsIsDataExtractor } from "./AsIsDataExtractor";
+import { GetDebugVisualizationDataExtractor } from "./GetDebugVisualizationDataExtractor";
 
-export { AsIsDataExtractor, TypeScriptAstDataExtractor };
+export {
+	AsIsDataExtractor,
+	TypeScriptAstDataExtractor,
+	GetDebugVisualizationDataExtractor,
+};
 
 export function registerAll() {
 	const api = getDataExtractorApi();
 	for (const item of [
 		new TypeScriptAstDataExtractor(),
 		new AsIsDataExtractor(),
+		new GetDebugVisualizationDataExtractor(),
 	]) {
 		api.registerExtractor(item);
 	}