Date+TimeOfDay.swift

Date+TimeOfDay.swift

//
//  Date+TimeOfDay.swift
//  Created by: Jader Feijo
//

import Foundation

extension Date {
	enum TimeOfDay {
		case morning
		case afternoon
		case evening

		var description: String {
			switch self {
			case .morning:
				return "Morning"
			case .afternoon:
				return "Afternoon"
			case .evening:
				return "Evening"
			}
		}
	}

	var timeOfDay: TimeOfDay {
		if hour >= 3 && hour < 12 {
			return .morning
		} else if hour >= 12 && hour < 18 {
			return .afternoon
		} else {
			return .evening
		}
	}
}