package exh.widget.preference import android.app.Dialog import android.os.Bundle import android.view.LayoutInflater import androidx.core.os.bundleOf import androidx.core.view.isVisible import com.afollestad.materialdialogs.MaterialDialog import com.afollestad.materialdialogs.customview.customView import com.bluelinelabs.conductor.ControllerChangeHandler import com.bluelinelabs.conductor.ControllerChangeType import com.dd.processbutton.iml.ActionProcessButton import eu.kanade.tachiyomi.R import eu.kanade.tachiyomi.data.preference.PreferencesHelper import eu.kanade.tachiyomi.data.track.TrackManager import eu.kanade.tachiyomi.databinding.PrefSiteLoginTwoFactorAuthBinding import eu.kanade.tachiyomi.source.Source import eu.kanade.tachiyomi.source.SourceManager import eu.kanade.tachiyomi.source.online.all.MangaDex import eu.kanade.tachiyomi.ui.base.controller.DialogController import eu.kanade.tachiyomi.util.system.toast import exh.source.getMainSource import kotlinx.coroutines.CoroutineScope import kotlinx.coroutines.Dispatchers import kotlinx.coroutines.Job import kotlinx.coroutines.cancel import kotlinx.coroutines.launch import uy.kohesive.injekt.Injekt import uy.kohesive.injekt.api.get import uy.kohesive.injekt.injectLazy class MangadexLoginDialog(bundle: Bundle? = null) : DialogController(bundle) { val source = Injekt.get().get(args.getLong("key", 0))?.getMainSource() as? MangaDex val service = Injekt.get().mdList val preferences: PreferencesHelper by injectLazy() val scope = CoroutineScope(Job() + Dispatchers.Main) lateinit var binding: PrefSiteLoginTwoFactorAuthBinding constructor(source: MangaDex) : this( bundleOf( "key" to source.id ) ) override fun onCreateDialog(savedViewState: Bundle?): Dialog { binding = PrefSiteLoginTwoFactorAuthBinding.inflate(LayoutInflater.from(activity!!)) val dialog = MaterialDialog(activity!!) .customView(view = binding.root, scrollable = false) onViewCreated() return dialog } fun onViewCreated() { binding.login.setMode(ActionProcessButton.Mode.ENDLESS) binding.login.setOnClickListener { checkLogin() } setCredentialsOnView() binding.twoFactorCheck.setOnCheckedChangeListener { _, isChecked -> binding.twoFactorHolder.isVisible = isChecked } } private fun setCredentialsOnView() { binding.username.setText(service.getUsername()) binding.password.setText(service.getPassword()) } private fun checkLogin() { with(binding) { if (username.text.isNullOrBlank() || password.text.isNullOrBlank() || (twoFactorCheck.isChecked && twoFactorEdit.text.isNullOrBlank())) { errorResult() root.context.toast(R.string.fields_cannot_be_blank) return } login.progress = 1 dialog?.setCancelable(false) dialog?.setCanceledOnTouchOutside(false) scope.launch { try { val result = source?.login( username.text.toString(), password.text.toString(), twoFactorEdit.text.toString() ) ?: false if (result) { dialog?.dismiss() preferences.setTrackCredentials(Injekt.get().mdList, username.toString(), password.toString()) root.context.toast(R.string.login_success) } else { errorResult() } } catch (error: Exception) { errorResult() error.message?.let { root.context.toast(it) } } } } } private fun errorResult() { with(binding) { dialog?.setCancelable(true) dialog?.setCanceledOnTouchOutside(true) login.progress = -1 login.setText(R.string.unknown_error) } } override fun onChangeStarted(handler: ControllerChangeHandler, type: ControllerChangeType) { super.onChangeStarted(handler, type) if (!type.isEnter) { onDialogClosed() } } private fun onDialogClosed() { scope.cancel() if (activity != null) { (activity as? Listener)?.siteLoginDialogClosed(source!!) } else { (targetController as? Listener)?.siteLoginDialogClosed(source!!) } } interface Listener { fun siteLoginDialogClosed(source: Source) } }