package allin.data.postgres

import allin.data.ParticipationDataSource
import allin.data.postgres.entities.*
import allin.model.Participation
import org.ktorm.database.Database
import org.ktorm.dsl.and
import org.ktorm.dsl.eq
import org.ktorm.dsl.insert
import org.ktorm.entity.*

class PostgresParticipationDataSource(private val database: Database) : ParticipationDataSource {

    override fun addParticipation(participation: Participation) {
        database.insert(ParticipationsEntity) {
            set(it.id, participation.id)
            set(it.betId, participation.betId)
            set(it.username, participation.username)
            set(it.answer, participation.answer)
            set(it.stake, participation.stake)
        }

        val betInfo = database.betInfos.find { it.id eq participation.betId } ?: BetInfoEntity {
            this.id = participation.betId
            this.totalStakes = 0
            this.totalParticipants = 0
        }

        betInfo.totalStakes += participation.stake
        betInfo.totalParticipants++
        database.betInfos.update(betInfo)

        database.betAnswerInfos.filter { it.betId eq participation.betId }.forEach {
            if (it.response == participation.answer) {
                it.totalStakes += participation.stake
            }
            val probability = it.totalStakes / betInfo.totalStakes.toFloat()
            it.odds = 1 / probability
            it.flushChanges()
        }
    }

    override fun getParticipationFromBetId(betid: String): List<Participation> =
        database.participations.filter { it.betId eq betid }.map { it.toParticipation() }

    override fun getParticipationFromUserId(username: String, betid: String): List<Participation> =
        database.participations.filter {
            (ParticipationsEntity.betId eq betid) and (ParticipationsEntity.username eq username)
        }.map { it.toParticipation() }

    override fun deleteParticipation(id: String): Boolean {
        val participation = database.participations.find { it.id eq id } ?: return false
        database.betInfos.find { it.id eq participation.bet.id }?.let { betInfo ->
            betInfo.totalStakes -= participation.stake
            betInfo.totalParticipants--

            database.betAnswerInfos.filter { it.betId eq participation.bet.id }.forEach {
                if (it.response == participation.answer) {
                    it.totalStakes -= participation.stake
                }
                val probability = it.totalStakes / betInfo.totalStakes.toFloat()
                it.odds = 1 / probability
                it.flushChanges()
            }

            betInfo.flushChanges()
        }
        return participation.delete() > 0
    }

    override fun getBestWinFromUserid(userId: String) =
        database.participations.filter { it.id eq userId }.maxBy { it.stake }


}