-
Notifications
You must be signed in to change notification settings - Fork 6
/
Copy pathcommon.js
89 lines (83 loc) · 2.39 KB
/
common.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
const puppeteer = require('puppeteer')
const path = require('path')
const sourceDir = 'src'
const timeout = 10000
/**
* @function getBrowser
* @return {puppeteer.Browser}
*/
async function getBrowser() {
const pathToExtension = path.join(process.cwd(), sourceDir)
console.log('pathToExtension:', pathToExtension)
return await puppeteer.launch({
args: [
`--disable-extensions-except=${pathToExtension}`,
`--load-extension=${pathToExtension}`,
'--no-sandbox',
// '--disable-blink-features=AutomationControlled',
// '--disable-features=ChromeUserPermPrompt',
],
dumpio: true,
// headless: false,
// slowMo: 50,
})
}
/**
* @function getWorker
* @global browser
* @global timeout
* @return {Promise<puppeteer.Page>}
*/
async function getWorker(browser) {
const workerTarget = await browser.waitForTarget(
(target) =>
target.type() === 'service_worker' &&
target.url().endsWith('service-worker.js'),
{ timeout }
)
return await workerTarget.worker()
}
/**
* @function getPage
* @global browser
* @global timeout
* @param {puppeteer.Browser} browser
* @param {String} name
* @param {Boolean=} log
* @param {String=} size
* @return {Promise<puppeteer.Page>}
*/
async function getPage(browser, name, log, size) {
console.debug(`getPage: ${name}`, log, size)
const target = await browser.waitForTarget(
(target) => target.type() === 'page' && target.url().includes(name),
{ timeout }
)
const newPage = await target.asPage()
await newPage.emulateMediaFeatures([
{ name: 'prefers-color-scheme', value: 'dark' },
])
newPage.setDefaultTimeout(timeout)
if (size) {
const [width, height] = size.split('x').map((x) => parseInt(x))
await newPage.setViewport({ width, height })
}
if (log) {
console.debug(`Adding Logger: ${name}`)
newPage.on('console', (msg) =>
console.log(`console: ${name}:`, msg.text())
)
}
return newPage
}
async function scrollPage(page) {
await page.evaluate(() => {
window.scrollBy({
top: window.innerHeight,
left: 0,
behavior: 'instant',
})
})
await new Promise((resolve) => setTimeout(resolve, 500))
}
module.exports = { getBrowser, getWorker, getPage, scrollPage }