Efficient digital root

digitalRoot.swift

import UIKit

func digitalRoot(of number: Int) -> Int {
    if (number <= 0) {
        return number
    } else if (number % 9 == 0) {
        return 9
    } else {
        return number % 9
    }
}