fix: heatmaps with segment integration (#1218)

PostHog · Jun 3, 2024 · bb8065e · bb8065e
1 parent 6c5d1d4
commit bb8065e
Showing 1 changed file with 13 additions and 9 deletions.
diff --git a/src/posthog-core.ts b/src/posthog-core.ts
@@ -771,14 +771,7 @@ export class PostHog {
         let data: CaptureResult = {
             uuid: uuidv7(),
             event: event_name,
-            properties: this._calculate_event_properties(event_name, properties || {}),
-        }
-
-        if (!options?._noHeatmaps) {
-            const heatmapsBuffer = this.heatmaps?.getAndClearBuffer()
-            if (heatmapsBuffer) {
-                data.properties['$heatmap_data'] = heatmapsBuffer
-            }
+            properties: this._calculate_event_properties(event_name, properties || {}, options),
         }
 
         if (clientRateLimitContext) {
@@ -831,7 +824,11 @@ export class PostHog {
         this.on('eventCaptured', (data) => callback(data.event))
     }
 
-    _calculate_event_properties(event_name: string, event_properties: Properties): Properties {
+    _calculate_event_properties(
+        event_name: string,
+        event_properties: Properties,
+        options?: CaptureOptions
+    ): Properties {
         if (!this.persistence || !this.sessionPersistence) {
             return event_properties
         }
@@ -911,6 +908,13 @@ export class PostHog {
 
         properties['$is_identified'] = this._isIdentified()
 
+        if (!options?._noHeatmaps) {
+            const heatmapsBuffer = this.heatmaps?.getAndClearBuffer()
+            if (heatmapsBuffer) {
+                properties['$heatmap_data'] = heatmapsBuffer
+            }
+        }
+
         if (isArray(this.config.property_denylist)) {
             each(this.config.property_denylist, function (denylisted_prop) {
                 delete properties[denylisted_prop]