52 lines
2.1 KiB
JavaScript
52 lines
2.1 KiB
JavaScript
|
chrome.runtime.onMessage.addListener((message, sender, sendResponse) => {
|
||
|
|
if (message.type === 'phishing-detected') {
|
||
|
|
const emails = message.emails;
|
||
|
|
let notificationTitle = 'You are safe!';
|
||
|
|
let notificationMessage = 'No phishing emails detected.';
|
||
|
|
|
||
|
|
const phishingEmails = emails.filter(email => email.is_phishing);
|
||
|
|
if (phishingEmails.length > 0) {
|
||
|
|
notificationTitle = 'You are in danger!';
|
||
|
|
notificationMessage = `Email from ${phishingEmails[0].from} titled "${phishingEmails[0].subject}" has been identified as phishing.`;
|
||
|
|
|
||
|
|
chrome.windows.create({
|
||
|
|
url: 'notification.html',
|
||
|
|
type: 'popup',
|
||
|
|
width: 300,
|
||
|
|
height: 200
|
||
|
|
}, function(window) {
|
||
|
|
chrome.storage.local.set({
|
||
|
|
notificationTitle: notificationTitle,
|
||
|
|
notificationMessage: notificationMessage,
|
||
|
|
emailId: phishingEmails[0].id
|
||
|
|
});
|
||
|
|
});
|
||
|
|
} else {
|
||
|
|
chrome.windows.create({
|
||
|
|
url: 'notification.html',
|
||
|
|
type: 'popup',
|
||
|
|
width: 300,
|
||
|
|
height: 200
|
||
|
|
}, function(window) {
|
||
|
|
chrome.storage.local.set({
|
||
|
|
notificationTitle: notificationTitle,
|
||
|
|
notificationMessage: notificationMessage,
|
||
|
|
emailId: null
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
} else if (message.type === 'mark-safe') {
|
||
|
|
fetch(`http://localhost:5000/mark_safe/${message.emailId}`, {
|
||
|
|
method: 'POST'
|
||
|
|
}).then(response => response.json())
|
||
|
|
.then(data => console.log(data))
|
||
|
|
.catch(error => console.error('Error:', error));
|
||
|
|
} else if (message.type === 'move-trash') {
|
||
|
|
fetch(`http://localhost:5000/move_trash/${message.emailId}`, {
|
||
|
|
method: 'POST'
|
||
|
|
}).then(response => response.json())
|
||
|
|
.then(data => console.log(data))
|
||
|
|
.catch(error => console.error('Error:', error));
|
||
|
|
}
|
||
|
|
});
|