Tusker/Tusker/Screens/Main/MainTabBarViewController.swift

//
//  MainTabBarViewController.swift
//  Tusker
//
//  Created by Shadowfacts on 8/21/18.
//  Copyright © 2018 Shadowfacts. All rights reserved.
//

import UIKit

class MainTabBarViewController: UITabBarController, UITabBarControllerDelegate {
    
    weak var mastodonController: MastodonController!
    
    private var composePlaceholder: UIViewController!
    
    var selectedTab: Tab {
        return Tab(rawValue: selectedIndex)!
    }

    override var supportedInterfaceOrientations: UIInterfaceOrientationMask {
        if UIDevice.current.userInterfaceIdiom == .phone {
            return .portrait
        } else {
            return .all
        }
    }
    
    init(mastodonController: MastodonController) {
        self.mastodonController = mastodonController
        
        super.init(nibName: nil, bundle: nil)
    }
    
    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
	}
    
    override func viewDidLoad() {
        super.viewDidLoad()

        self.delegate = self
        
        composePlaceholder = UIViewController()
        composePlaceholder.title = "Compose"
        composePlaceholder.tabBarItem.image = UIImage(systemName: "pencil")
        
        viewControllers = [
            embedInNavigationController(Tab.timelines.createViewController(mastodonController)),
            embedInNavigationController(Tab.notifications.createViewController(mastodonController)),
            composePlaceholder,
            embedInNavigationController(Tab.explore.createViewController(mastodonController)),
            embedInNavigationController(Tab.myProfile.createViewController(mastodonController)),
        ]
    }
    
    func embedInNavigationController(_ vc: UIViewController) -> UINavigationController {
        if let vc = vc as? UINavigationController {
            return vc
        } else {
            return EnhancedNavigationViewController(rootViewController: vc)
        }
    }
    
    func tabBarController(_ tabBarController: UITabBarController, shouldSelect viewController: UIViewController) -> Bool {
        if viewController == composePlaceholder {
            presentCompose()
            return false
        }
        return true
    }
    
    func setViewController(_ viewController: UIViewController, forTab tab: Tab) {
        viewControllers![tab.rawValue] = viewController
    }
    
    func viewController(for tab: Tab) -> UIViewController {
        return viewControllers![tab.rawValue]
    }
}

extension MainTabBarViewController {
    enum Tab: Int, Hashable, CaseIterable {
        case timelines
        case notifications
        case compose
        case explore
        case myProfile
        
        func createViewController(_ mastodonController: MastodonController) -> UIViewController {
            switch self {
            case .timelines:
                return TimelinesPageViewController(mastodonController: mastodonController)
            case .notifications:
                return NotificationsPageViewController(mastodonController: mastodonController)
            case .compose:
                return ComposeViewController(mastodonController: mastodonController)
            case .explore:
                return ExploreViewController(mastodonController: mastodonController)
            case .myProfile:
                return MyProfileTableViewController(mastodonController: mastodonController)
            }
        }
    }
    
    func getTabController(tab: Tab) -> UIViewController? {
        if tab == .compose {
            return nil
        } else {
            return viewControllers![tab.rawValue]
        }
    }
}

extension MainTabBarViewController: TuskerRootViewController {
    func presentCompose() {
        let compose = ComposeViewController(mastodonController: mastodonController)
        let navigationController = embedInNavigationController(compose)
        navigationController.presentationController?.delegate = compose
        present(navigationController, animated: true)
    }
    
    func select(tab: Tab) {
        if tab == .compose {
            presentCompose()
        } else {
            selectedIndex = tab.rawValue
        }
    }
}