SmsRetriever.kt

@Composable
fun SmsRetriever(
    smsCodeLength: Int = SMS_CODE_LENGTH,
    onSmsReceived: (message: String, code: String) -> Unit,
) {
    val context = LocalContext.current

    var shouldRegisterReceiver by remember { mutableStateOf(false) }

    LaunchedEffect(Unit) {
        SmsRetriever.getClient(context)
            .startSmsRetriever()
            .addOnSuccessListener {
                shouldRegisterReceiver = true
            }
    }

    if (shouldRegisterReceiver) {
        SystemBroadcastReceiver(
            systemAction = SmsRetriever.SMS_RETRIEVED_ACTION,
            broadCastPermission = SmsRetriever.SEND_PERMISSION,
        ) { intent ->
            if (SmsRetriever.SMS_RETRIEVED_ACTION == intent?.action) {
                val extras = intent.extras

                val smsRetrieverStatus = extras?.get(SmsRetriever.EXTRA_STATUS) as Status
                when (smsRetrieverStatus.statusCode) {
                    CommonStatusCodes.SUCCESS -> {
                        val message: String? = extras.getString(SmsRetriever.EXTRA_SMS_MESSAGE)
                        message?.let {
                            val verificationCode = getVerificationCodeFromSms(message, smsCodeLength)
                            onSmsReceived(message, verificationCode) 
                        }
                        shouldRegisterReceiver = false
                    }

                    CommonStatusCodes.TIMEOUT -> Timber.d("Timeout in sms verification receiver")
                }
            }
        }
    }
}