//
//  SleepManager.swift
//  QuickDrop
//
//  Created by Leon Böttger on 03.05.25.
//

import Foundation

#if os(macOS)
import IOKit
import IOKit.pwr_mgt
#elseif os(iOS)
import UIKit
#endif
import LUI

class SleepManager {
    public static let shared = SleepManager()

    #if os(macOS)
    private var assertionID: IOPMAssertionID = 0
    #else
    private var assertionID: Int = 0
    #endif

    private var timer: Timer?

    private init() {
        // Start periodic check
        timer = Timer.scheduledTimer(withTimeInterval: 5.0, repeats: true) { [weak self] _ in
            self?.updateSleepState()
        }

        RunLoop.main.add(timer!, forMode: .common)
    }

    private func disableSleep() {

        if assertionID != 0 { return }

        log("[SleepManager] Enabling Wakelock")
        
        #if os(macOS)
        IOPMAssertionCreateWithName(
            kIOPMAssertionTypePreventUserIdleDisplaySleep as CFString,
            IOPMAssertionLevel(kIOPMAssertionLevelOn),
            "QuickDrop Data Transfer" as CFString,
            &assertionID
        )
        #elseif os(iOS)
        UIApplication.shared.isIdleTimerDisabled = true
        assertionID = 1
        #endif
    }

    private func enableSleep() {
        if assertionID == 0 { return }

        log("[SleepManager] Disabling Wakelock")

        #if os(macOS)
        IOPMAssertionRelease(assertionID)
        #elseif os(iOS)
        UIApplication.shared.isIdleTimerDisabled = false
        #endif
        
        assertionID = 0
    }

    private func updateSleepState() {
        let activeConnections = NearbyConnectionManager.shared.getActiveConnectionsCount()
        if activeConnections > 0 {
            disableSleep()
        } else {
            enableSleep()
        }
    }
}