feat: add in more config factory for volatile

This commit is contained in:
0x330a 2023-02-06 14:28:00 +11:00
parent f7c5953f5d
commit 28e5794624
No known key found for this signature in database
GPG Key ID: 267811D6E6A2698C
5 changed files with 94 additions and 49 deletions

View File

@ -1,6 +1,7 @@
package org.thoughtcrime.securesms.dependencies
import network.loki.messenger.libsession_util.Contacts
import network.loki.messenger.libsession_util.ConversationVolatileConfig
import network.loki.messenger.libsession_util.UserProfile
import org.session.libsession.utilities.ConfigFactoryProtocol
import org.session.libsignal.protos.SignalServiceProtos.SharedConfigMessage
@ -12,16 +13,20 @@ class ConfigFactory(private val configDatabase: ConfigDatabase, private val mayb
fun keyPairChanged() { // this should only happen restoring or clearing data
_userConfig?.free()
_contacts?.free()
_convoVolatileConfig?.free()
_userConfig = null
_contacts = null
_convoVolatileConfig = null
}
private val userLock = Object()
private var _userConfig: UserProfile? = null
private val contactLock = Object()
private var _contacts: Contacts? = null
private val convoVolatileLock = Object()
private var _convoVolatileConfig: ConversationVolatileConfig? = null
override val userConfig: UserProfile? = synchronized(userLock) {
override val user: UserProfile? = synchronized(userLock) {
if (_userConfig == null) {
val (secretKey, publicKey) = maybeGetUserInfo() ?: return@synchronized null
val userDump = configDatabase.retrieveConfig(SharedConfigMessage.Kind.USER_PROFILE.name, publicKey)
@ -47,9 +52,22 @@ class ConfigFactory(private val configDatabase: ConfigDatabase, private val mayb
_contacts
}
override val convoVolatile: ConversationVolatileConfig? = synchronized(convoVolatileLock) {
if (_convoVolatileConfig == null) {
val (secretKey, publicKey) = maybeGetUserInfo() ?: return@synchronized null
val convoDump = configDatabase.retrieveConfig(SharedConfigMessage.Kind.CONVO_INFO_VOLATILE.name, publicKey)
_convoVolatileConfig = if (convoDump != null) {
ConversationVolatileConfig.newInstance(secretKey, convoDump)
} else {
ConversationVolatileConfig.newInstance(secretKey)
}
}
_convoVolatileConfig
}
override fun saveUserConfigDump() {
val dumped = userConfig?.dump() ?: return
val dumped = user?.dump() ?: return
val (_, publicKey) = maybeGetUserInfo() ?: return
configDatabase.storeConfig(SharedConfigMessage.Kind.USER_PROFILE.name, publicKey, dumped)
}
@ -60,4 +78,9 @@ class ConfigFactory(private val configDatabase: ConfigDatabase, private val mayb
configDatabase.storeConfig(SharedConfigMessage.Kind.CONTACTS.name, publicKey, dumped)
}
override fun saveConvoVolatileConfigDump() {
val dumped = convoVolatile?.dump() ?: return
val (_, publicKey) = maybeGetUserInfo() ?: return
configDatabase.storeConfig(SharedConfigMessage.Kind.CONVO_INFO_VOLATILE.name, publicKey, dumped)
}
}

View File

@ -17,7 +17,7 @@ import org.session.libsession.snode.RawResponse
import org.session.libsession.snode.SnodeAPI
import org.session.libsession.snode.SnodeModule
import org.session.libsession.utilities.ConfigFactoryProtocol
import org.session.libsignal.protos.SignalServiceProtos
import org.session.libsignal.utilities.JsonUtil
import org.session.libsignal.utilities.Log
import org.session.libsignal.utilities.Snode
import java.security.SecureRandom
@ -104,10 +104,30 @@ class Poller(private val configFactory: ConfigFactoryProtocol) {
}
}
private fun processUserConfig(rawMessages: List<Pair<SignalServiceProtos.Envelope, String?>>) {
private fun processPersonalMessages(snode: Snode, rawMessages: RawResponse) {
val messages = SnodeAPI.parseRawMessagesResponse(rawMessages, snode, userPublicKey)
val parameters = messages.map { (envelope, serverHash) ->
MessageReceiveParameters(envelope.toByteArray(), serverHash = serverHash)
}
parameters.chunked(BatchMessageReceiveJob.BATCH_DEFAULT_NUMBER).forEach { chunk ->
val job = BatchMessageReceiveJob(chunk)
JobQueue.shared.add(job)
}
}
private fun processUserConfig(snode: Snode, rawMessages: RawResponse) {
SnodeAPI.parseRawMessagesResponse(rawMessages, snode, userPublicKey)
}
private fun processContactsConfig(snode: Snode, rawMessages: RawResponse) {
}
private fun processConvoVolatileConfig(snode: Snode, rawMessages: RawResponse) {
}
private fun poll(snode: Snode, deferred: Deferred<Unit, Exception>): Promise<Unit, Exception> {
if (!hasStarted) { return Promise.ofFail(PromiseCanceledException()) }
return task {
@ -116,7 +136,7 @@ class Poller(private val configFactory: ConfigFactoryProtocol) {
// get messages
SnodeAPI.buildAuthenticatedRetrieveBatchRequest(snode, userPublicKey),
// get user config namespace
configFactory.userConfig?.let { currentUserConfig ->
configFactory.user?.let { currentUserConfig ->
SnodeAPI.buildAuthenticatedRetrieveBatchRequest(
snode,
userPublicKey,
@ -130,6 +150,13 @@ class Poller(private val configFactory: ConfigFactoryProtocol) {
userPublicKey,
currentContacts.configNamespace()
)
},
// get the latest convo info volatile
configFactory.convoVolatile?.let { currentConvoVolatile ->
SnodeAPI.buildAuthenticatedRetrieveBatchRequest(
snode, userPublicKey,
currentConvoVolatile.configNamespace()
)
}
)
@ -138,15 +165,7 @@ class Poller(private val configFactory: ConfigFactoryProtocol) {
if (deferred.promise.isDone()) {
return@bind Promise.ofSuccess(Unit)
} else {
val messageResponse = (rawResponses["results"] as List<*>).first() as RawResponse
val messages = SnodeAPI.parseRawMessagesResponse(messageResponse, snode, userPublicKey)
val parameters = messages.map { (envelope, serverHash) ->
MessageReceiveParameters(envelope.toByteArray(), serverHash = serverHash)
}
parameters.chunked(BatchMessageReceiveJob.BATCH_DEFAULT_NUMBER).forEach { chunk ->
val job = BatchMessageReceiveJob(chunk)
JobQueue.shared.add(job)
}
Log.d("Loki-DBG", JsonUtil.toJson(rawResponses))
poll(snode, deferred)
}

View File

@ -1,11 +1,14 @@
package org.session.libsession.utilities
import network.loki.messenger.libsession_util.Contacts
import network.loki.messenger.libsession_util.ConversationVolatileConfig
import network.loki.messenger.libsession_util.UserProfile
interface ConfigFactoryProtocol {
val userConfig: UserProfile?
val user: UserProfile?
val contacts: Contacts?
val convoVolatile: ConversationVolatileConfig?
fun saveUserConfigDump()
fun saveContactConfigDump()
fun saveConvoVolatileConfigDump()
}

View File

@ -243,8 +243,8 @@ message SharedConfigMessage {
enum Kind {
USER_PROFILE = 1;
CONTACTS = 2;
CONVERSATION_INFO = 3;
LEGACY_CLOSED_GROUPS = 4;
CONVO_INFO_VOLATILE = 3;
GROUPS = 4;
CLOSED_GROUP_INFO = 5;
CLOSED_GROUP_MEMBERS = 6;
ENCRYPTION_KEYS = 7;

View File

@ -22582,13 +22582,13 @@ public final class SignalServiceProtos {
*/
CONTACTS(1, 2),
/**
* <code>CONVERSATION_INFO = 3;</code>
* <code>CONVO_INFO_VOLATILE = 3;</code>
*/
CONVERSATION_INFO(2, 3),
CONVO_INFO_VOLATILE(2, 3),
/**
* <code>LEGACY_CLOSED_GROUPS = 4;</code>
* <code>GROUPS = 4;</code>
*/
LEGACY_CLOSED_GROUPS(3, 4),
GROUPS(3, 4),
/**
* <code>CLOSED_GROUP_INFO = 5;</code>
*/
@ -22612,13 +22612,13 @@ public final class SignalServiceProtos {
*/
public static final int CONTACTS_VALUE = 2;
/**
* <code>CONVERSATION_INFO = 3;</code>
* <code>CONVO_INFO_VOLATILE = 3;</code>
*/
public static final int CONVERSATION_INFO_VALUE = 3;
public static final int CONVO_INFO_VOLATILE_VALUE = 3;
/**
* <code>LEGACY_CLOSED_GROUPS = 4;</code>
* <code>GROUPS = 4;</code>
*/
public static final int LEGACY_CLOSED_GROUPS_VALUE = 4;
public static final int GROUPS_VALUE = 4;
/**
* <code>CLOSED_GROUP_INFO = 5;</code>
*/
@ -22639,8 +22639,8 @@ public final class SignalServiceProtos {
switch (value) {
case 1: return USER_PROFILE;
case 2: return CONTACTS;
case 3: return CONVERSATION_INFO;
case 4: return LEGACY_CLOSED_GROUPS;
case 3: return CONVO_INFO_VOLATILE;
case 4: return GROUPS;
case 5: return CLOSED_GROUP_INFO;
case 6: return CLOSED_GROUP_MEMBERS;
case 7: return ENCRYPTION_KEYS;
@ -27227,30 +27227,30 @@ public final class SignalServiceProtos {
"d\030\006 \001(\010\022\024\n\014didApproveMe\030\007 \001(\010\"y\n\026Message" +
"RequestResponse\022\022\n\nisApproved\030\001 \002(\010\022\022\n\np" +
"rofileKey\030\002 \001(\014\0227\n\007profile\030\003 \001(\0132&.signa" +
"lservice.DataMessage.LokiProfile\"\211\002\n\023Sha" +
"lservice.DataMessage.LokiProfile\"\375\001\n\023Sha" +
"redConfigMessage\0225\n\004kind\030\001 \002(\0162\'.signals" +
"ervice.SharedConfigMessage.Kind\022\r\n\005seqno" +
"\030\002 \002(\003\022\014\n\004data\030\003 \002(\014\"\235\001\n\004Kind\022\020\n\014USER_PR",
"OFILE\020\001\022\014\n\010CONTACTS\020\002\022\025\n\021CONVERSATION_IN" +
"FO\020\003\022\030\n\024LEGACY_CLOSED_GROUPS\020\004\022\025\n\021CLOSED" +
"_GROUP_INFO\020\005\022\030\n\024CLOSED_GROUP_MEMBERS\020\006\022" +
"\023\n\017ENCRYPTION_KEYS\020\007\"u\n\016ReceiptMessage\0220" +
"\n\004type\030\001 \002(\0162\".signalservice.ReceiptMess" +
"age.Type\022\021\n\ttimestamp\030\002 \003(\004\"\036\n\004Type\022\014\n\010D" +
"ELIVERY\020\000\022\010\n\004READ\020\001\"\354\001\n\021AttachmentPointe" +
"r\022\n\n\002id\030\001 \002(\006\022\023\n\013contentType\030\002 \001(\t\022\013\n\003ke" +
"y\030\003 \001(\014\022\014\n\004size\030\004 \001(\r\022\021\n\tthumbnail\030\005 \001(\014" +
"\022\016\n\006digest\030\006 \001(\014\022\020\n\010fileName\030\007 \001(\t\022\r\n\005fl",
"ags\030\010 \001(\r\022\r\n\005width\030\t \001(\r\022\016\n\006height\030\n \001(\r" +
"\022\017\n\007caption\030\013 \001(\t\022\013\n\003url\030e \001(\t\"\032\n\005Flags\022" +
"\021\n\rVOICE_MESSAGE\020\001\"\365\001\n\014GroupContext\022\n\n\002i" +
"d\030\001 \001(\014\022.\n\004type\030\002 \001(\0162 .signalservice.Gr" +
"oupContext.Type\022\014\n\004name\030\003 \001(\t\022\017\n\007members" +
"\030\004 \003(\t\0220\n\006avatar\030\005 \001(\0132 .signalservice.A" +
"ttachmentPointer\022\016\n\006admins\030\006 \003(\t\"H\n\004Type" +
"\022\013\n\007UNKNOWN\020\000\022\n\n\006UPDATE\020\001\022\013\n\007DELIVER\020\002\022\010" +
"\n\004QUIT\020\003\022\020\n\014REQUEST_INFO\020\004B3\n\034org.sessio" +
"n.libsignal.protosB\023SignalServiceProtos"
"\030\002 \002(\003\022\014\n\004data\030\003 \002(\014\"\221\001\n\004Kind\022\020\n\014USER_PR",
"OFILE\020\001\022\014\n\010CONTACTS\020\002\022\027\n\023CONVO_INFO_VOLA" +
"TILE\020\003\022\n\n\006GROUPS\020\004\022\025\n\021CLOSED_GROUP_INFO\020" +
"\005\022\030\n\024CLOSED_GROUP_MEMBERS\020\006\022\023\n\017ENCRYPTIO" +
"N_KEYS\020\007\"u\n\016ReceiptMessage\0220\n\004type\030\001 \002(\016" +
"2\".signalservice.ReceiptMessage.Type\022\021\n\t" +
"timestamp\030\002 \003(\004\"\036\n\004Type\022\014\n\010DELIVERY\020\000\022\010\n" +
"\004READ\020\001\"\354\001\n\021AttachmentPointer\022\n\n\002id\030\001 \002(" +
"\006\022\023\n\013contentType\030\002 \001(\t\022\013\n\003key\030\003 \001(\014\022\014\n\004s" +
"ize\030\004 \001(\r\022\021\n\tthumbnail\030\005 \001(\014\022\016\n\006digest\030\006" +
" \001(\014\022\020\n\010fileName\030\007 \001(\t\022\r\n\005flags\030\010 \001(\r\022\r\n",
"\005width\030\t \001(\r\022\016\n\006height\030\n \001(\r\022\017\n\007caption\030" +
"\013 \001(\t\022\013\n\003url\030e \001(\t\"\032\n\005Flags\022\021\n\rVOICE_MES" +
"SAGE\020\001\"\365\001\n\014GroupContext\022\n\n\002id\030\001 \001(\014\022.\n\004t" +
"ype\030\002 \001(\0162 .signalservice.GroupContext.T" +
"ype\022\014\n\004name\030\003 \001(\t\022\017\n\007members\030\004 \003(\t\0220\n\006av" +
"atar\030\005 \001(\0132 .signalservice.AttachmentPoi" +
"nter\022\016\n\006admins\030\006 \003(\t\"H\n\004Type\022\013\n\007UNKNOWN\020" +
"\000\022\n\n\006UPDATE\020\001\022\013\n\007DELIVER\020\002\022\010\n\004QUIT\020\003\022\020\n\014" +
"REQUEST_INFO\020\004B3\n\034org.session.libsignal." +
"protosB\023SignalServiceProtos"
};
com.google.protobuf.Descriptors.FileDescriptor.InternalDescriptorAssigner assigner =
new com.google.protobuf.Descriptors.FileDescriptor.InternalDescriptorAssigner() {