KCD/TemporaryDataStore.swift

   1 //
   2 //  TemporaryDataStore.swift
   3 //  KCD
   4 //
   5 //  Created by Hori,Masaki on 2017/01/06.
   6 //  Copyright © 2017年 Hori,Masaki. All rights reserved.
   7 //
   8
   9 import Cocoa
  10
  11 extension CoreDataIntormation {
  12     static let temporary = CoreDataIntormation("Temporary",
  13                                                fileName: ":memory:",
  14                                                options: [:],
  15                                                type: NSInMemoryStoreType
  16     )
  17 }
  18 extension CoreDataCore {
  19     static let temporary = CoreDataCore(.temporary)
  20 }
  21
  22 class TemporaryDataStore: CoreDataAccessor, CoreDataManager {
  23     static var `default` = TemporaryDataStore(type: .reader)
  24     class func oneTimeEditor() -> TemporaryDataStore {
  25         return TemporaryDataStore(type: .editor)
  26     }
  27
  28     required init(type: CoreDataManagerType) {
  29         managedObjectContext =
  30             type == .reader ? core.parentContext : core.editorContext()
  31     }
  32     deinit {
  33         saveActionCore()
  34     }
  35
  36     let core = CoreDataCore.temporary
  37     let managedObjectContext: NSManagedObjectContext
  38 }
  39
  40 extension TemporaryDataStore {
  41     func battle() -> Battle? {
  42         return battles().first
  43     }
  44     func battles() -> [Battle] {
  45         guard let battles = try? self.objects(with: Battle.entity)
  46             else { return [] }
  47         return battles
  48     }
  49     func resetBattle() {
  50         battles().forEach { delete($0) }
  51     }
  52     func createBattle() -> Battle? {
  53         return insertNewObject(for: Battle.entity)
  54     }
  55
  56     func sortedDamagesById() -> [Damage] {
  57         let sortDescriptor = NSSortDescriptor(key: "id", ascending: true)
  58         guard let damages = try? objects(with: Damage.entity, sortDescriptors: [sortDescriptor])
  59             else { return [] }
  60         return damages
  61     }
  62     func damages() -> [Damage] {
  63         guard let damages = try? objects(with: Damage.entity)
  64             else { return [] }
  65         return damages
  66     }
  67     func createDamage() -> Damage? {
  68         return insertNewObject(for: Damage.entity)
  69     }
  70
  71     func guardEscaped() -> [GuardEscaped] {
  72         guard let escapeds = try? objects(with: GuardEscaped.entity)
  73             else { return [] }
  74         return escapeds
  75     }
  76     func ensuredGuardEscaped(byShipId shipId: Int) -> GuardEscaped? {
  77         let p = NSPredicate(format: "shipID = %ld AND ensured = TRUE", shipId)
  78         guard let escapes = try? objects(with: GuardEscaped.entity, predicate: p)
  79             else { return nil }
  80         return escapes.first
  81     }
  82     func notEnsuredGuardEscaped() -> [GuardEscaped] {
  83         let predicate = NSPredicate(format: "ensured = FALSE")
  84         guard let escapeds = try? objects(with: GuardEscaped.entity, predicate: predicate)
  85             else { return [] }
  86         return escapeds
  87     }
  88     func createGuardEscaped() -> GuardEscaped? {
  89         return insertNewObject(for: GuardEscaped.entity)
  90     }
  91 }