122 lines
3.9 KiB
Kotlin
122 lines
3.9 KiB
Kotlin
package dev.inmo.tgbotapi.types
|
|
|
|
import dev.inmo.micro_utils.language_codes.IetfLanguageCode
|
|
import dev.inmo.micro_utils.language_codes.IetfLanguageCodeSerializer
|
|
import dev.inmo.tgbotapi.CommonAbstracts.WithOptionalLanguageCode
|
|
import dev.inmo.tgbotapi.types.chat.abstracts.PrivateChat
|
|
import dev.inmo.tgbotapi.types.chat.extended.ExtendedPrivateChatImpl
|
|
import dev.inmo.tgbotapi.utils.*
|
|
import kotlinx.serialization.*
|
|
import kotlinx.serialization.descriptors.SerialDescriptor
|
|
import kotlinx.serialization.encoding.Decoder
|
|
import kotlinx.serialization.encoding.Encoder
|
|
import kotlinx.serialization.json.*
|
|
|
|
@Serializable(UserSerializer::class)
|
|
sealed class User : PrivateChat
|
|
|
|
@Serializable
|
|
data class CommonUser(
|
|
override val id: UserId,
|
|
@SerialName(firstNameField)
|
|
override val firstName: String,
|
|
@SerialName(lastNameField)
|
|
override val lastName: String = "",
|
|
@SerialName(usernameField)
|
|
override val username: Username? = null,
|
|
@SerialName(languageCodeField)
|
|
@Serializable(IetfLanguageCodeSerializer::class)
|
|
override val ietfLanguageCode: IetfLanguageCode? = null
|
|
) : User(), WithOptionalLanguageCode {
|
|
constructor(
|
|
id: UserId,
|
|
firstName: String,
|
|
lastName: String = "",
|
|
username: Username? = null,
|
|
languageCode: String
|
|
) : this(id, firstName, lastName, username, IetfLanguageCode(languageCode))
|
|
}
|
|
|
|
@PreviewFeature
|
|
typealias ExtendedUser = ExtendedPrivateChatImpl
|
|
|
|
@Serializable(UserSerializer::class)
|
|
sealed class Bot : User() {
|
|
abstract override val username: Username
|
|
}
|
|
|
|
@Serializable
|
|
data class CommonBot(
|
|
override val id: UserId,
|
|
@SerialName(usernameField)
|
|
override val username: Username,
|
|
@SerialName(firstNameField)
|
|
override val firstName: String,
|
|
@SerialName(lastNameField)
|
|
override val lastName: String = ""
|
|
) : Bot() {
|
|
@SerialName(isBotField)
|
|
private val isBot = true
|
|
}
|
|
|
|
@Serializable
|
|
data class ExtendedBot(
|
|
override val id: UserId,
|
|
@SerialName(usernameField)
|
|
override val username: Username,
|
|
@SerialName(firstNameField)
|
|
override val firstName: String,
|
|
@SerialName(lastNameField)
|
|
override val lastName: String = "",
|
|
@SerialName(canJoinGroupsField)
|
|
val canJoinGroups: Boolean = false,
|
|
@SerialName(canReadAllGroupMessagesField)
|
|
val canReadAllGroupMessages: Boolean = false,
|
|
@SerialName(supportInlineQueriesField)
|
|
val supportsInlineQueries: Boolean = false
|
|
) : Bot() {
|
|
@SerialName(isBotField)
|
|
private val isBot = true
|
|
}
|
|
|
|
|
|
@RiskFeature
|
|
object UserSerializer : KSerializer<User> {
|
|
private val internalSerializer = JsonObject.serializer()
|
|
override val descriptor: SerialDescriptor = internalSerializer.descriptor
|
|
override fun deserialize(decoder: Decoder): User {
|
|
val asJson = internalSerializer.deserialize(decoder)
|
|
|
|
return when {
|
|
asJson[isBotField] ?.jsonPrimitive ?.booleanOrNull != true -> nonstrictJsonFormat.decodeFromJsonElement(
|
|
CommonUser.serializer(),
|
|
asJson
|
|
)
|
|
else -> {
|
|
if ((asJson[canJoinGroupsField]
|
|
?: asJson[canReadAllGroupMessagesField]
|
|
?: asJson[supportInlineQueriesField]) != null
|
|
) {
|
|
nonstrictJsonFormat.decodeFromJsonElement(
|
|
ExtendedBot.serializer(),
|
|
asJson
|
|
)
|
|
} else {
|
|
nonstrictJsonFormat.decodeFromJsonElement(
|
|
CommonBot.serializer(),
|
|
asJson
|
|
)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
override fun serialize(encoder: Encoder, value: User) {
|
|
when (value) {
|
|
is CommonUser -> CommonUser.serializer().serialize(encoder, value)
|
|
is CommonBot -> CommonBot.serializer().serialize(encoder, value)
|
|
is ExtendedBot -> ExtendedBot.serializer().serialize(encoder, value)
|
|
}
|
|
}
|
|
}
|