proxyUrl and headers have been added

coco50511 · coco50511 · commit f2f60c698f7b · 2020-02-19T07:47:37.000+08:00
diff --git a/lib/get-request.js b/lib/get-request.js
@@ -21,11 +21,26 @@ const TESTNET_API_URL_MAP = {
   homestead: 'https://api.etherscan.io'
 };
 
-module.exports = function(chain, timeout) {
-  var client = axios.create({
-    baseURL: pickChainUrl(chain),
+module.exports = function(chain, timeout, proxyUrl, headers) {
+  var param = {
     timeout: timeout
-  });
+  };
+
+  var baseUrl = pickChainUrl(chain);
+  if (proxyUrl && 0 < proxyUrl.length) {
+    if (proxyUrl.charAt(proxyUrl.length - 1) == '/') {
+      baseUrl = proxyUrl + '/' + baseUrl;
+    } else {
+      baseUrl = proxyUrl + baseUrl;
+    }
+  }
+  param[baseUrl] = baseUrl;
+
+  if (headers) {
+    param[headers] = headers;
+  }
+
+  var client = axios.create(param);
 
   /**
    * @param query
diff --git a/lib/init.js b/lib/init.js
@@ -14,8 +14,10 @@ const account = require('./account');
  * @param {string} apiKey - (optional) Your Etherscan APIkey
  * @param {string} chain - (optional) Testnet chain keys [ropsten, rinkeby, kovan]
  * @param {number} timeout - (optional) Timeout in milliseconds for requests, default 10000
+ * @param {string} proxyUrl - (optional) The URL of proxy server to avoid CORS
+ * @param {object} headers - (optional) The object containing custom headers
  */
-module.exports = function(apiKey, chain, timeout) {
+module.exports = function(apiKey, chain, timeout, proxyUrl, headers) {
 
   if (!apiKey) {
     apiKey = 'YourApiKeyToken';
@@ -26,7 +28,7 @@ module.exports = function(apiKey, chain, timeout) {
     timeout = 10000;
   }
 
-  var getRequest = require('./get-request')(chain, timeout);
+  var getRequest = require('./get-request')(chain, timeout, proxyUrl, headers);
 
   /** @lends module:etherscan/api */
   return {
diff --git a/package-lock.json b/package-lock.json
