From c57818bec78d7f51a26ecc20fddf85dc42ea9a84 Mon Sep 17 00:00:00 2001
From: Alan Hamlett <alan.hamlett@gmail.com>
Date: Tue, 17 Oct 2023 21:14:35 +0200
Subject: [PATCH] Use last branch same as last project

---
 src/core/WakaTimeCore.ts | 1 +
 src/types/heartbeats.ts  | 1 +
 src/types/user.ts        | 3 ++-
 3 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/core/WakaTimeCore.ts b/src/core/WakaTimeCore.ts
index 321ad46..c74a2c8 100644
--- a/src/core/WakaTimeCore.ts
+++ b/src/core/WakaTimeCore.ts
@@ -354,6 +354,7 @@ class WakaTimeCore {
     };
 
     payload.project = heartbeat.project ?? '<<LAST_PROJECT>>';
+    payload.branch= heartbeat.branch ?? '<<LAST_BRANCH>>';
 
     return payload;
   }
diff --git a/src/types/heartbeats.ts b/src/types/heartbeats.ts
index 1522fdb..4c0bb20 100644
--- a/src/types/heartbeats.ts
+++ b/src/types/heartbeats.ts
@@ -30,6 +30,7 @@ export interface Datum {
 export interface SendHeartbeat {
   hostname: string;
   project: string | null;
+  branch: string | null;
   url: string;
 }
 
diff --git a/src/types/user.ts b/src/types/user.ts
index c5edc51..d19e892 100644
--- a/src/types/user.ts
+++ b/src/types/user.ts
@@ -28,7 +28,8 @@ export interface User {
   last_heartbeat_at: string;
   last_plugin: string;
   last_plugin_name: string;
-  last_project: string;
+  last_project?: string;
+  last_branch?: string;
   location: string;
   logged_time_public: boolean;
   modified_at: string;