From a4baf7cf8600bc9c637a568f5c12208987a8f8d4 Mon Sep 17 00:00:00 2001
From: Sebastian Velez <sebasvelez@gmail.com>
Date: Fri, 27 Jan 2023 13:18:25 -0500
Subject: [PATCH] chore update user agent payload

---
 src/core/WakaTimeCore.ts | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/core/WakaTimeCore.ts b/src/core/WakaTimeCore.ts
index dcf5274..1747436 100644
--- a/src/core/WakaTimeCore.ts
+++ b/src/core/WakaTimeCore.ts
@@ -225,14 +225,18 @@ class WakaTimeCore {
    */
   preparePayload(heartbeat: SendHeartbeat, type: string): Record<string, unknown> {
     let browserName = 'chrome';
+    let userAgent;
     if (navigator.userAgent.includes('Firefox')) {
       browserName = 'firefox';
+      userAgent = navigator.userAgent.match(/Firefox\/\S+/g)![0];
+    } else {
+      userAgent = navigator.userAgent.match(/Chrome\/\S+/g)![0];
     }
     const payload: Record<string, unknown> = {
       entity: heartbeat.url,
       time: moment().format('X'),
       type: type,
-      user_agent: `${navigator.userAgent} ${browserName}-wakatime/${config.version}`,
+      user_agent: `${userAgent} ${browserName}-wakatime/${config.version}`,
     };
 
     if (heartbeat.project) {