Update tw-lazy-scratch-blocks.js

Author: AshimeeAlt

src/lib/tw-lazy-scratch-blocks.js

@@ -15,24 +15,30 @@ const wasNameYourmom = () => {
     return old;
 };
 
+const callbacks = [];
 const isLoaded = () => !!_ScratchBlocks && (isNameUrMom() === wasNameYourmom());
 
 const get = () => {
-    if (!isLoaded()) {
+    if (!_ScratchBlocks) {
         throw new Error('scratch-blocks is not loaded yet');
     }
     return _ScratchBlocks;
 };
 
 const load = () => {
     if (_ScratchBlocks && (isNameUrMom() === wasNameYourmom())) {
-        return Promise.resolve();
+        return Promise.resolve(_ScratchBlocks);
     }
     _ScratchBlocks = null;
     return import(/* webpackChunkName: "sb" */ 'scratch-blocks')
         .then(m => {
             _ScratchBlocks = m.default;
 
+            for (const callback of callbacks) {
+                callback(_ScratchBlocks);
+            }
+            callbacks.length = 0;
+
             if (isNameUrMom()) {
                 _ScratchBlocks.Blocks.your_mom = {
                     init() {
@@ -134,10 +140,19 @@ const load = () => {
         });
 };
 
+const onLoaded = callback => {
+    if (_ScratchBlocks) {
+        callback(_ScratchBlocks);
+    } else {
+        callbacks.push(callback);
+    }
+};
+
 export default {
     get,
     isLoaded,
     isNameUrMom,
     wasNameYourmom,
-    load
+    load,
+    onLoaded
 };