session-android/src/org/thoughtcrime/securesms/loki/LinkedDevicesActivity.kt

package org.thoughtcrime.securesms.loki

import android.os.AsyncTask
import android.os.Bundle
import android.view.MenuItem
import android.widget.Toast
import org.thoughtcrime.securesms.*
import org.thoughtcrime.securesms.util.DynamicTheme
import org.thoughtcrime.securesms.util.DynamicLanguage
import network.loki.messenger.R
import nl.komponents.kovenant.then
import org.thoughtcrime.securesms.database.DatabaseFactory
import org.thoughtcrime.securesms.sms.MessageSender
import org.thoughtcrime.securesms.util.TextSecurePreferences
import org.thoughtcrime.securesms.util.Util
import org.whispersystems.signalservice.loki.api.LokiStorageAPI
import org.whispersystems.signalservice.loki.api.PairingAuthorisation

class LinkedDevicesActivity : PassphraseRequiredActionBarActivity(), DeviceLinkingDelegate {

  companion object {
    private val TAG = DeviceActivity::class.java.simpleName
  }

  private val dynamicTheme = DynamicTheme()
  private val dynamicLanguage = DynamicLanguage()
  private lateinit var deviceListFragment: DeviceListFragment

  public override fun onPreCreate() {
    dynamicTheme.onCreate(this)
    dynamicLanguage.onCreate(this)
  }

  override fun onCreate(savedInstanceState: Bundle?, ready: Boolean) {
    super.onCreate(savedInstanceState, ready)
    supportActionBar?.setDisplayHomeAsUpEnabled(true)
    supportActionBar?.setTitle(R.string.AndroidManifest__linked_devices)
    this.deviceListFragment = DeviceListFragment()
    this.deviceListFragment.setAddDeviceButtonListener {
      DeviceLinkingDialog.show(this, DeviceLinkingView.Mode.Master, this)
    }
    this.deviceListFragment.setHandleDisconnectDevice { devicePublicKey ->
      // Purge the device pairing from our database
      val ourPublicKey = TextSecurePreferences.getLocalNumber(this)
      val database = DatabaseFactory.getLokiAPIDatabase(this)
      database.removePairingAuthorisation(ourPublicKey, devicePublicKey)
      // Update mapping on the file server
      LokiStorageAPI.shared.updateUserDeviceMappings().success {
        // Send an unpair request to let the device know that it has been revoked
        MessageSender.sendUnpairRequest(this, devicePublicKey)
      }
      // Refresh the list
      this.deviceListFragment.refresh()
      Toast.makeText(this, R.string.DeviceListActivity_unlinked_device, Toast.LENGTH_LONG).show()
      return@setHandleDisconnectDevice null
    }
    this.deviceListFragment.setHandleDeviceNameChange { pair ->
      DatabaseFactory.getLokiUserDatabase(this).setDisplayName(pair.first, pair.second)
      this.deviceListFragment.refresh()
      return@setHandleDeviceNameChange null
    }
    initFragment(android.R.id.content, deviceListFragment, dynamicLanguage.currentLocale)
  }

  public override fun onResume() {
    super.onResume()
    dynamicTheme.onResume(this)
    dynamicLanguage.onResume(this)
  }

  override fun onOptionsItemSelected(item: MenuItem): Boolean {
    if (item.itemId == android.R.id.home) {
      finish()
      return true
    }
    return false
  }

  override fun sendPairingAuthorizedMessage(pairingAuthorisation: PairingAuthorisation) {
    AsyncTask.execute {
      signAndSendPairingAuthorisationMessage(this, pairingAuthorisation)
      Util.runOnMain { this.deviceListFragment.refresh() }
    }
  }
}