import SwiftUI
import UIKit
import Foundation
import ExpoModulesCore
import MobileWhipWhepClient

protocol OnTrackUpdateListener {
    func onTrackUpdate()
}

@objc(ReactNativeMobileWhepClientView)
public class ReactNativeMobileWhepClientView: ExpoView, OnTrackUpdateListener {
    func onTrackUpdate() {
        setupPlayer()
    }
    
    public var playerType: String? {
        didSet {
            setupPlayer()
        }
    }
  
  public var pipEnabled = false {
    didSet {
      setupPip()
    }
  }
  
  public var pipController: PictureInPictureController? {
    hostingController?.pipController
  }
      
  private var player: ClientBase? {
    guard let playerType = self.playerType else { return nil }
    if (playerType == "WHIP") {
        return ReactNativeMobileWhepClientModule.whipClient
    } else {
        return ReactNativeMobileWhepClientModule.whepClient
    }
  }
    private var hostingController: VideoViewController?

    required init(appContext: AppContext? = nil) {
        super.init(appContext: appContext)
        ReactNativeMobileWhepClientModule.onTrackUpdateListeners = []
        ReactNativeMobileWhepClientModule.onTrackUpdateListeners.append(self)
    }

    private func setupPlayer() {
        removeOldPlayer()
        
        guard let player = self.player else { return }
        let hostingController = VideoViewController()
        hostingController.player = player
        hostingController.view.backgroundColor = nil
        self.addSubview(hostingController.view)
        
        hostingController.view.translatesAutoresizingMaskIntoConstraints = false
        NSLayoutConstraint.activate([
            hostingController.view.topAnchor.constraint(equalTo: self.topAnchor),
            hostingController.view.bottomAnchor.constraint(equalTo: self.bottomAnchor),
            hostingController.view.leadingAnchor.constraint(equalTo: self.leadingAnchor),
            hostingController.view.trailingAnchor.constraint(equalTo: self.trailingAnchor)
        ])
        
        self.hostingController = hostingController
      
      setupPip()
    }
  
  private func setupPip() {
    if pipEnabled {
      hostingController?.setup(pictureInPictureWith: PictureInPictureController(sourceView: self))
    } else {
      hostingController?.disablePictureInPicture()
    }
  }
    
    private func removeOldPlayer() {
        hostingController?.view.removeFromSuperview()
    }
  }