NCMainNavigationController.swift

// SPDX-FileCopyrightText: Nextcloud GmbH
// SPDX-FileCopyrightText: 2025 Marino Faggiana
// SPDX-License-Identifier: GPL-3.0-or-later

import UIKit
import SwiftUI
import NextcloudKit

class NCMainNavigationController: UINavigationController, UINavigationControllerDelegate {
    let database = NCManageDatabase.shared
    let global = NCGlobal.shared
    let utility = NCUtility()
    let utilityFileSystem = NCUtilityFileSystem()
    let appDelegate = (UIApplication.shared.delegate as? AppDelegate)!
    private var timer: Timer?

    var controller: NCMainTabBarController? {
        self.tabBarController as? NCMainTabBarController
    }

    var collectionViewCommon: NCCollectionViewCommon? {
        topViewController as? NCCollectionViewCommon
    }

    var session: NCSession.Session {
        NCSession.shared.getSession(controller: controller)
    }

    let menuButtonTag = 100
    let assistantButtonTag = 101
    let notificationsButtonTag = 102
    let transfersButtonTag = 103

    let menuButton = UIButton(type: .system)
    var menuBarButtonItem: UIBarButtonItem {
        let item = UIBarButtonItem(customView: menuButton)
        item.tag = menuButtonTag
        return item
    }

    let assistantButton = UIButton(type: .system)
    var assistantButtonItem: UIBarButtonItem {
        let item = UIBarButtonItem(customView: assistantButton)
        item.tag = assistantButtonTag
        return item
    }

    let notificationsButton = UIButton(type: .system)
    var notificationsButtonItem: UIBarButtonItem {
        let item = UIBarButtonItem(customView: notificationsButton)
        item.tag = notificationsButtonTag
        return item
    }

    let transfersButton = UIButton(type: .system)
    var transfersButtonItem: UIBarButtonItem {
        let item = UIBarButtonItem(customView: transfersButton)
        item.tag = transfersButtonTag
        return item
    }

    // MARK: - View Life Cycle

    override func viewDidLoad() {
        super.viewDidLoad()
        self.delegate = self

        menuButton.setImage(UIImage(systemName: "ellipsis.circle"), for: .normal)
        menuButton.tintColor = NCBrandColor.shared.iconImageColor
        menuButton.menu = createRightMenu()
        menuButton.showsMenuAsPrimaryAction = true

        assistantButton.setImage(UIImage(systemName: "sparkles"), for: .normal)
        assistantButton.tintColor = NCBrandColor.shared.iconImageColor
        assistantButton.addAction(UIAction(handler: { _ in
            let assistant = NCAssistant()
                .environmentObject(NCAssistantModel(controller: self.controller))
            let hostingController = UIHostingController(rootView: assistant)
            self.present(hostingController, animated: true, completion: nil)
        }), for: .touchUpInside)

        notificationsButton.setImage(UIImage(systemName: "bell.fill"), for: .normal)
        notificationsButton.tintColor = NCBrandColor.shared.iconImageColor
        notificationsButton.addAction(UIAction(handler: { _ in
            if let navigationController = UIStoryboard(name: "NCNotification", bundle: nil).instantiateInitialViewController() as? UINavigationController,
               let viewController = navigationController.topViewController as? NCNotification {
                viewController.modalPresentationStyle = .pageSheet
                viewController.session = self.session
                self.present(navigationController, animated: true, completion: nil)
            }
        }), for: .touchUpInside)

        transfersButton.setImage(UIImage(systemName: "arrow.left.arrow.right.circle.fill"), for: .normal)
        transfersButton.tintColor = NCBrandColor.shared.iconImageColor
        transfersButton.addAction(UIAction(handler: { _ in
            if let navigationController = UIStoryboard(name: "NCTransfers", bundle: nil).instantiateInitialViewController() as? UINavigationController,
               let viewController = navigationController.topViewController as? NCTransfers {
                viewController.modalPresentationStyle = .pageSheet
                self.present(navigationController, animated: true, completion: nil)
            }
        }), for: .touchUpInside)

        navigationBar.prefersLargeTitles = true
        setNavigationBarHidden(false, animated: true)

        NotificationCenter.default.addObserver(forName: UIApplication.didEnterBackgroundNotification, object: nil, queue: nil) { _ in
            self.timer?.invalidate()
            self.timer = nil
        }

        NotificationCenter.default.addObserver(forName: UIApplication.didBecomeActiveNotification, object: nil, queue: nil) { _ in
            DispatchQueue.main.asyncAfter(deadline: .now() + 0.1) {
                if UIApplication.shared.applicationState == .active {
                    self.timer = Timer.scheduledTimer(withTimeInterval: 1, repeats: true, block: { _ in
                        self.updateRightBarButtonItems()
                    })
                }
            }
        }
    }

    func navigationController(_ navigationController: UINavigationController, willShow viewController: UIViewController, animated: Bool) {
        setNavigationBarAppearance()
        self.updateRightBarButtonItems()
    }

    // MARK: - Right

    func setNavigationRightItems() {
        guard let collectionViewCommon
        else {
            return
        }

        if collectionViewCommon.isEditMode {
            collectionViewCommon.tabBarSelect?.update(fileSelect: collectionViewCommon.fileSelect, metadatas: collectionViewCommon.getSelectedMetadatas(), userId: session.userId)
            collectionViewCommon.tabBarSelect?.show()

            let select = UIBarButtonItem(title: NSLocalizedString("_cancel_", comment: ""), style: .done) {
                collectionViewCommon.setEditMode(false)
                collectionViewCommon.collectionView.reloadData()
            }

            self.collectionViewCommon?.navigationItem.rightBarButtonItems = [select]

        } else {

            collectionViewCommon.tabBarSelect?.hide()
            self.updateRightBarButtonItems()
        }

        // fix, if the tabbar was hidden before the update, set it in hidden
        if self.tabBarController?.tabBar.isHidden ?? true,
           collectionViewCommon.tabBarSelect?.isHidden() ?? true {
            self.tabBarController?.tabBar.isHidden = true
        }
    }

    func updateRightBarButtonItems() {
        guard let collectionViewCommon,
              !collectionViewCommon.isEditMode,
              let controller
        else {
            return
        }
        let capabilities = NCCapabilities.shared.getCapabilities(account: session.account)
        let resultsCount = self.database.getResultsMetadatas(predicate: NSPredicate(format: "status != %i", NCGlobal.shared.metadataStatusNormal))?.count ?? 0
        var tempRightBarButtonItems = [self.menuBarButtonItem]
        var tempTotalTags = self.menuBarButtonItem.tag
        var totalTags = 0

        if let rightBarButtonItems = self.collectionViewCommon?.navigationItem.rightBarButtonItems,
            let menuBarButtonItem = rightBarButtonItems.first(where: { $0.tag == menuButtonTag }),
            let menuButton = menuBarButtonItem.customView as? UIButton {
            menuButton.menu = createRightMenu()
        }

        if let rightBarButtonItems = collectionViewCommon.navigationItem.rightBarButtonItems {
            for item in rightBarButtonItems {
                totalTags = totalTags + item.tag
            }
        }

        if capabilities.capabilityAssistantEnabled {
            tempRightBarButtonItems.append(self.assistantButtonItem)
            tempTotalTags = tempTotalTags + self.assistantButtonItem.tag
        }

        if controller.availableNotifications {
            tempRightBarButtonItems.append(self.notificationsButtonItem)
            tempTotalTags = tempTotalTags + self.notificationsButtonItem.tag
        }

        if resultsCount > 0 {
            tempRightBarButtonItems.append(self.transfersButtonItem)
            tempTotalTags = tempTotalTags + self.transfersButtonItem.tag
        }

        if totalTags != tempTotalTags {
            collectionViewCommon.navigationItem.rightBarButtonItems = tempRightBarButtonItems
        }
    }

    func createRightMenu() -> UIMenu? { return nil }

    func createRightMenuActions() -> (select: UIAction, viewStyleSubmenu: UIMenu, sortSubmenu: UIMenu, personalFilesOnlyAction: UIAction, showDescription: UIAction, showRecommendedFiles: UIAction)? {
        guard let collectionViewCommon,
              let layoutForView = database.getLayoutForView(account: session.account, key: collectionViewCommon.layoutKey, serverUrl: collectionViewCommon.serverUrl) else { return nil }

        let select = UIAction(title: NSLocalizedString("_select_", comment: ""),
                              image: utility.loadImage(named: "checkmark.circle"),
                              attributes: (collectionViewCommon.dataSource.isEmpty() || NCNetworking.shared.isOffline) ? .disabled : []) { _ in
            collectionViewCommon.setEditMode(true)
            collectionViewCommon.collectionView.reloadData()
        }

        let list = UIAction(title: NSLocalizedString("_list_", comment: ""), image: utility.loadImage(named: "list.bullet"), state: layoutForView.layout == global.layoutList ? .on : .off) { _ in

            layoutForView.layout = self.global.layoutList

            NotificationCenter.default.postOnMainThread(name: self.global.notificationCenterChangeLayout,
                                                        object: nil,
                                                        userInfo: ["account": self.session.account,
                                                                   "serverUrl": collectionViewCommon.serverUrl,
                                                                   "layoutForView": layoutForView])
        }

        let grid = UIAction(title: NSLocalizedString("_icons_", comment: ""), image: utility.loadImage(named: "square.grid.2x2"), state: layoutForView.layout == global.layoutGrid ? .on : .off) { _ in

            layoutForView.layout = self.global.layoutGrid

            NotificationCenter.default.postOnMainThread(name: self.global.notificationCenterChangeLayout,
                                                        object: nil,
                                                        userInfo: ["account": self.session.account,
                                                                   "serverUrl": collectionViewCommon.serverUrl,
                                                                   "layoutForView": layoutForView])
        }

        let mediaSquare = UIAction(title: NSLocalizedString("_media_square_", comment: ""), image: utility.loadImage(named: "square.grid.3x3"), state: layoutForView.layout == global.layoutPhotoSquare ? .on : .off) { _ in

            layoutForView.layout = self.global.layoutPhotoSquare

            NotificationCenter.default.postOnMainThread(name: self.global.notificationCenterChangeLayout,
                                                        object: nil,
                                                        userInfo: ["account": self.session.account,
                                                                   "serverUrl": collectionViewCommon.serverUrl,
                                                                   "layoutForView": layoutForView])
        }

        let mediaRatio = UIAction(title: NSLocalizedString("_media_ratio_", comment: ""), image: utility.loadImage(named: "rectangle.grid.3x2"), state: layoutForView.layout == self.global.layoutPhotoRatio ? .on : .off) { _ in

            layoutForView.layout = self.global.layoutPhotoRatio

            NotificationCenter.default.postOnMainThread(name: self.global.notificationCenterChangeLayout,
                                                        object: nil,
                                                        userInfo: ["account": self.session.account,
                                                                   "serverUrl": collectionViewCommon.serverUrl,
                                                                   "layoutForView": layoutForView])
        }

        let viewStyleSubmenu = UIMenu(title: "", options: .displayInline, children: [list, grid, mediaSquare, mediaRatio])

        let ascending = layoutForView.ascending
        let ascendingChevronImage = utility.loadImage(named: ascending ? "chevron.up" : "chevron.down")
        let isName = layoutForView.sort == "fileName"
        let isDate = layoutForView.sort == "date"
        let isSize = layoutForView.sort == "size"

        let byName = UIAction(title: NSLocalizedString("_name_", comment: ""), image: isName ? ascendingChevronImage : nil, state: isName ? .on : .off) { _ in

            if isName { // repeated press
                layoutForView.ascending = !layoutForView.ascending
            }
            layoutForView.sort = "fileName"

            NotificationCenter.default.postOnMainThread(name: self.global.notificationCenterChangeLayout,
                                                        object: nil,
                                                        userInfo: ["account": self.session.account,
                                                                   "serverUrl": collectionViewCommon.serverUrl,
                                                                   "layoutForView": layoutForView])
        }

        let byNewest = UIAction(title: NSLocalizedString("_date_", comment: ""), image: isDate ? ascendingChevronImage : nil, state: isDate ? .on : .off) { _ in

            if isDate { // repeated press
                layoutForView.ascending = !layoutForView.ascending
            }
            layoutForView.sort = "date"

            NotificationCenter.default.postOnMainThread(name: self.global.notificationCenterChangeLayout,
                                                        object: nil,
                                                        userInfo: ["account": self.session.account,
                                                                   "serverUrl": collectionViewCommon.serverUrl,
                                                                   "layoutForView": layoutForView])
        }

        let byLargest = UIAction(title: NSLocalizedString("_size_", comment: ""), image: isSize ? ascendingChevronImage : nil, state: isSize ? .on : .off) { _ in

            if isSize { // repeated press
                layoutForView.ascending = !layoutForView.ascending
            }
            layoutForView.sort = "size"

            NotificationCenter.default.postOnMainThread(name: self.global.notificationCenterChangeLayout,
                                                        object: nil,
                                                        userInfo: ["account": self.session.account,
                                                                   "serverUrl": collectionViewCommon.serverUrl,
                                                                   "layoutForView": layoutForView])
        }

        let sortSubmenu = UIMenu(title: NSLocalizedString("_order_by_", comment: ""), options: .displayInline, children: [byName, byNewest, byLargest])

        let personalFilesOnly = NCKeychain().getPersonalFilesOnly(account: self.session.account)
        let personalFilesOnlyAction = UIAction(title: NSLocalizedString("_personal_files_only_", comment: ""), image: utility.loadImage(named: "folder.badge.person.crop", colors: NCBrandColor.shared.iconImageMultiColors), state: personalFilesOnly ? .on : .off) { _ in

            NCKeychain().setPersonalFilesOnly(account: self.session.account, value: !personalFilesOnly)

            NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterReloadDataSource, userInfo: ["serverUrl": collectionViewCommon.serverUrl, "clearDataSource": true])
            self.setNavigationRightItems()
        }

        let showDescriptionKeychain = NCKeychain().showDescription
        let showDescription = UIAction(title: NSLocalizedString("_show_description_", comment: ""), state: showDescriptionKeychain ? .on : .off) { _ in

            NCKeychain().showDescription = !showDescriptionKeychain

            NotificationCenter.default.postOnMainThread(name: NCGlobal.shared.notificationCenterReloadDataSource, userInfo: ["serverUrl": collectionViewCommon.serverUrl, "clearDataSource": true])
            self.setNavigationRightItems()
        }

        let showRecommendedFilesKeychain = NCKeychain().showRecommendedFiles
        let capabilityRecommendations = NCCapabilities.shared.getCapabilities(account: session.account).capabilityRecommendations
        let showRecommendedFiles = UIAction(title: NSLocalizedString("_show_recommended_files_", comment: ""), attributes: !capabilityRecommendations ? .disabled : [], state: showRecommendedFilesKeychain ? .on : .off) { _ in

            NCKeychain().showRecommendedFiles = !showRecommendedFilesKeychain

            collectionViewCommon.collectionView.reloadData()
            self.setNavigationRightItems()
        }

        return (select, viewStyleSubmenu, sortSubmenu, personalFilesOnlyAction, showDescription, showRecommendedFiles)
    }

    // MARK: - Left

    func setNavigationLeftItems() { }
}