element-ios/Riot/Routers/TabRouters/SegmentedController.swift

116 lines
3.9 KiB
Swift

//
// Copyright 2021-2024 New Vector Ltd.
//
// SPDX-License-Identifier: AGPL-3.0-only
// Please see LICENSE in the repository root for full details.
//
import Foundation
struct SegmentedControllerTab {
let title: String?
let viewController: UIViewController
}
class SegmentedController: UIViewController, Themable {
// MARK: Outlets
@IBOutlet private var segmentedControl: UISegmentedControl!
@IBOutlet private var contentView: UIView!
// MARK: Private
private var currentController: UIViewController? {
didSet {
if let viewController = oldValue {
viewController.vc_removeFromParent(animated: true)
}
if let viewController = currentController {
vc_addChildViewController(viewController: viewController, onView: contentView, animated: true)
// Needed for swiftUI as navigation items are loaded while loading the view
DispatchQueue.main.asyncAfter(deadline: .now() + 0.3) {
self.navigationItem.rightBarButtonItem = viewController.navigationItem.rightBarButtonItem
self.navigationItem.rightBarButtonItems = viewController.navigationItem.rightBarButtonItems
self.navigationItem.leftBarButtonItem = viewController.navigationItem.leftBarButtonItem
self.navigationItem.leftBarButtonItems = viewController.navigationItem.leftBarButtonItems
if self.title == nil {
self.navigationItem.title = viewController.navigationItem.title
self.navigationItem.titleView = viewController.navigationItem.titleView
}
}
}
}
}
private var theme: Theme!
// MARK: Properties
var tabs: [SegmentedControllerTab] = [] {
didSet {
if isViewLoaded {
populateSegmentedControl()
}
}
}
// MARK: Setup
class func instantiate() -> SegmentedController {
let storyboard = UIStoryboard(name: "SegmentedController", bundle: nil)
let viewController = storyboard.instantiateInitialViewController() as? SegmentedController ?? SegmentedController()
viewController.theme = ThemeService.shared().theme
return viewController
}
// MARK: Lifecycle
override func viewDidLoad() {
super.viewDidLoad()
populateSegmentedControl()
update(theme: theme)
registerThemeServiceDidChangeThemeNotification()
}
// MARK: Actions
@IBAction private func segmentDidChange(sender: UISegmentedControl) {
self.currentController = tabs[sender.selectedSegmentIndex].viewController
}
// MARK: Themable
private func registerThemeServiceDidChangeThemeNotification() {
NotificationCenter.default.addObserver(self, selector: #selector(themeDidChange), name: .themeServiceDidChangeTheme, object: nil)
}
@objc private func themeDidChange() {
self.theme = ThemeService.shared().theme
self.update(theme: self.theme)
}
func update(theme: Theme) {
self.view.backgroundColor = theme.baseColor
self.contentView.backgroundColor = theme.baseColor
}
// MARK: Private
private func populateSegmentedControl() {
segmentedControl.removeAllSegments()
for tab in tabs {
let title = tab.title ?? tab.viewController.tabBarItem.title ?? tab.viewController.title
segmentedControl.insertSegment(withTitle: title, at: segmentedControl.numberOfSegments, animated: false)
}
if segmentedControl.numberOfSegments > 0 {
segmentedControl.selectedSegmentIndex = 0
self.currentController = tabs.first?.viewController
}
}
}