48 lines
1.3 KiB
Kotlin
48 lines
1.3 KiB
Kotlin
package com.keylesspalace.tusky.entity
|
|
|
|
import com.google.gson.annotations.SerializedName
|
|
import java.util.Date
|
|
|
|
data class Poll(
|
|
val id: String,
|
|
@SerializedName("expires_at") val expiresAt: Date?,
|
|
val expired: Boolean,
|
|
val multiple: Boolean,
|
|
@SerializedName("votes_count") val votesCount: Int,
|
|
@SerializedName("voters_count") val votersCount: Int?, // nullable for compatibility with Pleroma
|
|
val options: List<PollOption>,
|
|
val voted: Boolean,
|
|
@SerializedName("own_votes") val ownVotes: List<Int>?
|
|
) {
|
|
|
|
fun votedCopy(choices: List<Int>): Poll {
|
|
val newOptions = options.mapIndexed { index, option ->
|
|
if (choices.contains(index)) {
|
|
option.copy(votesCount = option.votesCount + 1)
|
|
} else {
|
|
option
|
|
}
|
|
}
|
|
|
|
return copy(
|
|
options = newOptions,
|
|
votesCount = votesCount + choices.size,
|
|
votersCount = votersCount?.plus(1),
|
|
voted = true
|
|
)
|
|
}
|
|
|
|
fun toNewPoll(creationDate: Date) = NewPoll(
|
|
options.map { it.title },
|
|
expiresAt?.let {
|
|
((it.time - creationDate.time) / 1000).toInt() + 1
|
|
} ?: 3600,
|
|
multiple
|
|
)
|
|
}
|
|
|
|
data class PollOption(
|
|
val title: String,
|
|
@SerializedName("votes_count") val votesCount: Int
|
|
)
|