Batch Download on creator, not post
目前为
// ==UserScript==
// @name Fanbox Batch Downloader
// @namespace http://tampermonkey.net/
// @version 0.625
// @description Batch Download on creator, not post
// @author https://github.com/amarillys QQ 719862760
// @require https://cdnjs.cloudflare.com/ajax/libs/jszip/3.2.2/jszip.min.js
// @require https://cdnjs.cloudflare.com/ajax/libs/dat-gui/0.7.6/dat.gui.min.js
// @match https://www.pixiv.net/fanbox/creator/*
// @grant GM_xmlhttpRequest
// @grant GM_addStyle
// @grant unsafeWindow
// @run-at document-end
// @license MIT
// ==/UserScript==
/**
* Update Log
* > 200328
* Improve file naming
* Fix bugs that may cause files being skipped
* Add text if exist in post
* > 200226
* Adapt to new Api! Add Error Tip!
* More frequentyle progress bar!
* More clearly status!
* > 200224
* More beautiful! UI Redesigned. --use dat.gui,
* Performence Improved. -- multi-thread supported.
* > 200222
* Bug Fixed - Psd files download failure <Change download type from blob to arraybuffer, which cause low performence>
* Bug Fixed - Display incorrect on partial download
* > 200222
* Bug Fixed - Post with '/' cause deep path in zip
* > 200102
* Bug Fixed - Caused by empty cover
* > 191228
* Bug Fixed
* Correct filenames
* > 191227
* Code Reconstruct
* Support downloading of artice
* Correct filenames
*
* // 中文注释
* 代码重构
* 新增对文章的下载支持
* > 200222
* 偷懒，以后不加中文注释
* > 191226
* Support downloading by batch(default: 100 files per batch)
* Support donwloading by specific index
* // 中文注释
* 新增支持分批下载的功能（默认100个文件一个批次）
* 新增支持按索引下载的功能
*
* > 191223
* Add support of files
* Improve the detect of file extension
* Change Download Request as await, for avoiding delaying.
* Add manual package while click button use middle button of mouse
* // 中文注释
* 增加对附件下载的支持
* 优化文件后缀名识别
* 修改下载方式为按顺序下载，避免超时
* 增加当鼠标中键点击时手动打包
**/
/* global JSZip GM_xmlhttpRequest */
;(function() {
'use strict'
const apiUserUri = 'https://fanbox.pixiv.net/api/creator.get'
const apiPostUri = 'https://fanbox.pixiv.net/api/post.listCreator'
// set style
GM_addStyle(`
.dg.main{
top: 16px;
position: fixed;
left: 20%;
filter: drop-shadow(2px 4px 6px black);
opacity: 0.8;
z-index: 999;
}
li.cr.number.has-slider:nth-child(2) {
pointer-events: none;
}
.slider-fg {
transition: width 0.5s ease-out;
}
`)
window = unsafeWindow
class ThreadPool {
constructor(poolSize) {
this.size = poolSize || 20
this.running = 0
this.waittingTasks = []
this.callback = []
this.tasks = []
this.counter = 0
this.sum = 0
this.finished = false
this.errorLog = ''
this.step = () => {}
this.timer = null
this.callback.push(() =>
console.log(this.errorLog)
)
}
status() {
return ((this.counter / this.sum) * 100).toFixed(1) + '%'
}
run() {
if (this.finished) return
if (this.waittingTasks.length === 0)
if (this.running <= 0) {
for (let m = 0; m < this.callback.length; ++m)
this.callback[m] && this.callback[m]()
this.finished = true
} else return
while (this.running < this.size) {
if (this.waittingTasks.length === 0) return
let curTask = this.waittingTasks[0]
curTask.do().then(
onSucceed => {
this.running--
this.counter++
this.step()
this.run()
typeof onSucceed === 'function' && onSucceed()
},
onFailed => {
this.errorLog += onFailed + '\n'
this.running--
this.counter++
this.step()
this.run()
curTask.err()
}
)
this.waittingTasks.splice(0, 1)
this.tasks.push(this.waittingTasks[0])
this.running++
}
}
add(fn, errFn) {
this.waittingTasks.push({ do: fn, err: errFn || (() => {}) })
this.sum++
clearTimeout(this.timer)
this.timer = setTimeout(() => {
this.run()
clearTimeout(this.timer)
}, this.autoStartTime)
}
setAutoStart(time) {
this.autoStartTime = time
}
finish(callback) {
this.callback.push(callback)
}
isFinished() {
return this.finished
}
}
class Zip {
constructor(title) {
this.title = title
this.zip = new JSZip()
this.size = 0
this.partIndex = 0
}
file(filename, blob) {
this.zip.file(filename, blob, {
compression: 'STORE'
})
this.size += blob.size
}
add(folder, name, blob) {
if (this.size + blob.size >= Zip.MAX_SIZE)
this.pack()
this.zip.folder(folder).file(name, blob, {
compression: 'STORE'
})
this.size += blob.size
}
pack() {
if (this.size === 0) return
let index = this.partIndex
this.zip
.generateAsync({
type: 'blob'
})
.then(zipBlob => saveBlob(zipBlob, `${this.title}-${index}.zip`))
this.partIndex++
this.zip = new JSZip()
this.size = 0
}
}
Zip.MAX_SIZE = 1048576000
const creatorId = parseInt(document.URL.split('/')[5])
let creatorInfo = null
let options = {
start: 1,
end: 1,
thread: 6,
batch: 200,
progress: 0,
speed: 0
}
const Text = {
batch: '分批 / Batch',
download: '点击这里下载',
download_en: 'Click to Download',
downloading: '下载中...',
downloading_en: 'Downloading...',
packing: '打包中...',
packing_en: 'Packing...',
packed: '打包完成',
packed_en: 'Packed!',
init: '初始化中...',
init_en: 'Initilizing...',
initFailed: '请求数据失败',
initFailed_en: 'Failed to get Data',
initFailed_0: '请检查网络',
initFailed_0_en: 'check network',
initFailed_1: '或Github联系作者',
initFailed_1_en: 'or connect at Github',
initFinished: '初始化完成',
initFinished_en: 'Initilized',
start: '起始 / start',
end: '结束 / end',
thread: '线程 / threads',
pack: '手动打包(不推荐)',
pack_en: 'manual pack(Not Rcm)',
progress: '进度 / Progress',
speed: '网速 / speed'
}
const EN_FIX = navigator.language.indexOf('zh') > -1 ? '' : '_en'
let label = null
const gui = new dat.GUI({
autoPlace: false,
useLocalStorage: false
})
const clickHandler = {
text() {},
download: () => {
console.log('startDownloading')
downloadByFanboxId(creatorInfo, creatorId)
},
pack() {
label.name(Text['packing' + EN_FIX])
zip.pack()
label.name(Text['packed' + EN_FIX])
}
}
label = gui.add(clickHandler, 'text').name(Text['init' + EN_FIX])
let progressCtl = null
let init = async () => {
let base = unsafeWindow.document.querySelector('#root')
base.appendChild(gui.domElement)
uiInited = true
try {
creatorInfo = await getAllPostsByFanboxId(creatorId)
label.name(Text['initFinished' + EN_FIX])
} catch (e) {
label.name(Text['initFailed' + EN_FIX])
gui.add(clickHandler, 'text').name(Text['initFailed_0' + EN_FIX])
gui.add(clickHandler, 'text').name(Text['initFailed_1' + EN_FIX])
return
}
// init dat gui
const sum = creatorInfo.posts.length
progressCtl = gui.add(options, 'progress', 0, 100, 0.01).name(Text.progress)
const startCtl = gui.add(options, 'start', 1, sum, 1).name(Text.start)
const endCtl = gui.add(options, 'end', 1, sum, 1).name(Text.end)
gui.add(options, 'thread', 1, 20, 1).name(Text.thread)
gui.add(options, 'batch', 10, 5000, 10).name(Text.batch)
gui.add(clickHandler, 'download').name(Text['download' + EN_FIX])
gui.add(clickHandler, 'pack').name(Text['pack' + EN_FIX])
endCtl.setValue(sum)
startCtl.onChange(() => (options.start = options.start > options.end ? options.end : options.start))
endCtl.onChange(() => (options.end = options.end < options.start ? options.start : options.end ))
gui.open()
}
// init global values
let zip = null
let amount = 1
let pool = null
let progressList = []
let uiInited = false
const fetchOptions = {
credentials: 'include',
headers: {
Accept: 'application/json, text/plain, */*'
}
}
const setProgress = amount => {
let currentProgress = progressList.reduce((p, q) => p + q, 0) / amount * 100
if (currentProgress > 0)
progressCtl.setValue(currentProgress)
}
window.onload = () => {
init()
let timer = setInterval(() => {
(!uiInited && document.querySelector('.dg.main') === null) ? init() : clearInterval(timer)
}, 3000)
}
function gmRequireImage(url, index) {
return new Promise((resolve, reject) =>
GM_xmlhttpRequest({
method: 'GET',
url,
overrideMimeType: 'application/octet-stream',
responseType: 'blob',
asynchrouns: true,
onload: res => {
progressList[index] = 1
setProgress(amount)
resolve(res.response)
},
onprogress: res => {
progressList[index] = res.done / res.total
setProgress(amount)
},
onerror: () =>
GM_xmlhttpRequest({
method: 'GET',
url,
overrideMimeType: 'application/octet-stream',
responseType: 'arraybuffer',
onload: res => {
progressList[index] = 1
setProgress(amount)
resolve(new Blob([res.response]))
},
onprogress: res => {
progressList[index] = res.done / res.total
setProgress(amount)
},
onerror: res => reject(res)
})
})
)
}
async function downloadByFanboxId(creatorInfo, creatorId) {
let processed = 0
amount = 1
label.name(Text['downloading' + EN_FIX])
progressCtl.setValue(0)
let { batch, end, start, thread } = options
options.progress = 0
zip = new Zip(`${creatorInfo.name}-${start}-${end}`)
let stepped = 0
creatorInfo.cover ? gmRequireImage(creatorInfo.cover, 0).then(blob => zip.file('cover.jpg', blob)) : null
// init pool
pool = new ThreadPool(thread)
pool.finish(() => {
label.name(Text['packing' + EN_FIX])
zip.pack()
label.name(Text['packed' + EN_FIX])
})
// for name exist detect
let titles = []
// boolean for folder created
let folderCreated = false
// start downloading
for (let i = start - 1, p = creatorInfo.posts; i < end; ++i) {
let folder = `${p[i].title.replace(/\//g, '-')}`
let titleExistLength = titles.filter(title => title === folder).length
if (titleExistLength > 0)
folder += `-${titleExistLength}`
titles.push(folder)
if (!p[i].body) continue
let { blocks, imageMap, fileMap, files, images, text } = p[i].body
let picIndex = 0
let imageList = []
let fileList = []
if (p[i].type === 'article') {
let article = `# ${p[i].title}\n`
for (let j = 0; j < blocks.length; ++j) {
switch (blocks[j].type) {
case 'p': {
article += `${blocks[j].text}\n\n`
break
}
case 'image': {
picIndex++
let image = imageMap[blocks[j].imageId]
imageList.push(image)
article += `![${p[i].title} - P${picIndex}](${folder}_${j}.${image.extension})\n\n`
break
}
case 'file': {
let file = fileMap[blocks[j].fileId]
fileList.push(file)
article += `[${p[i].title} - ${file.name}](${creatorId}-${folder}-${file.name}.${file.extension})\n\n`
break
}
}
}
zip.add(folder, 'article.md', new Blob([article]))
for (let j = 0; j < imageList.length; ++j) {
let image = imageList[j]
let index = amount
amount++
pool.add(() => new Promise((resolve, reject) => {
gmRequireImage(image.originalUrl, index).then(blob => {
processed++
zip.add(folder, `${folder}_${j}.${image.extension}`, blob)
stepped++
resolve()
}).catch(() => {
console.log(`Failed to download: ${image.originalUrl}`)
reject()
})
}))
}
for (let j = 0; j < fileList.length; ++j) {
let file = fileList[j]
let index = amount
amount++
pool.add(() => new Promise((resolve, reject) => {
let fileIndexText = ''
if (files.length > 1) fileIndexText = `-${j}`
gmRequireImage(file.url, index).then(blob => {
processed++
saveBlob(blob, `${creatorInfo.name}-${folder}${fileIndexText}-${file.name}.${file.extension}`)
stepped++
resolve()
}).catch(() => {
console.log(`Failed to download: ${file.url}`)
reject()
})
}))
}
}
if (files) {
for (let j = 0; j < files.length; ++j) {
let file = files[j]
let index = amount
amount++
pool.add(() => new Promise((resolve, reject) => {
gmRequireImage(file.url, index).then(blob => {
processed++
let fileIndexText = ''
if (files.length > 1) fileIndexText = `-${j}`
if (blob.size < 51200000 * 2) {
zip.add(folder, `${file.name}${fileIndexText}.${file.extension}`, blob)
folderCreated = true
}
else
saveBlob(blob, `${creatorInfo.name}@${folder}${fileIndexText}.${file.extension}`)
stepped++
resolve()
}).catch(() => {
console.log(`Failed to download: ${file.url}`)
reject()
})
}))
}
}
if (images) {
for (let j = 0; j < images.length; ++j) {
let image = images[j]
let index = amount
amount++
pool.add(() => new Promise((resolve, reject) => {
gmRequireImage(image.originalUrl, index).then(blob => {
processed++
zip.add(folder, `${folder}_${j}.${image.extension}`, blob)
stepped++
resolve()
}).catch(() => {
console.log(`Failed to download: ${image.url}`)
reject()
})
}))
}
}
if (text) {
let textBlob = new Blob([text], { type: 'text/plain' })
zip.add(folder, `${creatorInfo.name}-${folder}.txt`, textBlob)
}
}
progressList = new Array(amount).fill(0)
pool.step = () => {
console.log(` Progress: ${processed} / ${amount}, Pool: ${pool.running} @ ${pool.sum}`)
if (stepped >= batch) {
zip.pack()
stepped = 0
}
}
}
async function getAllPostsByFanboxId(creatorId) {
// request userinfo
const userUri = `${apiUserUri}?userId=${creatorId}`
const userData = await (await fetch(userUri, fetchOptions)).json()
let creatorInfo = {
cover: null,
posts: []
}
const limit = 50
creatorInfo.cover = userData.body.coverImageUrl
creatorInfo.name = userData.body.user.name
// request post info
let postData = await (await fetch(`${apiPostUri}?userId=${creatorId}&limit=${limit}`, fetchOptions)).json()
creatorInfo.posts.push(...postData.body.items.filter(p => p.body))
let nextPageUrl = postData.body.nextUrl
while (nextPageUrl) {
let nextData = await (await fetch(nextPageUrl, fetchOptions)).json()
creatorInfo.posts.push(...nextData.body.items.filter(p => p.body))
nextPageUrl = nextData.body.nextUrl
}
console.log(creatorInfo)
return creatorInfo
}
function saveBlob(blob, fileName) {
let downloadDom = document.createElement('a')
document.body.appendChild(downloadDom)
downloadDom.style = `display: none`
let url = window.URL.createObjectURL(blob)
downloadDom.href = url
downloadDom.download = fileName
downloadDom.click()
window.URL.revokeObjectURL(url)
}
})()