Fix missed onAuthorizationCodeReceived callback

octopus-energy · Feb 11, 2022 · 4cdbcc6 · 4cdbcc6
1 parent a8765b6
commit 4cdbcc6
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/teslauth/src/main/java/energy/octopus/octopusenergy/teslauth/TeslaAuth.kt b/teslauth/src/main/java/energy/octopus/octopusenergy/teslauth/TeslaAuth.kt
@@ -89,8 +89,11 @@ fun TeslAuth(
                 }
                 is ReceivedAccessToken -> onAccessTokenReceived?.invoke(it.token)
                 is Error -> onError(it.t)
-                is AuthorizationCodeReceived -> if (onBearerTokenReceived != null || onAccessTokenReceived != null) {
-                    viewModel.getBearerToken(it.code)
+                is AuthorizationCodeReceived -> {
+                    onAuthorizationCodeReceived?.invoke(it.code)
+                    if (onBearerTokenReceived != null || onAccessTokenReceived != null) {
+                        viewModel.getBearerToken(it.code)
+                    }
                 }
                 Dismiss -> onDismiss()
             }