From 295642e2fbd3df73733f2a2b205f6939c217f570 Mon Sep 17 00:00:00 2001
From: makinosp <makinosp@icloud.com>
Date: Mon, 4 Nov 2024 17:34:36 +0900
Subject: [PATCH] feat: improvement preview data

---
 harmonie/Previews/Models/World.swift                 | 2 ++
 harmonie/Previews/Services/WorldPreviewService.swift | 9 +--------
 2 files changed, 3 insertions(+), 8 deletions(-)

diff --git a/harmonie/Previews/Models/World.swift b/harmonie/Previews/Models/World.swift
index 3465e41..03c1018 100644
--- 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
index 4ef8426..5fda54c 100644
--- 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] {