// ==UserScript==
// @name nhentai helper
// @name:zh-CN nhentai 助手
// @name:zh-TW nhentai 助手
// @namespace https://github.com/Tsuk1ko
// @version 2.2.1
// @icon https://nhentai.net/favicon.ico
// @description Add a "download zip" button for nhentai gallery page and some useful feature
// @description:zh-CN 为 nhentai 增加 zip 打包下载方式以及一些辅助功能
// @description:zh-TW 爲 nhentai 增加 zip 打包下載方式以及一些輔助功能
// @author Jindai Kirin
// @include https://nhentai.net/*
// @connect i.nhentai.net
// @license GPL-3.0
// @grant GM_addStyle
// @grant GM_getValue
// @grant GM_setValue
// @grant GM_registerMenuCommand
// @grant GM_xmlhttpRequest
// @grant GM_getResourceText
// @resource notycss https://cdn.bootcss.com/noty/3.1.4/noty.css
// @require https://cdn.bootcss.com/jquery/3.3.1/jquery.min.js
// @require https://cdn.bootcss.com/jszip/3.1.4/jszip.min.js
// @require https://cdn.bootcss.com/FileSaver.js/1.3.2/FileSaver.min.js
// @require https://cdn.bootcss.com/jquery.pjax/2.0.1/jquery.pjax.min.js
// @require https://cdn.bootcss.com/vue/2.6.10/vue.min.js
// @require https://cdn.bootcss.com/noty/3.1.4/noty.min.js
// @run-at document-end
// @noframes
// @homepageURL https://github.com/Tsuk1ko/nhentai-helper
// @supportURL https://github.com/Tsuk1ko/nhentai-helper/issues
// ==/UserScript==
(function() {
'use strict';
Array.prototype.remove = function(index) {
if (index > -1) return this.splice(index, 1)[0];
};
GM_addStyle(GM_getResourceText('notycss'));
GM_addStyle(
'.download-zip:disabled{cursor:wait}.gallery>.download-zip{position:absolute;z-index:1;left:0;top:0;opacity:.8}.gallery:hover>.download-zip{opacity:1}#download-panel::-webkit-scrollbar{width:6px;background-color:rgba(0,0,0,.7)}#download-panel::-webkit-scrollbar-thumb{background-color:rgba(255,255,255,.6)}#download-panel{ overflow-x:hidden;position:fixed;top:20vh;right:0;width:calc(50vw - 620px);max-width:300px;min-width:150px;max-height:60vh;background-color:rgba(0,0,0,.7);z-index:100;font-size:12px;overflow-y:scroll}.download-item{position:relative;white-space:nowrap;padding:2px;overflow:visible}.download-item-cancel{cursor:pointer;position:absolute;top:0;right:-30px;color:#F44336;font-size:20px;line-height:30px;width:30px}.download-item:hover{width:calc(100% - 30px)}.download-item-title{overflow:hidden;text-overflow:ellipsis;text-align:left}.download-item-progress{background-color:rgba(0,0,255,.5);line-height:10px}.download-error .download-item-progress{background-color:rgba(255,0,0,.5)}.download-item-progress-text{transform:scale(.8)}#page-container{position:relative}#gp-view-mode-btn{position:absolute;right:0;top:0;margin:0}.btn-noty-green{background-color:#66BB6A!important}.btn-noty-blue{background-color:#42A5F5!important}.btn-noty:hover{filter:brightness(1.15)}.noty_buttons{padding-top:0!important}'
);
$('body').append('<div id="download-panel"></div>');
const notyOption = {
type: 'error',
layout: 'bottomRight',
theme: 'nest',
timeout: false,
closeWith: [],
};
const EXT = { p: 'png', j: 'jpg', g: 'gif' };
const getExtension = _t => {
if (!EXT[_t]) throw new Error(`Unknown type "${_t}"`);
return EXT[_t];
};
// 页面类型
const pageType = {
gallery: !!/^https:\/\/nhentai\.net\/g\/[0-9]+\/(\?.*)?$/.exec(window.location.href),
galleryPage: !!/^https:\/\/nhentai\.net\/g\/[0-9]+\/[0-9]+\/(\?.*)?$/.exec(window.location.href),
list: $('.gallery').length > 0,
};
// 下载队列
const queue = [];
const queueInfo = JSON.parse(sessionStorage.getItem('queueInfo')) || [];
const downloadHistory = JSON.parse(localStorage.getItem('downloadHistory')) || [];
const downloadStatus = {
running: false,
skip: false,
};
const startQueue = async () => {
if (!downloadStatus.running && queue.length > 0) {
downloadStatus.running = true;
do {
await queue[0]();
queue.shift();
} while (queue.length > 0);
downloadStatus.running = false;
}
};
// 下载面板
Vue.component('download-item', {
props: ['item', 'index'],
computed: {
width() {
const { page, done } = this.item;
return page && done ? ((100 * done) / page).toFixed(2) : 0;
},
},
watch: {
'item.error': function(error) {
if (error) {
const n = new Noty({
...notyOption,
text: `Error occurred, retry?`,
buttons: [
Noty.button('SKIP', 'btn btn-noty', () => {
n.close();
queue.shift();
queueInfo.shift();
downloadStatus.running = false;
startQueue();
}),
Noty.button('YES', 'btn btn-noty-green btn-noty', () => {
n.close();
this.item.error = false;
downloadStatus.running = false;
startQueue();
}),
],
});
n.show();
}
},
},
methods: {
cancel() {
if (this.index === 0) {
downloadStatus.skip = true;
queueInfo.shift();
} else {
queue.remove(this.index);
const q = queueInfo.remove(this.index);
if (q && typeof q.cancel === 'function') q.cancel();
}
},
},
template: '<div :class="`download-item ${item.error?\'download-error\':\'\'}`" :title="item.title"><div class="download-item-cancel" @click="cancel"><i class="fa fa-times"></i></div><div class="download-item-title">{{item.title}}</div><div class="download-item-progress" :style="`width:${width}%`"><div class="download-item-progress-text">{{width}}%</div></div></div>',
});
Vue.component('download-list', {
props: ['list'],
template: '<div id="download-panel"><download-item v-for="(item, index) in list" :item="item" :index="index" :key="index" /></div>',
});
new Vue({
el: '#download-panel',
data: { queueInfo, downloadHistory, downloadStatus },
watch: {
queueInfo(val) {
sessionStorage.setItem('queueInfo', JSON.stringify(val));
},
downloadHistory(val) {
while (val.length > 100) val.shift();
localStorage.setItem('downloadHistory', JSON.stringify(val));
},
},
template: '<download-list :list="queueInfo" />',
});
// 网络请求
const get = (url, responseType = 'json', retry = 3) =>
new Promise((resolve, reject) => {
GM_xmlhttpRequest({
method: 'GET',
url,
responseType,
onerror: e => {
if (retry === 0) reject(e);
else {
console.warn('Network error, retry.');
setTimeout(() => {
resolve(get(url, responseType, retry--));
}, 1000);
}
},
onload: r => resolve(r.response),
});
});
// 下载线程数
let THREAD = GM_getValue('thread_num', 8);
GM_registerMenuCommand('设置 nhentai 下载线程数', () => {
let num;
do {
num = prompt(`请输入下载线程数 (1~32) [当前:${THREAD}]`, THREAD);
if (num === null) return;
num = parseInt(num);
} while (num.toString() == 'NaN' || num < 1 || num > 32);
THREAD = num;
GM_setValue('thread_num', num);
});
// 伪多线程
const multiThread = (tasks, promiseFunc) => {
const threads = [];
let taskIndex = 0;
const run = threadID =>
new Promise(async resolve => {
while (true) {
let i = taskIndex++;
if (i >= tasks.length) break;
await promiseFunc(tasks[i], threadID);
}
resolve();
});
// 创建线程
for (let threadID = 0; threadID < THREAD; threadID++) {
threads.push(run(threadID));
}
return Promise.all(threads);
};
// 获取本子信息
const getGallery = async gid => {
const {
media_id,
title: { english, japanese },
images: { pages },
} = gid ? await get(`https://nhentai.net/api/gallery/${gid}`) : gallery;
const p = [];
pages.forEach((page, i) => {
p.push({
i: i + 1,
t: getExtension(page.t),
});
});
const info = {
mid: media_id,
title: japanese || english,
pages: p,
};
console.log({ gid, ...info });
return info;
};
// 下载本子
const downloadGallery = async ({ mid, title, pages }, $btn = null, $btnTxt = null, headTxt = '') => {
const info = queueInfo[0] || {};
info.done = 0;
const zip = new JSZip();
const btnUpdateProgress = () => {
if (!$btnTxt) return;
if (info.done >= pages.length) $btnTxt.html(`${headTxt}√`);
else $btnTxt.html(`${headTxt}${info.done}/${pages.length}`);
};
btnUpdateProgress();
const dlPromise = (page, threadID) => {
if (info.error || downloadStatus.skip) return;
const filename = `${page.i}.${page.t}`;
const url = `https://i.nhentai.net/galleries/${mid}/${filename}`;
console.log(`[${threadID}] ${url}`);
return get(url, 'blob')
.then(r => {
zip.file(filename, r);
info.done++;
btnUpdateProgress();
})
.catch(e => {
info.error = true;
throw e;
});
};
await multiThread(pages, dlPromise);
if (downloadStatus.skip) return {};
const data = await zip.generateAsync({
type: 'blob',
base64: true,
});
if ($btn) $btn.attr('disabled', false);
queueInfo.shift();
return {
name: `${title}.zip`,
data,
};
};
const downloadG = async (gid, $btn = null, $btnTxt = null, headTxt = '') => downloadGallery(await getGallery(gid), $btn, $btnTxt, headTxt);
// 语言过滤
const langFilter = lang => {
if (lang == 'none') $('.gallery').removeClass('hidden');
else {
$(`.gallery[lang=${lang}]`).removeClass('hidden');
$(`.gallery:not([lang=${lang}])`).addClass('hidden');
}
};
// 本子浏览模式
const applyGPViewStyle = gpViewMode => {
if (gpViewMode) $('body').append('<style id="gp-view-mode-style">#image-container img{width:auto;max-height:100vh}</style>');
else $('#gp-view-mode-style').remove();
};
// 功能初始化
const init = (first = false) => {
if (!first) {
$('.pagination a').each(function() {
const $this = $(this);
$this.attr('href', $this.attr('href').replace(/(&?)_pjax=[^&]*(&?)/, ''));
});
// pjax 后需要初始化页面以加载 lazyload 图片
N.init();
}
if (pageType.gallery) {
// 本子详情页
$('#info > .buttons').append('<button class="btn btn-secondary download-zip"><i class="fa fa-download"></i> <span class="download-zip-txt">Download zip</span></button>');
const $btn = $('.download-zip');
const $btnTxt = $('.download-zip-txt');
let zip;
$btn.click(async () => {
try {
if (!zip) {
$btn.attr('disabled', true);
zip = await downloadG(null, $btn, $btnTxt, 'Download zip ');
}
saveAs(zip.data, zip.name);
} catch (error) {
$btnTxt.html('Error');
console.error(error);
}
});
} else if (pageType.list) {
// 本子列表页
$('.gallery').each(function() {
const $this = $(this);
$this.prepend('<button class="btn btn-secondary download-zip"><i class="fa fa-download"></i> <span class="download-zip-txt"></span></button>');
const $a = $this.find('a.cover');
$a.attr('target', '_blank');
const gid = /[0-9]+/.exec($a.attr('href'))[0];
// 用于语言过滤
let language = '';
const dataTags = $this.attr('data-tags').split(' ');
if (dataTags.includes('6346')) language = 'jp';
else if (dataTags.includes('12227')) language = 'en';
else if (dataTags.includes('29963')) language = 'zh';
$this.attr('lang', language);
const $btn = $this.find('.download-zip');
const $btnTxt = $this.find('.download-zip-txt');
const cancel = () => {
$btn.attr('disabled', false);
$btnTxt.html('');
};
$btn.click(async () => {
$btn.attr('disabled', true);
$btnTxt.html('Wait');
const gallery = await getGallery(gid);
if (downloadHistory.includes(gallery.title) || queueInfo.some(({ title }) => title == gallery.title)) {
const abandon = new Promise(resolve => {
const n = new Noty({
...notyOption,
text: `"${gallery.title}" is already downloaded or in queue.<br>Do you want to download again?`,
buttons: [
Noty.button('YES', 'btn btn-noty', () => {
n.close();
resolve(false);
}),
Noty.button('NO', 'btn btn-noty-green btn-noty', () => {
n.close();
resolve(true);
$btn.attr('disabled', false);
$btnTxt.html('');
}),
],
});
n.show();
});
if (await abandon) return;
}
queueInfo.push({
gid,
title: gallery.title,
page: gallery.pages.length,
done: 0,
error: false,
cancel,
});
queue.push(async () => {
const { data, name } = await downloadGallery(gallery, $btn, $btnTxt);
if (downloadStatus.skip) {
downloadStatus.skip = false;
cancel();
return;
}
saveAs(data, name);
if (!downloadHistory.includes(gallery.title)) downloadHistory.push(gallery.title);
});
startQueue();
});
});
if (first) {
// 语言过滤
$('ul.menu.left').append('<li style="padding:0 10px">LANG filter: <select id="lang-filter"><option value="none">None</option><option value="zh">Chinese</option><option value="jp">Japanese</option><option value="en">English</option></select></li>');
$('#lang-filter').change(function() {
langFilter(this.value);
sessionStorage.setItem('lang-filter', this.value);
});
// 左右键翻页
$(document).keydown(function(event) {
switch (event.keyCode) {
case 37: // left
$('.pagination .previous').click();
break;
case 39: // right
$('.pagination .next').click();
break;
}
});
}
// 还原记住的语言过滤
const rememberedLANG = sessionStorage.getItem('lang-filter');
if (rememberedLANG) {
$('#lang-filter')[0].value = rememberedLANG;
langFilter(rememberedLANG);
}
// 还原下载队列
if (first) {
for (const { gid, title } of queueInfo) {
queue.push(async () => {
const { data, name } = await downloadG(gid);
if (downloadStatus.skip) {
downloadStatus.skip = false;
return;
}
saveAs(data, name);
if (!downloadHistory.includes(title)) downloadHistory.push(title);
});
}
}
startQueue();
} else if (pageType.galleryPage) {
// 本子在线阅读
const gpViewModeText = ['[off]', '[on]'];
let gpViewMode = GM_getValue('gp_view_mode', 0);
applyGPViewStyle(gpViewMode);
$('#page-container').prepend(`<button id="gp-view-mode-btn" class="btn btn-secondary"><i class="fa fa-arrows-v"></i> <span>100% view height</span> <span id="gp-view-mode-switch-text">${gpViewModeText[gpViewMode]}</span></button>`);
const $gpvmst = $('#gp-view-mode-switch-text');
$('#gp-view-mode-btn').click(() => {
gpViewMode = 1 - gpViewMode;
GM_setValue('gp_view_mode', gpViewMode);
$gpvmst.html(gpViewModeText[gpViewMode]);
applyGPViewStyle(gpViewMode);
});
}
};
$(document).pjax('.pagination a, .sort a', { container: '#content', fragment: '#content' });
$(document).on('pjax:end', () => init());
init(true);
})();