Would like to do the same with AWS Cognito. Did you find the way for the callback? or any kind of documentation?