summaryrefslogtreecommitdiff
path: root/src/main/kotlin/io/dico/parcels2/storage/exposed/ListTables.kt
blob: 20b36b1902d797634f498e5a506c1805dabb3cce (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
@file:Suppress("PropertyName", "LocalVariableName", "NOTHING_TO_INLINE")

package io.dico.parcels2.storage.exposed

import io.dico.parcels2.AddedStatus
import io.dico.parcels2.ParcelId
import io.dico.parcels2.ParcelOwner
import io.dico.parcels2.util.toByteArray
import io.dico.parcels2.util.toUUID
import kotlinx.coroutines.experimental.channels.SendChannel
import org.jetbrains.exposed.sql.*
import java.util.UUID

object AddedLocalT : AddedTable<ParcelId>("parcels_added_local", ParcelsT)
object AddedGlobalT : AddedTable<ParcelOwner>("parcels_added_global", OwnersT)

object ParcelOptionsT : Table("parcel_options") {
    val parcel_id = integer("parcel_id").primaryKey().references(ParcelsT.id, ReferenceOption.CASCADE)
    val interact_inventory = bool("interact_inventory").default(true)
    val interact_inputs = bool("interact_inputs").default(true)
}

typealias AddedStatusSendChannel<AttachT> = SendChannel<Pair<AttachT, MutableMap<UUID, AddedStatus>>>

sealed class AddedTable<AttachT>(name: String, val idTable: IdTransactionsTable<*, AttachT>) : Table(name) {
    val attach_id = integer("attach_id").references(idTable.id, ReferenceOption.CASCADE)
    val player_uuid = binary("player_uuid", 16)
    val allowed_flag = bool("allowed_flag")
    val index_pair = uniqueIndexR("index_pair", attach_id, player_uuid)

    fun setPlayerStatus(attachedOn: AttachT, player: UUID, status: AddedStatus) {
        val binaryUuid = player.toByteArray()

        if (status.isDefault) {
            idTable.getId(attachedOn)?.let { id ->
                deleteWhere { (attach_id eq id) and (player_uuid eq binaryUuid) }
            }
            return
        }

        val id = idTable.getOrInitId(attachedOn)
        upsert(conflictIndex = index_pair) {
            it[attach_id] = id
            it[player_uuid] = binaryUuid
            it[allowed_flag] = status.isAllowed
        }
    }

    fun readAddedData(id: Int): MutableMap<UUID, AddedStatus> {
        return slice(player_uuid, allowed_flag).select { attach_id eq id }
            .associateByTo(hashMapOf(), { it[player_uuid].toUUID() }, { it[allowed_flag].asAddedStatus() })
    }

    suspend fun sendAllAddedData(channel: AddedStatusSendChannel<AttachT>) {
        /*
        val iterator = selectAll().orderBy(attach_id).iterator()

        if (iterator.hasNext()) {
            val firstRow = iterator.next()
            var id: Int = firstRow[attach_id]
            var attach: SerializableT? = null
            var map: MutableMap<UUID, AddedStatus>? = null

            fun initAttachAndMap() {
                attach = idTable.getId(id)
                map = attach?.let { mutableMapOf() }
            }

            suspend fun sendIfPresent() {
                if (attach != null && map != null && map!!.isNotEmpty()) {
                    channel.send(attach!! to map!!)
                }
                attach = null
                map = null
            }

            initAttachAndMap()

            for (row in iterator) {
                val rowId = row[attach_id]
                if (rowId != id) {
                    sendIfPresent()
                    id = rowId
                    initAttachAndMap()
                }

                if (attach == null) {
                    continue // owner not found for this owner id
                }

                val player_uuid = row[player_uuid].toUUID()
                val status = row[allowed_flag].asAddedStatus()
                map!![player_uuid] = status
            }

            sendIfPresent()
        }*/
    }

    private inline fun Boolean?.asAddedStatus() = if (this == null) AddedStatus.DEFAULT else if (this) AddedStatus.ALLOWED else AddedStatus.BANNED

}