session-android/app/src/main/java/org/thoughtcrime/securesms/database/GroupMemberDatabase.kt

package org.thoughtcrime.securesms.database

import android.content.ContentValues
import android.content.Context
import android.database.Cursor
import org.session.libsession.messaging.open_groups.GroupMember
import org.session.libsession.messaging.open_groups.GroupMemberRole
import org.thoughtcrime.securesms.database.helpers.SQLCipherOpenHelper

class GroupMemberDatabase(context: Context, helper: SQLCipherOpenHelper) : Database(context, helper) {

    companion object {
        const val TABLE_NAME = "group_member"
        const val GROUP_ID = "group_id"
        const val PROFILE_ID = "profile_id"
        const val ROLE = "role"

        private val allColumns = arrayOf(GROUP_ID, PROFILE_ID, ROLE)

        @JvmField
        val CREATE_GROUP_MEMBER_TABLE_COMMAND = """
      CREATE TABLE $TABLE_NAME (
        $GROUP_ID TEXT NOT NULL,
        $PROFILE_ID TEXT NOT NULL,
        $ROLE TEXT NOT NULL,
        PRIMARY KEY ($GROUP_ID, $PROFILE_ID)
      )
    """.trimIndent()

        private fun readGroupMember(cursor: Cursor): GroupMember {
            return GroupMember(
                groupId = cursor.getString(cursor.getColumnIndexOrThrow(GROUP_ID)),
                profileId = cursor.getString(cursor.getColumnIndexOrThrow(PROFILE_ID)),
                role = GroupMemberRole.valueOf(cursor.getString(cursor.getColumnIndexOrThrow(ROLE))),
            )
        }
    }

    fun getGroupMemberRoles(groupId: String, profileId: String): List<GroupMemberRole> {
        val query = "$GROUP_ID = ? AND $PROFILE_ID = ?"
        val args = arrayOf(groupId, profileId)

        val mappings: MutableList<GroupMember> = mutableListOf()

        readableDatabase.query(TABLE_NAME, allColumns, query, args, null, null, null).use { cursor ->
            while (cursor.moveToNext()) {
                mappings += readGroupMember(cursor)
            }
        }

        return mappings.map { it.role }
    }

    fun setGroupMembers(members: List<GroupMember>) {
        writableDatabase.beginTransaction()
        try {
            val grouped = members.groupBy { it.role }
            grouped.forEach { (role, members) ->
                if (members.isEmpty()) return@forEach

                val toDeleteQuery = "$GROUP_ID = ? AND $ROLE = ?"
                val toDeleteArgs = arrayOf(members.first().groupId, role.name)

                writableDatabase.delete(TABLE_NAME, toDeleteQuery, toDeleteArgs)

                members.forEach { member ->
                    val values = ContentValues().apply {
                        put(GROUP_ID, member.groupId)
                        put(PROFILE_ID, member.profileId)
                        put(ROLE, member.role.name)
                    }
                    val query = "$GROUP_ID = ? AND $PROFILE_ID = ?"
                    val args = arrayOf(member.groupId, member.profileId)

                    writableDatabase.insertOrUpdate(TABLE_NAME, values, query, args)
                }
                writableDatabase.setTransactionSuccessful()
            }
        } finally {
            writableDatabase.endTransaction()
        }
    }

}