DateIntervalOperators/DateIntervalOperators.swift at master · tschmitz/DateIntervalOperators · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
//
//  DateIntervalOperators.swift
//  SyncTime
//
//  Created by Tim Schmitz on 2/2/15.
//  Copyright (c) 2015 Tap and Tonic. All rights reserved.
//

import Foundation

public extension Int {
	public func day() -> NSDateComponents {
		let components = NSDateComponents()
		components.day = self
		return components
	}
	public func days() -> NSDateComponents { return self.day() }

	public func week() -> NSDateComponents {
		let components = NSDateComponents()
		components.weekOfYear = self
		return components
	}
	public func weeks() -> NSDateComponents { return self.week() }

	public func month() -> NSDateComponents {
		let components = NSDateComponents()
		components.month = self
		return components
	}
	public func months() -> NSDateComponents { return self.month() }

	public func year() -> NSDateComponents {
		let components = NSDateComponents()
		components.year = self
		return components
	}
	public func years() -> NSDateComponents { return self.year() }

	public func second() -> NSDateComponents {
		let components = NSDateComponents()
		components.second = self
		return components
	}
	public func seconds() -> NSDateComponents { return self.second() }

	public func minute() -> NSDateComponents {
		let components = NSDateComponents()
		components.minute = self
		return components
	}
	public func minutes() -> NSDateComponents { return self.minute() }

	public func hour() -> NSDateComponents {
		let components = NSDateComponents()
		components.hour = self
		return components
	}
	public func hours() -> NSDateComponents { return self.hour() }
}

public func +(lhs: NSDate, rhs: NSDateComponents) -> NSDate {
	return NSCalendar.currentCalendar().dateByAddingComponents(rhs, toDate: lhs, options: nil) ?? lhs
}

public func -(lhs: NSDate, rhs: NSDateComponents) -> NSDate {
	if rhs.era != Int(NSUndefinedDateComponent) { rhs.era *= -1 }
	if rhs.year != Int(NSUndefinedDateComponent) { rhs.year *= -1 }
	if rhs.month != Int(NSUndefinedDateComponent) { rhs.month *= -1 }
	if rhs.day != Int(NSUndefinedDateComponent) { rhs.day *= -1 }
	if rhs.hour != Int(NSUndefinedDateComponent) { rhs.hour *= -1 }
	if rhs.minute != Int(NSUndefinedDateComponent) { rhs.minute *= -1 }
	if rhs.second != Int(NSUndefinedDateComponent) { rhs.second *= -1 }
	if rhs.nanosecond != Int(NSUndefinedDateComponent) { rhs.nanosecond *= -1 }
	if rhs.weekday != Int(NSUndefinedDateComponent) { rhs.weekday *= -1 }
	if rhs.weekdayOrdinal != Int(NSUndefinedDateComponent) { rhs.weekdayOrdinal *= -1 }
	if rhs.quarter != Int(NSUndefinedDateComponent) { rhs.quarter *= -1 }
	if rhs.weekOfMonth != Int(NSUndefinedDateComponent) { rhs.weekOfMonth *= -1 }
	if rhs.weekOfYear != Int(NSUndefinedDateComponent) { rhs.weekOfYear *= -1 }
	if rhs.yearForWeekOfYear != Int(NSUndefinedDateComponent) { rhs.yearForWeekOfYear *= -1 }

	return NSCalendar.currentCalendar().dateByAddingComponents(rhs, toDate: lhs, options: nil) ?? lhs
}