switch to node-fetch

This commit is contained in:
Will Nelson
2018-06-08 20:01:22 -07:00
parent fc96cfbc00
commit 2239b05e13
7 changed files with 67 additions and 52 deletions

View File

@@ -1,7 +1,8 @@
const querystring = require('querystring');
const snekfetch = require('snekfetch');
const FormData = require('form-data');
const https = require('https');
const { browser, UserAgent } = require('../util/Constants');
var fetch = require('node-fetch');
if (https.Agent) var agent = new https.Agent({ keepAlive: true });
@@ -9,7 +10,7 @@ class APIRequest {
constructor(rest, method, path, options) {
this.rest = rest;
this.client = rest.client;
this.method = method;
this.method = method.toUpperCase();
this.route = options.route;
this.options = options;
@@ -17,24 +18,34 @@ class APIRequest {
this.path = `${path}${queryString ? `?${queryString}` : ''}`;
}
gen() {
make() {
const API = this.options.versioned === false ? this.client.options.http.api :
`${this.client.options.http.api}/v${this.client.options.http.version}`;
const url = API + this.path;
let headers = {};
const request = snekfetch[this.method](`${API}${this.path}`, { agent });
if (this.options.auth !== false) request.set('Authorization', this.rest.getAuth());
if (this.options.reason) request.set('X-Audit-Log-Reason', encodeURIComponent(this.options.reason));
if (!browser) request.set('User-Agent', UserAgent);
if (this.options.headers) request.set(this.options.headers);
if (this.options.auth !== false) headers.Authorization = this.rest.getAuth();
if (this.options.reason) headers['X-Audit-Log-Reason'] = encodeURIComponent(this.options.reason);
if (!browser) headers['User-Agent'] = UserAgent;
if (this.options.headers) headers = Object.assign(headers, this.options.headers);
let body;
if (this.options.files) {
for (const file of this.options.files) if (file && file.file) request.attach(file.name, file.file, file.name);
if (typeof this.options.data !== 'undefined') request.attach('payload_json', JSON.stringify(this.options.data));
} else if (typeof this.options.data !== 'undefined') {
request.send(this.options.data);
body = new FormData();
for (const file of this.options.files) if (file && file.file) body.append(file.name, file.file, file.name);
if (typeof this.options.data !== 'undefined') body.append('payload_json', JSON.stringify(this.options.data));
if (!browser) headers = Object.assign(headers, body.getHeaders());
} else if (this.options.data != null) { // eslint-disable-line eqeqeq
body = JSON.stringify(this.options.data);
headers['Content-Type'] = 'application/json';
}
return request;
return fetch(url, {
method: this.method,
headers,
agent,
body,
});
}
}