wumpfetch/lib/model/WumpRequest.js

const http = require('http');
const https = require('https');
const { URL } = require('url');
const { join } = require('path');
const { stringify } = require('querystring');
const { createGunzip, createInflate } = require('zlib');

const WumpResponse = require('./WumpResponse');

const w = require('../index');
const c = [ 'gzip', 'deflate' ];

module.exports = class WumpRequest {
	constructor (url = {}, method = {}) {
		const o = (typeof url === 'string' ? method : url);
		const obj = (typeof o === 'object');

		if (typeof url !== 'string' && !o.hasOwnProperty('url')) throw new Error('Missing URL parameter');

		this.o = {
			'm': typeof o === 'string' ? o : obj && o.method ? o.method : 'GET',
			'url': typeof url === 'string' ? new URL(url) : obj && typeof o.url === 'string' ? new URL(o.url) : url,
			'SDA': obj && typeof o.sendDataAs === 'string' ? o.sendDataAs : obj && o.data && typeof o.data === 'object' ? 'json' : undefined,
			'data': obj && o.body ? o.body : obj && o.data ? o.data : obj && o.json ? o.json : obj && o.form ? stringify(o.form) : undefined,
			'parse': obj && o.parse ? o.parse : undefined,
			'follow': !!(obj && o.followRedirects),
			'rHeaders': obj && typeof o.headers === 'object' ? o.headers : {},
			'streamed': !!(obj && o.streamed),
			'compressed': !!(obj && o.compressed),
			'timeoutTime': obj && typeof o.timeout === 'number' ? o.timeout : null,
			'coreOptions': obj && typeof o.coreOptions === 'object' ? o.coreOptions : {}
		};

		if (typeof o.core === 'object') Object.keys(o.core).forEach((v) => this.option(v, o.core[v]));

		return this;
	}

	query (a, b) {
		if (typeof a === 'object') Object.keys(a).forEach((v) => this.o.url.searchParams.append(v, a[v]));
		else this.o.url.searchParams.append(a, b);

		return this;
	}

	body (data, SA) {
		this.o.SDA = typeof data === 'object' && !SA && !Buffer.isBuffer(data) ? 'json' : (SA ? SA.toLowerCase() : 'buffer');
		this.o.data = this.SDA === 'form' ? stringify(data) : (this.SDA === 'json' ? JSON.stringify(data) : data);

		return this;
	}

	header (a, b) {
		if (typeof a === 'object') Object.keys(a).forEach((v) => this.o.rHeaders[v.toLowerCase()] = a[v]);
		else this.o.rHeaders[a.toLowerCase()] = b;

		return this;
	}

	compress () {
		this.compressed = true;
		if (!this.o.rHeaders['accept-encoding']) this.o.rHeaders['accept-encoding'] = c.join(', ');

		return this;
	}
	
	path (p) {
		this.o.url.pathname = join(this.o.url.pathname, p);

		return this;
	}
	
	stream () {
		this.o.streamed = true;
		
		return this;
	}

	option (n, v) {
		this.o.coreOptions[n] = v;
		
		return this;
	}
	timeout (timeout) {
		this.o.timeoutTime = timeout;
		
		return this;
	}

	send () {
		return new Promise((resolve, reject) => {
			if (this.o.data) {
				if (!this.o.rHeaders.hasOwnProperty('user-agent')) this.o.rHeaders['User-Agent'] = w.userAgent || 'Wumpfetch/2.0.1 (https://github.com/PassTheWessel/wumpfetch)';
				
				if (this.o.SDA === 'json' && !this.o.rHeaders.hasOwnProperty('content-type')) this.o.rHeaders['Content-Type'] = 'application/json';
				if (this.o.SDA === 'form') {
					if (!this.o.rHeaders.hasOwnProperty('content-type')) this.o.rHeaders['Content-Type'] = 'application/x-www-form-urlencoded';
					if (!this.o.rHeaders.hasOwnProperty('content-length')) this.o.rHeaders['Content-Length'] = Buffer.byteLength(this.o.data);
				}
			}

			let req;
			const options = {
				'host': this.o.url.hostname,
				'port': this.o.url.port,
				'path': this.o.url.pathname + this.o.url.search,
				'method': this.o.m,
				'headers': this.o.rHeaders,
				'protocol': this.o.url.protocol,
				...this.o.coreOptions
			}

			const handler = async (res) => {
				let stream = res;

				if (this.o.compressed) {
					if (res.headers['content-encoding'] === 'gzip') stream = res.pipe(createGunzip());
					else if (res.headers[ 'content-encoding'] === 'deflate') stream = res.pipe(createInflate());
				}

				let wumpRes;

				if (this.o.streamed) resolve(stream);
				else {
					wumpRes = new WumpResponse(res);

					if (res.headers.hasOwnProperty('location') && this.o.follow) {
						this.o.url = (new URL(res.headers['location'], this.o.url)).toString();
						return await w(this.o);
					}

					stream.on('error', (e) => reject(e));
					stream.on('data', (c) => wumpRes._addChunk(c));
					stream.on('end', () => {
						if (this.o.parse) {
							if (this.o.parse === 'json') wumpRes.body = JSON.parse(wumpRes.body);
							else if (this.o.parse === 'text') wumpRes.body = wumpRes.body.toString();
							else wumpRes.body = wumpRes.body;
						}
						
						resolve(wumpRes);
					});
				}
			};

			if (this.o.url.protocol === 'http:') req = http.request(options, handler);
			else if (this.o.url.protocol === 'https:') req = https.request(options, handler);
			else throw new Error(`Bad URL protocol: ${this.o.url.protocol}`);

			if (this.o.timeoutTime) {
				req.setTimeout(this.o.timeoutTime, () => {
					req.abort();
					if (!this.o.streamed) reject(new Error('Timeout reached'));
				});
			}

			req.on('error', (e) => reject(e));

			if (this.o.data) {
				if (this.o.SDA === 'json') req.write(JSON.stringify(this.o.data));
				else {
					if (typeof this.o.data === 'object') req.write(JSON.stringify(this.o.data));
					else req.write(this.o.data);
				}
			}
			
			req.end();
		});
	}
};