[Link new device] Automatically rotate the QR Code - limit number of rotation to 10.

Author: bmarty

features/linknewdevice/impl/src/main/kotlin/io/element/android/features/linknewdevice/impl/LinkNewMobileHandler.kt

@@ -12,6 +12,7 @@ import dev.zacsweers.metro.SingleIn
 import io.element.android.libraries.core.log.logger.LoggerTag
 import io.element.android.libraries.di.SessionScope
 import io.element.android.libraries.matrix.api.MatrixClient
+import io.element.android.libraries.matrix.api.linknewdevice.ErrorType
 import io.element.android.libraries.matrix.api.linknewdevice.LinkMobileHandler
 import io.element.android.libraries.matrix.api.linknewdevice.LinkMobileStep
 import io.element.android.libraries.matrix.api.logs.LoggerTags
@@ -69,4 +70,11 @@ class LinkNewMobileHandler(
     fun rotateQrCode() {
         createAndStartNewHandler()
     }
+
+    fun onTooManyRotation() {
+        reset()
+        sessionScope.launch {
+            linkMobileStepFlow.emit(LinkMobileStep.Error(ErrorType.Expired("Too many QR code rotations")))
+        }
+    }
 }

features/linknewdevice/impl/src/main/kotlin/io/element/android/features/linknewdevice/impl/screens/qrcode/ShowQrCodePresenter.kt

@@ -9,7 +9,10 @@ package io.element.android.features.linknewdevice.impl.screens.qrcode
 
 import androidx.compose.runtime.Composable
 import androidx.compose.runtime.getValue
+import androidx.compose.runtime.mutableIntStateOf
 import androidx.compose.runtime.produceState
+import androidx.compose.runtime.remember
+import androidx.compose.runtime.setValue
 import dev.zacsweers.metro.Assisted
 import dev.zacsweers.metro.AssistedFactory
 import dev.zacsweers.metro.AssistedInject
@@ -35,16 +38,22 @@ class ShowQrCodePresenter(
 
     @Composable
     override fun present(): ShowQrCodeState {
+        var qrCodeRotationCounter by remember { mutableIntStateOf(MAX_QR_CODE_ROTATION) }
         val data by produceState<AsyncData<String>>(AsyncData.Success(initialData)) {
             linkNewMobileHandler.stepFlow.collect { step ->
                 when (step) {
                     is LinkMobileStep.QrReady -> {
                         value = AsyncData.Success(step.data)
                     }
                     is LinkMobileStep.QrRotating -> {
-                        Timber.tag(tag.value).d("Rotating QrCode")
-                        linkNewMobileHandler.rotateQrCode()
-                        value = AsyncData.Loading()
+                        if (qrCodeRotationCounter-- > 0) {
+                            Timber.tag(tag.value).d("Rotating QrCode")
+                            linkNewMobileHandler.rotateQrCode()
+                            value = AsyncData.Loading()
+                        } else {
+                            Timber.tag(tag.value).w("Max QR code rotation reached, not rotating anymore")
+                            linkNewMobileHandler.onTooManyRotation()
+                        }
                     }
                     else -> Unit
                 }
@@ -55,4 +64,8 @@ class ShowQrCodePresenter(
             data = data,
         )
     }
+
+    companion object {
+        const val MAX_QR_CODE_ROTATION = 10
+    }
 }