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         context = (type == .reader ? core.parentContext : core.editorContext())
  30     }
  31     deinit {
  32         save()
  33     }
  34
  35     let core = CoreDataCore.temporary
  36     let context: NSManagedObjectContext
  37 }
  38
  39 extension TemporaryDataStore {
  40     func battle() -> Battle? {
  41         return battles().first
  42     }
  43     func battles() -> [Battle] {
  44         guard let battles = try? self.objects(with: Battle.entity)
  45             else { return [] }
  46         return battles
  47     }
  48     func resetBattle() {
  49         battles().forEach { delete($0) }
  50     }
  51     func createBattle() -> Battle? {
  52         return insertNewObject(for: Battle.entity)
  53     }
  54
  55     func sortedDamagesById() -> [Damage] {
  56         let sortDescriptor = NSSortDescriptor(key: "id", ascending: true)
  57         guard let damages = try? objects(with: Damage.entity, sortDescriptors: [sortDescriptor])
  58             else { return [] }
  59         return damages
  60     }
  61     func damages() -> [Damage] {
  62         guard let damages = try? objects(with: Damage.entity)
  63             else { return [] }
  64         return damages
  65     }
  66     func createDamage() -> Damage? {
  67         return insertNewObject(for: Damage.entity)
  68     }
  69
  70     func guardEscaped() -> [GuardEscaped] {
  71         guard let escapeds = try? objects(with: GuardEscaped.entity)
  72             else { return [] }
  73         return escapeds
  74     }
  75     func ensuredGuardEscaped(byShipId shipId: Int) -> GuardEscaped? {
  76         let p = NSPredicate(format: "shipID = %ld AND ensured = TRUE", shipId)
  77         guard let escapes = try? objects(with: GuardEscaped.entity, predicate: p)
  78             else { return nil }
  79         return escapes.first
  80     }
  81     func notEnsuredGuardEscaped() -> [GuardEscaped] {
  82         let predicate = NSPredicate(format: "ensured = FALSE")
  83         guard let escapeds = try? objects(with: GuardEscaped.entity, predicate: predicate)
  84             else { return [] }
  85         return escapeds
  86     }
  87     func createGuardEscaped() -> GuardEscaped? {
  88         return insertNewObject(for: GuardEscaped.entity)
  89     }
  90 }