Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Provide the same functions of Permissions to MessageFlags #203

Merged
merged 3 commits into from
Mar 10, 2021
Merged

Provide the same functions of Permissions to MessageFlags #203

Changes from 2 commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
de8e116
Provide the same functions of Permissions to MessageFlags
HopeBaron Mar 5, 2021
70f7260
Apply suggestions
HopeBaron Mar 9, 2021
9de3a20
Change toString to indicate the type as well
HopeBaron Mar 9, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
55 changes: 45 additions & 10 deletions common/src/main/kotlin/entity/DiscordMessage.kt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -284,32 +284,57 @@ enum class MessageFlag(val code: Int) {
}

@Serializable(with = MessageFlags.Serializer::class)
data class MessageFlags internal constructor(val code: Int) {
class MessageFlags internal constructor(val code: Int) {

val flags = MessageFlag.values().filter { code and it.code != 0 }

operator fun contains(flag: MessageFlag) = flag in flags
operator fun contains(flag: MessageFlag) = flag.code and this.code == flag.code

operator fun plus(flags: MessageFlags): MessageFlags = when {
code and flags.code == flags.code -> this
else -> MessageFlags(this.code or flags.code)
}
operator fun contains(flags: MessageFlags) = flags.code and this.code == flags.code

operator fun plus(flags: MessageFlags): MessageFlags = MessageFlags(this.code or flags.code)

operator fun plus(flags: MessageFlag): MessageFlags = MessageFlags(this.code or flags.code)

operator fun minus(flags: MessageFlags): MessageFlags = MessageFlags(this.code xor flags.code)

operator fun minus(flags: MessageFlag): MessageFlags = MessageFlags(this.code xor flags.code)

operator fun minus(flag: MessageFlag): MessageFlags = when {
code and flag.code == flag.code -> MessageFlags(code xor flag.code)
else -> this
}

inline fun copy(block: Builder.() -> Unit): MessageFlags {
val builder = Builder(code)
builder.apply(block)
return builder.flags()
}

override fun toString(): String = flags.toString()

companion object {
inline operator fun invoke(builder: Builder.() -> Unit): MessageFlags {
return Builder().apply(builder).flags()
}

operator fun invoke(value: Int) = MessageFlags(value)

operator fun invoke(vararg flags: MessageFlag) = MessageFlags {
flags.forEach { +it }
}

operator fun invoke(vararg flags: MessageFlags) = MessageFlags {
flags.forEach { +it }
}

operator fun invoke(flags: Iterable<MessageFlag>) = MessageFlags {
flags.forEach { +it }
}


@JvmName("invokeWithFlags")
operator fun invoke(flags: Iterable<MessageFlags>) = MessageFlags {
flags.forEach { +it }
}


}

internal object Serializer : KSerializer<MessageFlags> {
Expand Down Expand Up @@ -337,6 +362,16 @@ data class MessageFlags internal constructor(val code: Int) {
}
}

operator fun MessageFlags.unaryPlus() {
[email protected] = [email protected] or code
}

operator fun MessageFlags.unaryMinus() {
if ([email protected] and code == code) {
[email protected] = [email protected] xor code
}
}

fun flags() = MessageFlags(code)
}

Expand Down