diff --git a/src/gmail/context/list_emails/list_emails.js b/src/gmail/context/list_emails/list_emails.js index 73c07bd..3d498ad 100644 --- a/src/gmail/context/list_emails/list_emails.js +++ b/src/gmail/context/list_emails/list_emails.js @@ -1,5 +1,5 @@ let emails = []; -for (const tbody of getTbodyWithMultipleRows()) { +for (const tbody of getEmailTables()) { emails.push(...Array.from(tbody.querySelectorAll('tr')).filter(row => row.id).map(row => { const email = {}; email["id"] = row.id; diff --git a/src/gmail/shared/gmail.js b/src/gmail/shared/gmail.js index 56c607f..dc8e813 100644 --- a/src/gmail/shared/gmail.js +++ b/src/gmail/shared/gmail.js @@ -1,10 +1,9 @@ -function getTbodyWithMultipleRows() { +function getEmailTables() { const navDiv = document.querySelector('div[role="navigation"]'); const nextDiv = navDiv ? navDiv.nextElementSibling : null; if (nextDiv && nextDiv.tagName === 'DIV') { - const tbodies = nextDiv.querySelectorAll('tbody'); - return Array.from(tbodies); + return Array.from(nextDiv.querySelectorAll('tbody')); } else { console.log("Next element is not a div or doesn't exist."); return []; diff --git a/src/gmail/tools/read_email/read_email.js b/src/gmail/tools/read_email/read_email.js index 5c74e2f..fb56786 100644 --- a/src/gmail/tools/read_email/read_email.js +++ b/src/gmail/tools/read_email/read_email.js @@ -19,7 +19,7 @@ if (theAction === "") { if (!actionDefined) { console.log("finding email to click") let emails = []; - for (const tbody of getTbodyWithMultipleRows()) { + for (const tbody of getEmailTables()) { emails.push(...Array.from(tbody.querySelectorAll('tr')).map(row => { const email = {}; email["id"] = row.id; diff --git a/src/gmail/tools/select_emails/select_emails.js b/src/gmail/tools/select_emails/select_emails.js index 8707f5b..cfe3c77 100644 --- a/src/gmail/tools/select_emails/select_emails.js +++ b/src/gmail/tools/select_emails/select_emails.js @@ -5,7 +5,7 @@ console.log("Emails to select: " + emailsToSelect); console.log("Emails to select: " + JSON.stringify(emailsToSelect)); let emails = []; -for (const tbody of getTbodyWithMultipleRows()) { +for (const tbody of getEmailTables()) { emails.push(...Array.from(tbody.querySelectorAll('tr')).map(row => { const email = {}; email["id"] = row.id;