story/highlight update 2

AppleGamer22 · Dec 27, 2020 · a463632 · a463632
1 parent b763657
commit a463632
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/src/commands/highlight.ts b/src/commands/highlight.ts
@@ -89,8 +89,8 @@ export async function detectFiles(browser: Browser, page: Page, highlight: strin
 		await page.click("svg[aria-label='Pause']");
 		var urls: string[] = [];
 		await page.waitForSelector("div.qbCDp", {visible: true});
-		const imageURL = (await page.$$eval("img.y-yJ5", images => images.map(image => image.getAttribute("src"))))[0];
-		if (imageURL) urls.push(imageURL);
+		const imageURL = (await page.$$eval("img.y-yJ5", images => images.map(image => image.getAttribute("srcset"))))[0];
+		if (imageURL) urls.push(imageURL.split(" ")[0]);
 		const videoURL = (await page.$$eval("video > source", sources => sources.map(source => source.getAttribute("src"))))[0];
 		if (videoURL) urls.push(videoURL);
 		return { urls, username };

diff --git a/src/commands/story.ts b/src/commands/story.ts
@@ -83,8 +83,8 @@ export async function detectFiles(browser: Browser, page: Page, user: string, it
 		await page.click("svg[aria-label='Pause']");
 		var urls: string[] = [];
 		await page.waitForSelector("div.qbCDp", {visible: true});
-		const imageURL = (await page.$$eval("img.y-yJ5", images => images.map(image => image.getAttribute("src"))))[0];
-		if (imageURL) urls.push(imageURL);
+		const imageURL = (await page.$$eval("img.y-yJ5", images => images.map(image => image.getAttribute("srcset"))))[0];
+		if (imageURL) urls.push(imageURL.split(" ")[0]);
 		const videoURL = (await page.$$eval("video > source", sources => sources.map(source => source.getAttribute("src"))))[0];
 		if (videoURL) urls.push(videoURL);
 		return { username: user, urls };