feat: improvement preview data

makinosp · Nov 4, 2024 · 295642e · 295642e
1 parent 59a4b19
commit 295642e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/harmonie/Previews/Models/World.swift b/harmonie/Previews/Models/World.swift
@@ -58,6 +58,8 @@ extension PreviewData {
         popularity: 5,
         heat: 3
     )
+
+    static let worldList = [ bar, casino, fuji, chinatown, nightCity ]
 }
 
 extension World {

diff --git a/harmonie/Previews/Services/WorldPreviewService.swift b/harmonie/Previews/Services/WorldPreviewService.swift
@@ -14,14 +14,7 @@ final actor WorldPreviewService: APIService, WorldServiceProtocol {
     let client: APIClient
 
     func fetchWorld(worldId: String) async throws -> World {
-        switch worldId {
-        case PreviewData.bar.id:
-            PreviewData.bar
-        case PreviewData.casino.id:
-            PreviewData.casino
-        default:
-            PreviewData.bar
-        }
+        PreviewData.worldList.first { $0.id == worldId } ?? PreviewData.bar
     }
 
     func fetchFavoritedWorlds() async throws -> [FavoriteWorld] {