musescore-downloader/src/file.ts
2020-12-06 00:51:51 -05:00

124 lines
3.2 KiB
TypeScript

/* eslint-disable no-extend-native */
/* eslint-disable @typescript-eslint/no-unsafe-return */
import { hookNative } from './anti-detection'
import { console } from './utils'
type FileType = 'img' | 'mp3' | 'midi'
const TYPE_REG = /id=(\d+)&type=(img|mp3|midi)/
/**
* I know this is super hacky.
*/
const magicHookConstr = (() => {
const l = {}
// get rid of `Disable Tampermonkey`
const observer = new MutationObserver((mutations) => {
mutations.forEach(x => [...x.addedNodes].forEach(e => {
if (!document.querySelector('.js-page')) {
(e as HTMLElement).replaceWith(...x.removedNodes)
}
}))
})
observer.observe(document, { childList: true, subtree: true })
try {
const p = Object.getPrototypeOf(document.body)
Object.setPrototypeOf(document.body, null)
hookNative(document.body, 'append', () => {
return function (...nodes: Node[]) {
p.append.call(this, ...nodes)
if (nodes[0].nodeName === 'IFRAME') {
const iframe = nodes[0] as HTMLIFrameElement
const w = iframe.contentWindow as Window
hookNative(w, 'fetch', () => {
return function (url, init) {
const token = init?.headers?.Authorization
if (typeof url === 'string' && token) {
const m = url.match(TYPE_REG)
if (m) {
const type = m[2]
// eslint-disable-next-line no-unused-expressions
l[type]?.(token)
}
}
return fetch(url, init)
}
})
}
}
})
Object.setPrototypeOf(document.body, p)
} catch (err) {
console.error(err)
}
return async (type: FileType) => {
return new Promise<string>((resolve) => {
l[type] = (token) => {
resolve(token)
magics[type] = token
}
})
}
})()
const magics: Record<FileType, Promise<string>> = {
img: magicHookConstr('img'),
midi: magicHookConstr('midi'),
mp3: magicHookConstr('mp3'),
}
const getApiUrl = (id: number, type: FileType, index: number): string => {
return `/api/jmuse?id=${id}&type=${type}&index=${index}&v2=1`
}
const getApiAuth = async (type: FileType, index: number): Promise<string> => {
// eslint-disable-next-line no-void
void index
const magic = magics[type]
if (magic instanceof Promise) {
// force to retrieve the MAGIC
switch (type) {
case 'midi': {
const el = document.querySelectorAll('.SD7H- > button')[3] as HTMLButtonElement
el.click()
break
}
case 'mp3': {
const el = document.querySelector('button[title="Toggle Play"]') as HTMLButtonElement
el.click()
break
}
case 'img': {
const imgE = document.querySelector('img[src*=score_]')
const nextE = imgE?.parentElement?.nextElementSibling
if (nextE) nextE.scrollIntoView()
break
}
}
}
return magic
}
export const getFileUrl = async (id: number, type: FileType, index = 0): Promise<string> => {
const url = getApiUrl(id, type, index)
const auth = await getApiAuth(type, index)
const r = await fetch(url, {
headers: {
Authorization: auth,
},
})
const { info } = await r.json()
return info.url as string
}