@turbowarp/sb3fix 0.3.2 → 0.3.3

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@turbowarp/sb3fix",
-  "version": "0.3.2",
+  "version": "0.3.3",
   "description": "Fix corrupted Scratch projects",
   "main": "src/sb3fix.js",
   "scripts": {
@@ -21,11 +21,11 @@
   },
   "homepage": "https://github.com/TurboWarp/sb3fix#readme",
   "dependencies": {
-    "@turbowarp/jszip": "^3.11.0"
+    "@turbowarp/jszip": "^3.11.1"
   },
   "devDependencies": {
     "copy-webpack-plugin": "^12.0.2",
-    "webpack": "^5.90.3",
+    "webpack": "^5.96.1",
     "webpack-cli": "^5.1.4"
   }
 }

package/src/sb3fix.js

@@ -351,11 +351,10 @@ const fixJSON = (data, options = {}) => {
       fixTargetInPlace(target);
     }
 
-    let stage;
     const allStages = targets.filter((target) => target.isStage);
     if (allStages.length === 0) {
       log('stage is missing; adding an empty one');
-      stage = {
+      targets.unshift({
         isStage: true,
         name: 'Stage',
         variables: {},
@@ -380,22 +379,31 @@ const fixJSON = (data, options = {}) => {
         videoTransparency: 50,
         videoState: "on",
         textToSpeechLanguage: null
-      };
-      targets.unshift(stage);
-    } else if (allStages.length === 1) {
-      const stageIndex = targets.findIndex((target) => target.isStage);
-      // stageIndex guaranteed to not be -1 by earlier filter check
-      stage = targets[stageIndex];
-      // stage must be the first target
-      if (stageIndex !== 0) {
-        log(`stage was at wrong index: ${stageIndex}`);
-        targets.splice(stageIndex, 1);
+      });
+    } else {
+      // We will accept the first stage in targets as the real stage
+      const firstStageIndex = targets.findIndex((target) => target.isStage);
+
+      // Stage must be the first target
+      if (firstStageIndex !== 0) {
+        log(`stage was at wrong index: ${firstStageIndex}`);
+        const stage = targets[firstStageIndex];
+        targets.splice(firstStageIndex, 1);
         targets.unshift(stage);
       }
-    } else {
-      throw new Error(`wrong number of stages: ${allStages.length}`);
+
+      // Remove all the other stages
+      for (let i = targets.length - 1; i > 0; i--) {
+        if (targets[i].isStage) {
+          log(`removing extra stage at index ${i}`);
+          targets.splice(i, 1);
+        }
+      }
     }
 
+    // Above checks ensure this invariant holds
+    const stage = targets[0];
+
     // stage's name must match exactly
     if (stage.name !== 'Stage') {
       log(`stage had wrong name: ${stage.name}`);