coomdev
2 years ago
13 changed files with 245 additions and 72 deletions
@ -0,0 +1,34 @@ |
|||||
|
import { Platform } from "./platform"; |
||||
|
|
||||
|
const obj = execution_mode == "chrome_api" ? chrome : browser; |
||||
|
type Methods<T> = { |
||||
|
// eslint-disable-next-line @typescript-eslint/ban-types
|
||||
|
[k in Exclude<keyof T, 'prototype'>]: T[k] extends Function ? T[k] : never; |
||||
|
}; |
||||
|
|
||||
|
obj.webRequest.onBeforeRequest.addListener((details) => { |
||||
|
const redirectUrl = details.url; |
||||
|
if (!redirectUrl.startsWith("https://loli.piss/")) |
||||
|
return; |
||||
|
const m = redirectUrl.match(/https:\/\/loli.piss\/(?<domain>.*?)(?<path>\/.*)\/(?<start>.*)\/(?<end>.*)/); |
||||
|
if (!m) |
||||
|
return; |
||||
|
const { domain, path, start, end } = m.groups!; |
||||
|
return { |
||||
|
redirectUrl: `https://${domain}${path}`, |
||||
|
requestHeaders: [{ |
||||
|
name: 'range', |
||||
|
value: `bytes=${start}-${end}` |
||||
|
}] |
||||
|
} as browser.webRequest.BlockingResponse; |
||||
|
}, { urls: ['*://loli.piss/*'] }, ['blocking']); |
||||
|
|
||||
|
obj.runtime.onConnect.addListener((c) => { |
||||
|
c.onMessage.addListener(async obj => { |
||||
|
const { id, name, args } = obj as {id: number, name: keyof Methods<typeof Platform>, args: Parameters<typeof Platform[keyof Methods<typeof Platform>]>}; |
||||
|
const res = await Platform[name](...args); |
||||
|
c.postMessage({ |
||||
|
id, res |
||||
|
}); |
||||
|
}); |
||||
|
}); |
||||
@ -0,0 +1,130 @@ |
|||||
|
import { Buffer } from 'ts-ebml/lib/tools'; |
||||
|
import 'web-ext-types'; |
||||
|
import { GM_fetch, GM_head, headerStringToObject } from './requests'; |
||||
|
|
||||
|
let port: browser.runtime.Port; |
||||
|
const lqueue: ((e: any) => boolean)[] = []; |
||||
|
|
||||
|
if (execution_mode != 'userscript' && !isBackground) { |
||||
|
port = browser.runtime.connect(); |
||||
|
port.onMessage.addListener((e: any) => { |
||||
|
const k = lqueue.map(f => f(e)); |
||||
|
for (let i = k.length - 1; i != -1; --i) { |
||||
|
if (k[i]) |
||||
|
lqueue.splice(i, 1); |
||||
|
} |
||||
|
}); |
||||
|
} |
||||
|
|
||||
|
let gid = 0; |
||||
|
const bridge = <U extends any[], V, T extends (...args: U) => V>(name: string, f: T) => { |
||||
|
if (execution_mode != 'userscript' && !isBackground) |
||||
|
return f; |
||||
|
return (...args: U) => { |
||||
|
const id = gid++; |
||||
|
const prom = new Promise<V>(_ => { |
||||
|
lqueue.push((e: any) => { |
||||
|
if (e.id != id) |
||||
|
return false; |
||||
|
_(e.res); |
||||
|
return true; |
||||
|
}); |
||||
|
port.postMessage({ |
||||
|
id, name, args |
||||
|
}); |
||||
|
}); |
||||
|
return prom; |
||||
|
}; |
||||
|
}; |
||||
|
|
||||
|
// eslint-disable-next-line @typescript-eslint/ban-types
|
||||
|
const Bridged = (ctor: any) => { |
||||
|
const keys = Object.getOwnPropertyNames(ctor).filter(k => typeof ctor[k] == "function"); |
||||
|
for (const k of keys) |
||||
|
ctor[k] = bridge(k, ctor[k]); |
||||
|
}; |
||||
|
|
||||
|
export function supportedAltDomain(s: string) { |
||||
|
if (execution_mode == 'userscript') |
||||
|
return GM.info.script.matches.slice(2).some(m => m.includes(s)); |
||||
|
return false; |
||||
|
} |
||||
|
|
||||
|
// Used to call background-only APIs from content scripts
|
||||
|
@Bridged |
||||
|
export class Platform { |
||||
|
static async openInTab(src: string, opts: { active: boolean, insert: boolean }) { |
||||
|
if (execution_mode == 'userscript') { |
||||
|
return GM.openInTab(src, opts); |
||||
|
} |
||||
|
const obj = execution_mode == "chrome_api" ? chrome : browser; |
||||
|
if (execution_mode == 'chrome_api') { |
||||
|
let i: number | undefined; |
||||
|
if (opts.insert) |
||||
|
i = (await obj.tabs.getCurrent()).index + 1; |
||||
|
return obj.tabs.create({ active: opts.active, url: src, index: i }); |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
|
||||
|
export async function getHeaders(s: string) { |
||||
|
if (execution_mode == 'userscript') |
||||
|
return headerStringToObject(await GM_head(s)); |
||||
|
const res = await fetch(s, { |
||||
|
method: "HEAD" |
||||
|
}); |
||||
|
return [...res.headers.entries()].reduce((a, b) => (a[b[0]] = b[1], a), {} as ReturnType<typeof headerStringToObject>); |
||||
|
} |
||||
|
|
||||
|
export async function ifetch(...[url, opt, lisn]: [...Parameters<typeof fetch>, EventTarget?]): ReturnType<typeof fetch> { |
||||
|
if (execution_mode != "userscript") |
||||
|
return fetch(url, opt); |
||||
|
return GM_fetch(url, opt, lisn); |
||||
|
} |
||||
|
|
||||
|
// most pngs are encoded with 65k idat chunks
|
||||
|
export async function* streamRemote(url: string, chunkSize = 72 * 1024, fetchRestOnNonCanceled = true) { |
||||
|
if (execution_mode != 'userscript') { |
||||
|
const res = await fetch(url); |
||||
|
const reader = res.body; |
||||
|
const stream = reader?.getReader(); |
||||
|
while (!stream?.closed) { |
||||
|
const buff = await stream?.read(); |
||||
|
if (buff?.done) { |
||||
|
break; |
||||
|
} |
||||
|
if (buff?.value) { |
||||
|
const e = (yield buff.value) as boolean; |
||||
|
if (e) { |
||||
|
stream?.cancel(); |
||||
|
reader?.cancel(); |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
|
stream?.releaseLock(); |
||||
|
return; |
||||
|
} |
||||
|
const headers = await getHeaders(url); |
||||
|
const size = +headers['content-length']; |
||||
|
let ptr = 0; |
||||
|
let fetchSize = chunkSize; |
||||
|
while (ptr != size) { |
||||
|
//console.log('doing a fetch of ', url, ptr, ptr + fetchSize - 1);
|
||||
|
const res = await ifetch(url, { headers: { range: `bytes=${ptr}-${ptr + fetchSize - 1}` } }) as any as Tampermonkey.Response<any>; |
||||
|
const obj = headerStringToObject(res.responseHeaders); |
||||
|
if (!('content-length' in obj)) { |
||||
|
console.warn("no content lenght???", url); |
||||
|
break; |
||||
|
} const len = +obj['content-length']; |
||||
|
ptr += len; |
||||
|
if (fetchRestOnNonCanceled) |
||||
|
fetchSize = size; |
||||
|
const val = Buffer.from(await (res as any).arrayBuffer()); |
||||
|
const e = (yield val) as boolean; |
||||
|
//console.log('yeieledd, a', e);
|
||||
|
if (e) { |
||||
|
break; |
||||
|
} |
||||
|
} |
||||
|
} |
||||
Loading…
Reference in new issue