editor animations addon -- now we can patch custom modals

Author: SharkPool-SP

src/addons/addons/editor-animations/userscript.js

@@ -2,11 +2,6 @@
 // By: SharkPool
 // By: reflow <https://github.com/mgikdev>
 
-/* TODO
-- patch custom modal api when added
-- patch adding modals from addons (they dont use react)
-*/
-
 export default async function({ addon }) {
   const mediaQuery = window.matchMedia("(prefers-reduced-motion: reduce)");
   const addonKey = "addonAnimations-";
@@ -40,8 +35,7 @@ export default async function({ addon }) {
     "emphasis": "linear(0 0%, 0 1.8%, 0.01 3.6%, 0.03 6.35%, 0.07 9.1%, 0.13 11.4%, 0.19 13.4%, 0.27 15%, 0.34 16.1%, 0.54 18.35%, 0.66 20.6%, 0.72 22.4%, 0.77 24.6%, 0.81 27.3%, 0.85 30.4%, 0.88 35.1%, 0.92 40.6%, 0.94 47.2%, 0.96 55%, 0.98 64%, 0.99 74.4%, 1 86.4%, 1 100%)",
   };
   const hasNoVariation = ["default", "fastInSlowOut", "smoothStep", "elastic", "bounce", "emphasis"];
-  
-  
+
   let needsInit = true, animateModals = true, animateLibraries = true, animateButtons = true,
     animationSpeed = 1, animationType = "default", animationDir = "InOut";
   let patchedBody = false, sbPatched = false, sbEverPatched = false, listenerAttached = false;
@@ -394,6 +388,14 @@ export default async function({ addon }) {
         attachCloseHijack("modal");
       }
 
+      const ogSBCustomPrompt = ScratchBlocks.prompt;
+      ScratchBlocks.customPrompt = function(...args) {
+        ogSBCustomPrompt.call(this, ...args);
+
+        handleOpenAnimation("modal");
+        attachCloseHijack("modal");
+      }
+
       if (sbEverPatched) return;
       sbEverPatched = true;