yattee/Extensions/CaseIterable+Next.swift

9 lines
263 B
Swift
Raw Normal View History

extension CaseIterable where Self: Equatable {
func next() -> Self {
let all = Self.allCases
let idx = all.firstIndex(of: self)!
let next = all.index(after: idx)
return all[next == all.endIndex ? all.startIndex : next]
}
}