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

// Host A's Activity
class HostAActivity : AppCompatActivity() {
    private val liveId = "hostA_roomID" // Host A's Room ID
    private val coHostStore = CoHostStore.create(liveId)

    // User clicks the "Co-host" button and selects Host B
    fun inviteHostB(targetHostLiveId: String) {
        val layout = CoHostLayoutTemplate.HOST_DYNAMIC_GRID // Select a layout template
        val timeout = 30 // Invitation timeout (seconds)

        coHostStore.requestHostConnection(
            targetHostLiveID = targetHostLiveId,
            layoutTemplate = layout,
            timeout = timeout,
            extraInfo = null,
            completion = object : CompletionHandler {
                override fun onSuccess() { 
                    println("Co-hosting invitation sent, waiting for response...")
                }

                override fun onFailure(code: Int, desc: String) { 
                    println("Invitation failed to send: $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 HostAActivity : AppCompatActivity() {
    private val liveId = "hostA_roomID" // Host A's Room ID
    private val coHostStore = CoHostStore.create(liveId)
    private val coHostListener = object : CoHostListener() {
        override fun onCoHostRequestAccepted(invitee: SeatUserInfo) {
            println("Host ${invitee.userName} accepted your co-hosting invitation")
        }

        override fun onCoHostRequestRejected(invitee: SeatUserInfo) {
            println("Host ${invitee.userName} rejected your invitation")
        }

        override fun onCoHostRequestTimeout(inviter: SeatUserInfo, invitee: SeatUserInfo) { 
            println("Invitation timed out, no response received")
        }
    }

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        // ... initialization code ...
        
        // Initialize 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

// Host B's Activity
class HostBActivity : AppCompatActivity() {
    private val liveId = "hostB_roomID" // Host B's Room ID
    private val coHostStore = CoHostStore.create(liveId)
    
    private val coHostListener = object : CoHostListener() {
        override fun onCoHostRequestReceived(inviter: SeatUserInfo, extensionInfo: String) {
            println("Received co-hosting invitation from Host ${inviter.userName}")
            // Display invitation dialog to respond
            // showInvitationDialog(inviter)
        }
    }
}

// Part of HostBActivity
fun acceptInvitation(fromHostLiveId: String) {
    coHostStore.acceptHostConnection(fromHostLiveID = fromHostLiveId, completion = null)
}

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

// Part of HostAActivity
class HostAActivity : AppCompatActivity() {
    private val liveId = "hostA_roomID" // Host A's Room ID
    private val battleStore = BattleStore.create(liveId)

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

// Part of HostAActivity
class HostAActivity : AppCompatActivity() {
    private val liveId = "hostA_roomID" // Host A's Room ID
    private val battleStore = BattleStore.create(liveId)
    
    private val battleListener = object : BattleListener() {
        override fun onBattleStarted(
            battleInfo: BattleInfo,
            inviter: SeatUserInfo,
            invitees: List<SeatUserInfo>
        ) {
            println("PK started")
        }

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

    override fun onCreate(savedInstanceState: Bundle?) {
        super.onCreate(savedInstanceState)
        // ... other initialization code ...
        battleStore.addBattleListener(battleListener)
    }
}

// Add to HostBActivity
class HostBActivity : AppCompatActivity() {
    private val liveId = "hostB_roomID" // Host B's Room ID
    private val battleStore = BattleStore.create(liveId)

    private val battleListener = object : BattleListener() {
        override fun onBattleRequestReceived(battleID: String, inviter: SeatUserInfo, invitee: SeatUserInfo) {
            println("Received PK challenge from Host ${inviter.userName}")
            // Pop up a dialog for Host B to choose "Accept" or "Reject"
            // showPKChallengeDialog(battleID, inviter)
            }
        }
    
        override fun onCreate(savedInstanceState: Bundle?) {
            super.onCreate(savedInstanceState)
            // ... other initialization code ...
            battleStore.addBattleListener(battleListener)
        }
}

// Part of HostBActivity
// User clicks "Accept Challenge"
fun acceptPK(battleId: String) {
    battleStore.acceptBattle(battleID = battleId, completion = null)
}

// User clicks "Reject Challenge"
fun rejectPK(battleId: String) {
    battleStore.rejectBattle(battleID = battleId, completion = null)
}

fun stopPK(battleId: String) {
    battleStore.exitBattle(battleID = battleId, completion = object : CompletionHandler {
        override fun onSuccess() {
            println("PK ended")
            // Handle UI refresh in onBattleEnded event
        }
        override fun onFailure(code: Int, desc: String) {
            println("Failed to end PK: $desc")
        }
    })
}

  fun stopConnection() {
    coHostStore.exitHostConnection(completion = object : CompletionHandler {
        override fun onSuccess() {
            println("Co-hosting disconnected, back to solo live")
            // UI will receive onCoHostUserLeft event
        }
        override fun onFailure(code: Int, desc: String) {
            println("Failed to disconnect co-hosting: $desc")
        }
    })
}

  private val battleListener = object : BattleListener() {
    override fun onBattleEnded(battleInfo: BattleInfo, reason: BattleEndedReason?) {
        println("Received PK end event, reason: $reason")
        // Remove PK score view, progress bar, etc.
        // runOnUiThread { removeBattleUI() }
    }
}

private val coHostListener = object : CoHostListener() {
    override fun onCoHostUserLeft(userInfo: SeatUserInfo) {
        println("Host ${userInfo.userName} has left the co-hosting session")
        // Remove the other host's video view and restore solo live layout
        // runOnUiThread { resetToSingleStreamLayout() }
    }
}

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

// Custom user information overlay view (Foreground)
class CustomSeatView(context: Context) : LinearLayout(context) {
    private val nameLabel: TextView

    init {
        orientation = VERTICAL
        setBackgroundColor(Color.parseColor("#80000000")) // Semi-transparent black background

        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 com.tencent.cloud.tuikit.engine.room.TUIRoomDefine
import android.view.View
import androidx.appcompat.app.AppCompatActivity
import io.trtc.tuikit.atomicxcore.api.ViewLayer
import io.trtc.tuikit.atomicxcore.api.VideoViewAdapter

// 1. In your Activity, implement the VideoViewAdapter interface
class YourActivity : AppCompatActivity(), VideoViewAdapter {

    // ... Other code ...

    // 2. Fully implement the interface method to handle both view layers
    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)
                // You can load the user's real avatar here using seatInfo.userAvatar
                avatarView
            }
            null -> null
        }
    }
}

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

// 1. In your Activity, implement the VideoViewAdapter interface
class YourActivity : AppCompatActivity(), VideoViewAdapter {

    // ... Other code ...

    // 2. Fully implement the interface method to handle both view layers
    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)
                // You can load the user's real avatar here using 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.BattleStore
import kotlinx.coroutines.CoroutineScope
import kotlinx.coroutines.Dispatchers
import kotlinx.coroutines.launch

// Custom PK User View
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 Layout
        setupUI()
        // Subscribe to score changes
        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)
    }

    // Subscribe to PK score changes
    private fun subscribeBattleState() {
        CoroutineScope(Dispatchers.Main).launch {
            battleStore.battleState.battleScore.collect { battleScore ->
                val score = battleScore[battleUser.userId] ?: 0
                // Update UI
                scoreLabel.text = score.toString()
            }
        }
    }
}

// 1. Have your Activity implement the VideoViewAdapter interface
class YourActivity : AppCompatActivity(), VideoViewAdapter {

    override fun createBattleView(battleUser: BattleUser?): View? {
        battleUser ?: return null
        // CustomBattleUserView is your custom PK user information view
        return CustomBattleUserView(this, liveId, battleUser)
    }

}

// Have your Activity implement the VideoViewAdapter interface and set the adapter
class YourActivity : AppCompatActivity(), VideoViewAdapter {
    override fun createBattleContainerView(): View? {
        return CustomBattleContainerView(this)
    }
}