import Foundation
import CoreBluetooth

/// BLE scan via CoreBluetooth. Mirrors Android `feature/ScanFeature.kt`.
///
/// Like the Android implementation, scanning is **not** delegated to the JieLi
/// SDK's built-in `scan()` (which has its own filterContent/strategy quirks);
/// instead we drive `CBCentralManager` directly and parse the advertisement
/// data ourselves. The published `deviceFound` event payload matches the
/// Android schema 1:1 so the Dart parser is unchanged:
///
/// ```
/// {
///   type: "deviceFound",
///   name, address, rssi,
///   edrAddr, deviceType, connectWay,        // null on iOS — see note below
///   rawAdv, advFlags, manufacturerCompanyId, manufacturerData,
///   serviceUuids: [String], advRecords: [{len, type, data}]
/// }
/// ```
///
/// `edrAddr` / `deviceType` / `connectWay` are populated on Android by feeding
/// the raw scan record through `ParseDataUtil.isFilterBleDevice(...)`. On iOS
/// the equivalent live in `JL_AdvParse`'s `JLAdvParse` Obj-C class; until the
/// exact entry point is wired we leave them null and rely on the connection
/// path's BLE-only fallback (matches the Android "fallback when SDK can't
/// parse" behavior).
class ScanFeature: NSObject, CBCentralManagerDelegate {

  private weak var server: JieliHomeServer?
  private weak var dispatcher: EventDispatcher?

  /// Dedicated central manager for scanning — separated from the SDK's own
  /// `JL_BLEMultiple` central so that filter knobs do not collide.
  private var centralManager: CBCentralManager?

  private var scanning: Bool = false
  private var stopWorkItem: DispatchWorkItem?

  private var currentNameList: [String] = []
  private var currentSkipUnnamed: Bool = true
  private var serviceFilters: [CBUUID] = []

  // Track which addresses have been published this scan to dedupe.
  private var publishedIds: Set<String> = []

  init(dispatcher: EventDispatcher) {
    self.dispatcher = dispatcher
    super.init()
  }

  func attach(server: JieliHomeServer) {
    self.server = server
    if centralManager == nil {
      centralManager = CBCentralManager(delegate: self, queue: .main)
    }
  }

  func detach() {
    stopScan()
    centralManager = nil
  }

  // MARK: - Public API (mirrors Android signatures)

  func startScan(
    timeoutMs: Int,
    nameList: [String],
    uuidList: [String],
    skipUnnamed: Bool
  ) throws {
    guard let central = centralManager else {
      throw PluginError("SCAN_FAILED", "BLE central unavailable")
    }
    if scanning { stopScan() }

    currentNameList = nameList.filter { !$0.isEmpty }
    currentSkipUnnamed = skipUnnamed
    serviceFilters = uuidList.compactMap { Self.parseUuid($0) }
    publishedIds.removeAll(keepingCapacity: true)

    if central.state != .poweredOn {
      // Match Android behavior — scan request from a powered-off adapter
      // surfaces the same shape of error.
      throw PluginError("SCAN_FAILED", "BLE adapter not powered on (state=\(central.state.rawValue))")
    }

    let services: [CBUUID]? = serviceFilters.isEmpty ? nil : serviceFilters
    let options: [String: Any] = [
      // Allow duplicate adverts so we keep getting RSSI updates.
      CBCentralManagerScanOptionAllowDuplicatesKey: false
    ]
    central.scanForPeripherals(withServices: services, options: options)
    scanning = true
    dispatcher?.send([
      "type": "scanStatus",
      "ble": true,
      "started": true,
    ])

    let work = DispatchWorkItem { [weak self] in self?.stopScan() }
    stopWorkItem = work
    DispatchQueue.main.asyncAfter(deadline: .now() + .milliseconds(timeoutMs), execute: work)
  }

  func stopScan() {
    if let work = stopWorkItem { work.cancel(); stopWorkItem = nil }
    if let central = centralManager, central.state == .poweredOn {
      central.stopScan()
    }
    if scanning {
      dispatcher?.send([
        "type": "scanStatus",
        "ble": true,
        "started": false,
      ])
    }
    scanning = false
    currentNameList = []
    serviceFilters = []
  }

  func isScanning() -> Bool { scanning }

  // MARK: - CBCentralManagerDelegate

  func centralManagerDidUpdateState(_ central: CBCentralManager) {
    let on = central.state == .poweredOn
    dispatcher?.send([
      "type": "adapterStatus",
      "enabled": on,
      "hasBle": true,
    ])
    if !on && scanning { stopScan() }
  }

  func centralManager(
    _ central: CBCentralManager,
    didDiscover peripheral: CBPeripheral,
    advertisementData: [String: Any],
    rssi RSSI: NSNumber
  ) {
    let name = peripheral.name
      ?? (advertisementData[CBAdvertisementDataLocalNameKey] as? String)

    // === 调试用：打印每一条广播（过滤前），完整 dump 广播 hex ===
    // iOS CoreBluetooth 不暴露原始广播字节，这里把 advertisementData
    // 里所有字段按 BLE AD type 拼出 hex 方便核对。
    let manufacturerData = advertisementData[CBAdvertisementDataManufacturerDataKey] as? Data
    let serviceUuids = advertisementData[CBAdvertisementDataServiceUUIDsKey] as? [CBUUID]
    let serviceData = advertisementData[CBAdvertisementDataServiceDataKey] as? [CBUUID: Data]
    let txPower = advertisementData[CBAdvertisementDataTxPowerLevelKey] as? NSNumber
    let localName = advertisementData[CBAdvertisementDataLocalNameKey] as? String
    let isConnectable = (advertisementData[CBAdvertisementDataIsConnectable] as? NSNumber)?.boolValue ?? false
    let solicited = advertisementData[CBAdvertisementDataSolicitedServiceUUIDsKey] as? [CBUUID]
    let overflow = advertisementData[CBAdvertisementDataOverflowServiceUUIDsKey] as? [CBUUID]

    let mfrHex = manufacturerData?.map { String(format: "%02X", $0) }.joined() ?? "nil"
    let svcUuids = serviceUuids?.map { $0.uuidString }.joined(separator: ",") ?? ""
    let svcDataHex = serviceData?.map { (k, v) in
      "\(k.uuidString)=" + v.map { String(format: "%02X", $0) }.joined()
    }.joined(separator: ";") ?? "nil"
    let txPowerStr = txPower.map { "\($0.intValue)" } ?? "nil"
    let solicitedStr = solicited?.map { $0.uuidString }.joined(separator: ",") ?? ""
    let overflowStr = overflow?.map { $0.uuidString }.joined(separator: ",") ?? ""

    NSLog("%@", "================ [JieliScan] didDiscover ================")
    NSLog("%@", "[JieliScan]   name        = \(name ?? "<nil>")")
    NSLog("%@", "[JieliScan]   localName   = \(localName ?? "<nil>")")
    NSLog("%@", "[JieliScan]   id          = \(peripheral.identifier.uuidString)")
    NSLog("%@", "[JieliScan]   rssi        = \(RSSI)  txPower=\(txPowerStr)  connectable=\(isConnectable)")
    NSLog("%@", "[JieliScan]   mfr(hex)    = \(mfrHex)")
    if let m = manufacturerData, m.count >= 2 {
      let cid = Int(m[0]) | (Int(m[1]) << 8)
      let payload = m.count > 2
        ? m[2..<m.count].map { String(format: "%02X", $0) }.joined()
        : ""
      NSLog("%@", "[JieliScan]   mfr.companyId = 0x\(String(format: "%04X", cid))  payload=\(payload)")
    }
    NSLog("%@", "[JieliScan]   svcUuids    = [\(svcUuids)]")
    NSLog("%@", "[JieliScan]   svcData     = \(svcDataHex)")
    if !solicitedStr.isEmpty { NSLog("%@", "[JieliScan]   solicited   = [\(solicitedStr)]") }
    if !overflowStr.isEmpty  { NSLog("%@", "[JieliScan]   overflow    = [\(overflowStr)]") }
    NSLog("%@", "[JieliScan] =========================================================")

    if currentSkipUnnamed && (name?.isEmpty ?? true) { return }
    if !currentNameList.isEmpty {
      guard let n = name, currentNameList.contains(where: { $0.caseInsensitiveCompare(n) == .orderedSame }) else { return }
    }

    let id = peripheral.identifier.uuidString
    if !publishedIds.insert(id).inserted { return }

    let serviceUuidsUpper = serviceUuids?.map { $0.uuidString.uppercased() } ?? []

    var advFlags: Int? = nil
    var mfrCompanyId: Int? = nil
    var mfrPayloadHex: String? = nil
    var advRecords: [[String: String]] = []

    // iOS 拿不到原始扫描记录字节，但 `JieliAdvParser` (Dart 侧) 为 iOS
    // 设计的偏移本就是从 manufacturer data（company id 开头）算起的：
    //   bytes[0..1]=companyId, [4..5]=PID, [7..12]=MAC^0x5A, [13..22]=code
    // 所以把 manufacturer 数据完整 hex 喂给 `rawAdv` 即可让绑定流程跑通。
    let rawAdvHex: String? = manufacturerData?
      .map { String(format: "%02X", $0) }
      .joined()

    if let m = manufacturerData, m.count >= 2 {
      mfrCompanyId = Int(m[0]) | (Int(m[1]) << 8)
      if m.count > 2 {
        mfrPayloadHex = m[2..<m.count].map { String(format: "%02X", $0) }.joined()
      }
    }

    // BLE Core Spec §11 ad-record reconstruction is best-effort on iOS:
    // CoreBluetooth surfaces parsed fields only. We synthesize a minimal
    // record list for parity with Android's structured payload.
    if let m = manufacturerData {
      // 0xFF — Manufacturer Specific Data
      advRecords.append([
        "len": String(format: "%02d", m.count + 1),
        "type": "0xFF",
        "data": "0x" + m.map { String(format: "%02X", $0) }.joined(),
      ])
    }
    if let n = name, !n.isEmpty {
      let bytes = Array(n.utf8)
      advRecords.append([
        "len": String(format: "%02d", bytes.count + 1),
        "type": "0x09", // Complete Local Name
        "data": "0x" + bytes.map { String(format: "%02X", $0) }.joined(),
      ])
    }

    var payload: [String: Any?] = [
      "type": "deviceFound",
      "name": name ?? "",
      // iOS does not expose the BLE MAC in CoreBluetooth — use the peripheral
      // identifier UUID as a stable address. Downstream ConnectFeature looks
      // it up by the same id.
      "address": id,
      "edrAddr": NSNull(),    // TODO: parse via JL_AdvParse / JLAdvParse if available
      "deviceType": NSNull(), // TODO: parse via JL_AdvParse / JLAdvParse if available
      "connectWay": NSNull(), // TODO: parse via JL_AdvParse / JLAdvParse if available
      "rssi": RSSI.intValue,
      "rawAdv": (rawAdvHex as Any?) ?? NSNull(),
      "advRecords": advRecords,
      "advFlags": advFlags as Any,
      "manufacturerCompanyId": mfrCompanyId as Any,
      "manufacturerData": mfrPayloadHex as Any,
      "serviceUuids": serviceUuidsUpper,
    ]
    dispatcher?.send(payload)
  }

  // MARK: - UUID parsing (matches Android `parseUuid`)
  static func parseUuid(_ raw: String) -> CBUUID? {
    let s = raw.trimmingCharacters(in: .whitespacesAndNewlines)
    guard !s.isEmpty else { return nil }
    return CBUUID(string: s)
  }
}