prehistoric-systems/FMHY-SafeGuard
1
0
Fork 0
mirror of https://github.com/fmhy/FMHY-SafeGuard.git synced 2026-03-11 08:55:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update background.js

Browse source 
Updated warning toggle
This commit is contained in:
Kenneth Hendricks 2024-11-08 17:23:29 -05:00 committed by GitHub
parent ca48b3238f
commit c7538ac28e
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 42 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
src/js/background.js
View file

@ -350,12 +350,16 @@ browserAPI.runtime.onMessage.addListener((message, sender, sendResponse) => {
});
async function openWarningPage(tabId, unsafeUrl) {
const normalizedUrl = normalizeUrl(unsafeUrl);
const tabApprovedUrls = approvedUrls.get(tabId) || [];
if (tabApprovedUrls.includes(normalizeUrl(unsafeUrl))) {
// Check if URL has already been approved for this tab
if (tabApprovedUrls.includes(normalizedUrl)) {
console.log(`URL ${unsafeUrl} was already approved for tab ${tabId}`);
return;
}
// Fetch the warning page setting
const { warningPage } = await browserAPI.storage.sync.get({
warningPage: true,
});
@ -365,12 +369,49 @@ async function openWarningPage(tabId, unsafeUrl) {
return;
}
// Redirect to the warning page if it is enabled in settings
const warningPageUrl = browserAPI.runtime.getURL(
`../pub/warning-page.html?url=${encodeURIComponent(unsafeUrl)}`
);
browserAPI.tabs.update(tabId, { url: warningPageUrl });
}
// Add listener for approval from the warning page
browserAPI.runtime.onMessage.addListener((message, sender, sendResponse) => {
if (message.action === "approveSite") {
const { tabId, url } = message;
const normalizedUrl = normalizeUrl(url);
// Store approval for this URL in the current tab
if (tabId && normalizedUrl) {
let tabApprovedUrls = approvedUrls.get(tabId) || [];
if (!tabApprovedUrls.includes(normalizedUrl)) {
tabApprovedUrls.push(normalizedUrl);
approvedUrls.set(tabId, tabApprovedUrls);
}
console.log(`Approval stored for ${url} in tab ${tabId}`);
sendResponse({ status: "approved" });
}
}
return true; // Indicates asynchronous response handling
});
// Listen for settings updates from the settings page
browserAPI.runtime.onMessage.addListener((message, sender, sendResponse) => {
if (message.type === "settingsUpdated") {
console.log(
"Received settingsUpdated message in background script:",
message.settings
);
// Acknowledge the message by sending a response
sendResponse({ status: "Settings updated successfully" });
// Return true to indicate that the response will be sent asynchronously
return true;
}
});
browserAPI.tabs.onUpdated.addListener((tabId, changeInfo, tab) => {
if (changeInfo.status === "complete" && tab.url) {
checkSiteAndUpdatePageAction(tabId, tab.url);