forked from nunch-dev/Kyobo-EBOOK-PDF
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbackground.js
91 lines (78 loc) · 2.29 KB
/
background.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
let requestIds = {};
chrome.webRequest.onBeforeSendHeaders.addListener(
(details) => {
if (!details.url.toLowerCase().endsWith(".pdf")) {
return;
}
let headers = {};
for (let i of details.requestHeaders) {
headers[i.name.toLowerCase()] = i.value;
}
if (headers["range"]) {
return;
}
requestIds[details.requestId] = headers;
setTimeout(() => {
delete requestIds[details.requestId];
}, 10000);
},
{ urls: ["*://wviewer.kyobobook.co.kr/content/web_ebook/web_pdf/*"] },
["requestHeaders", "extraHeaders"]
);
chrome.webRequest.onHeadersReceived.addListener(
(details) => {
let reqHeaders = requestIds[details.requestId];
if (!reqHeaders) {
return;
}
let headers = {};
for (let i of details.responseHeaders) {
headers[i.name.toLowerCase()] = i.value;
}
if (headers["content-range"]) {
return;
}
chrome.scripting.executeScript({
target: { tabId: details.tabId },
func: async (url, headers) => {
if (!confirm("PDF Found. Do you want to download?")) {
return;
}
let overlay = document.createElement("div");
overlay.style.position = "fixed";
overlay.style.top = "1em";
overlay.style.left = "1em";
overlay.style.zIndex = "999999999";
overlay.textContent = "Downloading... This may take a long time.";
try {
document.body.appendChild(overlay);
let resp = await fetch(url, {
headers,
});
let blob = await resp.blob();
let objectURL = URL.createObjectURL(blob);
let link = document.createElement("a");
link.href = objectURL;
link.download = +new Date() + ".pdf";
document.body.appendChild(link);
link.click();
document.body.removeChild(link);
URL.revokeObjectURL(objectURL);
} catch (err) {
alert(`error: ${err}`);
} finally {
document.body.removeChild(overlay);
}
},
args: [
details.url,
{
...reqHeaders,
range: `bytes=0-${headers["content-length"]}`,
},
],
});
},
{ urls: ["*://wviewer.kyobobook.co.kr/content/web_ebook/web_pdf/*"] },
["responseHeaders", "extraHeaders"]
);