import androidx.appcompat.app.AppCompatActivity
import io.trtc.tuikit.atomicxcore.api.CompletionHandler
import io.trtc.tuikit.atomicxcore.api.live.CoHostLayoutTemplate
import io.trtc.tuikit.atomicxcore.api.live.CoHostStore

// 主播A的Activity
class AnchorAActivity : AppCompatActivity() {
    private val liveId = "主播A的房间ID"
    private val coHostStore = CoHostStore.create(liveId)

    // 用户点击"连线"按钮，并选择了主播B
    fun inviteHostB(targetHostLiveId: String) {
        val layout = CoHostLayoutTemplate.HOST_DYNAMIC_GRID // 选择一个布局模板
        val timeout = 30 // 邀请超时时间（秒）

        coHostStore.requestHostConnection(
            targetHostLiveID = targetHostLiveId,
            layoutTemplate = layout,
            timeout = timeout,
            extraInfo = null,
            completion = object : CompletionHandler {
                override fun onSuccess() { 
                    println("连线邀请已发送，等待对方处理...")
                }

                override fun onFailure(code: Int, desc: String) { 
                    println("邀请发送失败: $desc")
                }
            }
        )
    }
}

import android.os.Bundle
import androidx.appcompat.app.AppCompatActivity
import io.trtc.tuikit.atomicxcore.api.live.CoHostListener
import io.trtc.tuikit.atomicxcore.api.live.CoHostStore
import io.trtc.tuikit.atomicxcore.api.live.SeatUserInfo

class AnchorAActivity : AppCompatActivity() {
    private val liveId = "主播A的房间ID"
    private val coHostStore = CoHostStore.create(liveId)
    private val coHostListener = object : CoHostListener() {
        override fun onCoHostRequestAccepted(invitee: SeatUserInfo) {
            println("主播 ${invitee.userName} 同意了你的连线邀请")
        }

        override fun onCoHostRequestRejected(invitee: SeatUserInfo) {
            println("主播 ${invitee.userName} 拒绝了你的邀请")
        }

        override fun onCoHostRequestTimeout(inviter: SeatUserInfo, invitee: SeatUserInfo) { 
            println("邀请超时，对方未回应")
        }
    }

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        setContentView(R.layout.activity_anchor_a)
        
        // 初始化CoHostStore
        coHostStore.addCoHostListener(coHostListener)
    }
}

import androidx.appcompat.app.AppCompatActivity
import io.trtc.tuikit.atomicxcore.api.live.CoHostListener
import io.trtc.tuikit.atomicxcore.api.live.CoHostStore
import io.trtc.tuikit.atomicxcore.api.live.SeatUserInfo

// 主播B的Activity
class AnchorBActivity : AppCompatActivity() {
    private val liveId = "主播B的房间ID"
    private val coHostStore = CoHostStore.create(liveId)
    
    private val coHostListener = object : CoHostListener() {
        override fun onCoHostRequestReceived(inviter: SeatUserInfo, extensionInfo: String) {
            println("收到主播 ${inviter.userName} 的连线邀请")
            // 显示邀请对话框,响应联线邀请
            // showInvitationDialog(inviter)
        }
    }
}

// AnchorBActivity 的一部分
fun acceptInvitation(fromHostLiveId: String) {
    coHostStore.acceptHostConnection(fromHostLiveID = fromHostLiveId, completion = null)
}

fun rejectInvitation(fromHostLiveId: String) {
    coHostStore.rejectHostConnection(fromHostLiveID = fromHostLiveId, completion = null)
}

// AnchorAActivity 的一部分
class AnchorAActivity : AppCompatActivity() {
    private val liveId = "主播A的房间ID"
    private val battleStore = BattleStore.create(liveId)

    fun startPK(opponentUserId: String) {
        val config = BattleConfig(duration = 300) // PK持续5分钟
        battleStore.requestBattle(
            config = config,
            userIDList = listOf(opponentUserId),
            timeout = 30,
            completion = null
        )
    }
}

 // AnchorAActivity 的一部分
class AnchorAActivity : AppCompatActivity() {
    private val liveId = "主播A的房间ID"
    private val battleStore = BattleStore.create(liveId)
    
    private val battleListener = object : BattleListener() {
        override fun onBattleStarted(
            battleInfo: BattleInfo,
            inviter: SeatUserInfo,
            invitees: List<SeatUserInfo>
        ) {
            println("PK 开始")
        }

        override fun onBattleEnded(battleInfo: BattleInfo, reason: BattleEndedReason?) {
            println("PK 结束")
        }
    }

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        // ... 其他初始化代码 ...
        battleStore.addBattleListener(battleListener)
    }
}

// 在 AnchorBActivity 中添加
class AnchorBActivity : AppCompatActivity() {
    private val liveId = "主播B的房间ID"
    private val battleStore = BattleStore.create(liveId)

    private val battleListener = object : BattleListener() {
        override fun onBattleRequestReceived(battleID: String, inviter: SeatUserInfo, invitee: SeatUserInfo) {
            println("收到主播 ${inviter.userName} 的PK挑战")
            // 弹出对话框，让主播B选择"接受"或"拒绝"
            // showPKChallengeDialog(battleID, inviter)
            }
        }
    
        override fun onCreate(savedInstanceState: Bundle?) {
            super.onCreate(savedInstanceState)
            // ... 其他初始化代码 ...
            battleStore.addBattleListener(battleListener)
        }
}

// AnchorBActivity 的一部分
// 用户点击"接受挑战"
fun acceptPK(battleId: String) {
    battleStore.acceptBattle(battleID = battleId, completion = null)
}

// 用户点击"拒绝挑战"
fun rejectPK(battleId: String) {
    battleStore.rejectBattle(battleID = battleId, completion = null)
}

// AnchorAActivity 的一部分

// 主播主动结束 PK
fun stopPK(battleId: String) {
    // 调用 exitBattle 接口退出当前的 PK 对战
    battleStore.exitBattle(battleID = battleId, completion = object : CompletionHandler {
        override fun onSuccess() {
            println("PK 已结束")
            // 此时 UI 会收到 onBattleEnded 事件，请在事件回调中处理 UI 刷新
        }

        override fun onFailure(code: Int, desc: String) {
            println("结束 PK 失败: $desc")
        }
    })
}

// AnchorAActivity 的一部分

// 主播主动断开连线
fun stopConnection() {
    // 调用 exitHostConnection 接口断开与其他主播的连接
    coHostStore.exitHostConnection(completion = object : CompletionHandler {
        override fun onSuccess() {
            println("连线已断开，恢复单人直播")
            // 此时 UI 会收到 onCoHostUserLeft 事件
        }

        override fun onFailure(code: Int, desc: String) {
            println("断开连线失败: $desc")
        }
    })
}

// 在 AnchorAActivity 中补充监听逻辑

// 1. 监听 PK 结束事件
private val battleListener = object : BattleListener() {
    // ... 其他回调 ...
    
    override fun onBattleEnded(battleInfo: BattleInfo, reason: BattleEndedReason?) {
        println("收到 PK 结束事件，原因: $reason")
        // 在此处移除 PK 分数视图、进度条等 UI
        // runOnUiThread { removeBattleUI() }
    }
}

// 2. 监听连线断开事件
private val coHostListener = object : CoHostListener() {
    // ... 其他回调 ...
    
    override fun onCoHostUserLeft(userInfo: SeatUserInfo) {
        println("主播 ${userInfo.userName} 已离开连线")
        // 在此处移除对方的视频画面视图，恢复单人直播布局
        // runOnUiThread { resetToSingleStreamLayout() }
    }
}

import android.content.Context
import android.graphics.Color
import android.view.Gravity
import android.widget.LinearLayout
import android.widget.TextView

// 自定义的用户信息悬浮视图（前景）
class CustomSeatView(context: Context) : LinearLayout(context) {
    private val nameLabel: TextView

    init {
        orientation = VERTICAL
        setBackgroundColor(Color.parseColor("#80000000")) // 半透明黑色背景

        nameLabel = TextView(context).apply {
            setTextColor(Color.WHITE)
            textSize = 14f
            gravity = Gravity.CENTER
        }

        addView(nameLabel)
        val layoutParams = nameLabel.layoutParams as LayoutParams
        layoutParams.setMargins(5, 0, 5, 5)
    }

    fun setUserName(userName: String) {
        nameLabel.text = userName
    }
}

import android.content.Context
import android.graphics.Color
import android.view.Gravity
import android.widget.ImageView
import android.widget.LinearLayout

// 自定义的头像占位视图（背景）
class CustomAvatarView(context: Context) : LinearLayout(context) {
    private val avatarImageView: ImageView

    init {
        orientation = VERTICAL
        gravity = Gravity.CENTER
        setBackgroundColor(Color.TRANSPARENT)

        avatarImageView = ImageView(context).apply {
            setColorFilter(Color.GRAY)
            scaleType = ImageView.ScaleType.CENTER_CROP
        }

        addView(avatarImageView)
        val layoutParams = avatarImageView.layoutParams as LayoutParams
        layoutParams.width = 120 // 60dp
        layoutParams.height = 120 // 60dp
    }
}

import com.tencent.cloud.tuikit.engine.room.TUIRoomDefine
import android.view.View
import androidx.appcompat.app.AppCompatActivity
import io.trtc.tuikit.atomicxcore.api.view.VideoViewAdapter
import io.trtc.tuikit.atomicxcore.api.view.ViewLayer

// 在您的Activity中，实现 VideoViewAdapter 接口
class YourActivity : AppCompatActivity(), VideoViewAdapter {

    // ... 其他代码 ...
    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        // 1. 设置适配器
        liveCoreView.setVideoViewAdapter(this)
    }

    // 2. 完整实现接口方法，处理两种 viewLayer
    override fun createCoHostView(coHostUser: TUIRoomDefine.SeatFullInfo?, viewLayer: ViewLayer?): View? {
        val seatInfo = coHostUser ?: return null
        val userId = seatInfo.userId
        if (userId.isNullOrEmpty()) {
            return null
        }

        return when (viewLayer) {
            ViewLayer.FOREGROUND -> {
                val seatView = CustomSeatView(this)
                seatView.setUserName(seatInfo.userName ?: "")
                seatView
            }
            ViewLayer.BACKGROUND -> {
                val avatarView = CustomAvatarView(this)
                // 您可以在这里通过 seatInfo.userAvatar 加载用户真实头像
                avatarView
            }
            null -> null
        }
    }
}

import android.content.Context
import android.graphics.Color
import android.view.Gravity
import android.widget.LinearLayout
import android.widget.TextView
import com.tencent.cloud.tuikit.engine.extension.TUILiveBattleManager.BattleUser
import io.trtc.tuikit.atomicxcore.api.live.BattleStore
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.launch

// 自定义PK 用户视图
class CustomBattleUserView(
    context: Context,
    private val liveId: String,
    private val battleUser: BattleUser
) : LinearLayout(context) {

    private lateinit var scoreView: LinearLayout
    private lateinit var scoreLabel: TextView
    private val battleStore: BattleStore

    init {
        orientation = VERTICAL
        gravity = Gravity.BOTTOM or Gravity.END
        setBackgroundColor(Color.TRANSPARENT)
        isClickable = false

        battleStore = BattleStore.create(liveId)

        // UI 布局
        setupUI()
        // 订阅分数变化
        subscribeBattleState()
    }

    private fun setupUI() {
        scoreView = LinearLayout(context).apply {
            setBackgroundColor(Color.parseColor("#66000000"))
            orientation = VERTICAL
            gravity = Gravity.CENTER
        }

        scoreLabel = TextView(context).apply {
            setTextColor(Color.WHITE)
            textSize = 14f
            gravity = Gravity.CENTER
        }

        scoreView.addView(scoreLabel)
        addView(scoreView)

        val layoutParams = scoreView.layoutParams as LayoutParams
        layoutParams.width = LayoutParams.WRAP_CONTENT
        layoutParams.height = 48 // 24dp
        layoutParams.setMargins(0, 0, 10, 10)
    }

    // 订阅 PK 分数变化
    private fun subscribeBattleState() {
        CoroutineScope(Dispatchers.Main).launch {
            battleStore.battleState.battleScore.collect { battleScore ->
                val score = battleScore[battleUser.userId] ?: 0
                // 更新 UI
                scoreLabel.text = score.toString()
            }
        }
    }
}

  // 让您的Activity实现 VideoViewAdapter 接口
class YourActivity : AppCompatActivity(), VideoViewAdapter {

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        // 1. 设置适配器
        liveCoreView.setVideoViewAdapter(this)
    }

    override fun createBattleView(battleUser: BattleUser?): View? {
        battleUser ?: return null
        // 2. CustomBattleUserView 是您自定义的PK用户信息视图
        return CustomBattleUserView(this, liveId, battleUser)
    }

}

// 让您的Activity实现 VideoViewAdapter 接口并设置适配器
class YourActivity : AppCompatActivity(), VideoViewAdapter {

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        // 1. 设置适配器
        liveCoreView.setVideoViewAdapter(this)
    }

    override fun createBattleContainerView(): View? {
        // 2. 自定义PK视图
        return CustomBattleContainerView(this)
    }
}