@ -5,7 +5,6 @@ import com.fasterxml.jackson.databind.annotation.JsonNaming
import com.fasterxml.jackson.databind.type.TypeFactory
import kotlinx.coroutines.*
import kotlinx.coroutines.flow.MutableSharedFlow
import nl.komponents.kovenant.Kovenant
import nl.komponents.kovenant.Promise
import nl.komponents.kovenant.functional.bind
import nl.komponents.kovenant.functional.map
@ -14,7 +13,6 @@ import okhttp3.HttpUrl
import okhttp3.MediaType
import okhttp3.RequestBody
import org.session.libsession.messaging.MessagingModuleConfiguration
import org.session.libsession.messaging.open_groups.OpenGroupAPIV2.Error
import org.session.libsession.snode.OnionRequestAPI
import org.session.libsession.utilities.AESGCM
import org.session.libsignal.service.loki.HTTP
@ -29,108 +27,83 @@ import org.whispersystems.curve25519.Curve25519
import java.util.*
object OpenGroupAPIV2 {
private val moderators : HashMap < String , Set < String > > = hashMapOf ( ) // Server URL to (channel ID to set of moderator IDs)
const val DEFAULT _SERVER = " http://116.203.70.33 "
private const val DEFAULT _SERVER _PUBLIC _KEY = " a03c383cf63c3c4efe67acc52112a6dd734b3a946b9545f488aaa93da7991238 "
val defaultRooms = MutableSharedFlow < List < DefaultGroup > > ( replay = 1 )
private val curve = Curve25519 . getInstance ( Curve25519 . BEST )
val defaultRooms = MutableSharedFlow < List < DefaultGroup > > ( replay = 1 )
sealed class Error : Exception ( ) {
object GENERIC : Error ( )
object PARSING _FAILED : Error ( )
object DECRYPTION _FAILED : Error ( )
object SIGNING _FAILED : Error ( )
object INVALID _URL : Error ( )
object NO _PUBLIC _KEY : Error ( )
fun errorDescription ( ) = when ( this ) {
Error . GENERIC -> " An error occurred. "
Error . PARSING _FAILED -> " Invalid response. "
Error . DECRYPTION _FAILED -> " Couldn't decrypt response. "
Error . SIGNING _FAILED -> " Couldn't sign message. "
Error . INVALID _URL -> " Invalid URL. "
Error . NO _PUBLIC _KEY -> " Couldn't find server public key. "
}
private const val DEFAULT _SERVER _PUBLIC _KEY = " a03c383cf63c3c4efe67acc52112a6dd734b3a946b9545f488aaa93da7991238 "
const val DEFAULT _SERVER = " http://116.203.70.33 "
sealed class Error ( message : String ) : Exception ( message ) {
object Generic : Error ( " An error occurred. " )
object ParsingFailed : Error ( " Invalid response. " )
object DecryptionFailed : Error ( " Couldn't decrypt response. " )
object SigningFailed : Error ( " Couldn't sign message. " )
object InvalidURL : Error ( " Invalid URL. " )
object NoPublicKey : Error ( " Couldn't find server public key. " )
}
data class DefaultGroup ( val id : String ,
val name : String ,
val image : ByteArray ? ) {
fun toJoinUrl ( ) : String = " $DEFAULT _SERVER/ $id ?public_key= $DEFAULT _SERVER_PUBLIC_KEY "
data class DefaultGroup ( val id : String , val name : String , val image : ByteArray ? ) {
val joinURL : String get ( ) = " $DEFAULT _SERVER/ $id ?public_key= $DEFAULT _SERVER_PUBLIC_KEY "
}
data class Info (
val id : String ,
val name : String ,
val imageID : String ?
)
data class Info ( val id : String , val name : String , val imageID : String ? )
@JsonNaming ( PropertyNamingStrategy . SnakeCaseStrategy :: class )
data class CompactPollRequest ( val roomId : String ,
val authToken : String ,
val fromDeletionServerId : Long ? ,
val fromMessageServerId : Long ?
)
data class CompactPollResult ( val messages : List < OpenGroupMessageV2 > ,
val deletions : List < Long > ,
val moderators : List < String >
)
data class CompactPollRequest ( val roomID : String , val authToken : String , val fromDeletionServerID : Long ? , val fromMessageServerID : Long ? )
data class CompactPollResult ( val messages : List < OpenGroupMessageV2 > , val deletions : List < Long > , val moderators : List < String > )
@JsonNaming ( PropertyNamingStrategy . SnakeCaseStrategy :: class )
data class MessageDeletion @JvmOverloads constructor ( val id : Long = 0 ,
val deletedMessageId : Long = 0
data class MessageDeletion
@JvmOverloads constructor ( val id : Long = 0 , val deletedMessageId : Long = 0
) {
companion object {
val EMPTY = MessageDeletion ( )
}
}
data class Request (
val verb : HTTP . Verb ,
val room : String ? ,
val server : String ,
val endpoint : String ,
val queryParameters : Map < String , String > = mapOf ( ) ,
val parameters : Any ? = null ,
val headers : Map < String , String > = mapOf ( ) ,
val isAuthRequired : Boolean = true ,
// Always `true` under normal circumstances. You might want to disable
// this when running over Lokinet.
val useOnionRouting : Boolean = true
val verb : HTTP . Verb ,
val room : String ? ,
val server : String ,
val endpoint : String ,
val queryParameters : Map < String , String > = mapOf ( ) ,
val parameters : Any ? = null ,
val headers : Map < String , String > = mapOf ( ) ,
val isAuthRequired : Boolean = true ,
/ * *
* Always `true` under normal circumstances . You might want to disable
* this when running over Lokinet .
* /
val useOnionRouting : Boolean = true
)
private fun createBody ( parameters : Any ? ) : RequestBody ? {
if ( parameters == null ) return null
val parametersAsJSON = JsonUtil . toJson ( parameters )
return RequestBody . create ( MediaType . get ( " application/json " ) , parametersAsJSON )
}
private fun send ( request : Request , isJsonRequired : Boolean = true ) : Promise < Map < * , * > , Exception > {
val parsed = HttpUrl . parse ( request . server ) ?: return Promise . ofFail ( Error . I NVALID_ URL)
val url = HttpUrl . parse ( request . server ) ?: return Promise . ofFail ( Error . I nvalid URL)
val urlBuilder = HttpUrl . Builder ( )
. scheme ( parsed . scheme ( ) )
. host ( parsed . host ( ) )
. port ( parsed . port ( ) )
. addPathSegments ( request . endpoint )
. scheme ( url . scheme ( ) )
. host ( url . host ( ) )
. port ( url . port ( ) )
. addPathSegments ( request . endpoint )
if ( request . verb == GET ) {
for ( ( key , value ) in request . queryParameters ) {
urlBuilder . addQueryParameter ( key , value )
}
}
fun execute ( token : String ? ) : Promise < Map < * , * > , Exception > {
val requestBuilder = okhttp3 . Request . Builder ( )
. url ( urlBuilder . build ( ) )
. headers ( Headers . of ( request . headers ) )
. url ( urlBuilder . build ( ) )
. headers ( Headers . of ( request . headers ) )
if ( request . isAuthRequired ) {
if ( token . isNullOrEmpty ( ) ) throw IllegalStateException ( " No auth token for request " )
if ( token . isNullOrEmpty ( ) ) throw IllegalStateException ( " No auth token for request . " )
requestBuilder . header ( " Authorization " , token )
}
when ( request . verb ) {
@ -139,25 +112,25 @@ object OpenGroupAPIV2 {
POST -> requestBuilder . post ( createBody ( request . parameters ) !! )
DELETE -> requestBuilder . delete ( createBody ( request . parameters ) )
}
if ( ! request . room . isNullOrEmpty ( ) ) {
requestBuilder . header ( " Room " , request . room )
}
if ( request . useOnionRouting ) {
val publicKey = MessagingModuleConfiguration . shared . storage . getOpenGroupPublicKey ( request . server )
?: return Promise . ofFail ( Error . N O_PUBLIC _KEY )
return OnionRequestAPI . sendOnionRequest ( requestBuilder . build ( ) , request . server , publicKey , isJSONRequired = isJsonRequired )
. fail { e ->
if ( e is OnionRequestAPI . HTTPRequestFailedAtDestinationException && e . statusCode == 401 ) {
val storage = MessagingModuleConfiguration . shared . storage
if ( request . room != null ) {
storage . removeAuthToken ( " ${request.server} . ${request.room} " )
} else {
storage . removeAuthToken ( request . server )
}
}
?: return Promise . ofFail ( Error . N oPublicKey )
return OnionRequestAPI . sendOnionRequest ( requestBuilder . build ( ) , request . server , publicKey , isJSONRequired = isJsonRequired ) . fail { e ->
// A 401 means that we didn't provide a (valid) auth token for a route that required one. We use this as an
// indication that the token we're using has expired. Note that a 403 has a different meaning; it means that
// we provided a valid token but it doesn't have a high enough permission level for the route in question.
if ( e is OnionRequestAPI . HTTPRequestFailedAtDestinationException && e . statusCode == 401 ) {
val storage = MessagingModuleConfiguration . shared . storage
if ( request . room != null ) {
storage . removeAuthToken ( " ${request.server} . ${request.room} " )
} else {
storage . removeAuthToken ( request . server )
}
}
}
} else {
return Promise . ofFail ( IllegalStateException ( " It's currently not allowed to send non onion routed requests. " ) )
}
@ -172,52 +145,51 @@ object OpenGroupAPIV2 {
fun downloadOpenGroupProfilePicture ( roomID : String , server : String ) : Promise < ByteArray , Exception > {
val request = Request ( verb = GET , room = roomID , server = server , endpoint = " rooms/ $roomID /image " , isAuthRequired = false )
return send ( request ) . map { json ->
val result = json [ " result " ] as ? String ?: throw Error . P ARSING_FAILED
val result = json [ " result " ] as ? String ?: throw Error . P arsingFailed
decode ( result )
}
}
// region Authorization
fun getAuthToken ( room : String , server : String ) : Promise < String , Exception > {
val storage = MessagingModuleConfiguration . shared . storage
return storage . getAuthToken ( room , server ) ?. let {
Promise . of ( it )
} ?: run {
requestNewAuthToken ( room , server )
. bind { claimAuthToken ( it , room , server ) }
. success { authToken ->
storage . setAuthToken ( room , server , authToken )
}
. bind { claimAuthToken ( it , room , server ) }
. success { authToken ->
storage . setAuthToken ( room , server , authToken )
}
}
}
fun requestNewAuthToken ( room : String , server : String ) : Promise < String , Exception > {
val ( publicKey , privateKey ) = MessagingModuleConfiguration . shared . storage . getUserKeyPair ( )
?: return Promise . ofFail ( Error . G ENERIC )
val queryParameters = mutableMapOf ( " public_key " to publicKey )
?: return Promise . ofFail ( Error . G eneric )
val queryParameters = mutableMapOf ( " public_key " to publicKey )
val request = Request ( GET , room , server , " auth_token_challenge " , queryParameters , isAuthRequired = false , parameters = null )
return send ( request ) . map { json ->
val challenge = json [ " challenge " ] as ? Map < * , * > ?: throw Error . PARSING _FAILED
val base64EncodedCiphertext = challenge [ " ciphertext " ] as ? String
?: throw Error . PARSING _FAILED
val base64EncodedEphemeralPublicKey = challenge [ " ephemeral_public_key " ] as ? String
?: throw Error . PARSING _FAILED
val challenge = json [ " challenge " ] as ? Map < * , * > ?: throw Error . ParsingFailed
val base64EncodedCiphertext = challenge [ " ciphertext " ] as ? String ?: throw Error . ParsingFailed
val base64EncodedEphemeralPublicKey = challenge [ " ephemeral_public_key " ] as ? String ?: throw Error . ParsingFailed
val ciphertext = decode ( base64EncodedCiphertext )
val ephemeralPublicKey = decode ( base64EncodedEphemeralPublicKey )
val symmetricKey = AESGCM . generateSymmetricKey ( ephemeralPublicKey , privateKey )
val tokenAsData = try {
AESGCM . decrypt ( ciphertext , symmetricKey )
} catch ( e : Exception ) {
throw Error . D ECRYPTION_FAILED
throw Error . D ecryptionFailed
}
tokenAsData . toHexString ( )
}
}
fun claimAuthToken ( authToken : String , room : String , server : String ) : Promise < String , Exception > {
val parameters = mapOf ( " public_key " to MessagingModuleConfiguration . shared . storage . getUserPublicKey ( ) !! )
val headers = mapOf ( " Authorization " to authToken )
val parameters = mapOf ( " public_key " to MessagingModuleConfiguration . shared . storage . getUserPublicKey ( ) !! )
val headers = mapOf ( " Authorization " to authToken )
val request = Request ( verb = POST , room = room , server = server , endpoint = " claim_auth_token " ,
parameters = parameters , headers = headers , isAuthRequired = false )
parameters = parameters , headers = headers , isAuthRequired = false )
return send ( request ) . map { authToken }
}
@ -227,33 +199,36 @@ object OpenGroupAPIV2 {
MessagingModuleConfiguration . shared . storage . removeAuthToken ( room , server )
}
}
// endregion
// region Sending
// region Upload/Download
fun upload ( file : ByteArray , room : String , server : String ) : Promise < Long , Exception > {
val base64EncodedFile = encodeBytes ( file )
val parameters = mapOf ( " file " to base64EncodedFile )
val parameters = mapOf ( " file " to base64EncodedFile )
val request = Request ( verb = POST , room = room , server = server , endpoint = " files " , parameters = parameters )
return send ( request ) . map { json ->
json [ " result " ] as ? Long ?: throw Error . P ARSING_FAILED
json [ " result " ] as ? Long ?: throw Error . P arsingFailed
}
}
fun download ( file : Long , room : String , server : String ) : Promise < ByteArray , Exception > {
val request = Request ( verb = GET , room = room , server = server , endpoint = " files/ $file " )
return send ( request ) . map { json ->
val base64EncodedFile = json [ " result " ] as ? String ?: throw Error . P ARSING_FAILED
decode ( base64EncodedFile ) ?: throw Error . P ARSING_FAILED
val base64EncodedFile = json [ " result " ] as ? String ?: throw Error . P arsingFailed
decode ( base64EncodedFile ) ?: throw Error . P arsingFailed
}
}
// endregion
// region Sending
fun send ( message : OpenGroupMessageV2 , room : String , server : String ) : Promise < OpenGroupMessageV2 , Exception > {
val signedMessage = message . sign ( ) ?: return Promise . ofFail ( Error . S IGNING_FAILED )
val signedMessage = message . sign ( ) ?: return Promise . ofFail ( Error . S igningFailed )
val jsonMessage = signedMessage . toJSON ( )
val request = Request ( verb = POST , room = room , server = server , endpoint = " messages " , parameters = jsonMessage )
return send ( request ) . map { json ->
@Suppress ( " UNCHECKED_CAST " ) val rawMessage = json [ " message " ] as ? Map < String , Any >
?: throw Error . P ARSING_FAILED
OpenGroupMessageV2 . fromJSON ( rawMessage ) ?: throw Error . P ARSING_FAILED
?: throw Error . P arsingFailed
OpenGroupMessageV2 . fromJSON ( rawMessage ) ?: throw Error . P arsingFailed
}
}
// endregion
@ -266,36 +241,41 @@ object OpenGroupAPIV2 {
queryParameters += " from_server_id " to lastId . toString ( )
}
val request = Request ( verb = GET , room = room , server = server , endpoint = " messages " , queryParameters = queryParameters )
return send ( request ) . map { jsonList ->
@Suppress ( " UNCHECKED_CAST " ) val rawMessages = jsonList [ " messages " ] as ? List < Map < String , Any > >
?: throw Error . PARSING _FAILED
val lastMessageServerId = storage . getLastMessageServerId ( room , server ) ?: 0
var currentMax = lastMessageServerId
val messages = rawMessages . mapNotNull { json ->
try {
val message = OpenGroupMessageV2 . fromJSON ( json ) ?: return @mapNotNull null
if ( message . serverID == null || message . sender . isNullOrEmpty ( ) ) return @mapNotNull null
val sender = message . sender
val data = decode ( message . base64EncodedData )
val signature = decode ( message . base64EncodedSignature )
val publicKey = Hex . fromStringCondensed ( sender . removing05PrefixIfNeeded ( ) )
val isValid = curve . verifySignature ( publicKey , data , signature )
if ( !is Valid ) {
Log . d ( " Loki " , " Ignoring message with invalid signature " )
return @mapNotNull null
}
if ( message . serverID > lastMessageServerId ) {
currentMax = message . serverID
}
message
} catch ( e : Exception ) {
null
return send ( request ) . map { json ->
@Suppress ( " UNCHECKED_CAST " ) val rawMessages = json [ " messages " ] as ? List < Map < String , Any > >
?: throw Error . ParsingFailed
parseMessages ( room , server , rawMessages )
}
}
private fun parseMessages ( room : String , server : String , rawMessages : List < Map < * , * > > ) : List < OpenGroupMessageV2 > {
val storage = MessagingModuleConfiguration . shared . storage
val lastMessageServerID = storage . getLastMessageServerId ( room , server ) ?: 0
var currentLastMessageServerID = lastMessageServerID
val messages = rawMessages . mapNotNull { json ->
json as Map < String , Any >
try {
val message = OpenGroupMessageV2 . fromJSON ( json ) ?: return @mapNotNull null
if ( message . serverID == null || message . sender . isNullOrEmpty ( ) ) return @mapNotNull null
val sender = message . sender
val data = decode ( message . base64EncodedData )
val signature = decode ( message . base64EncodedSignature )
val publicKey = Hex . fromStringCondensed ( sender . removing05PrefixIfNeeded ( ) )
val isValid = curve . verifySignature ( publicKey , data , signature )
if ( !is Valid ) {
Log . d ( " Loki " , " Ignoring message with invalid signature. " )
return @mapNotNull null
}
if ( message . serverID > lastMessageServerID ) {
currentLastMessageServerID = message . serverID
}
message
} catch ( e : Exception ) {
null
}
storage . setLastMessageServerId ( room , server , currentMax )
messages
}
storage . setLastMessageServerId ( room , server , currentLastMessageServerID )
return messages
}
// endregion
@ -304,7 +284,7 @@ object OpenGroupAPIV2 {
fun deleteMessage ( serverID : Long , room : String , server : String ) : Promise < Unit , Exception > {
val request = Request ( verb = DELETE , room = room , server = server , endpoint = " messages/ $serverID " )
return send ( request ) . map {
Log . d ( " Loki " , " Deleted server message " )
Log . d ( " Loki " , " Message deletion successful. " )
}
}
@ -318,7 +298,7 @@ object OpenGroupAPIV2 {
return send ( request ) . map { json ->
val type = TypeFactory . defaultInstance ( ) . constructCollectionType ( List :: class . java , MessageDeletion :: class . java )
val idsAsString = JsonUtil . toJson ( json [ " ids " ] )
val serverIDs = JsonUtil . fromJson < List < MessageDeletion > > ( idsAsString , type ) ?: throw Error . P ARSING_FAILED
val serverIDs = JsonUtil . fromJson < List < MessageDeletion > > ( idsAsString , type ) ?: throw Error . P arsingFailed
val lastMessageServerId = storage . getLastDeletionServerId ( room , server ) ?: 0
val serverID = serverIDs . maxByOrNull { it . id } ?: MessageDeletion . EMPTY
if ( serverID . id > lastMessageServerId ) {
@ -338,7 +318,7 @@ object OpenGroupAPIV2 {
val request = Request ( verb = GET , room = room , server = server , endpoint = " moderators " )
return send ( request ) . map { json ->
@Suppress ( " UNCHECKED_CAST " ) val moderatorsJson = json [ " moderators " ] as ? List < String >
?: throw Error . P ARSING_FAILED
?: throw Error . P arsingFailed
val id = " $server . $room "
handleModerators ( id , moderatorsJson )
moderatorsJson
@ -347,90 +327,77 @@ object OpenGroupAPIV2 {
@JvmStatic
fun ban ( publicKey : String , room : String , server : String ) : Promise < Unit , Exception > {
val parameters = mapOf ( " public_key " to publicKey )
val parameters = mapOf ( " public_key " to publicKey )
val request = Request ( verb = POST , room = room , server = server , endpoint = " block_list " , parameters = parameters )
return send ( request ) . map {
Log . d ( " Loki " , " Banned user $publicKey from $server . $room " )
Log . d ( " Loki " , " Banned user : $publicKey from : $server . $room . " )
}
}
fun unban ( publicKey : String , room : String , server : String ) : Promise < Unit , Exception > {
val request = Request ( verb = DELETE , room = room , server = server , endpoint = " block_list/ $publicKey " )
return send ( request ) . map {
Log . d ( " Loki " , " Unbanned user $publicKey from $server . $room " )
Log . d ( " Loki " , " Unbanned user : $publicKey from : $server . $room " )
}
}
@JvmStatic
fun isUserModerator ( publicKey : String , room : String , server : String ) : Boolean =
moderators [ " $server . $room " ] ?. contains ( publicKey ) ?: false
moderators [ " $server . $room " ] ?. contains ( publicKey ) ?: false
// endregion
// region General
@Suppress ( " UNCHECKED_CAST " )
fun getCompactPoll ( rooms : List < String > , server : String ) : Promise < Map < String , CompactPollResult > , Exception > {
val requestAuth s = rooms . associateWith { room -> getAuthToken ( room , server ) }
val authTokenRequest s = rooms . associateWith { room -> getAuthToken ( room , server ) }
val storage = MessagingModuleConfiguration . shared . storage
val requests = rooms . mapNotNull { room ->
val authToken = try {
requestAuth s[ room ] ?. get ( )
authTokenRequest s[ room ] ?. get ( )
} catch ( e : Exception ) {
Log . e ( " Loki " , " Failed to get auth token for $room , e )
Log . e ( " Loki " , " Failed to get auth token for $room . ", e )
null
} ?: return @mapNotNull null
CompactPollRequest ( roomId = room ,
authToken = authToken ,
fromDeletionServerId = storage . getLastDeletionServerId ( room , server ) ,
fromMessageServerId = storage . getLastMessageServerId ( room , server )
CompactPollRequest (
roomID = room ,
authToken = authToken ,
fromDeletionServerID = storage . getLastDeletionServerId ( room , server ) ,
fromMessageServerID = storage . getLastMessageServerId ( room , server )
)
}
val request = Request ( verb = POST , room = null , server = server , endpoint = " compact_poll " , isAuthRequired = false , parameters = mapOf ( " requests " to requests ) )
// build a request for all rooms
val request = Request ( verb = POST , room = null , server = server , endpoint = " compact_poll " , isAuthRequired = false , parameters = mapOf ( " requests " to requests ) )
return send ( request = request ) . map { json ->
val results = json [ " results " ] as ? List < * > ?: throw Error . P ARSING_FAILED
results . mapNotNull { roomJson ->
if ( roomJson !is Map < * , * > ) return @mapNotNull null
val roomId = roomJson [ " room_id " ] as ? String ?: return @mapNotNull null
// check the status was fine
val statusCode = roomJ son[ " status_code " ] as ? Int ?: return @mapNotNull null
val results = json [ " results " ] as ? List < * > ?: throw Error . P arsingFailed
results . mapNotNull { json ->
if ( json !is Map < * , * > ) return @mapNotNull null
val roomID = json [ " room_id " ] as ? String ?: return @mapNotNull null
// A 401 means that we didn't provide a (valid) auth token for a route that required one. We use this as an
// indication that the token we're using has expired. Note that a 403 has a different meaning; it means that
// we provided a valid token but it doesn't have a high enough permission level for the route in question.
val statusCode = j son[ " status_code " ] as ? Int ?: return @mapNotNull null
if ( statusCode == 401 ) {
// delete auth token and return null
storage . removeAuthToken ( roomI d , server )
storage . removeAuthToken ( roomI D , server )
}
// check and store mods
val moderators = roomJson [ " moderators " ] as ? List < String > ?: return @mapNotNull null
handleModerators ( " $server . $roomId " , moderators )
// get deletions
// Moderators
val moderators = json [ " moderators " ] as ? List < String > ?: return @mapNotNull null
handleModerators ( " $server . $roomID " , moderators )
// Deletions
val type = TypeFactory . defaultInstance ( ) . constructCollectionType ( List :: class . java , MessageDeletion :: class . java )
val idsAsString = JsonUtil . toJson ( roomJ son[ " deletions " ] )
val deletedServerIDs = JsonUtil . fromJson < List < MessageDeletion > > ( idsAsString , type ) ?: throw Error . P ARSING_FAILED
val lastDeletionServerI d = storage . getLastDeletionServerId ( roomId , server ) ?: 0
val serverID = deletedServerIDs . maxByOrNull { it . id } ?: MessageDeletion . EMPTY
if ( serverID . id > lastDeletionServerI d ) {
storage . setLastDeletionServerId ( roomI d , server , serverID . id )
val idsAsString = JsonUtil . toJson ( json [ " deletions " ] )
val deletedServerIDs = JsonUtil . fromJson < List < MessageDeletion > > ( idsAsString , type ) ?: throw Error . P arsingFailed
val lastDeletionServerI D = storage . getLastDeletionServerId ( roomID , server ) ?: 0
val serverID = deletedServerIDs . maxByOrNull { it . id } ?: MessageDeletion . EMPTY
if ( serverID . id > lastDeletionServerI D ) {
storage . setLastDeletionServerId ( roomI D , server , serverID . id )
}
// get messages
val rawMessages = roomJson [ " messages " ] as ? List < Map < String , Any > > ?: return @mapNotNull null // parsing failed
val lastMessageServerId = storage . getLastMessageServerId ( roomId , server ) ?: 0
var currentMax = lastMessageServerId
val messages = rawMessages . mapNotNull { rawMessage ->
val message = OpenGroupMessageV2 . fromJSON ( rawMessage ) ?. apply {
currentMax = maxOf ( currentMax , this . serverID ?: 0 )
}
message
}
storage . setLastMessageServerId ( roomId , server , currentMax )
roomId to CompactPollResult (
messages = messages ,
deletions = deletedServerIDs . map { it . deletedMessageId } ,
moderators = moderators
// Messages
val rawMessages = json [ " messages " ] as ? List < Map < String , Any > > ?: return @mapNotNull null
val messages = parseMessages ( roomID , server , rawMessages )
roomID to CompactPollResult (
messages = messages ,
deletions = deletedServerIDs . map { it . deletedMessageId } ,
moderators = moderators
)
} . toMap ( )
}
@ -443,7 +410,7 @@ object OpenGroupAPIV2 {
val earlyGroups = groups . map { group ->
DefaultGroup ( group . id , group . name , null )
}
// s ee if we have any cached rooms, and if they already have images, don't overwrite with early non-image results
// S ee if we have any cached rooms, and if they already have images don't overwrite them with early non-image results
defaultRooms . replayCache . firstOrNull ( ) ?. let { replayed ->
if ( replayed . none { it . image ?. isNotEmpty ( ) == true } ) {
defaultRooms . tryEmit ( earlyGroups )
@ -452,12 +419,11 @@ object OpenGroupAPIV2 {
val images = groups . map { group ->
group . id to downloadOpenGroupProfilePicture ( group . id , DEFAULT _SERVER )
} . toMap ( )
groups . map { group ->
val image = try {
images [ group . id ] !! . get ( )
} catch ( e : Exception ) {
// n o image or image failed to download
// N o image or image failed to download
null
}
DefaultGroup ( group . id , group . name , image )
@ -470,9 +436,9 @@ object OpenGroupAPIV2 {
fun getInfo ( room : String , server : String ) : Promise < Info , Exception > {
val request = Request ( verb = GET , room = null , server = server , endpoint = " rooms/ $room " , isAuthRequired = false )
return send ( request ) . map { json ->
val rawRoom = json [ " room " ] as ? Map < * , * > ?: throw Error . P ARSING_FAILED
val id = rawRoom [ " id " ] as ? String ?: throw Error . P ARSING_FAILED
val name = rawRoom [ " name " ] as ? String ?: throw Error . P ARSING_FAILED
val rawRoom = json [ " room " ] as ? Map < * , * > ?: throw Error . P arsingFailed
val id = rawRoom [ " id " ] as ? String ?: throw Error . P arsingFailed
val name = rawRoom [ " name " ] as ? String ?: throw Error . P arsingFailed
val imageID = rawRoom [ " image_id " ] as ? String
Info ( id = id , name = name , imageID = imageID )
}
@ -481,13 +447,13 @@ object OpenGroupAPIV2 {
fun getAllRooms ( server : String ) : Promise < List < Info > , Exception > {
val request = Request ( verb = GET , room = null , server = server , endpoint = " rooms " , isAuthRequired = false )
return send ( request ) . map { json ->
val rawRooms = json [ " rooms " ] as ? List < Map < * , * > > ?: throw Error . P ARSING_FAILED
val rawRooms = json [ " rooms " ] as ? List < Map < * , * > > ?: throw Error . P arsingFailed
rawRooms . mapNotNull {
val roomJson = it as ? Map < * , * > ?: return @mapNotNull null
val id = roomJson [ " id " ] as ? String ?: return @mapNotNull null
val name = roomJson [ " name " ] as ? String ?: return @mapNotNull null
val imageI d = roomJson [ " image_id " ] as ? String
Info ( id , name , imageI d )
val imageI D = roomJson [ " image_id " ] as ? String
Info ( id , name , imageI D )
}
}
}
@ -495,12 +461,11 @@ object OpenGroupAPIV2 {
fun getMemberCount ( room : String , server : String ) : Promise < Int , Exception > {
val request = Request ( verb = GET , room = room , server = server , endpoint = " member_count " )
return send ( request ) . map { json ->
val memberCount = json [ " member_count " ] as ? Int ?: throw Error . P ARSING_FAILED
val memberCount = json [ " member_count " ] as ? Int ?: throw Error . P arsingFailed
val storage = MessagingModuleConfiguration . shared . storage
storage . setUserCount ( room , server , memberCount )
memberCount
}
}
// endregion
}
Brice-W
4 years ago