feat(relaunch-data): cleanup relaunch data

ph1p · May 1, 2020 · 427a1d5 · 427a1d5
1 parent 52a6073
commit 427a1d5
Showing 1 changed file with 21 additions and 10 deletions.
diff --git a/src/code.ts b/src/code.ts
@@ -32,7 +32,7 @@ async function main() {
     await figma.clientStorage.setAsync('id', instanceId);
   }
 
-  if(!roomName && !secret) {
+  if (!roomName && !secret) {
     figma.root.setPluginData('ownerId', instanceId);
   }
 
@@ -52,9 +52,6 @@ async function main() {
     figma.root.setPluginData('history', history);
   }
 
-
-
-
   return {
     roomName,
     secret,
@@ -115,18 +112,32 @@ function iterateOverFile(node, cb) {
   }
 }
 
+let previousSelection = figma.currentPage.selection || [];
+
 main().then(({ roomName, secret, history, instanceId }) => {
   postMessage('ready');
 
   // events
   figma.on('selectionchange', () => {
-    iterateOverFile(figma.root, (node) => {
-      if (node.setRelaunchData && isValidShape(node)) {
-        node.setRelaunchData({
-          relaunch: '',
-        });
+    if (figma.currentPage.selection.length > 0) {
+      for (let node of figma.currentPage.selection) {
+        if (node.setRelaunchData && isValidShape(node)) {
+          node.setRelaunchData({
+            relaunch: '',
+          });
+        }
       }
-    });
+      previousSelection = figma.currentPage.selection;
+    } else {
+      if (previousSelection.length > 0) {
+        // tidy up 🧹
+        for (let node of previousSelection) {
+          if (node.setRelaunchData && isValidShape(node)) {
+            node.setRelaunchData({});
+          }
+        }
+      }
+    }
     if (triggerSelectionEvent) {
       sendSelection();
     }