Doutakuを導入

[kcd/KCD.git] / KCD / NyukyoDockMapper.swift

diff --git a/KCD/NyukyoDockMapper.swift b/KCD/NyukyoDockMapper.swift
index 2f1481c..b5032d6 100644 (file)
--- a/KCD/NyukyoDockMapper.swift
+++ b/KCD/NyukyoDockMapper.swift
@@ -8,30 +8,28 @@
 
 import Cocoa
 
-fileprivate enum DeckAPI: String {
-    case getMemberNDock = "/kcsapi/api_get_member/ndock"
-    case port = "/kcsapi/api_port/port"
-}
-
-fileprivate func dataKey(_ apiResponse: APIResponse) -> String {
-    guard let deckApi = DeckAPI(rawValue: apiResponse.api)
-        else { return "api_data" }
-    switch deckApi {
-    case .port: return "api_data.api_ndock"
-    default: return "api_data"
-    }
-}
-
-class NyukyoDockMapper: JSONMapper {
-    typealias ObjectType = NyukyoDock
-
+final class NyukyoDockMapper: JSONMapper {
+    
     let apiResponse: APIResponse
     let configuration: MappingConfiguration<NyukyoDock>
     
     required init(_ apiResponse: APIResponse) {
+        
         self.apiResponse = apiResponse
         self.configuration = MappingConfiguration(entity: NyukyoDock.entity,
-                                                  dataKey: dataKey(apiResponse),
+                                                  dataKeys: NyukyoDockMapper.dataKeys(apiResponse),
                                                   editorStore: ServerDataStore.oneTimeEditor())
     }
+    
+    private class func dataKeys(_ apiResponse: APIResponse) -> [String] {
+        
+        switch apiResponse.api.endpoint {
+            
+        case .port: return ["api_data", "api_ndock"]
+            
+        case .ndock: return ["api_data"]
+            
+        default: return Logger.shared.log("Missing API: \(apiResponse.api)", value: ["api_data"])
+        }
+    }
 }