import Foundation
import UIKit

// 利用協議優化通知
protocol Notifier {
    // 添加一個關聯的類型
    associatedtype Notification: RawRepresentable
}

extension Notifier where Notification.RawValue == String {
    
    static func nameFor(notification: Notification) -> String {
        
        return "\(notification.rawValue)"
    }
}

class Barista: Notifier {
    
    /// 發送通知
    static func post(notification: Notification, object:AnyObject? = nil) {
        
        let name = nameFor(notification: notification)
        NotificationCenter.default.post(name: NSNotification.Name(rawValue: name), object: object)
    }
    
    /// 增加觀察 - 接收通知
    static func add(observer: AnyObject, selector: Selector, notification: Notification, object:AnyObject? = nil) {
        
        let name = nameFor(notification: notification)
        NotificationCenter.default
            .addObserver(observer, selector: selector, name: NSNotification.Name(rawValue: name), object: object)
    }
    
    /// 移除觀察 - 移除通知
    static func remove(observer: AnyObject, notification: Notification, object:AnyObject? = nil) {
        
        let name = nameFor(notification: notification)
        NotificationCenter.default.removeObserver(observer, name: NSNotification.Name(rawValue: name), object: object)
    }
}

// 定義的通知名字
extension Barista {
    enum Notification: String {
        /// 開心
        case happy
        /// 傷心
        case sad
        /// 睡覺
        case sleep
    }
}

/*
 * 添加: _ = Barista.add(observer: self, selector: #selector(reload), notification: .happy)
 * 移除: Barista.remove(observer: self, notification: .happy)
 * 發出通知: Barista.post(notification: .happy)
 */