// taken from https://github.com/taskcluster/taskcluster-queue/blob/0dd2b45b860836a2ef6c651c5499c9abfb5670ad/utils/aws-sdk-promise.js
var aws = require('aws-sdk');
var Promise = require('promise');

// XXX: This is a terrible hack but it should not break anything in horrible
// ways.
aws.Request.prototype.promise = function() {
  return new Promise(function(accept, reject) {
    this.on('complete', function(response) {
      if (response.error) {
        reject(response.error);
      } else {
        accept(response);
      }
    });
    this.send();
  }.bind(this));
};

module.exports = aws;