KCD/HistoryTableViewController.swift

   1 //
   2 //  HistoryTableViewController.swift
   3 //  KCD
   4 //
   5 //  Created by Hori,Masaki on 2017/03/10.
   6 //  Copyright © 2017年 Hori,Masaki. All rights reserved.
   7 //
   8
   9 import Cocoa
  10
  11 private protocol Markable {
  12
  13     var mark: Bool { get set }
  14 }
  15
  16 private protocol HistoryObject {
  17
  18     var date: Date { get }
  19 }
  20
  21 extension DropShipHistory: Markable, HistoryObject {}
  22 extension KenzoHistory: Markable, HistoryObject {}
  23 extension KaihatuHistory: Markable, HistoryObject {}
  24
  25 enum MenuItemTag: Int {
  26
  27     case delete = 314
  28
  29     case addMark = 50000
  30 }
  31
  32 class HistoryTableViewController: NSViewController {
  33
  34     // Subbclass MUST override these.
  35     var predicateFormat: String { fatalError("Subbclass MUST implement.") }
  36     func objects(of predicate: NSPredicate?, in store: LocalDataStore) throws -> [NSManagedObject] {
  37
  38         fatalError("Subbclass MUST implement.")
  39     }
  40
  41     @IBOutlet var controller: NSArrayController!
  42     @IBOutlet var tableView: NSTableView!
  43
  44     @IBAction func delete(_ sender: AnyObject?) {
  45
  46         let store = LocalDataStore.oneTimeEditor()
  47
  48         guard let controller = controller else { return }
  49         guard let selection = controller.selectedObjects as? [NSManagedObject] else { return }
  50
  51         let selectedIndex = controller.selectionIndex
  52         selection
  53             .map { $0.objectID }
  54             .map(store.object(with:))
  55             .forEach(store.delete)
  56
  57         if selectedIndex > 1 {
  58
  59             controller.setSelectionIndex(selectedIndex - 1)
  60         }
  61     }
  62
  63     @IBAction func addMark(_ sender: AnyObject?) {
  64
  65         guard let clickedRow = tableView?.clickedRow else { return }
  66         guard let items = controller?.arrangedObjects as? [HistoryObject] else { return }
  67         guard case 0..<items.count = clickedRow else { return }
  68
  69         let clickedObject = items[clickedRow]
  70         let predicate = NSPredicate(#keyPath(KenzoMark.date), equal: clickedObject.date)
  71
  72         let store = LocalDataStore.oneTimeEditor()
  73
  74         if let items = try? objects(of: predicate, in: store),
  75             var history = items.first as? Markable {
  76
  77             history.mark = !history.mark
  78         }
  79
  80         store.save(errorHandler: store.presentOnMainThread)
  81     }
  82
  83     override func validateMenuItem(_ menuItem: NSMenuItem) -> Bool {
  84
  85         guard let itemTag = MenuItemTag(rawValue: menuItem.tag) else { return false }
  86
  87         switch itemTag {
  88         case .delete:
  89             return controller.selectionIndex != -1
  90
  91         case .addMark:
  92             menuItem.isEnabled = false
  93             guard let clickedRow = tableView?.clickedRow else { return false }
  94             guard let items = controller?.arrangedObjects as? [Markable] else { return false }
  95             guard case 0..<items.count = clickedRow else { return false }
  96
  97             let clickedObject = items[clickedRow]
  98             menuItem.isEnabled = true
  99             if clickedObject.mark {
 100
 101                 menuItem.title = LocalizedStrings.removeMark.string
 102
 103             } else {
 104
 105                 menuItem.title = LocalizedStrings.addMark.string
 106             }
 107         }
 108
 109         return true
 110     }
 111 }
 112
 113 extension HistoryTableViewController: NSTableViewDelegate {
 114
 115     func tableView(_ tableView: NSTableView, viewFor tableColumn: NSTableColumn?, row: Int) -> NSView? {
 116
 117         return tableColumn
 118             .flatMap { $0.identifier }
 119             .flatMap { tableView.makeView(withIdentifier: $0, owner: nil) }
 120     }
 121 }
 122
 123 final class KaihatsuHistoryTableViewController: HistoryTableViewController {
 124
 125     override var predicateFormat: String { return "name contains $value" }
 126     override func objects(of predicate: NSPredicate?, in store: LocalDataStore) throws -> [NSManagedObject] {
 127
 128         return try store.objects(of: KaihatuHistory.entity, predicate: predicate)
 129     }
 130 }
 131
 132 final class KenzoHistoryTableViewController: HistoryTableViewController {
 133
 134     override var predicateFormat: String { return "name contains $value" }
 135     override func objects(of predicate: NSPredicate?, in store: LocalDataStore) throws -> [NSManagedObject] {
 136
 137         return try store.objects(of: KenzoHistory.entity, predicate: predicate)
 138     }
 139 }
 140
 141 final class DropShipHistoryTableViewController: HistoryTableViewController {
 142
 143     override var predicateFormat: String { return "shipName contains $value" }
 144     override func objects(of predicate: NSPredicate?, in store: LocalDataStore) throws -> [NSManagedObject] {
 145
 146         return try store.objects(of: DropShipHistory.entity, predicate: predicate)
 147     }
 148 }