commonMain.contact.NormalMemberImpl.kt Maven / Gradle / Ivy

Go to download
/*
 * Copyright 2019-2022 Mamoe Technologies and contributors.
 *
 * 此源代码的使用受 GNU AFFERO GENERAL PUBLIC LICENSE version 3 许可证的约束, 可以在以下链接找到该许可证.
 * Use of this source code is governed by the GNU AGPLv3 license that can be found through the following link.
 *
 * https://github.com/mamoe/mirai/blob/dev/LICENSE
 */

@file:Suppress("EXPERIMENTAL_API_USAGE")

package net.mamoe.mirai.internal.contact

import kotlinx.coroutines.CancellationException
import kotlinx.coroutines.cancel
import kotlinx.coroutines.launch
import net.mamoe.mirai.LowLevelApi
import net.mamoe.mirai.contact.*
import net.mamoe.mirai.data.MemberInfo
import net.mamoe.mirai.event.broadcast
import net.mamoe.mirai.event.events.*
import net.mamoe.mirai.internal.message.protocol.outgoing.GroupTempMessageProtocolStrategy
import net.mamoe.mirai.internal.message.protocol.outgoing.MessageProtocolStrategy
import net.mamoe.mirai.internal.message.source.OnlineMessageSourceToTempImpl
import net.mamoe.mirai.internal.message.source.createMessageReceipt
import net.mamoe.mirai.internal.network.protocol.packet.chat.TroopManagement
import net.mamoe.mirai.message.MessageReceipt
import net.mamoe.mirai.message.data.Message
import net.mamoe.mirai.utils.cast
import net.mamoe.mirai.utils.currentTimeSeconds
import kotlin.contracts.ExperimentalContracts
import kotlin.contracts.contract
import kotlin.coroutines.CoroutineContext

@OptIn(LowLevelApi::class)
@Suppress("MemberVisibilityCanBePrivate")
internal class NormalMemberImpl constructor(
    group: GroupImpl,
    parentCoroutineContext: CoroutineContext,
    memberInfo: MemberInfo,
) : NormalMember, AbstractMember(group, parentCoroutineContext, memberInfo) {

    override val joinTimestamp: Int get() = info.joinTimestamp
    override val lastSpeakTimestamp: Int get() = info.lastSpeakTimestamp

    private val messageProtocolStrategy: MessageProtocolStrategy = GroupTempMessageProtocolStrategy

    override fun toString(): String = "NormalMember($id)"

    @Suppress("DuplicatedCode")
    override suspend fun sendMessage(message: Message): MessageReceipt {
        return asFriendOrNull()?.sendMessage(message)?.convert()
            ?: asStrangerOrNull()?.sendMessage(message)?.convert()
            ?: sendMessageImpl(
                message = message,
                preSendEventConstructor = ::GroupTempMessagePreSendEvent,
                postSendEventConstructor = ::GroupTempMessagePostSendEvent.cast(),
                messageProtocolStrategy = messageProtocolStrategy
            )
    }

    private fun MessageReceipt.convert(): MessageReceipt {
        return OnlineMessageSourceToTempImpl(source, this@NormalMemberImpl).createMessageReceipt(
            this@NormalMemberImpl,
            doLightRefine = false //we've already did
        )
    }


    @Suppress("PropertyName")
    internal var _nameCard: String = memberInfo.nameCard

    @Suppress("PropertyName")
    internal var _specialTitle: String = memberInfo.specialTitle

    @Suppress("PropertyName")
    var _muteTimestamp: Int = memberInfo.muteTimestamp

    @Suppress("PropertyName")
    var _nudgeTimestamp: Long = 0L

    override val muteTimeRemaining: Int
        get() = if (_muteTimestamp == 0 || _muteTimestamp == 0xFFFFFFFF.toInt()) {
            0
        } else {
            (_muteTimestamp - currentTimeSeconds().toInt()).coerceAtLeast(0)
        }

    override var nameCard: String
        get() = _nameCard
        set(newValue) {
            if (id != bot.id) {
                group.checkBotPermission(MemberPermission.ADMINISTRATOR)
            }
            if (_nameCard != newValue) {
                val oldValue = _nameCard
                _nameCard = newValue
                launch {
                    bot.network.sendWithoutExpect(
                        TroopManagement.EditGroupNametag(
                            bot.client,
                            this@NormalMemberImpl,
                            newValue,
                        )
                    )
                    MemberCardChangeEvent(oldValue, newValue, this@NormalMemberImpl).broadcast()
                }
            }
        }

    override var specialTitle: String
        get() = _specialTitle
        set(newValue) {
            group.checkBotPermission(MemberPermission.OWNER)
            if (_specialTitle != newValue) {
                val oldValue = _specialTitle
                _specialTitle = newValue
                launch {
                    bot.network.sendWithoutExpect(
                        TroopManagement.EditSpecialTitle(
                            bot.client,
                            this@NormalMemberImpl,
                            newValue,
                        )
                    )
                    MemberSpecialTitleChangeEvent(oldValue, newValue, this@NormalMemberImpl, null).broadcast()
                }
            }
        }

    override suspend fun mute(durationSeconds: Int) {
        check(this.id != bot.id) {
            "A bot can't mute itself."
        }
        require(durationSeconds > 0) {
            "durationSeconds must greater than zero"
        }
        checkBotPermissionHigherThanThis("mute")
        bot.network.sendAndExpect(
            TroopManagement.Mute(
                client = bot.client,
                groupCode = group.id,
                memberUin = [email protected],
                timeInSecond = durationSeconds,
            ), 5000, 2
        )

        @Suppress("RemoveRedundantQualifierName") // or unresolved reference
        (net.mamoe.mirai.event.events.MemberMuteEvent(this@NormalMemberImpl, durationSeconds, null).broadcast())
        this._muteTimestamp = currentTimeSeconds().toInt() + durationSeconds
    }

    override suspend fun unmute() {
        checkBotPermissionHigherThanThis("unmute")
        bot.network.sendAndExpect(
            TroopManagement.Mute(
                client = bot.client,
                groupCode = group.id,
                memberUin = [email protected],
                timeInSecond = 0,
            ), 5000, 2
        )

        @Suppress("RemoveRedundantQualifierName") // or unresolved reference
        (net.mamoe.mirai.event.events.MemberUnmuteEvent(this@NormalMemberImpl, null).broadcast())
        this._muteTimestamp = 0
    }

    override suspend fun kick(message: String, block: Boolean) {
        checkBotPermissionHigherThanThis("kick")
        check(group.members[this.id] != null) {
            "Member ${this.id} had already been kicked from group ${group.id}"
        }
        val response: TroopManagement.Kick.Response = bot.network.sendAndExpect(
            TroopManagement.Kick(
                client = bot.client,
                groupCode = group.groupCode,
                memberId = id,
                message = message,
                ban = block
            ), 5000, 2
        )

        // Note: when member not found, result is still true.

        if (response.ret == 255) error("Operation too fast") // https://github.com/mamoe/mirai/issues/1503
        check(response.success) { "kick failed: ${response.ret}" }

        @Suppress("INVISIBLE_MEMBER", "INVISIBLE_REFERENCE")
        group.members.delegate.removeIf { it.id == [email protected] }
        [email protected](CancellationException("Kicked by bot"))
        MemberLeaveEvent.Kick(this@NormalMemberImpl, null).broadcast()
    }

    override suspend fun modifyAdmin(operation: Boolean) {
        checkBotPermissionHighest("modifyAdmin")

        val origin = [email protected]
        val new = if (operation) {
            MemberPermission.ADMINISTRATOR
        } else {
            MemberPermission.MEMBER
        }

        if (origin == new) return

        val resp: TroopManagement.ModifyAdmin.Response = bot.network.sendAndExpect(
            TroopManagement.ModifyAdmin(
                client = bot.client,
                member = this@NormalMemberImpl,
                operation = operation,
            ), 5000, 2
        ) as TroopManagement.ModifyAdmin.Response

        check(resp.success) {
            "Failed to modify admin, cause: ${resp.msg}"
        }

        [email protected] = new

        MemberPermissionChangeEvent(this@NormalMemberImpl, origin, new).broadcast()
    }
}

internal fun Member.checkBotPermissionHighest(operationName: String) {
    check(group.botPermission == MemberPermission.OWNER) {
        throw PermissionDeniedException(
            "`$operationName` operation requires the OWNER permission, while bot has ${group.botPermission}",
        )
    }
}

internal fun Member.checkBotPermissionHigherThanThis(operationName: String) {
    check(group.botPermission > this.permission) {
        throw PermissionDeniedException(
            "`$operationName` operation requires a higher permission, while " +
                    "${group.botPermission} < ${this.permission}",
        )
    }
}

@OptIn(ExperimentalContracts::class)
internal fun Member.checkIsMemberImpl(): NormalMemberImpl {
    contract {
        returns() implies (this@checkIsMemberImpl is NormalMemberImpl)
    }
    check(this is NormalMemberImpl) { "A Member instance is not instance of MemberImpl. Don't interlace two protocol implementations together!" }
    return this
}