import kotlinx.coroutines.*
import kotlinx.coroutines.flow.*
import io.trtc.tuikit.atomicxcore.api.CompletionHandler
import io.trtc.tuikit.atomicxcore.api.live.LiveAudienceStore
import io.trtc.tuikit.atomicxcore.api.live.LiveUserInfo

class AudienceManager(
    private val liveId: String
) {
    private val audienceStore: LiveAudienceStore = LiveAudienceStore.create(liveId)
    private val scope = CoroutineScope(Dispatchers.Main)

    // 对外暴露【全量】观众列表的状态流，方便UI层订阅
    private val _audienceList = MutableStateFlow<List<LiveUserInfo>>(emptyList())
    val audienceList: StateFlow<List<LiveUserInfo>> = _audienceList.asStateFlow()
    
    // 对外暴露观众总数的状态流
    private val _audienceCount = MutableStateFlow(0)
    val audienceCount: StateFlow<Int> = _audienceCount.asStateFlow()
    
    init {
        // 1. 订阅状态和事件，实现见下一节
        subscribeToAudienceState()
        subscribeToAudienceEvents()

        // 2. 主动拉取首屏数据
        fetchInitialAudienceList()
    }

    /// 主动获取一次观众列表快照
    private fun fetchInitialAudienceList() {
        audienceStore.fetchAudienceList(object : CompletionHandler {
            override fun onSuccess() {
                println("首次拉取观众列表成功")
                // 成功后，数据会通过下面的 state 订阅通道自动更新
            }

            override fun onFailure(code: Int, desc: String) {
                println("首次拉取观众列表失败: $desc")
            }
        })
    }
    
    // ... 后续代码
}

class AudienceManager {
    /// 订阅 state，用于获取观众总数和列表快照
    private fun subscribeToAudienceState() {
        scope.launch {
            // audienceList 是一个全量快照
            audienceStore.liveAudienceState.audienceList.collect { audienceList ->
                _audienceList.value = audienceList
            }
        }

        scope.launch {   
            // audienceCount 是实时总数
            audienceStore.liveAudienceState.audienceCount.collect { count ->
                _audienceCount.value = count
            }
        }
    }

    private val liveAudienceListener = object : LiveAudienceListener() {
        override fun onAudienceJoined(audience: LiveUserInfo) {   
            println("观众 ${newAudience.userName} 加入了直播间")
            // 增量更新：在当前列表末尾添加新观众
            val currentList = _audienceList.value.toMutableList()
            if (!currentList.any { it.userID == newAudience.userID }) {
                currentList.add(newAudience)
                _audienceList.value = currentList
            }
        }
           
        override fun onAudienceLeft(departedAudience: LiveUserInfo) {
            println("观众 ${departedAudience.userName} 离开了直播间")
            // 增量更新：从当前列表中移除离开的观众
            val currentList = _audienceList.value.toMutableList()
            currentList.removeAll { it.userID == departedAudience.userID }
            _audienceList.value = currentList
        }

    }

    /// 订阅 event，用于处理观众的实时进出事件
    private fun subscribeToAudienceEvents() {
        audienceStore.addLiveAudienceListener(liveAudienceListener)
    }
}

class AudienceManager {
    fun kick(userId: String) {
        audienceStore.kickUserOutOfRoom(userId, object : CompletionHandler {
            override fun onSuccess() {
                println("成功将用户 $userId 踢出房间")
                // 踢出成功后，您会收到 onAudienceLeft 事件
            }

            override fun onFailure(code: Int, desc: String) {
                println("踢出用户 $userId 失败: $desc")
            }
        })
    }
}

class AudienceManager {
    /// 将用户设为管理员
    fun promoteToAdmin(userId: String) {
        audienceStore.setAdministrator(userId, object : CompletionHandler {
            override fun onSuccess() {
                println("成功将用户 $userId 设为管理员")
            }
            
            override fun onFailure(code: Int, desc: String) {
                println("设置管理员失败: $desc")
            }
        })
    }

    /// 撤销用户的管理员身份
    fun revokeAdmin(userId: String) {
        audienceStore.revokeAdministrator(userId, object : CompletionHandler {
            override fun onSuccess() {
                println("成功撤销用户 $userId 的管理员身份")
            }

            override fun onFailure(code: Int, desc: String) {
                println("撤销管理员失败: $desc")
            }
        })
    }
}

import io.trtc.tuikit.atomicxcore.api.live.LiveAudienceListener
import io.trtc.tuikit.atomicxcore.api.live.LiveAudienceStore
import io.trtc.tuikit.atomicxcore.api.live.LiveUserInfo
import io.trtc.tuikit.atomicxcore.api.barrage.Barrage
import io.trtc.tuikit.atomicxcore.api.barrage.BarrageStore
import io.trtc.tuikit.atomicxcore.api.barrage.BarrageType

class LiveRoomManager(
    private val liveId: String
) {

    init {
        // 初始化两个核心管理器
        setupWelcomeMessageFlow()
    }
    
    
    private val liveAudienceListener = object : LiveAudienceListener() {
        override fun onAudienceJoined(audience: LiveUserInfo) {
            // 3. 创建一条本地提示消息
            val welcomeTip = Barrage().apply {
                liveID = liveId
                messageType = BarrageType.TEXT
                textContent = "欢迎 ${audience.userName} 进入直播间！"
            }


            // 4. 获取 BarrageStore 的实例 (得益于内部机制，这会是同一个实例)
            val barrageStore = BarrageStore.create(liveId)


            // 5. 调用 BarrageStore 的接口将其插入弹幕列表
            barrageStore.appendLocalTip(welcomeTip)
        }
    }


    private fun setupWelcomeMessageFlow() {
        // 1. 获取 LiveAudienceStore 的实例
        val audienceStore = LiveAudienceStore.create(liveId)

        // 2. 订阅观众事件
        audienceStore.addLiveAudienceListener(liveAudienceListener)
    }
}