diff --git a/Apps/Locations/Sources/Screens/LocationsList/LocationsListViewModel.swift b/Apps/Locations/Sources/Screens/LocationsList/LocationsListViewModel.swift index 7b68701..ebf3e9d 100644 --- a/Apps/Locations/Sources/Screens/LocationsList/LocationsListViewModel.swift +++ b/Apps/Locations/Sources/Screens/LocationsList/LocationsListViewModel.swift @@ -48,6 +48,7 @@ extension LocationsListViewModel: LocationsListViewModeling { // MARK: Properties var viewStatusPublisher: Published.Publisher { $viewStatus } + var numberOfSectionsInData: Int { fetchedResultsController.sections?.count ?? 0 } // MARK: Functions @@ -71,6 +72,21 @@ extension LocationsListViewModel: LocationsListViewModeling { } } + func numberOfDataItems(in section: Int) -> Int { + guard + let sections = fetchedResultsController.sections, + sections.endIndex > section + else { + return 0 + } + + return sections[section].numberOfObjects + } + + func dataItem(at indexPath: IndexPath) -> Location { + fetchedResultsController.object(at: indexPath) + } + } // MARK: - Enumerations