From b7a25a6115a8ea40011ee9e837b3b3a8e0ec2ce6 Mon Sep 17 00:00:00 2001
From: Brendan Chen <me@bchen.dev>
Date: Thu, 16 Jan 2025 21:07:44 -0800
Subject: [PATCH] add repository methods to clear all data

---
 src/repositories/GetterSetterRepository.ts    |  7 ++++++
 .../UnoptimizedInMemoryRepository.ts          | 24 +++++++++++++++++++
 2 files changed, 31 insertions(+)

diff --git a/src/repositories/GetterSetterRepository.ts b/src/repositories/GetterSetterRepository.ts
index ce50567..ddcf68b 100644
--- a/src/repositories/GetterSetterRepository.ts
+++ b/src/repositories/GetterSetterRepository.ts
@@ -18,4 +18,11 @@ export interface GetterSetterRepository extends GetterRepository {
   addOrUpdateStop(stop: IStop): Promise<void>;
   addOrUpdateOrderedStop(orderedStop: IOrderedStop): Promise<void>;
   addOrUpdateEta(eta: IEta): Promise<void>;
+
+  clearSystemData(): Promise<void>;
+  clearRouteData(): Promise<void>;
+  clearShuttleData(): Promise<void>;
+  clearStopData(): Promise<void>;
+  clearOrderedStopData(): Promise<void>;
+  clearEtaData(): Promise<void>;
 }
diff --git a/src/repositories/UnoptimizedInMemoryRepository.ts b/src/repositories/UnoptimizedInMemoryRepository.ts
index e992531..c24a3c5 100644
--- a/src/repositories/UnoptimizedInMemoryRepository.ts
+++ b/src/repositories/UnoptimizedInMemoryRepository.ts
@@ -147,4 +147,28 @@ export class UnoptimizedInMemoryRepository implements GetterSetterRepository {
       this.etas.push(eta);
     }
   }
+
+  public async clearSystemData() {
+    this.systems = [];
+  }
+
+  public async clearShuttleData(): Promise<void> {
+    this.shuttles = [];
+  }
+
+  public async clearEtaData(): Promise<void> {
+    this.etas = [];
+  }
+
+  public async clearOrderedStopData(): Promise<void> {
+    this.orderedStops = [];
+  }
+
+  public async clearRouteData(): Promise<void> {
+    this.routes = [];
+  }
+
+  public async clearStopData(): Promise<void> {
+    this.stops = [];
+  }
 }
\ No newline at end of file