Update scratch3_procedures.js

PenguinMod · Aug 17, 2024 · cf2f7de · cf2f7de
1 parent 68a90f7
commit cf2f7de
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/blocks/scratch3_procedures.js b/src/blocks/scratch3_procedures.js
@@ -16,6 +16,7 @@ class Scratch3ProcedureBlocks {
         return {
             procedures_definition: this.definition,
             procedures_call: this.call,
+            procedures_set: this.set,
             argument_reporter_string_number: this.argumentReporterStringNumber,
             argument_reporter_boolean: this.argumentReporterBoolean
         };
@@ -68,6 +69,21 @@ class Scratch3ProcedureBlocks {
         }
     }
 
+    set(args, util) {
+      const contain = util.thread.blockContainer;
+      const block = contain.getBlock(util.thread.isCompiled ? util.thread.peekStack() : util.thread.peekStackFrame().op.id);
+      if (!block) return;
+      const thread = util.thread;
+      const param = contain.getBlock(block.inputs.PARAM?.block);
+      if (param) {
+        try {
+          const curParams = thread.stackFrames[0].params;
+          if (curParams !== null) thread.stackFrames[0].params[param.fields.VALUE.value] = args.VALUE;
+          else thread.stackFrames[0].params = { [param.fields.VALUE.value]: args.VALUE }
+        } catch { /* shouldn't happen */ }
+      }
+    }
+
     argumentReporterStringNumber(args, util) {
         const value = util.getParam(args.VALUE);
         if (value === null) {