Improved notification

Notification2.swift

import UIKit

protocol NotificationScheduler {
  func scheduleNotification(with title: String)
}

protocol TimeProvider {
  var now: Date { get }
  func calendar(with identifier: Calendar.Identifier) -> Calendar 
}

class NotificationManager {
  let notificationScheduler: NotificationScheduler 
  let timeProvider: TimeProvider
  
  init(notificationScheduler: NotificationScheduler, timeProvider: TimeProvider) {
    self.notificationScheduler = notificationScheduler
    self.timeProvider = timeProvider
  }
  
  private func getCurrentHour() -> Int? {
    let calendar = self.timeProvider.calendar(identifier: .gregorian)
    let date = self.timeProvider.now
    let hour = calendar.component(Calendar.Component.hour, from: date)
    return hour
  }
  
  func scheduleMorningNotification() {
    guard 
      let hour = self.getCurrentHour,
      hour < 11
    else {
      return
    }
    let title = "Good Morning!"
    self.notificationScheduler.scheduleNotification(with: title)
  }
  
  func scheduleNoonNotification() {
    guard 
      let hour = self.getCurrentHour,
      hour > 11,
      hour < 14
    else {
      return
    }
    let title = "Enjoy your lunch!"
    self.notificationScheduler.scheduleNotification(with: title)
  }
  
  func scheduleDinnerNotification() {
    guard 
      let hour = self.getCurrentHour,
      hour > 19
    else {
      return
    }
    let title = "Enjoy your dinner!"
    self.notificationScheduler.scheduleNotification(with: title)
  }
}