avoid resetting timer on selection change

416rehman · Jan 20, 2024 · 189f8f2 · 189f8f2
1 parent 7962a80
commit 189f8f2
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/src/main/java/pl/piotradamczyk/ghidracord/GhidracordPlugin.java b/src/main/java/pl/piotradamczyk/ghidracord/GhidracordPlugin.java
@@ -23,6 +23,8 @@
 )
 //@formatter:on
 public class GhidracordPlugin extends ProgramPlugin {
+    // keeps track of last time RPC was updated
+    private long runningSince = 0;
 
     /**
      * Plugin constructor.
@@ -56,12 +58,7 @@ private void updatePresenceInfo() {
             return;
         }
 
-        String desc = switch (this.getTool().getName()) {
-            case "CodeBrowser" -> "Examining \"";
-            case "Debugger" -> "Debugging \"";
-            case "Version Tracking" -> "Version Tracking \"";
-            default -> "Examining \"";
-        } + currentProgram.getName() + "\"";
+        String desc = this.getCurrentProgram().getName();
 
         String status = null;
         if (this.getTool().getToolName().equals("CodeBrowser")) {// current function name
@@ -119,14 +116,18 @@ protected void highlightChanged(ProgramSelection hl) {
     private void updateRichPresence(String programStatus, String subStatus) {
         String program = programStatus == null ? "Idle" : programStatus;
         String sub = subStatus == null ? "" : subStatus;
-        DiscordRichPresence.Builder rich = new DiscordRichPresence.Builder(subStatus)
-                .setBigImage("logo", "Ghidra")
-                .setStartTimestamps(System.currentTimeMillis());
-        this.setSmallIcon(rich);
 
-        rich.setDetails(programStatus);
+        if (runningSince == 0) {
+            runningSince = System.currentTimeMillis();
+        }
+
+        DiscordRichPresence.Builder rpc = new DiscordRichPresence.Builder(sub);
+        this.setSmallIcon(rpc);
+        rpc.setBigImage("logo", "Ghidra");
+        rpc.setStartTimestamps(runningSince);
+        rpc.setDetails(program);
 
-        DiscordRPC.discordUpdatePresence(rich.build());
+        DiscordRPC.discordUpdatePresence(rpc.build());
     }
 
     private void setSmallIcon(DiscordRichPresence.Builder builder) {