pixiv-img/index.js

Save iamge of pixiv

'use strict';
const path = require('path');
const fs = require('fs');
const got = require('got');

module.exports = function (imgUrl, output) {
	return new Promise((resolve, reject) => {
		if (typeof imgUrl !== 'string') {
			reject(new TypeError('Expected a string'));
		}

		output = output || path.basename(imgUrl);

		const options = {
			encoding: null,
			headers: {
				Referer: 'http://www.pixiv.net/'
			}
		};

		const gotStream = got.stream(imgUrl, options);
		gotStream.on('error', err => {
			reject(err);
		});

		gotStream.pipe(fs.createWriteStream(output)).on('close', () => {
			resolve(output);
		});
	});
};