metatext/View Controllers/NotificationsViewController.swift
2021-03-02 22:55:35 -08:00

130 lines
5.2 KiB
Swift

// Copyright © 2021 Metabolist. All rights reserved.
import Combine
import Mastodon
import UIKit
import ViewModels
final class NotificationsViewController: UIPageViewController {
private let segmentedControl = UISegmentedControl(items: [
NSLocalizedString("notifications.all", comment: ""),
NSLocalizedString("notifications.mentions", comment: "")
])
private let notificationViewControllers: [TableViewController]
private let viewModel: NavigationViewModel
private let rootViewModel: RootViewModel
private var cancellables = Set<AnyCancellable>()
init(viewModel: NavigationViewModel, rootViewModel: RootViewModel) {
self.viewModel = viewModel
self.rootViewModel = rootViewModel
var excludingAllExceptMentions = Set(MastodonNotification.NotificationType.allCasesExceptUnknown)
excludingAllExceptMentions.remove(.mention)
notificationViewControllers = [
TableViewController(viewModel: viewModel.notificationsViewModel(excludeTypes: []),
rootViewModel: rootViewModel),
TableViewController(viewModel: viewModel.notificationsViewModel(excludeTypes: excludingAllExceptMentions),
rootViewModel: rootViewModel)
]
super.init(transitionStyle: .scroll,
navigationOrientation: .horizontal,
options: [.interPageSpacing: CGFloat.defaultSpacing])
if let firstViewController = notificationViewControllers.first {
setViewControllers([firstViewController], direction: .forward, animated: false)
}
tabBarItem = NavigationViewModel.Tab.notifications.tabBarItem
}
@available(*, unavailable)
required init?(coder: NSCoder) {
fatalError("init(coder:) has not been implemented")
}
override func viewDidLoad() {
super.viewDidLoad()
dataSource = self
delegate = self
navigationItem.titleView = segmentedControl
segmentedControl.selectedSegmentIndex = 0
segmentedControl.addAction(
UIAction { [weak self] _ in
guard let self = self,
let currentViewController = self.viewControllers?.first as? TableViewController,
let currentIndex = self.notificationViewControllers.firstIndex(of: currentViewController),
self.segmentedControl.selectedSegmentIndex != currentIndex
else { return }
self.setViewControllers(
[self.notificationViewControllers[self.segmentedControl.selectedSegmentIndex]],
direction: self.segmentedControl.selectedSegmentIndex > currentIndex ? .forward : .reverse,
animated: !UIAccessibility.isReduceMotionEnabled)
},
for: .valueChanged)
}
}
extension NotificationsViewController: NavigationHandling {
func handle(navigation: Navigation) {
switch navigation {
case .notification:
guard let firstViewController = notificationViewControllers.first else { return }
segmentedControl.selectedSegmentIndex = 0
setViewControllers([firstViewController], direction: .reverse, animated: false)
firstViewController.handle(navigation: navigation)
default:
(viewControllers?.first as? TableViewController)?.handle(navigation: navigation)
}
}
}
extension NotificationsViewController: UIPageViewControllerDataSource {
func pageViewController(_ pageViewController: UIPageViewController,
viewControllerAfter viewController: UIViewController) -> UIViewController? {
guard
let viewController = viewController as? TableViewController,
let index = notificationViewControllers.firstIndex(of: viewController),
index + 1 < notificationViewControllers.count
else { return nil }
return notificationViewControllers[index + 1]
}
func pageViewController(_ pageViewController: UIPageViewController,
viewControllerBefore viewController: UIViewController) -> UIViewController? {
guard
let viewController = viewController as? TableViewController,
let index = notificationViewControllers.firstIndex(of: viewController),
index > 0
else { return nil }
return notificationViewControllers[index - 1]
}
}
extension NotificationsViewController: UIPageViewControllerDelegate {
func pageViewController(_ pageViewController: UIPageViewController,
didFinishAnimating finished: Bool,
previousViewControllers: [UIViewController],
transitionCompleted completed: Bool) {
guard let viewController = viewControllers?.first as? TableViewController,
let index = notificationViewControllers.firstIndex(of: viewController)
else { return }
segmentedControl.selectedSegmentIndex = index
}
}
extension NotificationsViewController: ScrollableToTop {
func scrollToTop(animated: Bool) {
(viewControllers?.first as? TableViewController)?.scrollToTop(animated: animated)
}
}