TrezorStore.kt

package to.bitkit.data

import android.content.Context
import androidx.datastore.core.DataStore
import androidx.datastore.dataStore
import dagger.hilt.android.qualifiers.ApplicationContext
import kotlinx.coroutines.flow.Flow
import kotlinx.coroutines.flow.first
import kotlinx.serialization.Serializable
import to.bitkit.data.serializers.TrezorDataSerializer
import to.bitkit.repositories.KnownDevice
import javax.inject.Inject
import javax.inject.Singleton

private val Context.trezorDataStore: DataStore<TrezorData> by dataStore(
    fileName = "trezor_device.json",
    serializer = TrezorDataSerializer
)

@Singleton
class TrezorStore @Inject constructor(
    @ApplicationContext private val context: Context,
) {
    private val store = context.trezorDataStore

    val data: Flow<TrezorData> = store.data

    suspend fun loadKnownDevices(): List<KnownDevice> =
        store.data.first().knownDevices

    suspend fun saveKnownDevices(devices: List<KnownDevice>) {
        store.updateData { it.copy(knownDevices = devices) }
    }

    suspend fun reset() {
        store.updateData { TrezorData() }
    }
}

@Serializable
data class TrezorData(
    val knownDevices: List<KnownDevice> = emptyList(),
)