diff --git a/Tusker/Screens/Asset Picker/AssetCollectionsListViewController.swift b/Tusker/Screens/Asset Picker/AssetCollectionsListViewController.swift index 1d351cecc6..2201cb2638 100644 --- a/Tusker/Screens/Asset Picker/AssetCollectionsListViewController.swift +++ b/Tusker/Screens/Asset Picker/AssetCollectionsListViewController.swift @@ -46,7 +46,7 @@ class AssetCollectionsListViewController: UITableViewController { }) var snapshot = NSDiffableDataSourceSnapshot() - snapshot.appendSections([.system, .albums, .smartAlbums]) + snapshot.appendSections([.system, .albums, .sharedAlbums, .smartAlbums]) snapshot.appendItems([.cameraRoll], toSection: .system) let smartAlbums = PHAssetCollection.fetchAssetCollections(with: .smartAlbum, subtype: .any, options: nil) @@ -61,12 +61,18 @@ class AssetCollectionsListViewController: UITableViewController { let albums = PHAssetCollection.fetchAssetCollections(with: .album, subtype: .any, options: nil) var albumItems = [Item]() + var sharedItems = [Item]() albums.enumerateObjects { (collection, _, _) in if collection.estimatedAssetCount > 0 { - albumItems.append(.album(collection)) + if collection.assetCollectionSubtype == .albumCloudShared { + sharedItems.append(.album(collection)) + } else { + albumItems.append(.album(collection)) + } } } snapshot.appendItems(albumItems, toSection: .albums) + snapshot.appendItems(sharedItems, toSection: .sharedAlbums) dataSource.apply(snapshot, animatingDifferences: false) @@ -103,6 +109,7 @@ extension AssetCollectionsListViewController { enum Section { case system case albums + case sharedAlbums case smartAlbums } enum Item: Hashable { @@ -121,12 +128,14 @@ extension AssetCollectionsListViewController { } class DataSource: UITableViewDiffableDataSource { override func tableView(_ tableView: UITableView, titleForHeaderInSection section: Int) -> String? { - let currentSnapshot = snapshot() - if currentSnapshot.indexOfSection(.albums) == section { + switch sectionIdentifier(for: section) { + case .albums: return NSLocalizedString("Albums", comment: "albums section title") - } else if currentSnapshot.indexOfSection(.smartAlbums) == section { + case .sharedAlbums: + return NSLocalizedString("Shared Albums", comment: "shared albums section title") + case .smartAlbums: return NSLocalizedString("Smart Albums", comment: "smart albums section title") - } else { + default: return nil } }