const AWS = require('aws-sdk');
const aws_access_key_id = "aws_access_key_id";
const aws_secret_access_key = "aws_secret_access_key";
const region = "region";
const UserPoolId = "UserPoolId";
const ClientId = "ClientId";

AWS.config.update({
  accessKeyId: aws_access_key_id,
  secretAccessKey: aws_secret_access_key,
  region: region
});

var cognitoServiceProvider = new AWS.CognitoIdentityServiceProvider();

function initiateLogin(username, password) {
  let params = {
    AuthFlow: "ADMIN_NO_SRP_AUTH",
    ClientId: ClientId,
    UserPoolId: UserPoolId,
    AuthParameters: {
      'USERNAME': username,
      "PASSWORD": password
    }
  };

  return new Promise((resolve, reject) => {
    cognitoServiceProvider.adminInitiateAuth(params, (err, data) => {
      if (err) {
        reject(new Error(err));
      } else {
        resolve(data);
      }
    });
  });
}


function confirmLogin(username, confirmationcode, session) {
  var params = {
    ChallengeName: "SMS_MFA",
    ClientId: ClientId,
    UserPoolId: UserPoolId,
    ChallengeResponses: {
      'USERNAME': username,
      'SMS_MFA_CODE': confirmationcode
    },
    Session: session
  };

  return new Promise((resolve, reject) => {
    cognitoServiceProvider.adminRespondToAuthChallenge(params, function (err, data) {
      if (err) reject(new Error(err));
      else resolve(data);
    });
  });
}

function verifyUserAttribute(username, confirmationcode, accessToken) {
  var params = {
    AccessToken: accessToken,
    AttributeName: 'phone_number',
    Code: confirmationcode
  };
  return new Promise((resolve, reject) => {
    cognitoServiceProvider.verifyUserAttribute(params, function (err, data) {
      if (err) reject(new Error(err));
      else resolve(data);
    });
  });
}

function adminSetUserMFAPreference(username) {
  let params = {
    UserPoolId: UserPoolId,
    Username: username,
    SMSMfaSettings: {
      Enabled: true,
      PreferredMfa: true
    }
  };
  return new Promise((resolve, reject) => {
    cognitoServiceProvider.adminSetUserMFAPreference(params, (err, data) => {
      if (err) reject(new Error(err));
      else resolve(data);
    });
  })
}

function updateUserAttributes(username, phonenumber) {
  let params = {
    UserAttributes: [{
      Name: 'phone_number',
      Value: phonenumber
    }],
    UserPoolId: UserPoolId,
    Username: username
  };
  return new Promise((resolve, reject) => {
    cognitoServiceProvider.adminUpdateUserAttributes(params, (err, data) => {
      if (err) reject(new Error(err));
      else resolve(data);
    });
  })
}

function getUserAttributeVerificationCode(accessToken) {
  var params = {
    AccessToken: accessToken,
    AttributeName: 'phone_number'
  };
  return new Promise((resolve, reject) => {
    cognitoServiceProvider.getUserAttributeVerificationCode(params, function (err, data) {
      if (err) reject(err);
      else resolve(data);
    });
  })
}

function signUp(username, password) {
  var params = {
    ClientId: ClientId,
    Username: username,
    Password: password,
    UserAttributes: [{
      Name: 'email',
      Value: username
    }]
  };

  return new Promise((resolve, reject) => {
    cognitoServiceProvider.signUp(params, (err, data) => {
      if (err) reject(new Error(err));
      else resolve(data);
    });
  })
};

function confirmSignup(username, confirmationCode) {
  var params = {
    ClientId: ClientId,
    ConfirmationCode: confirmationCode,
    Username: username
  };

  return new Promise((resolve, reject) => {
    cognitoServiceProvider.confirmSignUp(params, (err, data) => {
      if (err) p.reject(new Error(err));
      else resolve(data);
    });
  });
}

function forgotPassword(username) {
  var params = {
    ClientId: ClientId,
    Username: username
  };

  return new Promise((resolve, reject) => {
    cognitoServiceProvider.forgotPassword(params, (err, data) => {
      if (err) reject(new Error(err))
      else resolve(data)
    });
  });
}

function confirmForgotPassword(username, confirmationcode, password) {
  var params = {
    ClientId: ClientId,
    ConfirmationCode: confirmationcode,
    Password: password,
    Username: username,
  };
  return new Promise((resolve, reject) => {
    cognitoServiceProvider.confirmForgotPassword(params, function (err, data) {
      if (err) return err;
      else return data;
    });
  })
}

function resendConfirmationCode(data) {
  var params = {
    ClientId: ClientId,
    Username: data.username
  };
  return new Promise((resolve, reject) => {
    cognitoServiceProvider.resendConfirmationCode(params, (err, data) => {
      if (err) reject(new Error(err));
      else resolve(data);
    });
  })
}