console 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. #!/usr/bin/env node
  2. var repl = require('repl').start('aws-sdk> '),
  3. replEval = repl.eval,
  4. defaultOptions = {
  5. logger: process.stdout,
  6. region: process.env.AWS_REGION || 'us-east-1'
  7. };
  8. function customEval(cmd, context, filename, callback) {
  9. replEval(cmd, context, filename, function(err, value) {
  10. if (err) {
  11. callback(err, null);
  12. return;
  13. }
  14. function consoleDataExtraction(resp) {
  15. context.data = resp.data;
  16. context.error = resp.error;
  17. callback(resp.error, resp.data);
  18. }
  19. if (value && value.constructor === AWS.Request && !value.__hasBeenEval__) {
  20. try {
  21. value.__hasBeenEval__ = true;
  22. if (value.response) value.response.__hasBeenEval__ = true;
  23. context.request = value;
  24. context.response = value.response || null;
  25. context.data = null;
  26. context.error = null;
  27. if (value._asm.currentState === 'complete' && value.response) {
  28. context.data = value.response.data || null;
  29. context.error = value.response.error || null;
  30. callback(value.response.error, value.response.data);
  31. } else {
  32. value.on('complete', consoleDataExtraction);
  33. if (!value.__hasBeenSent__) {
  34. if (context.autoSend) {
  35. value.send();
  36. } else {
  37. callback(null, value);
  38. }
  39. }
  40. }
  41. } catch (err2) {
  42. callback(err2, null);
  43. return;
  44. }
  45. } else if (value && value.constructor === AWS.Response && !value.__hasBeenEval__) {
  46. try {
  47. value.__hasBeenEval__ = true;
  48. context.response = value;
  49. context.request = value.request || null;
  50. context.data = value.data || null;
  51. context.error = value.error || null;
  52. if (value.request) {
  53. value.request.__hasBeenEval__ = true;
  54. if (value.request._asm.currentState === 'complete') {
  55. callback(value.error, value.data);
  56. } else {
  57. value.request.on('complete', consoleDataExtraction);
  58. }
  59. }
  60. } catch (err2) {
  61. callback(err2, null);
  62. return;
  63. }
  64. } else {
  65. callback(null, value);
  66. }
  67. });
  68. }
  69. var AWS = repl.context.AWS = require('../lib/aws');
  70. repl.eval = customEval;
  71. // context variables
  72. repl.context.data = null;
  73. repl.context.error = null;
  74. repl.context.request = null;
  75. repl.context.response = null;
  76. // setup REPL history
  77. try {
  78. var replHistory = require('repl.history');
  79. replHistory(repl, process.env.HOME + '/.node_history');
  80. } catch (e) {
  81. console.log("Missing repl.history package, history will not be supported.");
  82. console.log(" Type `npm install repl.history` to enable history.");
  83. }
  84. // modify Request.prototype.send listener to track if the listener has been called
  85. var sendListener = AWS.Request.prototype.send;
  86. AWS.Request.prototype.send = function(callback) {
  87. this.__hasBeenSent__ = true;
  88. return sendListener.call(this, callback);
  89. };
  90. // flag to indicate that requests should be sent when callback is not provided
  91. // by default this is on, but can be turned off by setting `autoSend = false`
  92. repl.context.autoSend = true;
  93. // load services as defined instances
  94. for (var key in AWS) {
  95. var id = AWS[key].serviceIdentifier;
  96. if (id) {
  97. if (id === 'cloudsearchdomain' || id === 'iotdata') continue; // this required an explicit endpoint
  98. var svcClass = AWS[key];
  99. var svc = new svcClass(defaultOptions);
  100. svc.with = function(config) {
  101. return new this.constructor.__super__(AWS.util.merge(this.config, config));
  102. };
  103. repl.context[svcClass.serviceIdentifier] = svc;
  104. }
  105. }