Signal-Android/app/src/main/java/org/thoughtcrime/securesms/jobs/CheckServiceReachabilityJob.kt

package org.thoughtcrime.securesms.jobs

import org.signal.core.util.logging.Log
import org.thoughtcrime.securesms.BuildConfig
import org.thoughtcrime.securesms.dependencies.ApplicationDependencies
import org.thoughtcrime.securesms.jobmanager.Data
import org.thoughtcrime.securesms.jobmanager.Job
import org.thoughtcrime.securesms.jobmanager.impl.NetworkConstraint
import org.thoughtcrime.securesms.keyvalue.SignalStore
import org.thoughtcrime.securesms.stories.Stories
import org.whispersystems.signalservice.api.websocket.WebSocketConnectionState
import org.whispersystems.signalservice.internal.util.StaticCredentialsProvider
import org.whispersystems.signalservice.internal.websocket.WebSocketConnection
import java.util.Optional
import java.util.concurrent.TimeUnit

/**
 * Checks to see if a censored user can establish a websocket connection with an uncensored network configuration.
 */
class CheckServiceReachabilityJob private constructor(params: Parameters) : BaseJob(params) {

  constructor() : this(
    Parameters.Builder()
      .addConstraint(NetworkConstraint.KEY)
      .setLifespan(TimeUnit.HOURS.toMillis(12))
      .setMaxAttempts(1)
      .build()
  )

  companion object {
    private val TAG = Log.tag(CheckServiceReachabilityJob::class.java)

    const val KEY = "CheckServiceReachabilityJob"

    @JvmStatic
    fun enqueueIfNecessary() {
      val isCensored = ApplicationDependencies.getSignalServiceNetworkAccess().isCensored()
      val timeSinceLastCheck = System.currentTimeMillis() - SignalStore.misc().lastCensorshipServiceReachabilityCheckTime
      if (SignalStore.account().isRegistered && isCensored && timeSinceLastCheck > TimeUnit.DAYS.toMillis(1)) {
        ApplicationDependencies.getJobManager().add(CheckServiceReachabilityJob())
      }
    }
  }

  override fun serialize(): Data {
    return Data.EMPTY
  }

  override fun getFactoryKey(): String {
    return KEY
  }

  override fun onRun() {
    if (!SignalStore.account().isRegistered) {
      Log.w(TAG, "Not registered, skipping.")
      SignalStore.misc().lastCensorshipServiceReachabilityCheckTime = System.currentTimeMillis()
      return
    }

    if (!ApplicationDependencies.getSignalServiceNetworkAccess().isCensored()) {
      Log.w(TAG, "Not currently censored, skipping.")
      SignalStore.misc().lastCensorshipServiceReachabilityCheckTime = System.currentTimeMillis()
      return
    }

    SignalStore.misc().lastCensorshipServiceReachabilityCheckTime = System.currentTimeMillis()

    val uncensoredWebsocket = WebSocketConnection(
      "uncensored-test",
      ApplicationDependencies.getSignalServiceNetworkAccess().uncensoredConfiguration,
      Optional.of(
        StaticCredentialsProvider(
          SignalStore.account().aci,
          SignalStore.account().pni,
          SignalStore.account().e164,
          SignalStore.account().deviceId,
          SignalStore.account().servicePassword
        )
      ),
      BuildConfig.SIGNAL_AGENT,
      null,
      "",
      Stories.isFeatureEnabled()
    )

    try {
      val startTime = System.currentTimeMillis()

      val state: WebSocketConnectionState = uncensoredWebsocket.connect()
        .filter { it == WebSocketConnectionState.CONNECTED || it == WebSocketConnectionState.FAILED }
        .timeout(30, TimeUnit.SECONDS)
        .blockingFirst(WebSocketConnectionState.FAILED)

      if (state == WebSocketConnectionState.CONNECTED) {
        Log.i(TAG, "Established connection in ${System.currentTimeMillis() - startTime} ms! Service is reachable!")
        SignalStore.misc().isServiceReachableWithoutCircumvention = true
      } else {
        Log.w(TAG, "Failed to establish a connection in ${System.currentTimeMillis() - startTime} ms.")
        SignalStore.misc().isServiceReachableWithoutCircumvention = false
      }
    } catch (exception: Exception) {
      Log.w(TAG, "Failed to connect to the websocket.", exception)
      SignalStore.misc().isServiceReachableWithoutCircumvention = false
    } finally {
      uncensoredWebsocket.disconnect()
    }
  }

  override fun onShouldRetry(e: Exception): Boolean {
    return false
  }

  override fun onFailure() {
  }

  class Factory : Job.Factory<CheckServiceReachabilityJob> {
    override fun create(parameters: Parameters, data: Data): CheckServiceReachabilityJob {
      return CheckServiceReachabilityJob(parameters)
    }
  }
}