Handle device confirmation?

[nahojkap]

I have been testing with a user pool which is configured to ask the user whether the device being logged in should be remembered or not and the code is throwing a CognitoUserConfirmationNecessaryException.

Now, I had initially assumed that this exception was used to indicate the user needs confirmation using a pin but if it is also used to indicate device confirmation, there is no way to distinguish the two different flows. Should the device confirmation requirement be signalled using a different exception?

And safe to assume the methods to confirm a device are still work in progress?

Thanks,

Johan