import UIKit

class CustomInfoView: UIView {
    let nameLabel = UILabel()
    let muteIcon = UIImageView(image: UIImage(named: "mute_icon"))

    init(name: String, isMuted: Bool) {
        super.init(frame: .zero)
        nameLabel.text = name
        muteIcon.isHidden = !isMuted
    }

    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    // ... layout code ...
}

import UIKit

class EmptySeatView: UIView {
    let addIcon = UIImageView(image: UIImage(named: "add_icon"))
    let addLabel = UILabel()
    // ... layout code ...
}

import UIKit

class CustomAvatarView: UIView {
    let avatarView = UIImageView(frame: .zero)

    init(avatarURL: String) {
        super.init(frame: .zero)
        // load avatar URL
    }

    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    // ... layout code ...
}

class VideoWidgetProvider: VideoViewDelegate {
    /// seatInfo: Data for the seat (user info, audio/video status)
    /// viewLayer: View layer (.foreground for the top view / .background for the background view)
    func createCoGuestView(seatInfo: SeatInfo, viewLayer: ViewLayer) -> UIView? {
        let isUserOnSeat = !seatInfo.userInfo.userID.isEmpty

        switch viewLayer {
        case .foreground:
            if isUserOnSeat {
                // Non-empty seat: return custom foreground view
                let widget = CustomInfoView(name: seatInfo.userInfo.userName,
                                            isMuted: seatInfo.userInfo.microphoneStatus == .off)
                return widget
            }
            // Empty seat: return custom empty seat view
            return EmptySeatView()
        case .background:
            if isUserOnSeat {
                // Custom background view (shown when camera is off)
                let bgView = CustomAvatarView(avatarURL: seatInfo.userInfo.avatarURL)
                return bgView
            }
            return nil
        }
    }
}

class VideoWidgetProvider: VideoViewDelegate {
    /// seatInfo: Data for the seat (user info, audio/video status)
    /// viewLayer: View layer (.foreground for the top view / .background for the background view)
    func createCoHostView(seatInfo: SeatInfo, viewLayer: ViewLayer) -> UIView? {
        let isUserOnSeat = !seatInfo.userInfo.userID.isEmpty

        switch viewLayer {
        case .foreground:
            if isUserOnSeat {
                // Custom foreground view—can style differently from audience connecting
                let widget = CustomInfoView(name: seatInfo.userInfo.userName,
                                            isMuted: seatInfo.userInfo.microphoneStatus == .off)
                return widget
            }
            // Custom empty seat view—can style differently from audience connecting
            return EmptySeatView()
        case .background:
            if isUserOnSeat {
                // Custom background view (shown when camera is off)—can style differently from audience co-hosting
                let bgView = CustomAvatarView(avatarURL: seatInfo.userInfo.avatarURL)
                return bgView
            }
            return nil
        }
    }
}

// Single-user score bar example
class MyBattleScoreView: UIView {
    private let scoreView = UIView()
    // ... layout code ...
}

// Global VS panel example
class MyBattleContainer: UIView {
    private let battleTimeView = UIImageView(frame: .zero)
    // Implements countdown and VS animation
}

class VideoWidgetProvider: VideoViewDelegate {
    /// 1. Create [PK single-user info] widget (displayed above each host's video)
    func createBattleView(seatInfo: SeatInfo) -> UIView? {
        let scoreView = MyBattleScoreView()
        return scoreView
    }

    /// 2. Create [PK global container] widget (displayed above the entire video area)
    func createBattleContainerView() -> UIView? {
        let container = MyBattleContainer()
        return container
    }
}

import AtomicXCore
import SnapKit

class YourAnchorViewController: UIViewController {
    private let anchorView: AnchorView
    private let widgetProvider = VideoWidgetProvider()

    init(liveInfo: LiveInfo) {
        let videoView = LiveCoreView(viewType: .pushView)
        videoView.videoViewDelegate = widgetProvider // Assign delegate
        anchorView = AnchorView(liveInfo: liveInfo, coreView: videoView)
        super.init(nibName: nil, bundle: nil)
    }

    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }

    override func viewDidLoad() {
        super.viewDidLoad()
        view.addSubview(anchorView)

        anchorView.snp.makeConstraints { make in
            make.edges.equalToSuperview()
        }
    }
}

class YourAudienceViewController: UIViewController {
    private let audienceView: AudienceContainerView
    private let widgetProvider = VideoWidgetProvider()

    public init(roomId: String) {
        self.audienceView = AudienceContainerView(roomId: roomId)
        super.init(nibName: nil, bundle: nil)
        self.audienceView.delegate = self
    }

    required init?(coder: NSCoder) {
        fatalError("init(coder:) has not been implemented")
    }
}

extension YourAudienceViewController: AudienceContainerViewDelegate {
    func onCreateCoreView(for liveInfo: LiveInfo) -> LiveCoreView? {
        let view = LiveCoreView(viewType: .playView)
        view.videoViewDelegate = widgetProvider // Assign delegate
        return view
    }
}