From 8d9ff90bcb6d25f2c0d9592c6a5e9d03dab199ca Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gerg=C5=91=20M=C3=B3ricz?= <mo.geryy@gmail.com>
Date: Thu, 22 Aug 2024 23:37:23 +0200
Subject: [PATCH] feat(fire-engine): propagate sentry trace

---
 .../scraper/WebScraper/scrapers/fireEngine.ts | 49 +++++++++++--------
 1 file changed, 29 insertions(+), 20 deletions(-)

diff --git a/apps/api/src/scraper/WebScraper/scrapers/fireEngine.ts b/apps/api/src/scraper/WebScraper/scrapers/fireEngine.ts
index b520bfe2..aa86ad5e 100644
--- a/apps/api/src/scraper/WebScraper/scrapers/fireEngine.ts
+++ b/apps/api/src/scraper/WebScraper/scrapers/fireEngine.ts
@@ -5,6 +5,7 @@ import { generateRequestParams } from "../single_url";
 import { fetchAndProcessPdf } from "../utils/pdfProcessor";
 import { universalTimeout } from "../global";
 import { Logger } from "../../../lib/logger";
+import * as Sentry from "@sentry/node";
 
 /**
  * Scrapes a URL with Fire-Engine
@@ -92,27 +93,35 @@ export async function scrapWithFireEngine({
     });
 
     const startTime = Date.now();
-    const _response = await axiosInstance.post(
-      process.env.FIRE_ENGINE_BETA_URL + endpoint,
-      {
-        url: url,
-        wait: waitParam,
-        screenshot: screenshotParam,
-        fullPageScreenshot: fullPageScreenshotParam,
-        headers: headers,
-        pageOptions: pageOptions,
-        disableJsDom: pageOptions?.disableJsDom ?? false,
-        priority,
-        engine,
-        instantReturn: true,
-        ...fireEngineOptionsParam,
-      },
-      {
-        headers: {
-          "Content-Type": "application/json",
+    const _response = await Sentry.startSpan({
+      name: "Call to fire-engine"
+    }, async span => {
+      return await axiosInstance.post(
+        process.env.FIRE_ENGINE_BETA_URL + endpoint,
+        {
+          url: url,
+          wait: waitParam,
+          screenshot: screenshotParam,
+          fullPageScreenshot: fullPageScreenshotParam,
+          headers: headers,
+          pageOptions: pageOptions,
+          disableJsDom: pageOptions?.disableJsDom ?? false,
+          priority,
+          engine,
+          instantReturn: true,
+          ...fireEngineOptionsParam,
+        },
+        {
+          headers: {
+            "Content-Type": "application/json",
+            ...(Sentry.isInitialized() ? ({
+                "sentry-trace": Sentry.spanToTraceHeader(span),
+                "baggage": Sentry.spanToBaggageHeader(span),
+            }) : {}),
+          }
         }
-      }
-    );
+      );
+    });
 
     let checkStatusResponse = await axiosInstance.get(`${process.env.FIRE_ENGINE_BETA_URL}/scrape/${_response.data.jobId}`);
     while (checkStatusResponse.data.processing && Date.now() - startTime < universalTimeout + waitParam) {