added additional error handling

paul-hanneforth · Jun 15, 2021 · 7026e06 · 7026e06
1 parent 4e0cbc1
commit 7026e06
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/message.js b/message.js
@@ -13,7 +13,8 @@ const errorCode = {
     12: "Browser is not running!",
     13: "Failed to load session!",
     14: "The bot is already authenticated!",
-    15: "An error was thrown without reasoning by instagram."
+    15: "An error was thrown without reasoning by instagram.",
+    16: "Username couldn't be extracted!"
 };
 const errorTemplate = (code) => {
     return {
@@ -36,7 +37,8 @@ const errorMessage = {
     "browserNotRunning": errorTemplate(12),
     "failedToLoadSession": errorTemplate(13),
     "botAlreadyAuthenticated": errorTemplate(14),
-    "errorWithoutReasoning": errorTemplate(15)
+    "errorWithoutReasoning": errorTemplate(15),
+    "usernameNotExtracted": errorTemplate(16)
 };
 
 module.exports = { errorMessage };
diff --git a/scripts/data.js b/scripts/data.js
@@ -261,7 +261,9 @@ const getPostDetails = async (page, identifier) => {
 
     const postId = link.split("/")[4];
 
-    const username = await page.evaluate(() => [...document.querySelectorAll(".sqdOP.yWX7d._8A5w5.ZIAjV")][0].innerText);
+    const linksOnPage = await page.evaluate(() => [...document.querySelectorAll("a")].map(el => el.href));
+    if(linksOnPage.length == 0) throw new IBError(errorMessage.usernameNotExtracted.code, errorMessage.usernameNotExtracted.username);
+    const username = linksOnPage[0].split("/")[3];
     const user = new User(`https://www.instagram.com/${username}/`, username, null);
 
     const likesStr = await page.evaluate((postId) => {