mirror of
https://github.com/yattee/yattee.git
synced 2024-12-14 14:20:32 +05:30
14 lines
303 B
Swift
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]
|
|
}
|
|
}
|