add FCM to app for remote push notification
							parent
							
								
									562deec218
								
							
						
					
					
						commit
						b72048d9a4
					
				@ -0,0 +1,98 @@
 | 
			
		||||
package org.thoughtcrime.securesms.loki
 | 
			
		||||
 | 
			
		||||
import android.content.Context
 | 
			
		||||
import okhttp3.*
 | 
			
		||||
import org.thoughtcrime.securesms.util.TextSecurePreferences
 | 
			
		||||
import org.whispersystems.libsignal.logging.Log
 | 
			
		||||
import org.whispersystems.signalservice.internal.util.JsonUtil
 | 
			
		||||
import java.io.IOException
 | 
			
		||||
 | 
			
		||||
object LokiPushNotificationManager {
 | 
			
		||||
    //const val server = "https://live.apns.getsession.org/"
 | 
			
		||||
    const val server = "https://dev.apns.getsession.org/"
 | 
			
		||||
    const val tokenExpirationInterval = 2 * 24 * 60 * 60
 | 
			
		||||
    private val connection = OkHttpClient()
 | 
			
		||||
 | 
			
		||||
    fun disableRemoteNotification(token: String, context: Context?) {
 | 
			
		||||
        val parameters = mapOf("token" to token)
 | 
			
		||||
        val url = "${server}register"
 | 
			
		||||
        val body = RequestBody.create(MediaType.get("application/json"), JsonUtil.toJson(parameters))
 | 
			
		||||
        val request = Request.Builder().url(url).post(body).build()
 | 
			
		||||
        connection.newCall(request).enqueue(object : Callback {
 | 
			
		||||
 | 
			
		||||
            override fun onResponse(call: Call, response: Response) {
 | 
			
		||||
                when (response.code()) {
 | 
			
		||||
                    200 -> {
 | 
			
		||||
                        val bodyAsString = response.body()!!.string()
 | 
			
		||||
                        val json = JsonUtil.fromJson(bodyAsString, Map::class.java)
 | 
			
		||||
                        val code  = json?.get("code") as? Int
 | 
			
		||||
                        if (code != null && code != 0) {
 | 
			
		||||
                            TextSecurePreferences.setIsUsingRemoteNotification(context, false)
 | 
			
		||||
                        } else {
 | 
			
		||||
                            Log.d("Loki", "Couldn't disable remote notification due to error: ${json?.get("message") as? String}.")
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            override fun onFailure(call: Call, exception: IOException) {
 | 
			
		||||
                Log.d("Loki", "Couldn't disable remote notification.")
 | 
			
		||||
            }
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    @JvmStatic
 | 
			
		||||
    fun register(token: String, hexEncodedPublicKey: String, context: Context?) {
 | 
			
		||||
        val parameters = mapOf("token" to token, "pubKey" to hexEncodedPublicKey)
 | 
			
		||||
        val url = "${server}register"
 | 
			
		||||
        val body = RequestBody.create(MediaType.get("application/json"), JsonUtil.toJson(parameters))
 | 
			
		||||
        val request = Request.Builder().url(url).post(body).build()
 | 
			
		||||
        connection.newCall(request).enqueue(object : Callback {
 | 
			
		||||
 | 
			
		||||
            override fun onResponse(call: Call, response: Response) {
 | 
			
		||||
                when (response.code()) {
 | 
			
		||||
                    200 -> {
 | 
			
		||||
                        val bodyAsString = response.body()!!.string()
 | 
			
		||||
                        val json = JsonUtil.fromJson(bodyAsString, Map::class.java)
 | 
			
		||||
                        val code  = json?.get("code") as? Int
 | 
			
		||||
                        if (code != null && code != 0) {
 | 
			
		||||
                            TextSecurePreferences.setIsUsingRemoteNotification(context, true)
 | 
			
		||||
                        } else {
 | 
			
		||||
                            Log.d("Loki", "Couldn't register device token due to error: ${json?.get("message") as? String}.")
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            override fun onFailure(call: Call, exception: IOException) {
 | 
			
		||||
                Log.d("Loki", "Couldn't register device token.")
 | 
			
		||||
            }
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    fun acknowledgeDeliveryForMessageWith(hash: String, expiration: Int, hexEncodedPublicKey: String, context: Context?) {
 | 
			
		||||
        val parameters = mapOf("hash" to hash, "pubKey" to hexEncodedPublicKey, "expiration" to expiration)
 | 
			
		||||
        val url = "${server}acknowledge_message_delivery"
 | 
			
		||||
        val body = RequestBody.create(MediaType.get("application/json"), JsonUtil.toJson(parameters))
 | 
			
		||||
        val request = Request.Builder().url(url).post(body).build()
 | 
			
		||||
        connection.newCall(request).enqueue(object : Callback {
 | 
			
		||||
 | 
			
		||||
            override fun onResponse(call: Call, response: Response) {
 | 
			
		||||
                when (response.code()) {
 | 
			
		||||
                    200 -> {
 | 
			
		||||
                        val bodyAsString = response.body()!!.string()
 | 
			
		||||
                        val json = JsonUtil.fromJson(bodyAsString, Map::class.java)
 | 
			
		||||
                        val code  = json?.get("code") as? Int
 | 
			
		||||
                        if (code == null || code == 0) {
 | 
			
		||||
                            Log.d("Loki", "Couldn't acknowledge the delivery for message due to error: ${json?.get("message") as? String}.")
 | 
			
		||||
                        }
 | 
			
		||||
                    }
 | 
			
		||||
                }
 | 
			
		||||
            }
 | 
			
		||||
 | 
			
		||||
            override fun onFailure(call: Call, exception: IOException) {
 | 
			
		||||
                Log.d("Loki", "Couldn't acknowledge the delivery for message with last hash: ${hash}")
 | 
			
		||||
            }
 | 
			
		||||
        })
 | 
			
		||||
    }
 | 
			
		||||
}
 | 
			
		||||
@ -0,0 +1,39 @@
 | 
			
		||||
package org.thoughtcrime.securesms.service
 | 
			
		||||
 | 
			
		||||
import com.google.firebase.messaging.FirebaseMessagingService
 | 
			
		||||
import com.google.firebase.messaging.RemoteMessage
 | 
			
		||||
import org.thoughtcrime.securesms.ApplicationContext
 | 
			
		||||
import org.thoughtcrime.securesms.jobs.PushContentReceiveJob
 | 
			
		||||
import org.thoughtcrime.securesms.loki.LokiPushNotificationManager
 | 
			
		||||
import org.thoughtcrime.securesms.util.TextSecurePreferences
 | 
			
		||||
import org.whispersystems.libsignal.logging.Log
 | 
			
		||||
import org.whispersystems.signalservice.api.messages.SignalServiceEnvelope
 | 
			
		||||
import org.whispersystems.signalservice.internal.util.Base64
 | 
			
		||||
import org.whispersystems.signalservice.loki.messaging.LokiMessageWrapper
 | 
			
		||||
 | 
			
		||||
class PushNotificationService: FirebaseMessagingService() {
 | 
			
		||||
 | 
			
		||||
    override fun onNewToken(token: String) {
 | 
			
		||||
        super.onNewToken(token)
 | 
			
		||||
        Log.d("Loki", "new token ${token}")
 | 
			
		||||
        val userHexEncodedPublicKey = TextSecurePreferences.getLocalNumber(this)
 | 
			
		||||
        LokiPushNotificationManager.register(token, userHexEncodedPublicKey, this)
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    override fun onMessageReceived(message: RemoteMessage) {
 | 
			
		||||
        val base64EncodedData = message.data["ENCRYPTED_DATA"]
 | 
			
		||||
        val data = base64EncodedData?.let { Base64.decode(it) }
 | 
			
		||||
        if (data != null) {
 | 
			
		||||
            try {
 | 
			
		||||
                val envelope = LokiMessageWrapper.unwrap(data)
 | 
			
		||||
                PushContentReceiveJob(this).processEnvelope(SignalServiceEnvelope(envelope))
 | 
			
		||||
            } catch (e: Exception) {
 | 
			
		||||
                Log.d("Loki", "Failed to unwrap data for message.")
 | 
			
		||||
            }
 | 
			
		||||
        } else {
 | 
			
		||||
            Log.d("Loki", "Failed to decode data for message.")
 | 
			
		||||
        }
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
}
 | 
			
		||||
					Loading…
					
					
				
		Reference in New Issue