Beam/ElementX/Sources/Other/UserIndicator/UserIndicatorController.swift

//
// Copyright 2022-2024 New Vector Ltd.
//
// SPDX-License-Identifier: AGPL-3.0-only OR LicenseRef-Element-Commercial
// Please see LICENSE files in the repository root for full details.
//

import Combine
import SwiftUI

class UserIndicatorController: ObservableObject, UserIndicatorControllerProtocol {
    private var timerCancellable: AnyCancellable?
    private var displayTimes = [String: Date]()
    private var delayedIndicators = Set<String>()
    
    var nonPersistentDisplayDuration = 2.5
    var minimumDisplayDuration = 0.5
    
    @Published private(set) var activeIndicator: UserIndicator?
    private(set) var indicatorQueue = [UserIndicator]() {
        didSet {
            activeIndicator = indicatorQueue.last
            
            if let activeIndicator, !activeIndicator.persistent {
                timerCancellable?.cancel()
                timerCancellable = Task { [weak self, nonPersistentDisplayDuration] in
                    try await Task.sleep(for: .seconds(nonPersistentDisplayDuration))
                    self?.retractIndicatorWithId(activeIndicator.id)
                }.asCancellable()
            }
        }
    }
    
    @Published var alertInfo: AlertInfo<UUID>?
    
    var window: UIWindow? {
        didSet {
            let hostingController = UIHostingController(rootView: UserIndicatorPresenter(userIndicatorController: self).statusBarHidden(ProcessInfo.isRunningUITests))
            hostingController.view.backgroundColor = .clear
            window?.rootViewController = hostingController
        }
    }
    
    func submitIndicator(_ indicator: UserIndicator, delay: Duration?) {
        if let index = indicatorQueue.firstIndex(where: { $0.id == indicator.id }) {
            indicatorQueue[index] = indicator
            displayTimes[indicator.id] = .now
        } else {
            if let delay {
                delayedIndicators.insert(indicator.id)
                
                Task {
                    try await Task.sleep(for: .seconds(delay.seconds))
                    
                    guard delayedIndicators.contains(indicator.id) else {
                        return
                    }

                    enqueue(indicator: indicator)
                }
            } else {
                enqueue(indicator: indicator)
            }
        }
    }
    
    func retractAllIndicators() {
        for indicator in indicatorQueue {
            retractIndicatorWithId(indicator.id)
        }
    }
    
    func retractIndicatorWithId(_ id: String) {
        delayedIndicators.remove(id)
        
        guard let displayTime = displayTimes[id], abs(displayTime.timeIntervalSinceNow) <= minimumDisplayDuration else {
            indicatorQueue.removeAll { $0.id == id }
            return
        }
        
        Task {
            try? await Task.sleep(for: .seconds(minimumDisplayDuration))
            indicatorQueue.removeAll { $0.id == id }
            displayTimes[id] = nil
        }
    }
    
    // MARK: - Private

    private func enqueue(indicator: UserIndicator) {
        retractIndicatorWithId(indicator.id)
        indicatorQueue.append(indicator)
        displayTimes[indicator.id] = .now
    }
}