WebViewController+Kiosk.swift

import Combine
import Shared
import SwiftUI
import UIKit

// MARK: - Kiosk Mode Extension

extension WebViewController {
    /// Setup kiosk mode integration with KioskModeManager
    /// Call this from viewDidLoad
    func setupKioskMode() {
        let manager = KioskModeManager.shared

        // Wire up callbacks from KioskModeManager
        manager.onNavigate = { [weak self] path in
            self?.navigateToKioskPath(path)
        }

        manager.onRefresh = { [weak self] in
            self?.refresh()
        }

        manager.onKioskModeChange = { [weak self] enabled in
            self?.updateKioskModeLockdown(enabled: enabled)
        }

        manager.onShowScreensaver = { [weak self] mode in
            self?.showScreensaver(mode: mode)
        }

        manager.onHideScreensaver = { [weak self] in
            self?.hideScreensaver()
        }

        // Observe kiosk mode and settings changes using Combine (auto-cleanup on dealloc)
        var cancellables = Set<AnyCancellable>()

        manager.$isKioskModeActive
            .receive(on: DispatchQueue.main)
            .sink { [weak self] _ in
                self?.kioskModeDidChange()
            }
            .store(in: &cancellables)

        manager.$settings
            .receive(on: DispatchQueue.main)
            .sink { [weak self] _ in
                self?.kioskSettingsDidChange()
            }
            .store(in: &cancellables)

        kioskCancellables = cancellables

        // Setup the screensaver
        setupScreensaver()

        // Setup secret exit gesture (only when not showing screensaver)
        setupSecretExitGesture()

        // Apply initial state if already in kiosk mode
        if manager.isKioskModeActive {
            updateKioskModeLockdown(enabled: true)
        }
    }

    // MARK: - Screensaver

    private func setupScreensaver() {
        let controller = KioskScreensaverViewController()
        screensaverController = controller

        // Forward the callback for showing settings
        controller.onShowSettings = { [weak self] in
            self?.showKioskSettings()
        }

        addChild(controller)
        view.addSubview(controller.view)
        controller.view.translatesAutoresizingMaskIntoConstraints = false

        NSLayoutConstraint.activate([
            controller.view.topAnchor.constraint(equalTo: view.topAnchor),
            controller.view.bottomAnchor.constraint(equalTo: view.bottomAnchor),
            controller.view.leadingAnchor.constraint(equalTo: view.leadingAnchor),
            controller.view.trailingAnchor.constraint(equalTo: view.trailingAnchor),
        ])

        controller.didMove(toParent: self)
        controller.view.isHidden = true
    }

    private func setupSecretExitGesture() {
        let controller = KioskSecretExitGestureViewController()
        secretExitGestureController = controller

        controller.onShowSettings = { [weak self] in
            self?.showKioskSettings()
        }

        addChild(controller)
        view.addSubview(controller.view)
        controller.view.translatesAutoresizingMaskIntoConstraints = false

        NSLayoutConstraint.activate([
            controller.view.topAnchor.constraint(equalTo: view.topAnchor),
            controller.view.bottomAnchor.constraint(equalTo: view.bottomAnchor),
            controller.view.leadingAnchor.constraint(equalTo: view.leadingAnchor),
            controller.view.trailingAnchor.constraint(equalTo: view.trailingAnchor),
        ])

        controller.didMove(toParent: self)
    }

    private func showScreensaver(mode: ScreensaverMode) {
        guard let controller = screensaverController else { return }

        Current.Log.info("Showing screensaver: \(mode.rawValue)")

        controller.view.isHidden = false
        view.bringSubviewToFront(controller.view)
        controller.show(mode: mode)
    }

    private func hideScreensaver() {
        guard let controller = screensaverController else { return }

        Current.Log.info("Hiding screensaver")
        controller.hide()

        // Delay hiding the view until the animation completes
        DispatchQueue.main.asyncAfter(deadline: .now() + 0.3) {
            controller.view.isHidden = true
        }
    }

    // MARK: - Navigation

    private func navigateToKioskPath(_ path: String) {
        Current.Log.info("Kiosk navigating to: \(path)")

        // Use the existing navigateToPath method
        navigateToPath(path: path)
    }

    // MARK: - UI Lockdown

    private func updateKioskModeLockdown(enabled: Bool) {
        let settings = KioskModeManager.shared.settings

        // Update iOS system status bar and home indicator visibility
        // The navigation controller must be updated first so it re-queries its child
        if let navController = navigationController {
            navController.setNeedsStatusBarAppearanceUpdate()
            navController.setNeedsUpdateOfHomeIndicatorAutoHidden()
        }
        setNeedsStatusBarAppearanceUpdate()
        setNeedsUpdateOfHomeIndicatorAutoHidden()

        // Hide/show the custom status bar background view
        if let statusBarView {
            let shouldHide = enabled && settings.hideStatusBar
            statusBarView.isHidden = shouldHide
        }
    }

    // MARK: - Status Bar & Home Indicator

    /// Override in WebViewController to check kiosk mode
    var kioskPrefersStatusBarHidden: Bool {
        let manager = KioskModeManager.shared
        return manager.isKioskModeActive && manager.settings.hideStatusBar
    }

    /// Override in WebViewController to check kiosk mode
    var kioskPrefersHomeIndicatorAutoHidden: Bool {
        KioskModeManager.shared.isKioskModeActive
    }

    // MARK: - Settings

    private func showKioskSettings() {
        Current.Log.info("Showing kiosk settings")

        // Use UINavigationController to avoid SwiftUI NavigationView dismissal bugs on iOS 15
        // Pass an explicit dismiss closure since SwiftUI's @Environment(\.dismiss) doesn't work
        // reliably when UIHostingController is embedded in UINavigationController presented via UIKit
        let settingsView = KioskSettingsView(onDismiss: { [weak self] in
            self?.dismiss(animated: true) { [weak self] in
                self?.refreshStatusBarAppearance()
            }
        })
        let hostingController = UIHostingController(rootView: settingsView)
        let navController = UINavigationController(rootViewController: hostingController)
        navController.modalPresentationStyle = .pageSheet
        present(navController, animated: true)
    }

    /// Force a complete status bar appearance refresh after modal dismissal
    private func refreshStatusBarAppearance() {
        navigationController?.setNeedsStatusBarAppearanceUpdate()
        setNeedsStatusBarAppearanceUpdate()
        navigationController?.setNeedsUpdateOfHomeIndicatorAutoHidden()
        setNeedsUpdateOfHomeIndicatorAutoHidden()
    }

    // MARK: - Observers

    private func kioskModeDidChange() {
        let manager = KioskModeManager.shared
        Current.Log.info("Kiosk mode changed: \(manager.isKioskModeActive)")

        updateKioskModeLockdown(enabled: manager.isKioskModeActive)
    }

    private func kioskSettingsDidChange() {
        // Re-apply lockdown settings in case they changed
        let manager = KioskModeManager.shared
        if manager.isKioskModeActive {
            updateKioskModeLockdown(enabled: true)
        }
    }

    // MARK: - Touch Handling

    /// Call this when user touches the screen to record activity
    func recordKioskActivity() {
        KioskModeManager.shared.recordActivity(source: "touch")
    }
}