yattee/Extensions/Array+Next.swift
2021-06-28 12:43:07 +02:00

14 lines
303 B
Swift

extension Array where Element: Equatable {
func next(after element: Element) -> Element? {
let idx = firstIndex(of: element)
if idx == nil {
return first
}
let next = index(after: idx!)
return self[next == endIndex ? startIndex : next]
}
}