Device

Device.swift

struct Device {
    let ran: Bool
    
    @discardableResult static func when4sOrSmaller(_ block: () -> Void) -> Device {
        return runBlockIfTrue(UIScreen.is4sOrSmaller, block: block)
    }
    
    @discardableResult static func when5sOrSmaller(_ block: () -> Void) -> Device {
        return runBlockIfTrue(UIScreen.is5sOrSmaller, block: block)
    }
    
    func otherwise(_ block: () -> Void) {
        if !ran {
            block()
        }
    }
    
    static fileprivate func runBlockIfTrue(_ fn: () -> Bool, block: () -> Void) -> Device {
        let run = fn()
        if run {
            block()
        }
        return Device(ran: run)
    }
}