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

import ExpoModulesCore
import UIKit
import MachO

let onHandleNotification = "onHandleNotification"
let onHandleNotificationTimeout = "onHandleNotificationTimeout"

open class HandlerModule: Module, NotificationDelegate, SingleNotificationHandlerTaskDelegate {
  var tasksMap: [String: SingleNotificationHandlerTask] = [:]

  public func definition() -> ModuleDefinition {
    Name("ExpoNotificationsHandlerModule")

    Events([onHandleNotification, onHandleNotificationTimeout])

    OnStartObserving {
      NotificationCenterManager.shared.addDelegate(self)
    }

    OnStopObserving {
      NotificationCenterManager.shared.removeDelegate(self)
    }

    AsyncFunction("handleNotificationAsync") { (identifier: String, behavior: NotificationBehavior, promise: Promise) in
      guard let task = tasksMap[identifier] else {
        promise.reject("ERR_NOTIFICATION_HANDLED", "Failed to handle notification \(identifier) because it has already been handled")
        return
      }
      if task.processNotificationWithOptions(behavior.presentationOptions) {
        promise.resolve(nil)
      } else {
        promise.reject("ERR_NOTIFICATION_RESPONSE_TIMEOUT", "Notification has already been handled. Most probably the request has timed out.")
      }
    }
  }

  // MARK: - NotificationDelegate

  open func willPresent(_ notification: UNNotification, completionHandler: @escaping (UNNotificationPresentationOptions) -> Void) -> Bool {
    let task = SingleNotificationHandlerTask(notification: notification, completionHandler: completionHandler, delegate: self)
    tasksMap[task.identifier] = task
    task.start()
    return true
  }

  // MARK: - SingleNotificationHandlerTaskDelegate

  public func taskDidFinish(_ task: SingleNotificationHandlerTask) {
    tasksMap[task.identifier] = nil
  }

  public func handleNotification(_ notification: UNNotification) {
    sendEvent(onHandleNotification, [
      "id": notification.request.identifier,
      "notification": NotificationRecord(from: notification).toDictionary(appContext: appContext)
    ])
  }

  public func handleNotificationTimeout(_ notification: UNNotification) {
    sendEvent(onHandleNotificationTimeout, [
      "id": notification.request.identifier,
      "notification": NotificationRecord(from: notification).toDictionary(appContext: appContext)
    ])
  }
}

struct NotificationBehavior: Record {
  @Field var shouldShowAlert: Bool = false
  @Field var shouldShowBanner: Bool = false
  @Field var shouldShowList: Bool = false
  @Field var shouldPlaySound: Bool = false
  @Field var shouldSetBadge: Bool = false

  var presentationOptions: UNNotificationPresentationOptions {
    var options: UNNotificationPresentationOptions = []

    // Deprecated but kept for backward compatibility
    if shouldShowAlert {
      options.insert(.alert)
    }
    if shouldShowBanner {
      options.insert(.banner)
    }
    if shouldShowList {
      options.insert(.list)
    }
    if shouldPlaySound {
      options.insert(.sound)
    }
    if shouldSetBadge {
      options.insert(.badge)
    }

    return options
  }
}