//  Copyright © 2024 650 Industries. All rights reserved.

import ExpoModulesCore
import UIKit
internal import React

public class BadgeModule: Module {
  public func definition() -> ModuleDefinition {
    Name("ExpoBadgeModule")

    AsyncFunction("getBadgeCountAsync") { () -> Int in
      return RCTSharedApplication()?.applicationIconBadgeNumber ?? 0
    }
    .runOnQueue(.main)

    AsyncFunction("setBadgeCountAsync") { (badgeCount: Int) -> Bool in
      let center = UNUserNotificationCenter.current()

      let settings = await center.notificationSettings()

      if settings.badgeSetting == .enabled {
        if #available(iOS 16.0, *) {
          try await center.setBadgeCount(badgeCount)
        } else {
          await MainActor.run {
            RCTSharedApplication()?.applicationIconBadgeNumber = badgeCount
          }
        }
        return true
      }
      return false
    }
  }
}