feat: improve enumerator even more

reflektone-games · Dec 2, 2023 · 8b82cb7 · 8b82cb7
1 parent 48a4198
commit 8b82cb7
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/internal/enumerator.ts b/src/internal/enumerator.ts
@@ -5,8 +5,8 @@ export class Enumerator<T> implements Iterable<T> {
 
     constructor(private _source: Iterable<T>) {}
 
-    public get current(): T {
-        return this._current!;
+    public get current(): T | undefined {
+        return this._current;
     }
 
     public get index(): number {
@@ -37,7 +37,7 @@ export class Enumerator<T> implements Iterable<T> {
             next: () => {
                 this.moveNext();
                 return {
-                    value: this.current,
+                    value: this.current!,
                     done: this.done,
                 };
             },

diff --git a/src/internal/syntacticAnalysis/deserializer.ts b/src/internal/syntacticAnalysis/deserializer.ts
@@ -55,7 +55,7 @@ export class Deserializer {
         let manuallyMoved = false;
 
         while (!this.endOfFile && (manuallyMoved || this.moveNext())) {
-            const token = this.enumerator.current;
+            const token = this.enumerator.current!;
             manuallyMoved = false;
 
             switch (token.type) {