Fix State Machine Updates

Author: rcosta358

client/src/services/state-machine.ts

@@ -8,6 +8,9 @@ import { LJStateMachine } from "../types/fsm";
  */
 export async function updateStateMachine(document: vscode.TextDocument) {
     const sm: LJStateMachine = await extension.client?.sendRequest("liquidjava/fsm", { uri: document.uri.toString() });
-    extension.webview?.sendMessage({ type: "fsm", sm });
+
+    // dont update diagram if it hasnt changed to a new one
+    if (!sm || JSON.stringify(sm) === JSON.stringify(extension.stateMachine)) return;
     extension.stateMachine = sm;
+    extension.webview?.sendMessage({ type: "fsm", sm });
 }

client/src/webview/script.ts

@@ -227,10 +227,8 @@ export function getScript(vscode: any, document: any, window: any) {
                 if (!showAllDiagnostics && selectedTab === 'diagnostics') updateView();
                 break;
             case 'fsm':
-                if (msg.sm && msg.sm !== stateMachine) {
-                    stateMachine = msg.sm as LJStateMachine;
-                    if (selectedTab === 'fsm') updateView();
-                }
+                stateMachine = msg.sm as LJStateMachine;
+                if (selectedTab === 'fsm') updateView();
                 break;
             case 'context':
                 context = msg.context as LJContext;