import java.util.HashMap;
import java.util.Map;

import com.amazonaws.services.cognitoidp.AWSCognitoIdentityProviderClient;
import com.amazonaws.services.cognitoidp.model.*;


/**
 *  Demonstrates that it's possible to login with the original credentials
 *  after initiating the Forgot Password flow.
 */
public class ForgotPasswordExample
{
    public static void main(String[] argv)
    throws Exception
    {
        String poolId   = "UPDATE";
        String clientId = "UPDATE";
        String username = "UPDATE";
        String password = "UPDATE";

        AWSCognitoIdentityProviderClient cognitoClient = new AWSCognitoIdentityProviderClient();

        ForgotPasswordRequest request1 = new ForgotPasswordRequest()
                                           .withClientId(clientId)
                                           .withUsername(username);
        ForgotPasswordResult response1 = cognitoClient.forgotPassword(request1);
        System.out.println("result of marking password forgotten: " + response1);

        Map<String,String> authParams = new HashMap<String,String>();
        authParams.put("USERNAME", username);
        authParams.put("PASSWORD", password);

        AdminInitiateAuthRequest request2 = new AdminInitiateAuthRequest()
                                          .withAuthFlow(AuthFlowType.ADMIN_NO_SRP_AUTH)
                                          .withAuthParameters(authParams)
                                          .withClientId(clientId)
                                          .withUserPoolId(poolId);

        AdminInitiateAuthResult response2 = cognitoClient.adminInitiateAuth(request2);
        System.out.println("authentication result: " + response2);
    }
}