206 lines
8.7 KiB
Kotlin
206 lines
8.7 KiB
Kotlin
package org.thoughtcrime.securesms.backup
|
|
|
|
import android.app.Activity
|
|
import android.app.Application
|
|
import android.content.Intent
|
|
import android.graphics.Typeface
|
|
import android.net.Uri
|
|
import android.os.Bundle
|
|
import android.provider.OpenableColumns
|
|
import android.text.Spannable
|
|
import android.text.SpannableStringBuilder
|
|
import android.text.style.ClickableSpan
|
|
import android.text.style.StyleSpan
|
|
import android.view.View
|
|
import android.widget.Toast
|
|
import androidx.activity.result.ActivityResult
|
|
import androidx.activity.result.contract.ActivityResultContracts
|
|
import androidx.activity.viewModels
|
|
import androidx.lifecycle.AndroidViewModel
|
|
import androidx.lifecycle.MutableLiveData
|
|
import androidx.lifecycle.viewModelScope
|
|
import com.google.android.gms.common.util.Strings
|
|
import kotlinx.coroutines.Dispatchers
|
|
import kotlinx.coroutines.launch
|
|
import kotlinx.coroutines.withContext
|
|
import network.loki.messenger.R
|
|
import org.session.libsession.utilities.TextSecurePreferences
|
|
import org.session.libsignal.utilities.Log
|
|
import org.thoughtcrime.securesms.BaseActionBarActivity
|
|
import org.thoughtcrime.securesms.backup.FullBackupImporter.DatabaseDowngradeException
|
|
import org.thoughtcrime.securesms.crypto.AttachmentSecretProvider
|
|
import org.thoughtcrime.securesms.database.DatabaseFactory
|
|
import org.thoughtcrime.securesms.dependencies.DatabaseComponent
|
|
import org.thoughtcrime.securesms.home.HomeActivity
|
|
import org.thoughtcrime.securesms.notifications.NotificationChannels
|
|
import org.thoughtcrime.securesms.util.BackupUtil
|
|
import org.thoughtcrime.securesms.util.setUpActionBarSessionLogo
|
|
import org.thoughtcrime.securesms.util.show
|
|
|
|
class BackupRestoreActivity : BaseActionBarActivity() {
|
|
|
|
companion object {
|
|
private const val TAG = "BackupRestoreActivity"
|
|
}
|
|
|
|
private val viewModel by viewModels<BackupRestoreViewModel>()
|
|
|
|
private val fileSelectionResultLauncher = registerForActivityResult(ActivityResultContracts.StartActivityForResult()
|
|
) { result: ActivityResult ->
|
|
if (result.resultCode == Activity.RESULT_OK && result.data != null && result.data!!.data != null) {
|
|
viewModel.backupFile.value = result.data!!.data!!
|
|
}
|
|
}
|
|
|
|
override fun onCreate(savedInstanceState: Bundle?) {
|
|
super.onCreate(savedInstanceState)
|
|
setUpActionBarSessionLogo()
|
|
|
|
// val viewBinding = DataBindingUtil.setContentView<ActivityBackupRestoreBinding>(this, R.layout.activity_backup_restore)
|
|
// viewBinding.lifecycleOwner = this
|
|
// viewBinding.viewModel = viewModel
|
|
|
|
// viewBinding.restoreButton.setOnClickListener { viewModel.tryRestoreBackup() }
|
|
|
|
// viewBinding.buttonSelectFile.setOnClickListener {
|
|
// fileSelectionResultLauncher.launch(Intent(Intent.ACTION_OPEN_DOCUMENT).apply {
|
|
// //FIXME On some old APIs (tested on 21 & 23) the mime type doesn't filter properly
|
|
// // and the backup files are unavailable for selection.
|
|
//// type = BackupUtil.BACKUP_FILE_MIME_TYPE
|
|
// type = "*/*"
|
|
// })
|
|
// }
|
|
|
|
// viewBinding.backupCode.addTextChangedListener { text -> viewModel.backupPassphrase.value = text.toString() }
|
|
|
|
// Focus passphrase text edit when backup file is selected.
|
|
// viewModel.backupFile.observe(this, { backupFile ->
|
|
// if (backupFile != null) viewBinding.backupCode.post {
|
|
// viewBinding.backupCode.requestFocus()
|
|
// (getSystemService(INPUT_METHOD_SERVICE) as InputMethodManager)
|
|
// .showSoftInput(viewBinding.backupCode, InputMethodManager.SHOW_IMPLICIT)
|
|
// }
|
|
// })
|
|
|
|
// React to backup import result.
|
|
viewModel.backupImportResult.observe(this) { result ->
|
|
if (result != null) when (result) {
|
|
BackupRestoreViewModel.BackupRestoreResult.SUCCESS -> {
|
|
val intent = Intent(this, HomeActivity::class.java)
|
|
intent.flags = Intent.FLAG_ACTIVITY_NEW_TASK or Intent.FLAG_ACTIVITY_CLEAR_TASK
|
|
this.show(intent)
|
|
}
|
|
BackupRestoreViewModel.BackupRestoreResult.FAILURE_VERSION_DOWNGRADE ->
|
|
Toast.makeText(this, R.string.RegistrationActivity_backup_failure_downgrade, Toast.LENGTH_LONG).show()
|
|
BackupRestoreViewModel.BackupRestoreResult.FAILURE_UNKNOWN ->
|
|
Toast.makeText(this, R.string.RegistrationActivity_incorrect_backup_passphrase, Toast.LENGTH_LONG).show()
|
|
}
|
|
}
|
|
|
|
//region Legal info views
|
|
val termsExplanation = SpannableStringBuilder("By using this service, you agree to our Terms of Service and Privacy Policy")
|
|
termsExplanation.setSpan(StyleSpan(Typeface.BOLD), 40, 56, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE)
|
|
termsExplanation.setSpan(object : ClickableSpan() {
|
|
override fun onClick(widget: View) {
|
|
openURL("https://getsession.org/terms-of-service/")
|
|
}
|
|
}, 40, 56, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE)
|
|
termsExplanation.setSpan(StyleSpan(Typeface.BOLD), 61, 75, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE)
|
|
termsExplanation.setSpan(object : ClickableSpan() {
|
|
override fun onClick(widget: View) {
|
|
openURL("https://getsession.org/privacy-policy/")
|
|
}
|
|
}, 61, 75, Spannable.SPAN_EXCLUSIVE_EXCLUSIVE)
|
|
// viewBinding.termsTextView.movementMethod = LinkMovementMethod.getInstance()
|
|
// viewBinding.termsTextView.text = termsExplanation
|
|
//endregion
|
|
}
|
|
|
|
private fun openURL(url: String) {
|
|
try {
|
|
val intent = Intent(Intent.ACTION_VIEW, Uri.parse(url))
|
|
startActivity(intent)
|
|
} catch (e: Exception) {
|
|
Toast.makeText(this, R.string.invalid_url, Toast.LENGTH_SHORT).show()
|
|
}
|
|
}
|
|
}
|
|
|
|
class BackupRestoreViewModel(application: Application): AndroidViewModel(application) {
|
|
|
|
companion object {
|
|
private const val TAG = "BackupRestoreViewModel"
|
|
|
|
@JvmStatic
|
|
fun uriToFileName(view: View, fileUri: Uri?): String? {
|
|
fileUri ?: return null
|
|
|
|
view.context.contentResolver.query(fileUri, null, null, null, null).use {
|
|
val nameIndex = it!!.getColumnIndex(OpenableColumns.DISPLAY_NAME)
|
|
it.moveToFirst()
|
|
return it.getString(nameIndex)
|
|
}
|
|
}
|
|
|
|
@JvmStatic
|
|
fun validateData(fileUri: Uri?, passphrase: String?): Boolean {
|
|
return fileUri != null &&
|
|
!Strings.isEmptyOrWhitespace(passphrase) &&
|
|
passphrase!!.length == BackupUtil.BACKUP_PASSPHRASE_LENGTH
|
|
}
|
|
}
|
|
|
|
val backupFile = MutableLiveData<Uri>(null)
|
|
val backupPassphrase = MutableLiveData<String>(null)
|
|
|
|
val processingBackupFile = MutableLiveData<Boolean>(false)
|
|
val backupImportResult = MutableLiveData<BackupRestoreResult>(null)
|
|
|
|
fun tryRestoreBackup() = viewModelScope.launch {
|
|
if (processingBackupFile.value == true) return@launch
|
|
if (backupImportResult.value == BackupRestoreResult.SUCCESS) return@launch
|
|
if (!validateData(backupFile.value, backupPassphrase.value)) return@launch
|
|
|
|
val context = getApplication<Application>()
|
|
val backupFile = backupFile.value!!
|
|
val passphrase = backupPassphrase.value!!
|
|
|
|
val result: BackupRestoreResult
|
|
|
|
processingBackupFile.value = true
|
|
|
|
withContext(Dispatchers.IO) {
|
|
result = try {
|
|
val database = DatabaseComponent.get(context).openHelper().readableDatabase
|
|
FullBackupImporter.importFromUri(
|
|
context,
|
|
AttachmentSecretProvider.getInstance(context).getOrCreateAttachmentSecret(),
|
|
database,
|
|
backupFile,
|
|
passphrase
|
|
)
|
|
DatabaseFactory.upgradeRestored(context, database)
|
|
NotificationChannels.restoreContactNotificationChannels(context)
|
|
TextSecurePreferences.setRestorationTime(context, System.currentTimeMillis())
|
|
TextSecurePreferences.setHasViewedSeed(context, true)
|
|
TextSecurePreferences.setHasSeenWelcomeScreen(context, true)
|
|
|
|
BackupRestoreResult.SUCCESS
|
|
} catch (e: DatabaseDowngradeException) {
|
|
Log.w(TAG, "Failed due to the backup being from a newer version of Signal.", e)
|
|
BackupRestoreResult.FAILURE_VERSION_DOWNGRADE
|
|
} catch (e: Exception) {
|
|
Log.w(TAG, e)
|
|
BackupRestoreResult.FAILURE_UNKNOWN
|
|
}
|
|
}
|
|
|
|
processingBackupFile.value = false
|
|
|
|
backupImportResult.value = result
|
|
}
|
|
|
|
enum class BackupRestoreResult {
|
|
SUCCESS, FAILURE_VERSION_DOWNGRADE, FAILURE_UNKNOWN
|
|
}
|
|
} |