Count a number into 1 digit value

counting_1_digit.swift

import Foundation
import UIKit

var number = 55555

// means 5+5+5+5+5 = 30 , so 3+0 = 3

func countDigits(_ digit: Int) -> Int {
    
    var nums = digit
    
    if nums < 0 {
        nums = abs(nums)
    }
    
    var arrayInt = [Int]()
    
    arrayInt.append(nums % 10)

    while nums >= 10 {
        nums = nums / 10
        arrayInt.insert(nums % 10, at: 0)
    }

    var total = 0
    for n in arrayInt {
        total += n
    }
    
    if total > 9 {
        return countDigits(total)
    } else {
        return total
    }
    
}

print("TOTAL INTO 1 DIGIT = ", countDigits(number))

// --  TOTAL INTO 1 DIGIT =  3