diff options
Diffstat (limited to 'src/server/ApiManagers/UploadManager.ts')
| -rw-r--r-- | src/server/ApiManagers/UploadManager.ts | 17 |
1 files changed, 13 insertions, 4 deletions
diff --git a/src/server/ApiManagers/UploadManager.ts b/src/server/ApiManagers/UploadManager.ts index 10d7e3a34..756bde738 100644 --- a/src/server/ApiManagers/UploadManager.ts +++ b/src/server/ApiManagers/UploadManager.ts @@ -277,7 +277,9 @@ export default class UploadManager extends ApiManager { } } - +function delay(ms: number) { + return new Promise(resolve => setTimeout(resolve, ms)); +} /** * On success, returns a buffer containing the bytes of a screenshot * of the video (optionally, at a timecode) specified by @param targetUrl. @@ -289,12 +291,19 @@ async function captureYoutubeScreenshot(targetUrl: string): Promise<Opt<Buffer>> const page = await browser.newPage(); await page.setViewport({ width: 1920, height: 1080 }); - await page.goto(targetUrl, { waitUntil: 'networkidle0' }); + await page.goto(targetUrl, { waitUntil: 'domcontentloaded' as any }); + const videoPlayer = await page.$('.html5-video-player'); + videoPlayer && await page.focus("video"); + await delay(7000); + const ad = await page.$('.ytp-ad-skip-button-text'); + await ad?.click(); + await videoPlayer?.click(); + await delay(1000); // hide youtube player controls. - await page.evaluate(() => (document.querySelector('.ytp-chrome-bottom') as any).style.display = 'none'); + await page.evaluate(() => + (document.querySelector('.ytp-chrome-bottom') as any).style.display = 'none'); - const videoPlayer = await page.$('.html5-video-player'); const buffer = await videoPlayer?.screenshot({ encoding: "binary" }); await browser.close(); |