쥬스메이커 [STEP 2] 제이티, Ari

JuiceMaker/JuiceMaker/Controller/JuiceMakerViewController.swift

@@ -8,11 +8,128 @@ import UIKit
 
 class JuiceMakerViewController: UIViewController {
 
+ private let juiceMaker = JuiceMaker()
+
+ @IBOutlet weak var strawberryStockLabel: UILabel!
+ @IBOutlet weak var bananaStockLabel: UILabel!
+ @IBOutlet weak var pineappleStockLabel: UILabel!
+ @IBOutlet weak var kiwiStockLabel: UILabel!
+ @IBOutlet weak var mangoStockLabel: UILabel!
+
+ @IBOutlet weak var orderStrawberryBananaJuiceButton: UIButton!
+ @IBOutlet weak var orderMangoKiwiJuiceButton: UIButton!
+ @IBOutlet weak var orderStrawberryJuiceButton: UIButton!
+ @IBOutlet weak var orderBananaJuiceButton: UIButton!
+ @IBOutlet weak var orderPineappleJuiceButton: UIButton!
+ @IBOutlet weak var orderKiwiJuiceButton: UIButton!
+ @IBOutlet weak var orderMangoJuiceButton: UIButton!
+
  override func viewDidLoad() {
  super.viewDidLoad()
- // Do any additional setup after loading the view.
+ updateFruitLabels()
+
+ NotificationCenter.default.addObserver(self, selector: #selector(fruitLabelChanged(notification:)), name: .changedFruitStockNotification, object: nil)
+ }
+
+ @IBAction func orderJuiceButtonTapped(_ sender: UIButton) {
+ let juice: Juice
+
+ switch sender {
+ case orderStrawberryBananaJuiceButton:
+ juice = .strawberryBanana
+ case orderMangoKiwiJuiceButton:
+ juice = .mangoKiwi
+ case orderStrawberryJuiceButton:
+ juice = .strawberry
+ case orderBananaJuiceButton:
+ juice = .banana
+ case orderPineappleJuiceButton:
+ juice = .pineapple
+ case orderKiwiJuiceButton:
+ juice = .kiwi
+ case orderMangoJuiceButton:
+ juice = .mango
+ default:
+ showNotificationAlert(message: "잘못된 접근입니다.")
+ return
+ }
+
+ mixFruit(juice: juice)
+ }
+
+ func mixFruit(juice: Juice) {
+ do {
+ try juiceMaker.mixFruit(juice: juice)
+ showNotificationAlert(message: "\(juice) 쥬스 나왔습니다! 맛있게 드세요!")
+ } catch RequestError.fruitStockOut {
+ showOutOfStockAlert()
+ } catch let error as RequestError {
+ showNotificationAlert(message: error.errorDescription)
+ } catch {
+ showNotificationAlert(message: "알 수 없는 에러가 발생했습니다.")
+ }
+ }
+
+ @objc func fruitLabelChanged(notification: Notification) {
+ guard let fruit = notification.object as? Fruit else {
+ showNotificationAlert(message: "알 수 없는 에러가 발생했습니다.")
+ return
+ }
+ currentStockLabelUpdate(fruit: fruit, label: fruitlabel(of: fruit))
+ }
+
+ func updateFruitLabels() {
+ currentStockLabelUpdate(fruit: .strawberry, label: strawberryStockLabel)
+ currentStockLabelUpdate(fruit: .banana, label: bananaStockLabel)
+ currentStockLabelUpdate(fruit: .pineapple, label: pineappleStockLabel)
+ currentStockLabelUpdate(fruit: .kiwi, label: kiwiStockLabel)
+ currentStockLabelUpdate(fruit: .mango, label: mangoStockLabel)
+ }
+
+ func currentStockLabelUpdate(fruit: Fruit, label: UILabel) {
+ do {
+ let stock = try FruitStore.shared.stock(fruit: fruit)
+ label.text = String(stock)
+ } catch let error as RequestError {
+ showNotificationAlert(message: error.errorDescription)
+ } catch {
+ showNotificationAlert(message: "알 수 없는 에러가 발생했습니다.")
+ }
+ }
+
+ func fruitlabel(of fruit: Fruit) -> UILabel {
+ switch fruit {
+ case .strawberry:
+ return strawberryStockLabel
+ case .banana:
+ return bananaStockLabel
+ case .pineapple:
+ return pineappleStockLabel
+ case .kiwi:
+ return kiwiStockLabel
+ case .mango:
+ return mangoStockLabel
+ }
+ }
+
+ func showNotificationAlert(message: String, title: String = "OK") {
+ let alert = UIAlertController(title: nil, message: message, preferredStyle: .alert)
+ let cancel = UIAlertAction(title: title, style: .cancel, handler: nil)
+ alert.addAction(cancel)
+ present(alert, animated: true, completion: nil)
+ }
+
+ func showOutOfStockAlert() {
+ let alert = UIAlertController(title: nil, message: "재료가 모자라요. 재고를 수정할까요?", preferredStyle: .alert)
+ let cancelAction = UIAlertAction(title: "Cancel", style: .cancel, handler: nil)
+ let okAction = UIAlertAction(title: "OK", style: .default) { _ in
+ guard let viewController = self.storyboard?.instantiateViewController(withIdentifier: "FruitStoreView") else { return }
+ self.present(viewController, animated: true, completion: nil)
+ }
+ alert.addAction(cancelAction)
+ alert.addAction(okAction)
+ present(alert, animated: true, completion: nil)
  }
-
 
 }
 

JuiceMaker/JuiceMaker/Model/FruitStore.swift

@@ -8,18 +8,16 @@ import Foundation
 
 private let defaultFruitCount = 10
 
+enum Fruit: CaseIterable {
+ case strawberry
+ case banana
+ case pineapple
+ case kiwi
+ case mango
+}
+
 class FruitStore {
-
  static let shared: FruitStore = FruitStore()
-
- enum Fruit: CaseIterable {
- case strawberry
- case banana
- case pineapple
- case kiwi
- case mango
- }
-
  private var fruitBasket: [Fruit: Int]
 
  init(count: Int = defaultFruitCount) {
@@ -29,6 +27,13 @@ class FruitStore {
  self.fruitBasket = Dictionary(uniqueKeysWithValues: zip(allFruits, fruitscount))
  }
 
+ func stock(fruit: Fruit) throws -> Int {
+ guard let fruitStock = fruitBasket[fruit] else {
+ throw RequestError.fruitNotFound
+ }
+ return fruitStock
+ }
+
  func addFruitStock(fruit: Fruit, count: Int) throws {
  try changeAmount(count: count, of: fruit, by: +)
  }
@@ -49,6 +54,8 @@ class FruitStore {
  }
  let newFruitCount = calculator(oldFruitCount, count)
  fruitBasket[fruit] = newFruitCount
+
+ NotificationCenter.default.post(name: .changedFruitStockNotification, object: fruit)
  }
 
  func hasFruitStock(of fruit: Fruit, count fruitCountSubtracted: Int) -> Bool {
@@ -58,3 +65,6 @@ class FruitStore {
 }
 
 
+extension Notification.Name {
+ static let changedFruitStockNotification = Notification.Name("changeFruitStock")
+}

JuiceMaker/JuiceMaker/Model/JuiceMaker.swift

@@ -6,16 +6,20 @@
 
 import Foundation
 
-fileprivate typealias Recipe = [FruitStore.Fruit: Int]
+fileprivate typealias Recipe = [Fruit: Int]
 
-enum Juice {
- case strawberry
- case banana
- case kiwi
- case pineapple
- case strawberryBanana
- case mango
- case mangoKiwi
+enum Juice: String, CustomStringConvertible {
+ case strawberry = "딸기"
+ case banana = "바나나"
+ case kiwi = "키위"
+ case pineapple = "파인애플"
+ case strawberryBanana = "딸바"
+ case mango = "망고"
+ case mangoKiwi = "망키"
+
+ var description: String {
+ return rawValue
+ }
 
  fileprivate var recipe: Recipe {
  switch self {

JuiceMaker/JuiceMaker/Model/RequestError.swift

@@ -7,7 +7,7 @@
 
 import Foundation
 
-enum RequestError: Error, LocalizedError {
+enum RequestError: Error {
  case wrongCount
  case fruitNotFound
  case fruitStockOut