pxder/src/illust.js

Download illusts from pixiv.net P站插画批量下载器

let pixiv;

/**
 * 插画
 *
 * @class Illust
 */
class Illust {
	/**
	 *Creates an instance of Illust.
	 * @param {number} id PID
	 * @param {string} title 作品名
	 * @param {string} url 原画链接
	 * @param {string} file 文件名
	 * @memberof Illust
	 */
	constructor(id, title, url, file) {
		this.id = id;
		this.title = title;
		this.url = url;
		this.file = file;
	}

	static setPixiv(p) {
		pixiv = p;
	}

	getObject() {
		return {
			id: this.id,
			title: this.title,
			url: this.url,
			file: this.file,
		};
	}

	/**
	 * 从插画JSON对象中得到插画列表
	 *
	 * @param {*} illustJSON 插画JSON对象
	 * @returns 插画列表
	 */
	static async getIllusts(illustJSON) {
		const illusts = [];
		// 得到插画信息
		const title = illustJSON.title.replace(/[\x00-\x1F\x7F]/g, '');
		const fileName = title.replace(/[/\\:*?"<>|.&$]/g, ''); // 适合的文件名
		const id = illustJSON.id;
		// 动图的话是一个压缩包
		if (illustJSON.type == 'ugoira') {
			const ugoiraParams = [id, title, illustJSON.meta_single_page.original_image_url.replace('img-original', 'img-zip-ugoira').replace(/_ugoira0\.(.*)/, '_ugoira1920x1080.zip')];
			if (global.ugoiraMeta) {
				try {
					const uDelay = await pixiv.ugoiraMetaData(id).then(ret => ret.ugoira_metadata.frames[0].delay);
					illusts.push(new Illust(...ugoiraParams, `(${id})${fileName}@${uDelay}ms.zip`));
				} catch (error) {
					console.error('\nFailed to get ugoira meta data . If you get a rate limit error, please use ', '--no-ugoira-meta'.yellow, 'argument to avoid it.', error, '\n');
					illusts.push(new Illust(...ugoiraParams, `(${id})${fileName}.zip`));
				}
			} else illusts.push(new Illust(...ugoiraParams, `(${id})${fileName}.zip`));
		} else {
			if (illustJSON.meta_pages.length > 0) {
				// 组图
				for (const pi in illustJSON.meta_pages) {
					const url = illustJSON.meta_pages[pi].image_urls.original;
					const ext = url.substr(url.lastIndexOf('.')); // 图片扩展名
					illusts.push(new Illust(id, title + '_p' + pi, url, `(${id})${fileName}_p${pi}${ext}`));
				}
			} else if (illustJSON.meta_single_page.original_image_url) {
				const url = illustJSON.meta_single_page.original_image_url;
				const ext = url.substr(url.lastIndexOf('.')); // 图片扩展名
				// 单图
				illusts.push(new Illust(id, title, url, `(${id})${fileName}${ext}`));
			}
		}
		// 结果
		return illusts;
	}
}

module.exports = Illust;