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 index.js

Browse source
This commit is contained in:
Kenneth Hendricks 2024-11-03 16:05:34 -05:00 committed by GitHub
parent aba7d2cb27
commit e43f4b02a5
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 83 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

127
platform/chromium/pub/index.js
View file

@ -19,52 +19,78 @@ document.addEventListener("DOMContentLoaded", () => {
return `${urlObj.protocol}//${urlObj.hostname}`;
}
try {
// Get the active tab's URL
chrome.tabs.query({ active: true, currentWindow: true }, (tabs) => {
const activeTab = tabs[0];
if (!activeTab || !activeTab.url) {
throw new Error("No active tab found or URL is unavailable.");
// Function to apply theme based on settings
function applyTheme() {
chrome.storage.sync.get("theme", ({ theme }) => {
if (theme === "dark") {
document.body.setAttribute("data-theme", "dark");
} else if (theme === "light") {
document.body.setAttribute("data-theme", "light");
} else {
const prefersDark = window.matchMedia(
"(prefers-color-scheme: dark)"
).matches;
document.body.setAttribute(
"data-theme",
prefersDark ? "dark" : "light"
);
}
const currentUrl = normalizeUrl(activeTab.url);
const rootDomain = extractRootDomain(currentUrl);
console.log(`Active tab URL: ${currentUrl}, Root domain: ${rootDomain}`);
// Send a message to the background script to check the site's status
chrome.runtime.sendMessage(
{ action: "checkSiteStatus", url: currentUrl },
(response) => {
if (!response || !response.status) {
throw new Error(
"Failed to retrieve site status from the background script."
);
}
// Determine if the root domain or the full URL is marked as safe/starred
chrome.runtime.sendMessage(
{ action: "checkSiteStatus", url: rootDomain },
(isRootDomainMarked) => {
// Handle different site statuses and update the UI accordingly
if (
isRootDomainMarked.status === response.status &&
response.status !== "no_data"
) {
// If both the root domain and the current URL share the same status, show the root domain in the message
handleStatusUpdate(response.status, rootDomain);
} else {
// Otherwise, show the current URL in the message
handleStatusUpdate(response.status, currentUrl);
}
}
);
}
);
});
} catch (error) {
console.error("Error while checking site status:", error);
errorMessage.textContent = `Error: ${error.message}`;
}
// Apply theme on load
applyTheme();
// Get the active tab's URL
chrome.tabs.query({ active: true, currentWindow: true }, (tabs) => {
if (chrome.runtime.lastError || !tabs[0] || !tabs[0].url) {
handleError("No active tab found or URL is unavailable.");
return;
}
const currentUrl = normalizeUrl(tabs[0].url);
const rootDomain = extractRootDomain(currentUrl);
console.log(`Active tab URL: ${currentUrl}, Root domain: ${rootDomain}`);
// Send a message to the background script to check the site's status
chrome.runtime.sendMessage(
{ action: "checkSiteStatus", url: currentUrl },
(response) => {
if (chrome.runtime.lastError || !response || !response.status) {
handleError(
"Failed to retrieve site status from the background script."
);
return;
}
// Determine if the root domain or the full URL is marked as safe/starred
chrome.runtime.sendMessage(
{ action: "checkSiteStatus", url: rootDomain },
(isRootDomainMarked) => {
if (chrome.runtime.lastError || !isRootDomainMarked) {
handleError("Failed to check root domain status.");
return;
}
if (
isRootDomainMarked.status === response.status &&
response.status !== "no_data"
) {
// If both the root domain and the current URL share the same status, show the root domain in the message
handleStatusUpdate(response.status, rootDomain);
} else {
// Otherwise, show the current URL in the message
handleStatusUpdate(response.status, currentUrl);
}
}
);
}
);
});
function handleError(message) {
console.error(message);
errorMessage.textContent = `Error: ${message}`;
updateUI("error", "An error occurred while retrieving the site status.");
}
@ -87,6 +113,12 @@ document.addEventListener("DOMContentLoaded", () => {
`${displayUrl} is <strong>potentially unsafe</strong>. Proceed with caution.`
);
break;
case "fmhy":
updateUI(
"fmhy",
`${displayUrl} is an <strong>FMHY</strong> related site. Proceed confidently.`
);
break;
case "safe":
updateUI("safe", `${displayUrl} is <strong>safe</strong> to browse.`);
break;
@ -119,6 +151,7 @@ document.addEventListener("DOMContentLoaded", () => {
const icons = {
unsafe: "../res/icons/unsafe.png",
potentially_unsafe: "../res/icons/potentially_unsafe.png",
fmhy: "../res/icons/fmhy.png",
safe: "../res/icons/safe.png",
starred: "../res/icons/starred.png",
no_data: "../res/ext_icon_144.png",
@ -136,4 +169,10 @@ document.addEventListener("DOMContentLoaded", () => {
console.log(`UI updated: ${message}`);
}
// Add settings button functionality
document.getElementById("settingsButton").addEventListener("click", () => {
// Open the settings page in a new tab
chrome.runtime.openOptionsPage();
});
});