Draft: Add Z1013 decoder (2)

stefanschramm · Nov 12, 2023 · 9fa96c5 · 9fa96c5
1 parent 176fb85
commit 9fa96c5
Showing 1 changed file with 16 additions and 7 deletions.
diff --git a/retroload-lib/src/decoding/writer/z1013/Z1013BlockProcessor.ts b/retroload-lib/src/decoding/writer/z1013/Z1013BlockProcessor.ts
@@ -18,15 +18,12 @@ export class Z1013BlockProcessor {
       if (this.belongsToCurrentFile(bdr)) {
         this.blocks.push(bdr);
       } else {
-        yield new FileDecodingResult(
-          this.blocks.map((b) => b.data),
-          this.currentFileWasSuccessful() ? FileDecodingResultStatus.Success : FileDecodingResultStatus.Error,
-          this.blocks[0].blockBegin,
-          this.blocks[this.blocks.length - 1].blockEnd,
-        );
-        this.blocks = [];
+        yield this.finishFile();
       }
     }
+    if (this.blocks.length !== 0) {
+      yield this.finishFile();
+    }
   }
 
   private belongsToCurrentFile(bdr: BlockDecodingResult): boolean {
@@ -49,6 +46,18 @@ export class Z1013BlockProcessor {
 
     return true;
   }
+
+  private finishFile(): FileDecodingResult {
+    const fdr = new FileDecodingResult(
+      this.blocks.map((b) => b.data),
+      this.currentFileWasSuccessful() ? FileDecodingResultStatus.Success : FileDecodingResultStatus.Error,
+      this.blocks[0].blockBegin,
+      this.blocks[this.blocks.length - 1].blockEnd,
+    );
+    this.blocks = [];
+
+    return fdr;
+  }
 }
 
 export class FileDecodingResult {