-
Notifications
You must be signed in to change notification settings - Fork 4
/
content-script.js
87 lines (77 loc) · 2.52 KB
/
content-script.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
let _started = false;
const handleFromBackground = async (request, sender, sendResponse) => {
if (request.message === 'start') {
start();
sendResponse();
} else if (request.message === 'end') {
end();
sendResponse();
} else if (request.data) {
if (!_started) return true;
console.log(`process from ${request.from}`);
process(request.data);
sendResponse();
}
return true;
};
const handleFromWeb = async (event) => {
if (event.data.from) {
const data = event.data.data;
console.log(`process from ${event.data.from}`);
process(Object.values(data));
}
};
const injectScript = (filePath, tag) => {
var node = document.getElementsByTagName(tag)[0];
var script = document.createElement('script');
script.setAttribute('type', 'text/javascript');
script.setAttribute('src', filePath);
script.setAttribute('id', 'inject');
node.appendChild(script);
}
const removeInjectScript = () => {
const scriptElement = document.querySelector('#inject');
if (!scriptElement) return;
scriptElement.remove();
};
const process = async (items) => {
const storage = await chrome.storage.sync.get(['ts']);
console.log('Will post data is below.');
console.log(items);
const url = storage.ts.url;
Promise.all(items.map(async (i) => {
return fetch(url, {
method: "POST",
headers: {
'Content-Type': 'application/json'
},
body: JSON.stringify({ 'data': i })
}).catch((error) => {
chrome.runtime.sendMessage({ data: i }, (response) => { });
});
}))
};
const scrollToBottom = async (event) => {
document.scrollingElement.scrollBy(0, document.scrollingElement.scrollHeight)
};
const start = () => {
_started = true;
window.addEventListener('message', handleFromWeb);
window.addEventListener('scroll', scrollToBottom);
injectScript(chrome.runtime.getURL('web_accessible_resources.js'), 'body');
console.log('start tiktok scraping!!!');
}
const end = () => {
_started = false;
window.removeEventListener('message', handleFromWeb);
window.removeEventListener('scroll', scrollToBottom);
removeInjectScript();
console.log('end tiktok scraping');
};
chrome.runtime.onMessage.addListener(handleFromBackground);
(async () => {
const storage = await chrome.storage.sync.get(['ts']);
setInterval(() => {
window.dispatchEvent(new Event('scroll'));
}, storage.ts.interval);
})();