Refactor SmartFeedController to find SmartFeeds by FeedIdentifier instead of the string identifier.

This commit is contained in:
Maurice Parker
2019-11-23 12:30:18 -06:00
parent 01f86d8c1b
commit 6d18cfec7c
2 changed files with 20 additions and 15 deletions

View File

@@ -24,14 +24,19 @@ final class SmartFeedsController: DisplayNameProvider {
self.smartFeeds = [todayFeed, unreadFeed, starredFeed]
}
func find(by identifier: String) -> PseudoFeed? {
func find(by identifier: FeedIdentifier) -> PseudoFeed? {
switch identifier {
case String(describing: TodayFeedDelegate.self):
return todayFeed
case String(describing: UnreadFeed.self):
return unreadFeed
case String(describing: StarredFeedDelegate.self):
return starredFeed
case .smartFeed(let stringIdentifer):
switch stringIdentifer {
case String(describing: TodayFeedDelegate.self):
return todayFeed
case String(describing: UnreadFeed.self):
return unreadFeed
case String(describing: StarredFeedDelegate.self):
return starredFeed
default:
return nil
}
default:
return nil
}