Atoll/DynamicIsland/observers/FullscreenMediaDetection.swift

//
//  FullscreenMediaDetection.swift
//  DynamicIsland
//
//  Created by Richard Kunkli on 06/09/2024.
//

import Accessibility
import Cocoa
import CoreAudio
import Defaults
import MacroVisionKit
import SwiftUI

@MainActor
class FullscreenMediaDetector: ObservableObject {
    static let shared = FullscreenMediaDetector()
    private let detector: MacroVisionKit

    // changed from [NSScreen:Bool] to [screenName:Bool]
    @Published private(set) var fullscreenStatus: [String: Bool] = [:]

    private init() {
        self.detector = MacroVisionKit.shared
        detector.configuration.includeSystemApps = true
        setupNotificationObservers()
        updateFullScreenStatus()  // now on MainActor
    }

    private func setupNotificationObservers() {
        let nc = NSWorkspace.shared.notificationCenter
        nc.addObserver(self, selector: #selector(handleChange),
                       name: NSWorkspace.activeSpaceDidChangeNotification, object: nil)
        nc.addObserver(self, selector: #selector(handleChange),
                       name: NSApplication.didChangeScreenParametersNotification, object: nil)
    }

    @objc private func handleChange() {
        DispatchQueue.main.asyncAfter(deadline: .now() + 0.1) {
            self.updateFullScreenStatus()
        }
    }

    private func updateFullScreenStatus() {
        NSLog("🔄 Fullscreen status update triggered")
        guard Defaults[.enableFullscreenMediaDetection] else {
            NSLog("❌ Fullscreen media detection disabled")
            let reset = Dictionary(uniqueKeysWithValues: NSScreen.screens.map { ($0.localizedName, false) }
            )
            if reset != fullscreenStatus {
                fullscreenStatus = reset
                NSLog("⏸ Fullscreen disabled → reset all screens")
            }
            return
        }
        
        NSLog("🔄 Fullscreen media detection enabled")

        let apps = detector.detectFullscreenApps(debug: false)
        let names = NSScreen.screens.map { $0.localizedName }
        var newStatus: [String: Bool] = [:]
        for name in names {
            newStatus[name] = apps.contains { $0.screen.localizedName == name && $0.bundleIdentifier != "com.apple.finder" }
        }

        if newStatus != fullscreenStatus {
            fullscreenStatus = newStatus
            NSLog("✅ Fullscreen status: \(newStatus)")
        }
    }

    func isScreenInFullScreen(_ screen: NSScreen) -> Bool {
        let screenName = screen.localizedName
        return fullscreenStatus[screenName] ?? false
    }
    
    func isAnyScreenInFullScreen() -> Bool {
        return fullscreenStatus.values.contains(true)
    }

    private func cleanupNotificationObservers() {
        NSWorkspace.shared.notificationCenter.removeObserver(self)
    }

    deinit {
        NSWorkspace.shared.notificationCenter.removeObserver(self)
    }
}