@_spi(Experimental) import MapboxMaps

@objc(RNMBXModels)
open class RNMBXModels : UIView, RNMBXMapComponent {
  var modelIdToUrl: [String: String] = [:]
  
  @objc(models)
  public var models: NSDictionary {
    set {
      var newModels: [String: String] = [:]
      newValue.forEach { (key, value) in
        if let value = value as? NSDictionary, let key = key as? String {
          if let uri = value["uri"] as? String {
            newModels[key] = uri
          } else if let url = value["url"] as? String {
            newModels[key] = url
          } else {
            Logger.log(level: .error, message: "Unexpected value for model key: \(key) \(value) - no uri or url found")
          }
        }
      }
      self.modelIdToUrl = newModels
    }
    get {
      return modelIdToUrl as NSDictionary
    }
  }
  
  func setModels(_ models: NSDictionary) {

  }
  
  public func addToMap(_ map: RNMBXMapView, style: Style) {
    modelIdToUrl.forEach { (id, uri) in
      logged("Models.addStyleModel") {
        if let link = URL(string: uri), let scheme = link.scheme, let host = link.host,
            let port = link.port {

          // https://github.com/mapbox/mapbox-maps-ios/issues/2067
          let uriWithoutQuery = "\(scheme)://\(host):\(port)\(link.path)"

          try style.addStyleModel(modelId: id, modelUri: uriWithoutQuery)
        } else {
          try style.addStyleModel(modelId: id, modelUri: uri)
        }
      }
    }
  }
  
  public func removeFromMap(_ map: RNMBXMapView, reason: RemovalReason) -> Bool {
    modelIdToUrl.forEach { (id, _) in
      try? map._mapView?.mapboxMap.removeStyleModel(modelId: id)
    }
    return true
  }
  
  public func waitForStyleLoad() -> Bool {
    true
  }
}