//

// SwiftyTimer

//

// Copyright (c) 2015-2016 Radosław Pietruszewski

//

// Permission is hereby granted, free of charge, to any person obtaining a copy

// of this software and associated documentation files (the "Software"), to deal

// in the Software without restriction, including without limitation the rights

// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

// copies of the Software, and to permit persons to whom the Software is

// furnished to do so, subject to the following conditions:

//

// The above copyright notice and this permission notice shall be included in all

// copies or substantial portions of the Software.

//

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

// SOFTWARE.

//

import Foundation

import ObjectiveC

extension Timer {

// MARK: Schedule timers

private struct AssociatedKeys {

static var repeatCounterAddress = "repeat_counter_address"

static var isTimerRunning = "is_timer_running_address"

}

public var isRunning: Bool {

get {

return isValid && objc_getAssociatedObject(self, &AssociatedKeys.isTimerRunning) as? Bool ?? false

}

set {

objc_setAssociatedObject(self,

&AssociatedKeys.isTimerRunning,

newValue,

.OBJC_ASSOCIATION_RETAIN_NONATOMIC)

}

}

private var repeatCounter: Int {

get {

return objc_getAssociatedObject(self, &AssociatedKeys.repeatCounterAddress) as? Int ?? 0

}

set {

objc_setAssociatedObject(self,

&AssociatedKeys.repeatCounterAddress,

newValue,

.OBJC_ASSOCIATION_RETAIN_NONATOMIC)

}

}

public class func every(_ interval: TimeInterval, `for` times: Int, _ block: @escaping (Timer, Int) -> Void) -> Timer {

let timer = Timer.new(every: interval, for: times, block)

timer.start()

return timer

}

/// Create and schedule a timer that will call `block` once after the specified time.

@discardableResult

public class func after(_ interval: TimeInterval, _ block: @escaping () -> Void) -> Timer {

let timer = Timer.new(after: interval, block)

timer.start()

return timer

}

/// Create and schedule a timer that will call `block` repeatedly in specified time intervals.

@discardableResult

public class func every(_ interval: TimeInterval, _ block: @escaping () -> Void) -> Timer {

let timer = Timer.new(every: interval, block)

timer.start()

return timer

}

/// Create and schedule a timer that will call `block` repeatedly in specified time intervals.

/// (This variant also passes the timer instance to the block)

@nonobjc @discardableResult

public class func every(_ interval: TimeInterval, _ block: @escaping (Timer) -> Void) -> Timer {

let timer = Timer.new(every: interval, block)

timer.start()

return timer

}

// MARK: Create timers without scheduling

/// Create a timer that will call `block` once after the specified time.

///

/// - Note: The timer won't fire until it's scheduled on the run loop.

/// Use `NSTimer.after` to create and schedule a timer in one step.

/// - Note: The `new` class function is a workaround for a crashing bug when using convenience initializers (rdar://18720947)

public class func new(after interval: TimeInterval, _ block: @escaping () -> Void) -> Timer {

return CFRunLoopTimerCreateWithHandler(kCFAllocatorDefault, CFAbsoluteTimeGetCurrent() + interval, 0, 0, 0) { timer in

(timer as! Timer).isRunning = false

block()

}

}

/// Create a timer that will call `block` repeatedly in specified time intervals.

///

/// - Note: The timer won't fire until it's scheduled on the run loop.

/// Use `NSTimer.every` to create and schedule a timer in one step.

/// - Note: The `new` class function is a workaround for a crashing bug when using convenience initializers (rdar://18720947)

public class func new(every interval: TimeInterval, _ block: @escaping () -> Void) -> Timer {

return CFRunLoopTimerCreateWithHandler(kCFAllocatorDefault, CFAbsoluteTimeGetCurrent() + interval, interval, 0, 0) { _ in

block()

}

}

@nonobjc public class func new(every interval: TimeInterval, `for` times: Int, _ block: @escaping () -> Void) -> Timer {

var timer: Timer!

timer = CFRunLoopTimerCreateWithHandler(kCFAllocatorDefault, CFAbsoluteTimeGetCurrent() + interval, interval, 0, 0) { _ in

if times > 0 {

if timer.repeatCounter == (times-1) {

timer.invalidate()

}

timer.repeatCounter += 1

}

block()

}

return timer

}

/// Create a timer that will call `block` repeatedly in specified time intervals.

/// (This variant also passes the timer instance to the block)

///

/// - Note: The timer won't fire until it's scheduled on the run loop.

/// Use `NSTimer.every` to create and schedule a timer in one step.

/// - Note: The `new` class function is a workaround for a crashing bug when using convenience initializers (rdar://18720947)

@nonobjc public class func new(every interval: TimeInterval, `for` times: Int, _ block: @escaping (Timer, Int) -> Void) -> Timer {

var timer: Timer!

timer = CFRunLoopTimerCreateWithHandler(kCFAllocatorDefault, CFAbsoluteTimeGetCurrent() + interval, interval, 0, 0) { _ in

if times > 0 {

if timer.repeatCounter == (times-1) {

timer.invalidate()

}

timer.repeatCounter += 1

}

block(timer, timer.repeatCounter)

}

return timer

}

@nonobjc public class func new(every interval: TimeInterval, _ block: @escaping (Timer) -> Void) -> Timer {

var timer: Timer!

timer = CFRunLoopTimerCreateWithHandler(kCFAllocatorDefault, CFAbsoluteTimeGetCurrent() + interval, interval, 0, 0) { _ in

block(timer)

}

return timer

}

// MARK: Manual scheduling

/// Schedule this timer on the run loop

///

/// By default, the timer is scheduled on the current run loop for the default mode.

/// Specify `runLoop` or `modes` to override these defaults.

public func start(runLoop: RunLoop = .current, modes: RunLoopMode...) {

isRunning = true

let modes = modes.isEmpty ? [.defaultRunLoopMode] : modes

for mode in modes {

runLoop.add(self, forMode: mode)

}

}

}

// MARK: - Time extensions

public extension Double {

public var millisecond: TimeInterval { return self / 1000 }

public var milliseconds: TimeInterval { return self / 1000 }

public var ms: TimeInterval { return self / 1000 }

public var second: TimeInterval { return self }

public var seconds: TimeInterval { return self }

public var minute: TimeInterval { return self * 60 }

public var minutes: TimeInterval { return self * 60 }

public var hour: TimeInterval { return self * 3600 }

public var hours: TimeInterval { return self * 3600 }

public var day: TimeInterval { return self * 3600 * 24 }

public var days: TimeInterval { return self * 3600 * 24 }

}