OSDN Git Service

Doutakuを導入
[kcd/KCD.git] / KCD / GuardShelterCommand.swift
index 6be5a79..8558782 100644 (file)
 
 import Cocoa
 
-
-enum GuardEscapeAPI: String {
-    
-    case goback = "/kcsapi/api_req_combined_battle/goback_port"
-}
-
 extension Notification.Name {
     
     static let DidUpdateGuardEscape = Notification.Name("com.masakih.KCD.Notification.DidUpdateGuardEscape")
 }
 
 final class GuardShelterCommand: JSONCommand {
-    
-    override class func canExecuteAPI(_ api: String) -> Bool {
         
-        return GuardEscapeAPI(rawValue: api) != nil ? true : false
+    override class func canExecuteAPI(_ api: API) -> Bool {
+        
+        return api.type == .guardEscape
     }
     
     override func execute() {
         
-        if let b = BattleAPI(rawValue: api) {
+        switch api.type {
             
-            switch b {
-            case .battleResult, .combinedBattleResult:
-                registerReserve()
-                
-            default:
-                break
-            }
-        }
-        if let m = MapAPI(rawValue: api), m == .next {
+        case .battleResult:
+            reserveEscapeShips()
             
+        case .map:
             removeInvalidEntry()
             
-        }
-        if let _ = GuardEscapeAPI(rawValue: api) {
-            
+        case .guardEscape:
             ensureGuardShelter()
             
-        }
-        if let _ = PortAPI(rawValue: api) {
-            
+        case .port:
             removeAllEntry()
             
+        default: return Logger.shared.log("Missing API: \(apiResponse.api)")
         }
     }
     
-    private func fleetMembers(fleetId: Int) -> [Int]? {
+    private func damagedShipId(damagedPos: Int) -> Int? {
         
-        guard let deck = ServerDataStore.default.deck(by: fleetId)
-            else { return nil }
+        let tempStore = TemporaryDataStore.default
+        let firstDeckId = tempStore.sync { tempStore.battle()?.deckId ?? 1 }
         
-        return [deck.ship_0, deck.ship_1, deck.ship_2,
-                deck.ship_3, deck.ship_4, deck.ship_5]
-    }
-    
-    private func damagedShipId(damagedPos: Int) -> Int? {
+        let store = ServerDataStore.default
         
-        func fleetAndPos(_ pos: Int) -> ([Int]?, Int) {
-            switch pos {
-            case 1...6: return (fleetMembers(fleetId: 1), pos - 1)
-                
-            case 7...12: return (fleetMembers(fleetId: 2), pos - 6 - 1)
-                
-            default: return (nil, -1)
+        switch firstDeckId {
+        case 1:
+            switch damagedPos {
+            case 1...6: return store.sync { store.deck(by: 1)?.shipId(of: damagedPos - 1) }
+            case 7...12: return store.sync { store.deck(by: 2)?.shipId(of: damagedPos - 6 - 1) }
+            default: return nil
             }
+        case 3:
+            return store.sync { store.deck(by: 3)?.shipId(of: damagedPos - 1) }
+        default:
+            return nil
         }
-        
-        let (fleet, pos) = fleetAndPos(damagedPos)
-        
-        return fleet?[pos]
     }
     
-    private func registerReserve() {
+    private func reserveEscapeShips() {
         
         let escape = data["api_escape"]
         
-        guard let guardianPos = escape["api_tow_idx"][0].int
-            else { return }
-        
-        let fixedGuardianPos = guardianPos - 6 - 1
-        
-        guard case 0..<6 = fixedGuardianPos,
-            let guardianId = fleetMembers(fleetId: 2)?[fixedGuardianPos]
-            else { return print("guardianPos is wrong") }
-        
-        guard let escapeIdx = escape["api_escape_idx"][0].int,
-            let damagedId = damagedShipId(damagedPos: escapeIdx)
-            else { return print("damagedPos is wrong") }
+        guard let escapeIdx = escape["api_escape_idx"][0].int else { return }
+        guard let damagedId = damagedShipId(damagedPos: escapeIdx) else {
+            
+            return  Logger.shared.log("damagedPos is wrong")
+        }
         
         let store = TemporaryDataStore.oneTimeEditor()
-        
-        guard let guardian = store.createGuardEscaped()
-            else { return print("Can not create GuardEscaped for guardinan") }
-        
-        guardian.shipID = guardianId
-        guardian.ensured = false
-        
-        guard let damaged = store.createGuardEscaped()
-            else { return print("Can not create GuardEscaped for damaged") }
-        
-        damaged.shipID = damagedId
-        damaged.ensured = false
+        store.sync {
+            
+            guard let damaged = store.createGuardEscaped() else {
+                
+                return Logger.shared.log("Can not create GuardEscaped for damaged")
+            }
+            
+            damaged.shipID = damagedId
+            damaged.ensured = false
+            
+            // store guardian if needs
+            guard let guardianPos = escape["api_tow_idx"][0].int else { return }
+            
+            let fixedGuardianPos = guardianPos - 6 - 1
+            
+            let sStore = ServerDataStore.default
+            guard let guardianId = sStore.sync(execute: { sStore.deck(by: 2)?.shipId(of: fixedGuardianPos) }) else {
+                
+                return Logger.shared.log("guardianPos is wrong")
+            }
+            
+            guard let guardian = store.createGuardEscaped() else {
+                
+                return Logger.shared.log("Can not create GuardEscaped for guardinan")
+            }
+            
+            guardian.shipID = guardianId
+            guardian.ensured = false
+        }
     }
     
     private func removeInvalidEntry() {
         
         let store = TemporaryDataStore.oneTimeEditor()
-        
-        store.notEnsuredGuardEscaped().forEach { store.delete($0) }
-        store.save()
+        store.sync {
+            store.notEnsuredGuardEscaped().forEach(store.delete)
+            store.save()
+        }
         Thread.sleep(forTimeInterval: 0.1)
         notify()
     }
@@ -126,19 +117,21 @@ final class GuardShelterCommand: JSONCommand {
     private func removeAllEntry() {
         
         let store = TemporaryDataStore.oneTimeEditor()
-        
-        store.guardEscaped().forEach { store.delete($0) }
-        store.save()
+        store.sync {
+            store.guardEscaped().forEach(store.delete)
+            store.save()
+        }
         Thread.sleep(forTimeInterval: 0.1)
         notify()
     }
     
-    func ensureGuardShelter() {
+    private func ensureGuardShelter() {
         
         let store = TemporaryDataStore.oneTimeEditor()
-        
-        store.guardEscaped().forEach { $0.ensured = true }
-        store.save()
+        store.sync {
+            store.guardEscaped().forEach { $0.ensured = true }
+            store.save()
+        }
         Thread.sleep(forTimeInterval: 0.1)
         notify()
     }