metatext/Extensions/CollectionSection+Extensions.swift

28 lines
825 B
Swift
Raw Permalink Normal View History

2021-01-23 06:15:52 +00:00
// Copyright © 2021 Metabolist. All rights reserved.
import UIKit
import ViewModels
extension CollectionSection {
struct Identifier: Hashable {
let index: Int
2021-01-31 01:43:48 +00:00
let searchScope: SearchScope?
2021-01-23 06:15:52 +00:00
}
}
extension Array where Element == CollectionSection {
func snapshot() -> NSDiffableDataSourceSnapshot<CollectionSection.Identifier, CollectionItem> {
var snapshot = NSDiffableDataSourceSnapshot<CollectionSection.Identifier, CollectionItem>()
for (index, section) in enumerated() {
let identifier = CollectionSection.Identifier(
index: index,
2021-01-31 01:43:48 +00:00
searchScope: section.searchScope)
2021-01-23 06:15:52 +00:00
snapshot.appendSections([identifier])
snapshot.appendItems(section.items, toSection: identifier)
}
return snapshot
}
}