fix: Hopefully fix race condition.

xdy · Dec 28, 2021 · 3ffaedb · 3ffaedb
1 parent 2a9b3f4
commit 3ffaedb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 5 deletions.
diff --git a/src/module/app/mystify-token/index.ts b/src/module/app/mystify-token/index.ts
@@ -123,7 +123,8 @@ export async function doMystification(token: Token, active: boolean) {
                 })
             );
     }
-    (game as Game).scenes?.active?.updateEmbeddedDocuments("Token", updates);
+
+    return updates;
 }
 
 export function renderNameHud(data: TokenData, html: JQuery) {
@@ -141,7 +142,8 @@ export function renderNameHud(data: TokenData, html: JQuery) {
             const hudElement = $(e.currentTarget);
             const active = hudElement.hasClass("active");
             if (token !== null && isTokenMystified(token) === active) {
-                await doMystification(token, active);
+                const updates = await doMystification(token, active);
+                await (game as Game).scenes?.active?.updateEmbeddedDocuments("Token", updates);
             }
             hudElement.toggleClass("active");
         });

diff --git a/src/module/keybinds.ts b/src/module/keybinds.ts
@@ -1,7 +1,7 @@
 // Keyboard key controlling whether the actor should be mystified, if this feature is toggled on
 import { MODULENAME } from "./xdy-pf2e-workbench";
-import { doMystification, isTokenMystified } from "./app/mystify-token";
 import { moveSelectedAheadOfCurrent } from "./app/moveCombatant";
+import { doMystification, isTokenMystified } from "./app/mystify-token";
 
 export let mystifyModifierKey: string;
 
@@ -35,8 +35,13 @@ export function registerKeybindings() {
                 key: "KeyM",
             },
         ],
-        onDown: () => {
-            canvas?.tokens?.controlled.filter(async (token) => await doMystification(token, isTokenMystified(token)));
+        onDown: async () => {
+            const updates = [];
+            for (const token of canvas?.tokens?.controlled || []) {
+                updates.push(...(await doMystification(token, isTokenMystified(token))));
+            }
+
+            await (game as Game).scenes?.active?.updateEmbeddedDocuments("Token", updates);
         },
         restricted: false,
         reservedModifiers: [],