comparison src/app/Session.ts @ 462:2624bb55dbf6

Fix ordering of adding to history (after an action). Add functions for stepping through the history.

 };
 
 export class PersistentStack<T> {
   private stack: T[];
   private history: T[][];
+  private historyOffset: number;
 
   constructor() {
     this.stack = [];
-    this.history = [];
+    this.history = [[]];
+    this.historyOffset = 0;
   }
 
   shift(): T {
-    this.history.push([...this.stack]);
     const item = this.stack[0];
     this.stack = this.stack.slice(1);
+    this.history.push([...this.stack]);
     return item;
   }
 
   unshift(item: T): number {
+    this.stack = [item, ...this.stack];
     this.history.push([...this.stack]);
-    this.stack = [item, ...this.stack];
     return this.stack.length;
   }
 
   findIndex(predicate: (value: T,
                         index: number,

   get(index: number): T {
     return this.stack[index];
   }
 
   set(index: number, value: T) {
-    this.history.push([...this.stack]);
     this.stack = [
       ...this.stack.slice(0, index),
       value,
       ...this.stack.slice(index + 1)
     ];
+    this.history.push([...this.stack]);
   }
 
   map<U>(transform: (value: T, index: number, array: T[]) => U): U[] {
     return this.stack.map(transform);
   }

             initialValue: U): U {
     return this.stack.reduce(reducer, initialValue);
   }
 
   remove(...indices: number[]) {
-    this.history.push([...this.stack]);
     this.stack = this.stack.reduce((acc, item, i) => {
       if (!indices.includes(i)) {
         acc.push(item);
       }
       return acc;
     }, [] as T[]);
+    this.history.push([...this.stack]);
+  }
+
+  stepBack(): void {
+    const latest = this.history.length - 1;
+    if (++this.historyOffset <= latest) {
+      this.stack = this.history[latest - this.historyOffset];
+    } else {
+      this.historyOffset = latest;
+    }
+  }
+
+  stepForward(): void {
+    const latest = this.history.length - 1;
+    if (--this.historyOffset >= 0) {
+      this.stack = this.history[latest - this.historyOffset];
+    } else {
+      this.historyOffset = 0;
+    }
   }
 
   toIterable(): Iterable<T> {
     return this.stack;
   }