funkwhale-app-android/app/src/main/java/audio/funkwhale/ffa/activities/LoginActivity.kt

package audio.funkwhale.ffa.activities

import android.content.Context
import android.content.Intent
import android.content.res.Configuration
import android.net.Uri
import android.os.Bundle
import android.text.Editable
import android.view.ViewGroup
import androidx.activity.result.contract.ActivityResultContracts.StartActivityForResult
import androidx.appcompat.app.AppCompatActivity
import androidx.core.view.doOnLayout
import androidx.lifecycle.lifecycleScope
import audio.funkwhale.ffa.R
import audio.funkwhale.ffa.databinding.ActivityLoginBinding
import audio.funkwhale.ffa.fragments.LoginDialog
import audio.funkwhale.ffa.utils.AppContext
import audio.funkwhale.ffa.utils.FuelResult
import audio.funkwhale.ffa.utils.OAuth
import audio.funkwhale.ffa.utils.Userinfo
import com.github.kittinunf.fuel.Fuel
import com.github.kittinunf.fuel.coroutines.awaitObjectResponseResult
import com.github.kittinunf.fuel.gson.gsonDeserializerOf
import com.github.kittinunf.result.Result
import com.preference.PowerPreference
import kotlinx.coroutines.Dispatchers.Main
import kotlinx.coroutines.launch
import kotlinx.coroutines.runBlocking
import org.koin.java.KoinJavaComponent.inject

data class FwCredentials(val token: String, val non_field_errors: List<String>?)

class LoginActivity : AppCompatActivity() {

  private lateinit var binding: ActivityLoginBinding
  private val oAuth: OAuth by inject(OAuth::class.java)

  override fun onCreate(savedInstanceState: Bundle?) {
    super.onCreate(savedInstanceState)

    binding = ActivityLoginBinding.inflate(layoutInflater)
    setContentView(binding.root)
    limitContainerWidth()
  }

  private var resultLauncher =
    registerForActivityResult(StartActivityForResult()) { result ->
      result.data?.let {
        oAuth.exchange(this, it) {
          PowerPreference
            .getFileByName(AppContext.PREFS_CREDENTIALS)
            .setBoolean("anonymous", false)

          lifecycleScope.launch(Main) {
            Userinfo.get(this@LoginActivity, oAuth)?.let {
              startActivity(Intent(this@LoginActivity, MainActivity::class.java))

              return@launch finish()
            }
            throw Exception(getString(R.string.login_error_userinfo))
          }
        }
      }
    }

  override fun onResume() {
    super.onResume()
    with(binding) {
      val preferences = getPreferences(Context.MODE_PRIVATE)
      val hn = preferences?.getString("hostname", "")
      if (hn != null && !hn.isEmpty()) {
        hostname.text = Editable.Factory.getInstance().newEditable(hn)
      }
      cleartext.setChecked(preferences?.getBoolean("cleartext", false) ?: false)
      anonymous.setChecked(preferences?.getBoolean("anonymous", false) ?: false)
      login.setOnClickListener {
        var hostname = hostname.text.toString().trim().trim('/')

        try {
          validateHostname(hostname, cleartext.isChecked)?.let {
            hostnameField.error = it
            return@setOnClickListener
          }

          val uri = Uri.parse(hostname)
          if (uri.scheme == null) {
            hostname = when (cleartext.isChecked) {
              true -> "http://$hostname"
              false -> "https://$hostname"
            }
          }

          hostnameField.error = ""

          val fuelResult = when (anonymous.isChecked) {
            false -> authedLogin(hostname)
            true -> anonymousLogin(hostname)
          }

          hostnameField.error = mapFuelResultToError(fuelResult)
        } catch (e: Exception) {
          val message =
            if (e.message?.isEmpty() == true) getString(R.string.login_error_hostname)
            else e.message

          hostnameField.error = message
        }
        if (hostnameField.error == null) {
          val preferences = getPreferences(Context.MODE_PRIVATE)
          preferences?.edit()?.putString("hostname", hostname)?.commit()
          preferences?.edit()?.putBoolean("cleartext", cleartext.isChecked)?.commit()
          preferences?.edit()?.putBoolean("anonymous", anonymous.isChecked)?.commit()
        }
      }
    }
  }

  private fun mapFuelResultToError(fuelResult: FuelResult) = when {
    fuelResult.httpStatus == 404 ->
      getString(R.string.login_error_funkwhale_not_found)

    !fuelResult.success ->
      getString(R.string.login_error, fuelResult.message)

    else -> ""
  }

  override fun onConfigurationChanged(newConfig: Configuration) {
    super.onConfigurationChanged(newConfig)

    limitContainerWidth()
  }

  private fun validateHostname(hostname: String, cleartext: Boolean): String? {
    if (hostname.isEmpty()) {
      return getString(R.string.login_error_hostname)
    }
    if (!cleartext && hostname.startsWith("http://")) {
      return getString(R.string.login_error_hostname_https)
    }
    return null
  }

  private fun authedLogin(hostname: String): FuelResult {
    oAuth.init(hostname)
    return oAuth.register {
      PowerPreference.getFileByName(AppContext.PREFS_CREDENTIALS).setString("hostname", hostname)
      resultLauncher.launch(oAuth.authorizeIntent(this))
    }
  }

  private fun anonymousLogin(hostname: String): FuelResult {
    val dialog = LoginDialog().apply {
      show(supportFragmentManager, "LoginDialog")
    }

    val uri = "$hostname/api/v1/tracks/"
    val (_, _, result) = runBlocking {
      Fuel.get(uri).awaitObjectResponseResult(gsonDeserializerOf(FwCredentials::class.java))
    }

    when (result) {
      is Result.Success -> {
        PowerPreference.getFileByName(AppContext.PREFS_CREDENTIALS).apply {
          setString("hostname", hostname)
          setBoolean("anonymous", true)
        }

        dialog.dismiss()
        startActivity(Intent(this@LoginActivity, MainActivity::class.java))
        finish()
        return FuelResult.ok()
      }

      is Result.Failure -> {
        dialog.dismiss()
        return FuelResult.from(result)
      }
    }
  }

  private fun limitContainerWidth() {
    binding.container.doOnLayout {
      if (resources.configuration.orientation == Configuration.ORIENTATION_LANDSCAPE && binding.container.width >= 1440) {
        binding.container.layoutParams.width = 1440
      } else {
        binding.container.layoutParams.width = ViewGroup.LayoutParams.MATCH_PARENT
      }

      binding.container.requestLayout()
    }
  }
}